// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2024 NXP.
 */

#include <linux/jiffies.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/rtc.h>
#include <linux/scmi_protocol.h>
#include <linux/scmi_imx_protocol.h>

struct scmi_imx_bbm {
	const struct scmi_imx_bbm_proto_ops *ops;
	struct rtc_device *rtc_dev;
	struct scmi_protocol_handle *ph;
	struct notifier_block nb;
};

static int scmi_imx_bbm_read_time(struct device *dev, struct rtc_time *tm)
{
	struct scmi_imx_bbm *bbnsm = dev_get_drvdata(dev);
	struct scmi_protocol_handle *ph = bbnsm->ph;
	u64 val;
	int ret;

	ret = bbnsm->ops->rtc_time_get(ph, 0, &val);
	if (ret)
		return ret;

	rtc_time64_to_tm(val, tm);

	return 0;
}

static int scmi_imx_bbm_set_time(struct device *dev, struct rtc_time *tm)
{
	struct scmi_imx_bbm *bbnsm = dev_get_drvdata(dev);
	struct scmi_protocol_handle *ph = bbnsm->ph;
	u64 val;

	val = rtc_tm_to_time64(tm);

	return bbnsm->ops->rtc_time_set(ph, 0, val);
}

static int scmi_imx_bbm_alarm_irq_enable(struct device *dev, unsigned int enable)
{
	struct scmi_imx_bbm *bbnsm = dev_get_drvdata(dev);
	struct scmi_protocol_handle *ph = bbnsm->ph;

	/* scmi_imx_bbm_set_alarm enables the irq, just handle disable here */
	if (!enable)
		return bbnsm->ops->rtc_alarm_set(ph, 0, false, 0);

	return 0;
}

static int scmi_imx_bbm_set_alarm(struct device *dev, struct rtc_wkalrm *alrm)
{
	struct scmi_imx_bbm *bbnsm = dev_get_drvdata(dev);
	struct scmi_protocol_handle *ph = bbnsm->ph;
	struct rtc_time *alrm_tm = &alrm->time;
	u64 val;

	val = rtc_tm_to_time64(alrm_tm);

	return bbnsm->ops->rtc_alarm_set(ph, 0, true, val);
}

static const struct rtc_class_ops smci_imx_bbm_rtc_ops = {
	.read_time = scmi_imx_bbm_read_time,
	.set_time = scmi_imx_bbm_set_time,
	.set_alarm = scmi_imx_bbm_set_alarm,
	.alarm_irq_enable = scmi_imx_bbm_alarm_irq_enable,
};

static int scmi_imx_bbm_rtc_notifier(struct notifier_block *nb, unsigned long event, void *data)
{
	struct scmi_imx_bbm *bbnsm = container_of(nb, struct scmi_imx_bbm, nb);
	struct scmi_imx_bbm_notif_report *r = data;

	if (r->is_rtc)
		rtc_update_irq(bbnsm->rtc_dev, 1, RTC_AF | RTC_IRQF);
	else
		pr_err("Unexpected bbm event: %s\n", __func__);

	return 0;
}

static int scmi_imx_bbm_rtc_init(struct scmi_device *sdev)
{
	const struct scmi_handle *handle = sdev->handle;
	struct device *dev = &sdev->dev;
	struct scmi_imx_bbm *bbnsm = dev_get_drvdata(dev);
	int ret;

	bbnsm->rtc_dev = devm_rtc_allocate_device(dev);
	if (IS_ERR(bbnsm->rtc_dev))
		return PTR_ERR(bbnsm->rtc_dev);

	bbnsm->rtc_dev->ops = &smci_imx_bbm_rtc_ops;
	bbnsm->rtc_dev->range_max = U32_MAX;

	bbnsm->nb.notifier_call = &scmi_imx_bbm_rtc_notifier;
	ret = handle->notify_ops->devm_event_notifier_register(sdev, SCMI_PROTOCOL_IMX_BBM,
							       SCMI_EVENT_IMX_BBM_RTC,
							       NULL, &bbnsm->nb);
	if (ret)
		return ret;

	return devm_rtc_register_device(bbnsm->rtc_dev);
}

static int scmi_imx_bbm_rtc_probe(struct scmi_device *sdev)
{
	const struct scmi_handle *handle = sdev->handle;
	struct device *dev = &sdev->dev;
	struct scmi_protocol_handle *ph;
	struct scmi_imx_bbm *bbnsm;
	int ret;

	if (!handle)
		return -ENODEV;

	bbnsm = devm_kzalloc(dev, sizeof(*bbnsm), GFP_KERNEL);
	if (!bbnsm)
		return -ENOMEM;

	bbnsm->ops = handle->devm_protocol_get(sdev, SCMI_PROTOCOL_IMX_BBM, &ph);
	if (IS_ERR(bbnsm->ops))
		return PTR_ERR(bbnsm->ops);

	bbnsm->ph = ph;

	device_init_wakeup(dev, true);

	dev_set_drvdata(dev, bbnsm);

	ret = scmi_imx_bbm_rtc_init(sdev);
	if (ret)
		device_init_wakeup(dev, false);

	return ret;
}

static const struct scmi_device_id scmi_id_table[] = {
	{ SCMI_PROTOCOL_IMX_BBM, "imx-bbm-rtc" },
	{ },
};
MODULE_DEVICE_TABLE(scmi, scmi_id_table);

static struct scmi_driver scmi_imx_bbm_rtc_driver = {
	.name = "scmi-imx-bbm-rtc",
	.probe = scmi_imx_bbm_rtc_probe,
	.id_table = scmi_id_table,
};
module_scmi_driver(scmi_imx_bbm_rtc_driver);

MODULE_AUTHOR("Peng Fan <peng.fan@nxp.com>");
MODULE_DESCRIPTION("IMX SM BBM RTC driver");
MODULE_LICENSE("GPL");