// SPDX-License-Identifier: GPL-2.0+

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

#define A500_EC_LED_DELAY_USEC  (100 * 1000)

enum {
        REG_RESET_LEDS = 0x40,
        REG_POWER_LED_ON = 0x42,
        REG_CHARGE_LED_ON = 0x43,
        REG_ANDROID_LEDS_OFF = 0x5a,
};

struct a500_led {
        struct led_classdev cdev;
        const struct reg_sequence *enable_seq;
        struct a500_led *other;
        struct regmap *rmap;
};

static const struct reg_sequence a500_ec_leds_reset_seq[] = {
        REG_SEQ(REG_RESET_LEDS, 0x0, A500_EC_LED_DELAY_USEC),
        REG_SEQ(REG_ANDROID_LEDS_OFF, 0x0, A500_EC_LED_DELAY_USEC),
};

static const struct reg_sequence a500_ec_white_led_enable_seq[] = {
        REG_SEQ(REG_POWER_LED_ON, 0x0, A500_EC_LED_DELAY_USEC),
};

static const struct reg_sequence a500_ec_orange_led_enable_seq[] = {
        REG_SEQ(REG_CHARGE_LED_ON, 0x0, A500_EC_LED_DELAY_USEC),
};

static int a500_ec_led_brightness_set(struct led_classdev *led_cdev,
                                      enum led_brightness value)
{
        struct a500_led *led = container_of(led_cdev, struct a500_led, cdev);
        struct reg_sequence control_seq[2];
        unsigned int num_regs = 1;

        if (value) {
                control_seq[0] = led->enable_seq[0];
        } else {
                /*
                 * There is no separate controls which can disable LEDs
                 * individually, there is only RESET_LEDS command that turns
                 * off both LEDs.
                 *
                 * RESET_LEDS turns off both LEDs, thus restore other LED if
                 * it's turned ON.
                 */
                if (led->other->cdev.brightness)
                        num_regs = 2;

                control_seq[0] = a500_ec_leds_reset_seq[0];
                control_seq[1] = led->other->enable_seq[0];
        }

        return regmap_multi_reg_write(led->rmap, control_seq, num_regs);
}

static int a500_ec_leds_probe(struct platform_device *pdev)
{
        struct a500_led *white_led, *orange_led;
        struct regmap *rmap;
        int err;

        rmap = dev_get_regmap(pdev->dev.parent, "KB930");
        if (!rmap)
                return -EINVAL;

        /* reset and turn off LEDs */
        regmap_multi_reg_write(rmap, a500_ec_leds_reset_seq, 2);

        white_led = devm_kzalloc(&pdev->dev, sizeof(*white_led), GFP_KERNEL);
        if (!white_led)
                return -ENOMEM;

        white_led->cdev.name = "power:white";
        white_led->cdev.brightness_set_blocking = a500_ec_led_brightness_set;
        white_led->cdev.flags = LED_CORE_SUSPENDRESUME;
        white_led->cdev.max_brightness = 1;
        white_led->enable_seq = a500_ec_white_led_enable_seq;
        white_led->rmap = rmap;

        orange_led = devm_kzalloc(&pdev->dev, sizeof(*orange_led), GFP_KERNEL);
        if (!orange_led)
                return -ENOMEM;

        orange_led->cdev.name = "power:orange";
        orange_led->cdev.brightness_set_blocking = a500_ec_led_brightness_set;
        orange_led->cdev.flags = LED_CORE_SUSPENDRESUME;
        orange_led->cdev.max_brightness = 1;
        orange_led->enable_seq = a500_ec_orange_led_enable_seq;
        orange_led->rmap = rmap;

        white_led->other = orange_led;
        orange_led->other = white_led;

        err = devm_led_classdev_register(&pdev->dev, &white_led->cdev);
        if (err) {
                dev_err(&pdev->dev, "failed to register white LED\n");
                return err;
        }

        err = devm_led_classdev_register(&pdev->dev, &orange_led->cdev);
        if (err) {
                dev_err(&pdev->dev, "failed to register orange LED\n");
                return err;
        }

        return 0;
}

static struct platform_driver a500_ec_leds_driver = {
        .driver = {
                .name = "acer-a500-iconia-leds",
        },
        .probe = a500_ec_leds_probe,
};
module_platform_driver(a500_ec_leds_driver);

MODULE_DESCRIPTION("LED driver for Acer Iconia Tab A500 Power Button");
MODULE_AUTHOR("Dmitry Osipenko <digetx@gmail.com>");
MODULE_ALIAS("platform:acer-a500-iconia-leds");
MODULE_LICENSE("GPL");