#include <avr/io.h>

  #include <avr/interrupt.h>

  #include <util/delay.h>
  #include <stdio.h>

  #include <string.h>
  #include <stdlib.h>
  #include <stdbool.h>

  #include "serial.h"
  #include "ethernet.h"
  #include "w5100.h"
  #include "socket.h"
  
  #define MAC_SIZE	6
  #define IPV4_SIZE	4

  #define PRESCALER 1024
  #define NB_TICK 6250
  #define DELAY 300
  

  bool isComMem = false; // Commande déjà enregistrée ou non?
  bool sleep = true; //Etat de l'interface : true = mode sommeil / false = mode eveillé

  uint8 comMem[2]; //Commande en mémoire

  char msg[30]={'\0'};

  void init_led(void)
  {
    DDRD = 0xFF;
    PORTD = 0x08;
  }

  void init_timer(void)
  {
          TCCR1B |= _BV(WGM12);
          TCCR1B |= _BV(CS12);  // Facteur de division 1024
          TCCR1B |= _BV(CS10);
          OCR1A  = NB_TICK;
          TIMSK1 |= _BV(OCIE1A);
  }

  ISR(TIMER1_COMPA_vect)
  {
    if(sleep == false)
      PORTD ^= 0x20;
  }
  
  void requestTCP(SOCKET s,uint8 rq [2])

  {

    
    if (!isComMem) //Aucune commande en mémoire
      {
        isComMem = true;
        comMem[0] = rq[0]; comMem[1] = rq [1];
      }
    

    uint8 x = rq [0] & 0xE0; //Récupération de la commande
    x = x >> 5;
    uint16 y = rq[0] & 0x1F; //Récupération du pourcentage
    y = y << 8;
    y += rq[1];

    switch(x)

      {

      case 0: // GETSTATUS
        if (sleep==false)
  	{
  	  strcpy(msg,"\x20\x01");
  	}
        else
  	{
  	  strcpy(msg,"\x20\x00");				    
  	}
        send(s,msg,sizeof(char)*strlen(msg)); // REPONSE JE DORS OU NON
        disconnect(s);
        comMem[0] = rq[0]; comMem[1] = rq [1]; // SAVE
        break;

      case 3:
        printf("Request ComMem\n");
        strcpy(msg,comMem);
        send(s,msg,sizeof(char)*strlen(msg));
        comMem[0] = rq[0]; comMem[1] = rq [1]; // SAVE
        break;
      }
  }
  
  void requestUDP(uint8_t addr[IPV4_SIZE],uint8 rq [2])
  {
    SOCKET s = 2; //Socket TCP pour réponse à la broadcast UDP
    if (!isComMem) //Aucune commande en mémoire
      {
        isComMem = true;
        comMem[0] = rq[0]; comMem[1] = rq [1];
      }
    
    uint8 x = rq [0] & 0xE0; //Récupération de la commande
    x = x >> 5;
    uint16 y = rq[0] & 0x1F; //Récupération du pourcentage
    y = y << 8;
    y += rq[1];
    float per = y*0.0122;
    
    switch(x)
      {
        
      case 0: // GETSTATUS ( Via UDP broadcast, on répond en TCP ! )
        if(socket(s,Sn_MR_TCP,2020,0) == 0)
  	{
  	  printf("SOCKET NON INITILIASEE \n");
  	}
        if (connect(s,addr,2020))
  	{
  	  _delay_ms(DELAY); //l'IT doit attendre un peu avant d'envoyer
  	  if (sleep==false)
  	    {
  	      strcpy(msg,"\x20\x01");
  	    }
  	  else
  	    {
  	      strcpy(msg,"\x20\x00");
  	    }
  	  send(s,msg,sizeof(char)*strlen(msg));
  	  disconnect(s);
  	}

        comMem[0] = rq[0]; comMem[1] = rq [1]; //save

        break;
  

  	 
      case 2: // SLEEP MODE ON/OFF
        if ((y == 0x0001) & (sleep == true))

  	{

  	  sleep = false;
  	  printf("MODE EVEILLE\n");
  	  PORTD = PORTD ^ 0x08;

  	}

        else if ((y == 0x0000) & (sleep == false))

  	{

  	  sleep = true;
  	  printf("MODE SOMMEIL\n");
  	  PORTD = 0x08;

  	}
        comMem[0] = rq[0]; comMem[1] = rq [1]; //save 

        break;
        

      case 5: // SETCOMMANDE
        if (sleep == false)
  	{
  	  OCR1A = NB_TICK - 53*(int)per;
  	  comMem[0] = rq[0]; comMem[1] = rq [1]; //save 
  	}

        break;
      }
  

  }
  

  
  int main(void)
  {
    /*variables*/
    SOCKET sUDP=0,sTCP=1;

    uint8_t mac[MAC_SIZE] = {0xA0,0xBD,0xCD,0xDD,0xED,0xFD}; // <- Adresse MAC de l'IT
    uint8_t ip[IPV4_SIZE] = {192,168,1,205}; // <- Adresse IPV4 de l'IT
    uint8_t gateway[4] = {192,168,1,1}; // <- Adresse passerelle 
    uint8_t mask[4] = {255,255,255,0}; // <- Masque de sous réseau
    uint8 buf[2]; // <- Sert à stocker les deux octets recus via TCP & UDP
    uint16 datasize; // <- Sert à stocker la longueur du datagramme
    uint8_t addr[IPV4_SIZE]; // <- Sert à stocker l'adresse IP de l'émetteur
    uint16_t port; // <- Sert à stocker le port utilisé par l'émetteur
    bool isTCP_received = false; // <- Msg TCP reçu ou non? Permet la réouverture de la socket TCP pour permettre la réécoute TCP
  

    /*init*/
    init_led(); // <- Initialise les LEDS 
    init_timer(); // <- Initialise le compteur pour la fréquence du clignotement LED
    init_printf(); // <- Initialise la communication série
    ethernet_init(mac,ip,gateway,mask); // <- Initialise le shield ethernet
    
    

    if (!socket(sUDP,Sn_MR_UDP,2020,0))
      {
        return -1;
      }
    if (!socket(sTCP,Sn_MR_TCP,2020,0))
      {
        return -1;
      }

    while(1)
      {

        socket(sTCP,Sn_MR_TCP,2020,0);
        listen(sTCP);
        isTCP_received=false;
        while(!isTCP_received)
  	{
  	  sei();
  	  if (IINCHIP_READ(Sn_SR(sTCP)) == SOCK_ESTABLISHED) // Quelqu'un s'est connecté en TCP
  	    {
  	      cli();
  	      while(!isTCP_received) // On boucle en attendant le message du client
  		{
  		  if(recv(sTCP,buf,2)>0)
  		  {
  		      printf("MESSAGE RECU VIA TCP %x %x \n",buf[0],buf[1]);
  		      requestTCP(sTCP,buf); // On traite le message du client
  		      isTCP_received = true;
  		  }
  		}
  	    }
  	  if((datasize=recvfrom(sUDP,buf,sizeof(buf),addr,&port)) == 8) // Quelqu'un a envoyé 2 octets via UDP
  	    {
  	      cli();
  	      printf("MESSAGE RECU VIA UDP %x %x \n",buf[0],buf[1]);
  	      requestUDP(addr,buf); // On traite le message du client
  	      isTCP_received = true;
  	    }
  	}

      }
    close(sTCP);
    close(sUDP);
    return 0;

  }