/*
 * SPDX-FileCopyrightText: 2015-2024 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <stdbool.h>
#include "soc/soc_caps.h"
#include "esp_private/sleep_console.h"
#include "esp_attr.h"

#if SOC_USB_SERIAL_JTAG_SUPPORTED
#include "hal/usb_serial_jtag_ll.h"

static sleep_console_usj_enable_state_t s_usj_state = {0};

void sleep_console_usj_pad_backup_and_disable(void)
{
    // This function can be called in sleep process only, and sleep process code
    // is in critical region and thread safe already, so to avoid build errors/warnings
    // declare __DECLARE_RCC_ATOMIC_ENV here.
    int __DECLARE_RCC_ATOMIC_ENV __attribute__ ((unused));

    s_usj_state.usj_clock_enabled = usb_serial_jtag_ll_module_is_enabled();
    if (!s_usj_state.usj_clock_enabled) {
        // Enable USJ clock and clear reset
        usb_serial_jtag_ll_enable_bus_clock(true);
        usb_serial_jtag_ll_reset_register();
    }
    s_usj_state.usj_pad_enabled = usb_serial_jtag_ll_phy_is_pad_enabled();
    usb_serial_jtag_ll_phy_enable_pad(false);
    // Disable USJ clock
    usb_serial_jtag_ll_enable_bus_clock(false);
}

void sleep_console_usj_pad_restore(void)
{
    // This function can be called in sleep process only, and sleep process code
    // is in critical region and thread safe already, so to avoid build errors/warnings
    // declare __DECLARE_RCC_ATOMIC_ENV here.
    int __DECLARE_RCC_ATOMIC_ENV __attribute__ ((unused));

    usb_serial_jtag_ll_enable_bus_clock(true);
    usb_serial_jtag_ll_phy_enable_pad(s_usj_state.usj_pad_enabled);
    if (!s_usj_state.usj_clock_enabled) {
        usb_serial_jtag_ll_enable_bus_clock(false);
    }
}
#endif