// SPDX-License-Identifier: GPL-2.0-only OR MIT
/*
 * Apple SPMI NVMEM driver
 *
 * Copyright The Asahi Linux Contributors
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/nvmem-provider.h>
#include <linux/of.h>
#include <linux/spmi.h>
#include <linux/regmap.h>

static const struct regmap_config apple_spmi_regmap_config = {
	.reg_bits	= 16,
	.val_bits	= 8,
	.max_register	= 0xffff,
};

static int apple_spmi_nvmem_probe(struct spmi_device *sdev)
{
	struct regmap *regmap;
	struct nvmem_config nvmem_cfg = {
		.dev = &sdev->dev,
		.name = "spmi_nvmem",
		.id = NVMEM_DEVID_AUTO,
		.word_size = 1,
		.stride = 1,
		.size = 0xffff,
		.reg_read = (void *)regmap_bulk_read,
		.reg_write = (void *)regmap_bulk_write,
	};

	regmap = devm_regmap_init_spmi_ext(sdev, &apple_spmi_regmap_config);
	if (IS_ERR(regmap))
		return PTR_ERR(regmap);

	nvmem_cfg.priv = regmap;

	return PTR_ERR_OR_ZERO(devm_nvmem_register(&sdev->dev, &nvmem_cfg));
}

static const struct of_device_id apple_spmi_nvmem_id_table[] = {
	{ .compatible = "apple,spmi-nvmem" },
	{ },
};
MODULE_DEVICE_TABLE(of, apple_spmi_nvmem_id_table);

static struct spmi_driver apple_spmi_nvmem_driver = {
	.probe = apple_spmi_nvmem_probe,
	.driver = {
		.name = "apple-spmi-nvmem",
		.of_match_table	= apple_spmi_nvmem_id_table,
	},
};

module_spmi_driver(apple_spmi_nvmem_driver);

MODULE_LICENSE("Dual MIT/GPL");
MODULE_AUTHOR("Hector Martin <marcan@marcan.st>");
MODULE_DESCRIPTION("Apple SPMI NVMEM driver");