/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Definitions for the Linux I2C OF component prober
 *
 * Copyright (C) 2024 Google LLC
 */

#ifndef _LINUX_I2C_OF_PROBER_H
#define _LINUX_I2C_OF_PROBER_H

#include <linux/kconfig.h>
#include <linux/types.h>

struct device;
struct device_node;

/**
 * struct i2c_of_probe_ops - I2C OF component prober callbacks
 *
 * A set of callbacks to be used by i2c_of_probe_component().
 *
 * All callbacks are optional. Callbacks are called only once per run, and are
 * used in the order they are defined in this structure.
 *
 * All callbacks that have return values shall return %0 on success,
 * or a negative error number on failure.
 *
 * The @dev parameter passed to the callbacks is the same as @dev passed to
 * i2c_of_probe_component(). It should only be used for dev_printk() calls
 * and nothing else, especially not managed device resource (devres) APIs.
 */
struct i2c_of_probe_ops {
 /**
  * @enable: Retrieve and enable resources so that the components respond to probes.
  *
  * It is OK for this callback to return -EPROBE_DEFER since the intended use includes
  * retrieving resources and enables them. Resources should be reverted to their initial
  * state and released before returning if this fails.
 */
 int (*enable)(struct device *dev, struct device_node *bus_node, void *data);

 /**
  * @cleanup_early: Release exclusive resources prior to calling probe() on a
  *     detected component.
  *
  * Only called if a matching component is actually found. If none are found,
  * resources that would have been released in this callback should be released in
  * @free_resourcs_late instead.
 */
 void (*cleanup_early)(struct device *dev, void *data);

 /**
  * @cleanup: Opposite of @enable to balance refcounts and free resources after probing.
  *
  * Should check if resources were already freed by @cleanup_early.
 */
 void (*cleanup)(struct device *dev, void *data);
};

/**
 * struct i2c_of_probe_cfg - I2C OF component prober configuration
 * @ops: Callbacks for the prober to use.
 * @type: A string to match the device node name prefix to probe for.
 */
struct i2c_of_probe_cfg {
 const struct i2c_of_probe_ops *ops;
 const char *type;
};

#if IS_ENABLED(CONFIG_OF_DYNAMIC)

int i2c_of_probe_component(struct device *dev, const struct i2c_of_probe_cfg *cfg, void *ctx);

/**
 * DOC: I2C OF component prober simple helpers
 *
 * Components such as trackpads are commonly connected to a devices baseboard
 * with a 6-pin ribbon cable. That gives at most one voltage supply and one
 * GPIO (commonly a "enable" or "reset" line) besides the I2C bus, interrupt
 * pin, and common ground. Touchscreens, while integrated into the display
 * panel's connection, typically have the same set of connections.
 *
 * A simple set of helpers are provided here for use with the I2C OF component
 * prober. This implementation targets such components, allowing for at most
 * one regulator supply.
 *
 * The following helpers are provided:
 * * i2c_of_probe_simple_enable()
 * * i2c_of_probe_simple_cleanup_early()
 * * i2c_of_probe_simple_cleanup()
 */

/**
 * struct i2c_of_probe_simple_opts - Options for simple I2C component prober callbacks
 * @res_node_compatible: Compatible string of device node to retrieve resources from.
 * @supply_name: Name of regulator supply.
 * @gpio_name: Name of GPIO. NULL if no GPIO line is used. Empty string ("") if GPIO
 *        line is unnamed.
 * @post_power_on_delay_ms: Delay after regulators are powered on. Passed to msleep().
 * @post_gpio_config_delay_ms: Delay after GPIO is configured. Passed to msleep().
 * @gpio_assert_to_enable: %true if GPIO should be asserted, i.e. set to logical high,
 *      to enable the component.
 *
 * This describes power sequences common for the class of components supported by the
 * simple component prober:
 * * @gpio_name is configured to the non-active setting according to @gpio_assert_to_enable.
 * * @supply_name regulator supply is enabled.
 * * Wait for @post_power_on_delay_ms to pass.
 * * @gpio_name is configured to the active setting according to @gpio_assert_to_enable.
 * * Wait for @post_gpio_config_delay_ms to pass.
 */
struct i2c_of_probe_simple_opts {
 const char *res_node_compatible;
 const char *supply_name;
 const char *gpio_name;
 unsigned int post_power_on_delay_ms;
 unsigned int post_gpio_config_delay_ms;
 bool gpio_assert_to_enable;
};

struct gpio_desc;
struct regulator;

struct i2c_of_probe_simple_ctx {
 /* public: provided by user before helpers are used. */
 const struct i2c_of_probe_simple_opts *opts;
 /* private: internal fields for helpers. */
 struct regulator *supply;
 struct gpio_desc *gpiod;
};

int i2c_of_probe_simple_enable(struct device *dev, struct device_node *bus_node, void *data);
void i2c_of_probe_simple_cleanup_early(struct device *dev, void *data);
void i2c_of_probe_simple_cleanup(struct device *dev, void *data);

extern struct i2c_of_probe_ops i2c_of_probe_simple_ops;

#endif /* IS_ENABLED(CONFIG_OF_DYNAMIC) */

#endif /* _LINUX_I2C_OF_PROBER_H */