// Copyright 2020 Espressif Systems (Shanghai) PTE LTD
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// from an initial work of : https://github.com/david-cermak/MadsHHLunds_esp_idf_eth_w6100

#include <string.h>
#include <stdlib.h>
#include <sys/cdefs.h>
#include "esp_log.h"
#include "esp_eth.h"
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "driver/gpio.h"
#include "w6100.h"

static const char *TAG = "w6100-phy";
#define PHY_CHECK(a, str, goto_tag, ...)                                          \
    do                                                                            \
    {                                                                             \
        if (!(a))                                                                 \
        {                                                                         \
            ESP_LOGE(TAG, "%s(%d): " str, __FUNCTION__, __LINE__, ##__VA_ARGS__); \
            goto goto_tag;                                                        \
        }                                                                         \
    } while (0)

/***************Vendor Specific Register***************/
/**
 * @brief PHYCFGR(PHY Configuration Register)
 *
 */
typedef union {
    struct {
        uint8_t link: 1;   /*!< Link status */
        uint8_t speed: 1;  /*!< Speed status */
        uint8_t duplex: 1; /*!< Duplex status */
        uint8_t opmode: 3; /*!< Operation mode */
        uint8_t reserved: 1;  /*!< */
        uint8_t cableOffBite: 1;  /* */
    };
    uint8_t val;
} phystatus_reg_t;


typedef union {
    struct {
        uint8_t opmode: 3; /*!< Operation mode */
        uint8_t reserved: 5;  /*!< Operation select */
        
    };
    uint8_t val;
} phycr0_reg_t;


typedef union {
    struct {
       uint8_t rst: 1; /*!< Operation mode */
       uint8_t reserved21: 2;  /**/
       uint8_t te: 1; /*!< Operation mode */
       uint8_t reserved4: 1;  /**/
       uint8_t pwdn: 1; /*!< Operation mode */
       uint8_t bit6: 1; /*always 1 */
       uint8_t reserved7: 1;  /*!< Operation select */
        
    };
    uint8_t val;
} phycr1_reg_t;

typedef struct {
    esp_eth_phy_t parent;
    esp_eth_mediator_t *eth;
    uint32_t addr;
    uint32_t reset_timeout_ms;
    uint32_t autonego_timeout_ms;
    eth_link_t link_status;
    int reset_gpio_num;
} phy_w5500_t;

typedef enum {
    W6100_OP_MODE_AUTO_NEGOCIATION = 0,
    W6100_OP_MODE_100BASE_TX_FDX = 4,
    W6100_OP_MODE_100BASE_TX_HDX = 5,
    W6100_OP_MODE_10BASE_TX_FDX = 6,
    W6100_OP_MODE_10BASE_TX_HDX = 7,
} phy_w5500_op_mode_e;

static esp_err_t w6100_update_link_duplex_speed(phy_w5500_t *w5500)
{
    esp_eth_mediator_t *eth = w5500->eth;
    eth_speed_t speed = ETH_SPEED_10M;
    eth_duplex_t duplex = ETH_DUPLEX_HALF;
    phystatus_reg_t phycfg;

    PHY_CHECK(eth->phy_reg_read(eth, w5500->addr, W6100_REG_PHYSR, (uint32_t *) & (phycfg.val)) == ESP_OK, "read PHYCFG failed", err);
  
    eth_link_t link = phycfg.link ? ETH_LINK_UP : ETH_LINK_DOWN;
    /* check if link status changed */
    if (w5500->link_status != link) {
        /* when link up, read negotiation result */
        if (link == ETH_LINK_UP) {
            if (phycfg.speed) {
                speed = ETH_SPEED_100M;
            } else {
                speed = ETH_SPEED_10M;
            }
            if (phycfg.duplex) {
                duplex = ETH_DUPLEX_FULL;
            } else {
                duplex = ETH_DUPLEX_HALF;
            }
       
            PHY_CHECK(eth->on_state_changed(eth, ETH_STATE_SPEED, (void *)speed) == ESP_OK,    "change speed failed", err);
            PHY_CHECK(eth->on_state_changed(eth, ETH_STATE_DUPLEX, (void *)duplex) == ESP_OK,  "change duplex failed", err);
        }

        PHY_CHECK(eth->on_state_changed(eth, ETH_STATE_LINK, (void *)link) == ESP_OK, "change link failed", err);
        w5500->link_status = link;
    }
    return ESP_OK;
err:
    return ESP_FAIL;
}

static esp_err_t w6100_set_mediator(esp_eth_phy_t *phy, esp_eth_mediator_t *eth)
{
    PHY_CHECK(eth, "can't set mediator to null", err);
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    w5500->eth = eth;
    return ESP_OK;
err:
    return ESP_ERR_INVALID_ARG;
}

static esp_err_t w6100_get_link(esp_eth_phy_t *phy)
{
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    /* Updata information about link, speed, duplex */
    PHY_CHECK(w6100_update_link_duplex_speed(w5500) == ESP_OK, "update link duplex speed failed", err);
    return ESP_OK;
err:
    return ESP_FAIL;
}

static esp_err_t w6100_reset(esp_eth_phy_t *phy)
{
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    w5500->link_status = ETH_LINK_DOWN;
    esp_eth_mediator_t *eth = w5500->eth;

    phycr0_reg_t phycr0;
    phycr0.opmode=0;        // Auto negotiation
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR0, phycr0.val) == ESP_OK, "write phycr0 failed", err);

    phycr1_reg_t phycr1;
    phycr1.rst=1;               // Phy Reset
    phycr1.bit6=1;
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycr1.val) == ESP_OK, "write phycr1 failed", err);
    return ESP_OK;
err:
    return ESP_FAIL;
}

static esp_err_t w6100_reset_hw(esp_eth_phy_t *phy)
{
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);

    // set reset_gpio_num to a negative value can skip hardware reset phy chip
    if (w5500->reset_gpio_num >= 0) {
        esp_rom_gpio_pad_select_gpio(w5500->reset_gpio_num); //! ajout
        gpio_set_direction(w5500->reset_gpio_num, GPIO_MODE_OUTPUT);
        gpio_set_level(w5500->reset_gpio_num, 0);
        vTaskDelay(pdMS_TO_TICKS(150));      //  esp_rom_delay_us(100); // insert min input assert time
        gpio_set_level(w5500->reset_gpio_num, 1);
        vTaskDelay(pdMS_TO_TICKS(100));
    }
    return ESP_OK;
}

static esp_err_t w6100_negotiate(esp_eth_phy_t *phy)
{
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    esp_eth_mediator_t *eth = w5500->eth;

    // Auto Negotiation......
    phycr0_reg_t phycr0;
    phycr0.opmode=0;

    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR0, phycr0.val) == ESP_OK, "write PHYCFG failed", err);

    /* Update information about link, speed, duplex */
    PHY_CHECK(w6100_update_link_duplex_speed(w5500) == ESP_OK, "update link duplex speed failed", err);

    return ESP_OK;
err:
    return ESP_FAIL;
}

static esp_err_t w6100_autonego_ctrl(esp_eth_phy_t *phy, eth_phy_autoneg_cmd_t cmd, bool *autonego_en_stat)
{
    esp_err_t ret = ESP_FAIL;
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    esp_eth_mediator_t *eth = w5500->eth;

    phystatus_reg_t phycfg;
    phycr0_reg_t phycfg0;
    phycr1_reg_t phycfg1;
    PHY_CHECK(eth->phy_reg_read(eth, w5500->addr, W6100_REG_PHYSR, (uint32_t *) &(phycfg.val)) == ESP_OK, "read PHYCFG failed", err);

    switch (cmd) {
    case ESP_ETH_PHY_AUTONEGO_RESTART:
        PHY_CHECK(phycfg.opmode == W6100_OP_MODE_AUTO_NEGOCIATION, "auto negotiation is disabled", err);
        /* in case any link status has changed, let's assume we're in link down status */
        w5500->link_status = ETH_LINK_DOWN;

        phycfg1.rst = 1; // PHY needs to be reset after configuring opsel and opmode
        PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycfg1.val) == ESP_OK, "write PHYCR1 failed", err);
        vTaskDelay(pdMS_TO_TICKS(50));
        phycfg1.rst = 0; // reset flag needs to be put back to 0
        PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycfg1.val) == ESP_OK, "write PHYCR1 failed", err);

        *autonego_en_stat = phycfg.opmode == W6100_OP_MODE_AUTO_NEGOCIATION;
        break;
    case ESP_ETH_PHY_AUTONEGO_DIS:
        /* W5500 autonegotiation cannot be separately disabled, only specific speed/duplex mode needs to be configured. Hence set the
        last used configuration */
        if (phycfg.duplex == W6100_ETH_DUPLEX_FULL) { // Full duplex
            if (phycfg.speed == W6100_ETH_SPEED_100M) { // 100 Mbps speed
                phycfg0.opmode = W6100_OP_MODE_100BASE_TX_FDX;
            } else {
                phycfg0.opmode = W6100_OP_MODE_10BASE_TX_FDX;
            }
        } else {
            if (phycfg.speed == W6100_ETH_SPEED_100M) { // 100 Mbps speed
                phycfg0.opmode = W6100_OP_MODE_100BASE_TX_HDX;
            } else {
                phycfg0.opmode = W6100_OP_MODE_10BASE_TX_HDX;
            }
        }
        PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR0, phycfg0.val) == ESP_OK, "write PHYCR1 failed", err);
        *autonego_en_stat = false;
        break;
    case ESP_ETH_PHY_AUTONEGO_EN:                         // PHY working mode configured by register
        phycfg0.opmode = W6100_OP_MODE_AUTO_NEGOCIATION; // all capable, auto-negotiation enabled
        PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR0, phycfg0.val) == ESP_OK, "write PHYCR0 failed", err);
        *autonego_en_stat = true;
        break;
    case ESP_ETH_PHY_AUTONEGO_G_STAT:
        *autonego_en_stat = phycfg.opmode == W6100_OP_MODE_AUTO_NEGOCIATION;
        break;
    default:
        return ESP_ERR_INVALID_ARG;
    }

    return ESP_OK;
err:
    return ret;
}

static esp_err_t w6100_pwrctl(esp_eth_phy_t *phy, bool enable)
{
    // power control is not supported for W5500 internal PHY
    return ESP_OK;
}

static esp_err_t w6100_set_addr(esp_eth_phy_t *phy, uint32_t addr)
{
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    w5500->addr = addr;
    return ESP_OK;
}

static esp_err_t w6100_get_addr(esp_eth_phy_t *phy, uint32_t *addr)
{
    PHY_CHECK(addr, "addr can't be null", err);
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    *addr = w5500->addr;
    return ESP_OK;
err:
    return ESP_ERR_INVALID_ARG;
}

static esp_err_t w6100_del(esp_eth_phy_t *phy)
{
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    free(w5500);
    return ESP_OK;
}

static esp_err_t w6100_advertise_pause_ability(esp_eth_phy_t *phy, uint32_t ability)
{
    // pause ability advertisement is not supported for W5500 internal PHY
    return ESP_OK;
}

static esp_err_t w5500_loopback(esp_eth_phy_t *phy, bool enable)
{
    // Loopback is not supported for W5500 internal PHY
    return ESP_ERR_NOT_SUPPORTED;
}

static esp_err_t w5500_set_speed(esp_eth_phy_t *phy, eth_speed_t speed)
{
    esp_err_t ret = ESP_OK;
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    esp_eth_mediator_t *eth = w5500->eth;

    if (w5500->link_status == ETH_LINK_UP) {
        /* Since the link is going to be reconfigured, consider it down for a while */
        w5500->link_status = ETH_LINK_DOWN;
        /* Indicate to upper stream apps the link is cosidered down */
        PHY_CHECK(eth->on_state_changed(eth, ETH_STATE_LINK, (void *)w5500->link_status) == ESP_OK, "change link failed", err);
    }

    phystatus_reg_t phycfg;
    phycr0_reg_t phycfg0;
    phycr1_reg_t phycfg1;
    PHY_CHECK(eth->phy_reg_read(eth, w5500->addr, W6100_REG_PHYSR, (uint32_t *) & (phycfg.val)) == ESP_OK, "read PHYSR failed", err);
    if (phycfg.duplex == W6100_ETH_DUPLEX_FULL) { // Full duplex
        // inverse 0 / 1
        if (speed == ETH_SPEED_100M) {
            phycfg0.opmode = W6100_OP_MODE_100BASE_TX_FDX;
        } else {
            phycfg0.opmode = W6100_OP_MODE_10BASE_TX_FDX;
        }
    } else {
        if (speed == ETH_SPEED_100M) {
            phycfg0.opmode = W6100_OP_MODE_100BASE_TX_HDX;
        } else {
            phycfg0.opmode = W6100_OP_MODE_10BASE_TX_HDX;
        }
    }
    phycfg1.rst = 0;  // PHY needs to be reset after configuring opsel and opmode
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR0, phycfg0.val) == ESP_OK, "write PHYCR0 failed", err);
    vTaskDelay(pdMS_TO_TICKS(50));
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycfg1.val) == ESP_OK, "write PHYCR1 failed", err);
    vTaskDelay(pdMS_TO_TICKS(50));
    phycfg1.rst = 1; // reset flag needs to be put back to 1
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycfg1.val) == ESP_OK, "write PHYCR1 failed", err);

err:
    return ret;
}

static esp_err_t w5500_set_duplex(esp_eth_phy_t *phy, eth_duplex_t duplex)
{
    esp_err_t ret = ESP_OK;
    phy_w5500_t *w5500 = __containerof(phy, phy_w5500_t, parent);
    esp_eth_mediator_t *eth = w5500->eth;

    if (w5500->link_status == ETH_LINK_UP) {
        /* Since the link is going to be reconfigured, consider it down for a while */
        w5500->link_status = ETH_LINK_DOWN;
        /* Indicate to upper stream apps the link is cosidered down */
        PHY_CHECK(eth->on_state_changed(eth, ETH_STATE_LINK, (void *)w5500->link_status) == ESP_OK, "change link failed", err);
    }

    phystatus_reg_t phycfg;
    phycr0_reg_t phycfg0;
    phycr1_reg_t phycfg1;
    PHY_CHECK(eth->phy_reg_read(eth, w5500->addr, W6100_REG_PHYSR, (uint32_t *) & (phycfg.val)) == ESP_OK, "read PHYSR failed", err);
    if (phycfg.speed == W6100_ETH_SPEED_100M) { // 100Mbps
        if (duplex == ETH_DUPLEX_FULL) {
            phycfg0.opmode = W6100_OP_MODE_100BASE_TX_FDX;
        } else {
            phycfg0.opmode = W6100_OP_MODE_100BASE_TX_HDX;
        }
    } else {
        if (duplex == ETH_DUPLEX_FULL) {
            phycfg0.opmode = W6100_OP_MODE_10BASE_TX_FDX;
        } else {
            phycfg0.opmode = W6100_OP_MODE_10BASE_TX_HDX;
        }
    }
    phycfg1.rst = 0;  // PHY needs to be reset after configuring opsel and opmode
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR0, phycfg0.val) == ESP_OK, "write PHYCR0 failed", err);
    vTaskDelay(pdMS_TO_TICKS(50));
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycfg1.val) == ESP_OK, "write PHYCR1 failed", err);
    vTaskDelay(pdMS_TO_TICKS(50));
    phycfg1.rst = 1; // reset flag needs to be put back to 1
    PHY_CHECK(eth->phy_reg_write(eth, w5500->addr, W6100_REG_PHYCR1, phycfg1.val) == ESP_OK, "write PHYCR1 failed", err);
err:
    return ret;
}

static esp_err_t w6100_init(esp_eth_phy_t *phy)
{
    /* Power on Ethernet PHY */
    PHY_CHECK(w6100_pwrctl(phy, true) == ESP_OK, "power control failed", err);
    /* Reset Ethernet PHY */
    PHY_CHECK(w6100_reset(phy) == ESP_OK, "reset failed", err);
    return ESP_OK;
err:
    return ESP_FAIL;
}

static esp_err_t w6100_deinit(esp_eth_phy_t *phy)
{
    /* Power off Ethernet PHY */
    PHY_CHECK(w6100_pwrctl(phy, false) == ESP_OK, "power control failed", err);
    return ESP_OK;
err:
    return ESP_FAIL;
}

esp_eth_phy_t *esp_eth_phy_new_w6100(const eth_phy_config_t *config)
{
    PHY_CHECK(config, "invalid arguments", err);
    phy_w5500_t *w5500 = calloc(1, sizeof(phy_w5500_t));
    PHY_CHECK(w5500, "no mem for PHY instance", err);
    w5500->addr = config->phy_addr;
    w5500->reset_timeout_ms = config->reset_timeout_ms;
    w5500->reset_gpio_num = config->reset_gpio_num;
    w5500->link_status = ETH_LINK_DOWN;
    w5500->autonego_timeout_ms = config->autonego_timeout_ms;
    w5500->parent.reset = w6100_reset;
    w5500->parent.reset_hw = w6100_reset_hw;
    w5500->parent.init = w6100_init;
    w5500->parent.deinit = w6100_deinit;
    w5500->parent.set_mediator = w6100_set_mediator;
    w5500->parent.autonego_ctrl = w6100_autonego_ctrl;
    w5500->parent.get_link = w6100_get_link;
    w5500->parent.pwrctl = w6100_pwrctl;
    w5500->parent.get_addr = w6100_get_addr;
    w5500->parent.set_addr = w6100_set_addr;
    w5500->parent.advertise_pause_ability = w6100_advertise_pause_ability;//    Mads 2021
    // ------------------------------------------
    w5500->parent.loopback = w5500_loopback;
    w5500->parent.set_speed = w5500_set_speed;
    w5500->parent.set_duplex = w5500_set_duplex;
    // ------------------------------------------
    w5500->parent.del = w6100_del;
    return &(w5500->parent);
err:
    return NULL;
}