/*
 *  © 2020 Gregor Baues. All rights reserved.
 *  
 *  This is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  It is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with CommandStation.  If not, see <https://www.gnu.org/licenses/>.
 */

#include <Arduino.h>
#include "NetworkDiag.h"
#include "EthernetSetup.h"

byte EthernetSetup::setup() 
{

    INFO(F("Initialize Ethernet with DHCP"));
    if (Ethernet.begin(mac) == 0)
    {
        WARN(F("Failed to configure Ethernet using DHCP ... Trying with fixed IP"));
        Ethernet.begin(mac, IPAddress(IP_ADDRESS)); // default ip address

        if (Ethernet.hardwareStatus() == EthernetNoHardware)
        {
            ERR(F("Ethernet shield was not found. Sorry, can't run without hardware. :("));
            return 0;
        };
        if (Ethernet.linkStatus() == LinkOFF)
        {
            ERR(F("Ethernet cable is not connected."));
            return 0;
        }
    }

    maxConnections = MAX_SOCK_NUM;

    if (Ethernet.hardwareStatus() == EthernetW5100)
    {
        INFO(F("W5100 Ethernet controller detected."));
        maxConnections = 4;  // Max supported officaly by the W5100 but i have been running over 8 as well. Perf has to be evaluated though comparing 4 vs. 8 connections
    }
    else if (Ethernet.hardwareStatus() == EthernetW5200)
    {
        INFO(F("W5200 Ethernet controller detected."));
        maxConnections = 8;
    }
    else if (Ethernet.hardwareStatus() == EthernetW5500)
    {
        INFO(F("W5500 Ethernet controller detected."));
        maxConnections = 8;
    }

   INFO(F("Network Protocol:      [%s]"), protocol ? "UDP" : "TCP");
    switch (protocol)
    {
        case UDPR:
        { 
            udp = new EthernetUDP();
            byte udpState = udp->begin(port);
            if (udpState) 
            {
                TRC(F("UDP status: %d"), udpState);
                maxConnections = 1;             // there is only one UDP object listening for incomming data
                connected = true;
            }
            else
            {
                ERR(F("UDP failed to start"));
                connected = false;
            }
            break;
        };
        case TCP:
        {
            server = new EthernetServer(port);
            server->begin();
            connected = true;
            break;
        };
        case MQTT:
        {
            // do the MQTT setup stuff ...
        };
        default:
        {
            ERR(F("\nUnkown Ethernet protocol; Setup failed"));
            connected = false;
            break;
        }
    }
    if (connected)
    {
        ip = Ethernet.localIP();
        INFO(F("Local IP address:      [%d.%d.%d.%d]"), ip[0], ip[1], ip[2], ip[3]);
        INFO(F("Listening on port:     [%d]"), port);
        dnsip = Ethernet.dnsServerIP();
        INFO(F("DNS server IP address: [%d.%d.%d.%d] "), dnsip[0], dnsip[1], dnsip[2], dnsip[3]);
        INFO(F("Number of connections: [%d]"), maxConnections);
    }
    return connected;
}

EthernetSetup::EthernetSetup() {}
EthernetSetup::EthernetSetup(uint16_t p, protocolType pt ) { port = p; protocol = pt; }
EthernetSetup::~EthernetSetup() {}