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

#include "battery.h"
#include "button.h"
#include "charge_ramp.h"
#include "charger.h"
#include "common.h"
#include "compile_time_macros.h"
#include "console.h"
#include "cros_board_info.h"
#include "gpio.h"
#include "gpio_signal.h"
#include "hooks.h"
#include "driver/als_tcs3400.h"
#include "driver/charger/isl9241.h"
#include "driver/retimer/bb_retimer.h"
#include "fw_config.h"
#include "hooks.h"
#include "keyboard_customization.h"
#include "keyboard_scan.h"
#include "lid_switch.h"
#include "power_button.h"
#include "power.h"
#include "registers.h"
#include "switch.h"
#include "throttle_ap.h"
#include "usbc_config.h"
#include "watchdog.h"

#include "gpio_list.h" /* Must come after other header files. */

/* Console output macros */
#define CPRINTF(format, args...) cprintf(CC_CHARGER, format, ##args)
#define CPRINTS(format, args...) cprints(CC_CHARGER, format, ##args)

/*
 * USBA card connect to chromebook the USB_3_CONNECTION
 * bit would be enable.
 * It will increase BBR power consumption, so clear
 * USB3_Connection bit in S0ix and enable when return S0.
 */
void set_bb_retimer_usb3_state(bool enable)
{
	mux_state_t mux_state = 0;

	for (int i = 0; i < CONFIG_USB_PD_PORT_MAX_COUNT; i++) {
		const struct usb_mux *mux = &usb_muxes[i];

		mux_state = usb_mux_get(i);

		if ((mux_state & USB_PD_MUX_USB_ENABLED)) {
			bb_retimer_set_usb3(mux, enable);
		}
	}
}

/* Called on AP S3 -> S0 transition */
static void board_chipset_resume(void)
{
	if (chipset_in_state(CHIPSET_STATE_ON))
		set_bb_retimer_usb3_state(true);
}
DECLARE_HOOK(HOOK_CHIPSET_RESUME, board_chipset_resume, HOOK_PRIO_DEFAULT);

/* Called on AP S0 -> S3 transition */
static void board_chipset_suspend(void)
{
	set_bb_retimer_usb3_state(false);
}
DECLARE_HOOK(HOOK_CHIPSET_SUSPEND, board_chipset_suspend, HOOK_PRIO_DEFAULT);

void board_set_charger_current_limit_deferred(void)
{
	int action;
	int rv;

	if (extpower_is_present() &&
	    (battery_get_disconnect_state() != BATTERY_NOT_DISCONNECTED))
		/* AC only or AC+DC but battery is disconnect */
		action = MASK_SET;
	else
		action = MASK_CLR;

	rv = i2c_update16(chg_chips[CHARGER_SOLO].i2c_port,
			  chg_chips[CHARGER_SOLO].i2c_addr_flags,
			  ISL9241_REG_CONTROL3,
			  ISL9241_CONTROL3_INPUT_CURRENT_LIMIT, action);

	if (rv)
		CPRINTF("Could not set charger input current limit! Error: %d\n",
			rv);
}

DECLARE_DEFERRED(board_set_charger_current_limit_deferred);
DECLARE_HOOK(HOOK_SECOND, board_set_charger_current_limit_deferred,
	     HOOK_PRIO_DEFAULT);

void battery_present_interrupt(enum gpio_signal signal)
{
	hook_call_deferred(&board_set_charger_current_limit_deferred_data, 0);
}

static uint32_t board_id;
static void configure_keyboard(void)
{
	uint32_t cbi_val;

	/* Board ID */
	if (cbi_get_board_version(&cbi_val) != EC_SUCCESS ||
	    cbi_val > UINT8_MAX)
		CPRINTS("CBI: Read Board ID failed");
	else
		board_id = cbi_val;

	CPRINTS("Read Board ID: %d", board_id);

	if (board_id == 0) {
		/* keyboard_col2_inverted on board id 0 */
		gpio_set_flags(GPIO_EC_KSO_04_INV, GPIO_ODR_HIGH);
		gpio_set_flags(GPIO_EC_KSO_05_INV, GPIO_ODR_HIGH);
		gpio_set_alternate_function(GPIO_PORT_1, (BIT(4) | BIT(5)),
					    GPIO_ALT_FUNC_DEFAULT);
	} else if (board_id == 1) {
		/* keyboard_col4_inverted on board id 1 */
		gpio_set_flags(GPIO_EC_KSO_02_INV, GPIO_ODR_HIGH);
		gpio_set_flags(GPIO_EC_KSO_05_INV, GPIO_ODR_HIGH);
		gpio_set_alternate_function(GPIO_PORT_1, (BIT(4) | BIT(7)),
					    GPIO_ALT_FUNC_DEFAULT);
	} else {
		/* keyboard_col5_inverted on board id 2 and later */
		gpio_set_flags(GPIO_EC_KSO_02_INV, GPIO_ODR_HIGH);
		gpio_set_flags(GPIO_EC_KSO_04_INV, GPIO_ODR_HIGH);
		gpio_set_alternate_function(GPIO_PORT_1, (BIT(5) | BIT(7)),
					    GPIO_ALT_FUNC_DEFAULT);
		key_typ.col_refresh = KEYBOARD_COL_ID2_REFRESH;
		key_typ.row_refresh = KEYBOARD_ROW_ID2_REFRESH;
	}

	board_id_keyboard_col_inverted((int)board_id);
}

void board_init(void)
{
	gpio_enable_interrupt(GPIO_EC_BATT_PRES_ODL);
	hook_call_deferred(&board_set_charger_current_limit_deferred_data, 0);
}
DECLARE_HOOK(HOOK_INIT, board_init, HOOK_PRIO_DEFAULT);

__override void board_pre_task_i2c_peripheral_init(void)
{
	/* Configure board specific keyboard */
	configure_keyboard();

	/* Workaround for b:238683420 with board id >= 2 */

#ifdef SECTION_IS_RO
	if (board_id >= 2) {
		udelay(500 * MSEC);
		watchdog_reload();
		CPRINTS("Add delay to check boot key");
	}
#endif
}

__override uint8_t board_keyboard_row_refresh(void)
{
	if (board_id < 2)
		return KEYBOARD_ROW_ID1_REFRESH;
	else
		return KEYBOARD_ROW_ID2_REFRESH;
}