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

#include <linux/gpio/consumer.h>
#include <linux/module.h>
#include <linux/regulator/consumer.h>
#include <sound/soc.h>

struct aw8738_priv {
        struct gpio_desc *gpiod_mode;
        unsigned int mode;
};

static int aw8738_drv_event(struct snd_soc_dapm_widget *w,
                            struct snd_kcontrol *kcontrol, int event)
{
        struct snd_soc_component *c = snd_soc_dapm_to_component(w->dapm);
        struct aw8738_priv *aw = snd_soc_component_get_drvdata(c);
        int i;

        switch (event) {
        case SND_SOC_DAPM_POST_PMU:
                for (i = 0; i < aw->mode; i++) {
                        gpiod_set_value_cansleep(aw->gpiod_mode, 0);
                        udelay(2);
                        gpiod_set_value_cansleep(aw->gpiod_mode, 1);
                        udelay(2);
                }
                msleep(40);
                break;
        case SND_SOC_DAPM_PRE_PMD:
                gpiod_set_value_cansleep(aw->gpiod_mode, 0);
                usleep_range(1000, 2000);
                break;
        default:
                WARN(1, "Unexpected event");
                return -EINVAL;
        }

        return 0;
}

static const struct snd_soc_dapm_widget aw8738_dapm_widgets[] = {
        SND_SOC_DAPM_INPUT("IN"),
        SND_SOC_DAPM_OUT_DRV_E("DRV", SND_SOC_NOPM, 0, 0, NULL, 0, aw8738_drv_event,
                               SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_PRE_PMD),
        SND_SOC_DAPM_OUTPUT("OUT"),
};

static const struct snd_soc_dapm_route aw8738_dapm_routes[] = {
        { "DRV", NULL, "IN" },
        { "OUT", NULL, "DRV" },
};

static const struct snd_soc_component_driver aw8738_component_driver = {
        .dapm_widgets = aw8738_dapm_widgets,
        .num_dapm_widgets = ARRAY_SIZE(aw8738_dapm_widgets),
        .dapm_routes = aw8738_dapm_routes,
        .num_dapm_routes = ARRAY_SIZE(aw8738_dapm_routes),
};

static int aw8738_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct aw8738_priv *aw;
        int ret;

        aw = devm_kzalloc(dev, sizeof(*aw), GFP_KERNEL);
        if (!aw)
                return -ENOMEM;
        platform_set_drvdata(pdev, aw);

        aw->gpiod_mode = devm_gpiod_get(dev, "mode", GPIOD_OUT_LOW);
        if (IS_ERR(aw->gpiod_mode))
                return dev_err_probe(dev, PTR_ERR(aw->gpiod_mode),
                                     "Failed to get 'mode' gpio");

        ret = device_property_read_u32(dev, "awinic,mode", &aw->mode);
        if (ret)
                return -EINVAL;

        return devm_snd_soc_register_component(&pdev->dev,
                                               &aw8738_component_driver,
                                               NULL, 0);
}

#ifdef CONFIG_OF
static const struct of_device_id aw8738_of_match[] = {
        { .compatible = "awinic,aw8738" },
        { }
};
MODULE_DEVICE_TABLE(of, aw8738_of_match);
#endif

static struct platform_driver aw8738_driver = {
        .probe  = aw8738_probe,
        .driver = {
                .name = "aw8738",
                .of_match_table = of_match_ptr(aw8738_of_match),
        },
};
module_platform_driver(aw8738_driver);

MODULE_DESCRIPTION("Awinic AW8738 Amplifier Driver");
MODULE_LICENSE("GPL v2");