// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2019 Sven Van Asbroeck
 */

#include <linux/leds.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/mfd/tps6105x.h>
#include <linux/regmap.h>

struct tps6105x_priv {
        struct regmap *regmap;
        struct led_classdev cdev;
        struct fwnode_handle *fwnode;
};

static void tps6105x_handle_put(void *data)
{
        struct tps6105x_priv *priv = data;

        fwnode_handle_put(priv->fwnode);
}

static int tps6105x_brightness_set(struct led_classdev *cdev,
                                  enum led_brightness brightness)
{
        struct tps6105x_priv *priv = container_of(cdev, struct tps6105x_priv,
                                                        cdev);

        return regmap_update_bits(priv->regmap, TPS6105X_REG_0,
                                  TPS6105X_REG0_TORCHC_MASK,
                                  brightness << TPS6105X_REG0_TORCHC_SHIFT);
}

static int tps6105x_led_probe(struct platform_device *pdev)
{
        struct tps6105x *tps6105x = dev_get_platdata(&pdev->dev);
        struct tps6105x_platform_data *pdata = tps6105x->pdata;
        struct led_init_data init_data = { };
        struct tps6105x_priv *priv;
        int ret;

        /* This instance is not set for torch mode so bail out */
        if (pdata->mode != TPS6105X_MODE_TORCH) {
                dev_info(&pdev->dev,
                        "chip not in torch mode, exit probe");
                return -EINVAL;
        }

        priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
        if (!priv)
                return -ENOMEM;
        /* fwnode/devicetree is optional. NULL is allowed for priv->fwnode */
        priv->fwnode = device_get_next_child_node(pdev->dev.parent, NULL);
        ret = devm_add_action_or_reset(&pdev->dev, tps6105x_handle_put, priv);
        if (ret)
                return ret;
        priv->regmap = tps6105x->regmap;
        priv->cdev.brightness_set_blocking = tps6105x_brightness_set;
        priv->cdev.max_brightness = 7;
        init_data.devicename = "tps6105x";
        init_data.default_label = ":torch";
        init_data.fwnode = priv->fwnode;

        ret = regmap_update_bits(tps6105x->regmap, TPS6105X_REG_0,
                                 TPS6105X_REG0_MODE_MASK |
                                        TPS6105X_REG0_TORCHC_MASK,
                                 TPS6105X_REG0_MODE_TORCH <<
                                        TPS6105X_REG0_MODE_SHIFT);
        if (ret)
                return ret;

        return devm_led_classdev_register_ext(&pdev->dev, &priv->cdev,
                                              &init_data);
}

static struct platform_driver led_driver = {
        .probe = tps6105x_led_probe,
        .driver = {
                .name = "tps6105x-leds",
        },
};

module_platform_driver(led_driver);

MODULE_DESCRIPTION("TPS6105x LED driver");
MODULE_AUTHOR("Sven Van Asbroeck <TheSven73@gmail.com>");
MODULE_LICENSE("GPL v2");