/*
 * An hwmon driver for accton as5812_54x Power Module
 *
 * Copyright (C) 2015 Accton Technology Corporation.
 * Brandon Chuang <brandon_chuang@accton.com.tw>
 *
 * Based on ad7414.c
 * Copyright 2006 Stefan Roese <sr at denx.de>, DENX Software Engineering
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <linux/module.h>
#include <linux/jiffies.h>
#include <linux/i2c.h>
#include <linux/hwmon.h>
#include <linux/hwmon-sysfs.h>
#include <linux/err.h>
#include <linux/mutex.h>
#include <linux/sysfs.h>
#include <linux/slab.h>


#define PSU_STATUS_I2C_ADDR			0x60
#define PSU_STATUS_I2C_REG_OFFSET	0x2

#define IS_POWER_GOOD(id, value)	(!!(value & BIT(id*4 + 1)))
#define IS_PRESENT(id, value)		(!(value & BIT(id*4)))

static ssize_t show_index(struct device *dev, struct device_attribute *da, char *buf);
static ssize_t show_status(struct device *dev, struct device_attribute *da, char *buf);
static ssize_t show_model_name(struct device *dev, struct device_attribute *da, char *buf);
static ssize_t show_serial_number(struct device *dev, struct device_attribute *da,char *buf);
static int as5812_54x_psu_read_block(struct i2c_client *client, u8 command, u8 *data,int data_len);
extern int as5812_54x_cpld_read(unsigned short cpld_addr, u8 reg);
static int as5812_54x_psu_model_name_get(struct device *dev , int get_serial);

/* Addresses scanned
 */
static const unsigned short normal_i2c[] = { I2C_CLIENT_END };

/* Each client has this additional data
 */
struct as5812_54x_psu_data {
    struct device      *hwmon_dev;
    struct mutex        update_lock;
    char                valid;           /* !=0 if registers are valid */
    unsigned long       last_updated;    /* In jiffies */
    u8  index;           /* PSU index */
    u8  status;          /* Status(present/power_good) register read from CPLD */
    char model_name[14]; /* Model name, read from eeprom */
    char serial[16]; /* Model name, read from eeprom */
};

static struct as5812_54x_psu_data *as5812_54x_psu_update_device(struct device *dev);

enum as5812_54x_psu_sysfs_attributes {
    PSU_INDEX,
    PSU_PRESENT,
    PSU_MODEL_NAME,
    PSU_POWER_GOOD,
    PSU_SERIAL_NUMBER
};

/* sysfs attributes for hwmon
 */
static SENSOR_DEVICE_ATTR(psu_index,      S_IRUGO, show_index,     NULL, PSU_INDEX);
static SENSOR_DEVICE_ATTR(psu_present,    S_IRUGO, show_status,    NULL, PSU_PRESENT);
static SENSOR_DEVICE_ATTR(psu_model_name, S_IRUGO, show_model_name,NULL, PSU_MODEL_NAME);
static SENSOR_DEVICE_ATTR(psu_serial, S_IRUGO, show_serial_number, NULL, PSU_SERIAL_NUMBER);
static SENSOR_DEVICE_ATTR(psu_power_good, S_IRUGO, show_status,    NULL, PSU_POWER_GOOD);

static struct attribute *as5812_54x_psu_attributes[] = {
    &sensor_dev_attr_psu_index.dev_attr.attr,
    &sensor_dev_attr_psu_present.dev_attr.attr,
    &sensor_dev_attr_psu_model_name.dev_attr.attr,
    &sensor_dev_attr_psu_serial.dev_attr.attr,
    &sensor_dev_attr_psu_power_good.dev_attr.attr,
    NULL
};

static ssize_t show_index(struct device *dev, struct device_attribute *da,
             char *buf)
{
    struct i2c_client *client = to_i2c_client(dev);
    struct as5812_54x_psu_data *data = i2c_get_clientdata(client);

    return sprintf(buf, "%d\n", data->index);
}

static ssize_t show_status(struct device *dev, struct device_attribute *da,
             char *buf)
{
    struct sensor_device_attribute *attr = to_sensor_dev_attr(da);
    struct as5812_54x_psu_data *data = as5812_54x_psu_update_device(dev);
    u8 status = 0;

    if (!data->valid) {
        return sprintf(buf, "0\n");
    }

    if (attr->index == PSU_PRESENT) {
        status = IS_PRESENT(data->index, data->status);
    }
    else { /* PSU_POWER_GOOD */
        status = IS_POWER_GOOD(data->index, data->status);
    }

    return sprintf(buf, "%d\n", status);
}

static ssize_t show_serial_number(struct device *dev, struct device_attribute *da,
             char *buf)
{
    struct as5812_54x_psu_data *data = as5812_54x_psu_update_device(dev);

    if (!data->valid) {
        return 0;
    }

    if (!IS_PRESENT(data->index, data->status)) {
        return 0;
    }

    if (as5812_54x_psu_model_name_get(dev, 1) < 0) {
        return -ENXIO;
    }
    return sprintf(buf, "%s\n", data->serial);
}

static ssize_t show_model_name(struct device *dev, struct device_attribute *da,
             char *buf)
{
    struct as5812_54x_psu_data *data = as5812_54x_psu_update_device(dev);

    if (!data->valid) {
        return 0;
    }

    if (!IS_PRESENT(data->index, data->status)) {
        return 0;
    }

    if (as5812_54x_psu_model_name_get(dev, 0) < 0) {
        return -ENXIO;
    }

    return sprintf(buf, "%s\n", data->model_name);
}

static const struct attribute_group as5812_54x_psu_group = {
    .attrs = as5812_54x_psu_attributes,
};

static int as5812_54x_psu_probe(struct i2c_client *client,
            const struct i2c_device_id *dev_id)
{
    struct as5812_54x_psu_data *data;
    int status;

    if (!i2c_check_functionality(client->adapter, I2C_FUNC_SMBUS_I2C_BLOCK)) {
        status = -EIO;
        goto exit;
    }

    data = kzalloc(sizeof(struct as5812_54x_psu_data), GFP_KERNEL);
    if (!data) {
        status = -ENOMEM;
        goto exit;
    }

    i2c_set_clientdata(client, data);
    data->valid = 0;
    data->index = dev_id->driver_data;
    mutex_init(&data->update_lock);

    dev_info(&client->dev, "chip found\n");

    /* Register sysfs hooks */
    status = sysfs_create_group(&client->dev.kobj, &as5812_54x_psu_group);
    if (status) {
        goto exit_free;
    }

    data->hwmon_dev = hwmon_device_register_with_info(&client->dev, "as5812_54x_psu",
                                                      NULL, NULL, NULL);
    if (IS_ERR(data->hwmon_dev)) {
        status = PTR_ERR(data->hwmon_dev);
        goto exit_remove;
    }

    dev_info(&client->dev, "%s: psu '%s'\n",
         dev_name(data->hwmon_dev), client->name);

    return 0;

exit_remove:
    sysfs_remove_group(&client->dev.kobj, &as5812_54x_psu_group);
exit_free:
    kfree(data);
exit:

    return status;
}

static int as5812_54x_psu_remove(struct i2c_client *client)
{
    struct as5812_54x_psu_data *data = i2c_get_clientdata(client);

    hwmon_device_unregister(data->hwmon_dev);
    sysfs_remove_group(&client->dev.kobj, &as5812_54x_psu_group);
    kfree(data);

    return 0;
}

enum psu_index 
{ 
    as5812_54x_psu1, 
    as5812_54x_psu2
};

static const struct i2c_device_id as5812_54x_psu_id[] = {
    { "as5812_54x_psu1", as5812_54x_psu1 },
    { "as5812_54x_psu2", as5812_54x_psu2 },
    {}
};
MODULE_DEVICE_TABLE(i2c, as5812_54x_psu_id);

static struct i2c_driver as5812_54x_psu_driver = {
    .class        = I2C_CLASS_HWMON,
    .driver = {
        .name     = "as5812_54x_psu",
    },
    .probe        = as5812_54x_psu_probe,
    .remove       = as5812_54x_psu_remove,
    .id_table     = as5812_54x_psu_id,
    .address_list = normal_i2c,
};

static int as5812_54x_psu_read_block(struct i2c_client *client, u8 command, u8 *data,
              int data_len)
{
    int result = i2c_smbus_read_i2c_block_data(client, command, data_len, data);

    if (unlikely(result < 0))
        goto abort;
    if (unlikely(result != data_len)) {
        result = -EIO;
        goto abort;
    }

    result = 0;

abort:
    return result;
}

enum psu_type {
    PSU_YM_2401_JCR,    /* AC110V - F2B */
    PSU_YM_2401_JDR,    /* AC110V - B2F */
    PSU_CPR_4011_4M11,  /* AC110V - F2B */
    PSU_CPR_4011_4M21,  /* AC110V - B2F */
    PSU_CPR_6011_2M11,  /* AC110V - F2B */
    PSU_CPR_6011_2M21,  /* AC110V - B2F */
    PSU_UM400D_01G,     /* DC48V  - F2B */
    PSU_UM400D01_01G    /* DC48V  - B2F */
};

struct serial_number_info {
    enum psu_type type;
    u8 offset;
    u8 length;
};

struct serial_number_info serials[] = {
{PSU_YM_2401_JCR,   0x20, 11},
{PSU_YM_2401_JDR,   0x20, 11},
{PSU_CPR_4011_4M11, 0x47, 15},
{PSU_CPR_4011_4M21, 0x47, 15},
{PSU_CPR_6011_2M11, 0x46, 15},
{PSU_CPR_6011_2M21, 0x46, 15},
{PSU_UM400D_01G,    0x50,  9},
{PSU_UM400D01_01G,  0x50, 12},
};

static int as5812_54x_psu_serial_number_get(struct device *dev, enum psu_type type)
{
    struct i2c_client *client = to_i2c_client(dev);
    struct as5812_54x_psu_data *data = i2c_get_clientdata(client);
    int i, status;

    switch (type) {
    case PSU_CPR_4011_4M11:
    case PSU_CPR_4011_4M21:
    case PSU_CPR_6011_2M11:
    case PSU_CPR_6011_2M21:
        {
            if(type >= sizeof(serials)/sizeof(struct serial_number_info))
                return -EINVAL;
            status = as5812_54x_psu_read_block(client, serials[type].offset,
                                               data->serial, serials[type].length);
            if (status < 0) {
                data->serial[0] = '\0';
                dev_dbg(&client->dev, "unable to read serial number from (0x%x) offset(0x%x)\n", 
                                      client->addr, serials[type].offset);
                return status;
            }
            else {
                data->serial[serials[type].length] = '\0';
                return 0;
            }
        }
        break;
    default:
        break;
    }
    
    return -ENODATA;
}                                            


struct model_name_info {
    enum psu_type type;
    u8 offset;
    u8 length;
    char* model_name;
};

struct model_name_info models[] = {
{PSU_YM_2401_JCR,   0x20, 11, "YM-2401JCR"},
{PSU_YM_2401_JDR,   0x20, 11, "YM-2401JDR"},
{PSU_CPR_4011_4M11, 0x26, 13, "CPR-4011-4M11"},
{PSU_CPR_4011_4M21, 0x26, 13, "CPR-4011-4M21"},
{PSU_CPR_6011_2M11, 0x26, 13, "CPR-6011-2M11"},
{PSU_CPR_6011_2M21, 0x26, 13, "CPR-6011-2M21"},
{PSU_UM400D_01G,    0x50,  9, "um400d01G"},
{PSU_UM400D01_01G,  0x50, 12, "um400d01-01G"},
};

static int as5812_54x_psu_model_name_get(struct device *dev , int get_serial)
{
    struct i2c_client *client = to_i2c_client(dev);
    struct as5812_54x_psu_data *data = i2c_get_clientdata(client);
    int i, status;

    for (i = 0; i < ARRAY_SIZE(models); i++) {
        memset(data->model_name, 0, sizeof(data->model_name));

        status = as5812_54x_psu_read_block(client, models[i].offset,
                                           data->model_name, models[i].length);
        if (status < 0) {
            data->model_name[0] = '\0';
            dev_dbg(&client->dev, "unable to read model name from (0x%x) offset(0x%x)\n", 
                                  client->addr, models[i].offset);
            return status;
        }
        else {
            data->model_name[models[i].length] = '\0';
        }

        if (i == PSU_YM_2401_JCR || i == PSU_YM_2401_JDR) {
            /* Skip the meaningless data byte 8*/
            data->model_name[8] = data->model_name[9];
            data->model_name[9] = data->model_name[10];
            data->model_name[10] = '\0';
        }

        /* Determine if the model name is known, if not, read next index
         */
        if (strncmp(data->model_name, models[i].model_name, models[i].length) == 0) {
            return get_serial ? as5812_54x_psu_serial_number_get(dev, i) : 0;
        }
        else {
            data->model_name[0] = '\0';
        }
    }

    return -ENODATA;
}

static struct as5812_54x_psu_data *as5812_54x_psu_update_device(struct device *dev)
{
    struct i2c_client *client = to_i2c_client(dev);
    struct as5812_54x_psu_data *data = i2c_get_clientdata(client);

    mutex_lock(&data->update_lock);

    if (time_after(jiffies, data->last_updated + HZ + HZ / 2)
        || !data->valid) {
        int status = -1;

        dev_dbg(&client->dev, "Starting as5812_54x update\n");
        data->valid = 0;


        /* Read psu status */
        status = as5812_54x_cpld_read(PSU_STATUS_I2C_ADDR, PSU_STATUS_I2C_REG_OFFSET);

        if (status < 0) {
            dev_dbg(&client->dev, "cpld reg (0x%x) err %d\n", PSU_STATUS_I2C_ADDR, status);
            goto exit;
        }
        else {
            data->status = status;
        }

        data->last_updated = jiffies;
        data->valid = 1;
    }

exit:
    mutex_unlock(&data->update_lock);

    return data;
}

module_i2c_driver(as5812_54x_psu_driver);

MODULE_AUTHOR("Brandon Chuang <brandon_chuang@accton.com.tw>");
MODULE_DESCRIPTION("accton as5812_54x_psu driver");
MODULE_LICENSE("GPL");

