/* Copyright 2022 The Chromium OS Authors. All rights reserved.
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#ifndef __ZEPHYR_SHIM_TCPCI_USB_MUX_H
#define __ZEPHYR_SHIM_TCPCI_USB_MUX_H

#include "dt-bindings/usbc_mux.h"
#include "tcpm/ps8xxx_public.h"
#include "tcpm/tcpci.h"

#define TCPCI_TCPM_USB_MUX_COMPAT cros_ec_usbc_mux_tcpci
#define PS8XXX_USB_MUX_COMPAT parade_usbc_mux_ps8xxx

/**
 * Add I2C configuration and USB_MUX_FLAG_NOT_TCPC to enforce it when
 * mux_read()/mux_write() functions are used.
 */
#define USB_MUX_CONFIG_TCPCI_TCPM_WITH_I2C(mux_id, port_id, idx)              \
	{                                                                     \
		USB_MUX_COMMON_FIELDS_WITH_FLAGS(mux_id, port_id, idx,        \
						 USB_MUX_FLAG_NOT_TCPC,       \
						 USB_MUX_FLAG_NOT_TCPC),      \
			.driver = &tcpci_tcpm_usb_mux_driver,                 \
			.hpd_update =                                         \
				USB_MUX_CALLBACK_OR_NULL(mux_id, hpd_update), \
			.i2c_port = I2C_PORT(DT_PHANDLE(mux_id, port)),       \
			.i2c_addr_flags = DT_PROP(mux_id, i2c_addr_flags),    \
	}

/** Use I2C configuration from TCPC */
#define USB_MUX_CONFIG_TCPCI_TCPM_WO_I2C(mux_id, port_id, idx)                \
	{                                                                     \
		USB_MUX_COMMON_FIELDS(mux_id, port_id, idx),                  \
			.driver = &tcpci_tcpm_usb_mux_driver,                 \
			.hpd_update =                                         \
				USB_MUX_CALLBACK_OR_NULL(mux_id, hpd_update), \
	}

/** This macro will fail if only port or i2c_addr_flags property is present */
#define USB_MUX_CONFIG_TCPCI_TCPM(mux_id, port_id, idx)                     \
	COND_CODE_1(                                                        \
		UTIL_OR(DT_NODE_HAS_PROP(mux_id, port),                     \
			DT_NODE_HAS_PROP(mux_id, i2c_addr_flags)),          \
		(USB_MUX_CONFIG_TCPCI_TCPM_WITH_I2C(mux_id, port_id, idx)), \
		(USB_MUX_CONFIG_TCPCI_TCPM_WO_I2C(mux_id, port_id, idx)))

#endif /* __ZEPHYR_SHIM_TCPCI_USB_MUX_H */