// SPDX-License-Identifier: GPL-2.0
/*
 * Microchip KSZ9477 series register access through I2C
 *
 * Copyright (C) 2018-2024 Microchip Technology Inc.
 */

#include <linux/i2c.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/regmap.h>

#include "ksz_common.h"

KSZ_REGMAP_TABLE(ksz9477, not_used, 16, 0, 0);

static int ksz9477_i2c_probe(struct i2c_client *i2c)
{
        const struct ksz_chip_data *chip;
        struct device *ddev = &i2c->dev;
        struct regmap_config rc;
        struct ksz_device *dev;
        int i, ret;

        dev = ksz_switch_alloc(&i2c->dev, i2c);
        if (!dev)
                return -ENOMEM;

        chip = device_get_match_data(ddev);
        if (!chip)
                return -EINVAL;

        /* Save chip id to do special initialization when probing. */
        dev->chip_id = chip->chip_id;
        for (i = 0; i < __KSZ_NUM_REGMAPS; i++) {
                rc = ksz9477_regmap_config[i];
                rc.lock_arg = &dev->regmap_mutex;
                dev->regmap[i] = devm_regmap_init_i2c(i2c, &rc);
                if (IS_ERR(dev->regmap[i])) {
                        return dev_err_probe(&i2c->dev, PTR_ERR(dev->regmap[i]),
                                             "Failed to initialize regmap%i\n",
                                             ksz9477_regmap_config[i].val_bits);
                }
        }

        if (i2c->dev.platform_data)
                dev->pdata = i2c->dev.platform_data;

        dev->irq = i2c->irq;

        ret = ksz_switch_register(dev);

        /* Main DSA driver may not be started yet. */
        if (ret)
                return ret;

        i2c_set_clientdata(i2c, dev);

        return 0;
}

static void ksz9477_i2c_remove(struct i2c_client *i2c)
{
        struct ksz_device *dev = i2c_get_clientdata(i2c);

        if (dev)
                ksz_switch_remove(dev);
}

static void ksz9477_i2c_shutdown(struct i2c_client *i2c)
{
        struct ksz_device *dev = i2c_get_clientdata(i2c);

        if (!dev)
                return;

        ksz_switch_shutdown(dev);

        i2c_set_clientdata(i2c, NULL);
}

static const struct i2c_device_id ksz9477_i2c_id[] = {
        { "ksz9477-switch" },
        {}
};

MODULE_DEVICE_TABLE(i2c, ksz9477_i2c_id);

static const struct of_device_id ksz9477_dt_ids[] = {
        {
                .compatible = "microchip,ksz9477",
                .data = &ksz_switch_chips[KSZ9477]
        },
        {
                .compatible = "microchip,ksz9896",
                .data = &ksz_switch_chips[KSZ9896]
        },
        {
                .compatible = "microchip,ksz9897",
                .data = &ksz_switch_chips[KSZ9897]
        },
        {
                .compatible = "microchip,ksz9893",
                .data = &ksz_switch_chips[KSZ9893]
        },
        {
                .compatible = "microchip,ksz9563",
                .data = &ksz_switch_chips[KSZ9563]
        },
        {
                .compatible = "microchip,ksz8563",
                .data = &ksz_switch_chips[KSZ8563]
        },
        {
                .compatible = "microchip,ksz8567",
                .data = &ksz_switch_chips[KSZ8567]
        },
        {
                .compatible = "microchip,ksz9567",
                .data = &ksz_switch_chips[KSZ9567]
        },
        {
                .compatible = "microchip,lan9646",
                .data = &ksz_switch_chips[LAN9646]
        },
        {},
};
MODULE_DEVICE_TABLE(of, ksz9477_dt_ids);

static DEFINE_SIMPLE_DEV_PM_OPS(ksz_i2c_pm_ops,
                                ksz_switch_suspend, ksz_switch_resume);

static struct i2c_driver ksz9477_i2c_driver = {
        .driver = {
                .name   = "ksz9477-switch",
                .of_match_table = ksz9477_dt_ids,
                .pm = &ksz_i2c_pm_ops,
        },
        .probe = ksz9477_i2c_probe,
        .remove = ksz9477_i2c_remove,
        .shutdown = ksz9477_i2c_shutdown,
        .id_table = ksz9477_i2c_id,
};

module_i2c_driver(ksz9477_i2c_driver);

MODULE_AUTHOR("Tristram Ha <Tristram.Ha@microchip.com>");
MODULE_DESCRIPTION("Microchip KSZ9477 Series Switch I2C access Driver");
MODULE_LICENSE("GPL v2");