#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 sleep = false; //Etat de l'interface : true = mode sommeil / false = mode eveillé
unsigned char comMem[2] = {'\0'}; //Commande en mémoire
unsigned char msg[2]; //Message qui sera envoyé via TCP
typedef enum type_msg {TCP,UDP}type_msg;
bool isTCP_received = false; // <- Msg TCP reçu ou non? Permet la réouverture de la socket TCP pour la réécoute TCP

void init_led(void)
{
  DDRD = 0xFF;
  if (sleep == true) // si IT dort
    {
      PORTD = 0x08;
    }
  else
    {
      PORTD = 0x00;
    }
}

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 request(SOCKET sock,uint8 rq[2],uint8 addr[IPV4_SIZE],type_msg type)
{
  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;

  if (type == UDP)
    {
      SOCKET s = 2; //Creation socket TCP pour réponse à la broadcast UDP
      switch(x)
	{
	  
	case 0: // GETSTATUS + connection ( Reçu via UDP broadcast, mais on répond en TCP ! )
	  printf("Reception UDP getstatus\n");
	  if(socket(s,Sn_MR_TCP,2020,0) == 0)
	    {
	      printf("SOCKET TCP NON INITILIASEE \n");
	      break;
	    }
	  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);
	    }	  
	  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;
	    }
	  break;

	case 3: // GETCOMMAND
	  if(socket(s,Sn_MR_TCP,2020,0) == 0)
	    {
	      printf("SOCKET TCP NON INITILIASEE \n");
	      break;
	    }
	  if (connect(s,addr,2020))
	    {
	      _delay_ms(DELAY); //l'IT doit attendre un peu avant d'envoyer
	      msg[0]=comMem[0];
	      msg[1]=comMem[1];	      
	      send(s,msg,sizeof(char)*strlen(msg));
	      disconnect(s);
	    }
	  break;
	  
	case 5: // SETCOMMANDE
	  if (sleep == false)
	    {
	      OCR1A = NB_TICK - 53*(int)per;
	      comMem[0] = rq[0]; comMem[1] = rq [1]; //save
	    }
	  break;
	  
	}
    }
  else if (type == TCP)
    {
      switch(x)
	{
	  
	case 0: // GETSTATUS
	  if (sleep==false)
	    {
	      strcpy(msg,"\x20\x01");
	    }
	  else
	    {
	      strcpy(msg,"\x20\x00");
	    }
	  send(sock,msg,sizeof(char)*strlen(msg));
	  disconnect(sock);
	  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;
	    }
	  disconnect(sock);
	  break;
	  
	case 3: // GETCOMMAND
	  printf("Request ComMem\n");
	  strcpy(msg,comMem);
	  send(sock,msg,sizeof(char)*strlen(msg));
	  disconnect(sock);
	  break;
	  
	case 5: // SETCOMMANDE
	  if (sleep == false)
	    {
	      OCR1A = NB_TICK - 53*(int)per;
	      comMem[0] = rq[0]; comMem[1] = rq [1]; //save
	    }
	  disconnect(sock);
	  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] = {172,26,145,205}; // <- Adresse IPV4 de l'IT
  uint8_t gateway[4] = {172,26,145,35}; // <- Adresse passerelle 
  uint8_t mask[4] = {255,255,255,0}; // <- Masque de sous réseau
  uint8_t buf[2]; // <- Sert à stocker les deux octets recus via TCP & UDP
  uint16_t 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
  type_msg type; // <- Enumeration : type de message recu {TCP,UDP}

  /* INITIALISATION */
    
  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;
    }
  
  comMem[0] = 0xA0; comMem[1] = 0x00; // Commande en mémoire (default)
  
  listen(sTCP);

  /* BOUCLE PRINCIPALE */

  while(1)
    {
      sei();	
      if (isTCP_received == true)
	{
	  socket(sTCP,Sn_MR_TCP,2020,0);
	  listen(sTCP);
	  isTCP_received=false;
	}

      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)
		{
		  type=TCP;
		  request(sTCP,buf,NULL,type); // 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();
	  type=UDP;
	  printf("MESSAGE RECU VIA UDP %x %x \n",buf[0],buf[1]);
	  request(sUDP,buf,addr,type); // On traite le message du client
	}
	
    }
  close(sTCP);
  close(sUDP);
  return 0;
}