/*
 * This file is part of the Micro Python project, http://micropython.org/
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2018 hathach for Adafruit Industries
 * Copyright (c) 2019 Lucian Copeland for Adafruit Industries
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */


#include "supervisor/usb.h"
#include "shared/readline/readline.h"

#include "py/mpconfig.h"

#include "common-hal/microcontroller/Pin.h"

STATIC void init_usb_vbus_sense(void) {

    #if (BOARD_NO_VBUS_SENSE)
    // Disable VBUS sensing
    #ifdef USB_OTG_GCCFG_VBDEN
    // Deactivate VBUS Sensing B
    USB_OTG_FS->GCCFG &= ~USB_OTG_GCCFG_VBDEN;

    #if (BOARD_NO_USB_OTG_ID_SENSE)
    USB_OTG_FS->GUSBCFG &= ~USB_OTG_GUSBCFG_FHMOD;
    USB_OTG_FS->GUSBCFG |= USB_OTG_GUSBCFG_FDMOD;
    #endif

    // B-peripheral session valid override enable
    USB_OTG_FS->GOTGCTL |= USB_OTG_GOTGCTL_BVALOEN;
    USB_OTG_FS->GOTGCTL |= USB_OTG_GOTGCTL_BVALOVAL;
    #else
    USB_OTG_FS->GCCFG |= USB_OTG_GCCFG_NOVBUSSENS;
    USB_OTG_FS->GCCFG &= ~USB_OTG_GCCFG_VBUSBSEN;
    USB_OTG_FS->GCCFG &= ~USB_OTG_GCCFG_VBUSASEN;
    #endif
    #else
    // Enable VBUS hardware sensing
    #ifdef USB_OTG_GCCFG_VBDEN
    USB_OTG_FS->GCCFG |= USB_OTG_GCCFG_VBDEN;
    #else
    USB_OTG_FS->GCCFG &= ~USB_OTG_GCCFG_NOVBUSSENS;
    USB_OTG_FS->GCCFG |= USB_OTG_GCCFG_VBUSBSEN;     // B Device sense
    #endif
    #endif
}

void init_usb_hardware(void) {

    /* Enable USB power on Pwrctrl CR2 register */
    #ifdef PWR_CR2_USV
    HAL_PWREx_EnableVddUSB();
    #endif

    // TODO: if future chips overload this with options, move to peripherals management.

    GPIO_InitTypeDef GPIO_InitStruct = {0};
    /**USB_OTG_FS GPIO Configuration
    PA10     ------> USB_OTG_FS_ID
    PA11     ------> USB_OTG_FS_DM
    PA12     ------> USB_OTG_FS_DP
    */
    __HAL_RCC_GPIOA_CLK_ENABLE();

    /* Configure DM DP Pins */
    GPIO_InitStruct.Pin = GPIO_PIN_11 | GPIO_PIN_12;
    GPIO_InitStruct.Speed = GPIO_SPEED_HIGH;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    #if CPY_STM32H7
    GPIO_InitStruct.Alternate = GPIO_AF10_OTG1_FS;
    #elif CPY_STM32F4 || CPY_STM32F7 || CPY_STM32L4
    GPIO_InitStruct.Alternate = GPIO_AF10_OTG_FS;
    #endif
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);
    never_reset_pin_number(0, 11);
    never_reset_pin_number(0, 12);
    claim_pin(0, 11);
    claim_pin(0, 12);

    /* Configure VBUS Pin */
    #if  !(BOARD_NO_VBUS_SENSE)
    GPIO_InitStruct.Pin = GPIO_PIN_9;
    GPIO_InitStruct.Mode = GPIO_MODE_INPUT;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);
    never_reset_pin_number(0, 9);
    claim_pin(0, 9);
    #endif

    /* This for ID line debug */
    #if !(BOARD_NO_USB_OTG_ID_SENSE)
    GPIO_InitStruct.Pin = GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_OD;
    GPIO_InitStruct.Pull = GPIO_PULLUP;
    GPIO_InitStruct.Speed = GPIO_SPEED_HIGH;
    #if CPY_STM32H7
    GPIO_InitStruct.Alternate = GPIO_AF10_OTG1_FS;
    #elif CPY_STM32F4 || CPY_STM32F7 || CPY_STM32L4
    GPIO_InitStruct.Alternate = GPIO_AF10_OTG_FS;
    #endif
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);
    never_reset_pin_number(0, 10);
    claim_pin(0, 10);
    #endif

    #ifdef STM32F412Zx
    /* Configure POWER_SWITCH IO pin (F412 ONLY)*/
    GPIO_InitStruct.Pin = GPIO_PIN_8;
    GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_OD;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    HAL_GPIO_Init(GPIOG, &GPIO_InitStruct);
    never_reset_pin_number(0, 8);
    claim_pin(0, 8);
    #endif

    #if CPY_STM32H7
    HAL_PWREx_EnableUSBVoltageDetector();
    __HAL_RCC_USB2_OTG_FS_CLK_ENABLE();
    #else
    /* Peripheral clock enable */
    __HAL_RCC_USB_OTG_FS_CLK_ENABLE();
    #endif


    init_usb_vbus_sense();
}

void OTG_FS_IRQHandler(void) {
    usb_irq_handler(0);
}
