// 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");