#include <Arduino.h>
#include <SPI.h>
#include <Ethernet.h>

byte mac[] = { 0xDE, 0xAD, 0xBE, 0xEF, 0xFE, 0xED }; // Mac address
IPAddress server(208,94,118,151); // Public IRC server;
const char *nick = "MrArduino"; // Nickname on the IRC server
const char *chan = "#oc"; // Channel on the IRC server

// Local settings
IPAddress ip(192,168,1,104); // IP address of the Arduino
IPAddress dnss(255,255,255,0); // IP address of the DNS server 
IPAddress gw(192,168,1,1); // IP address of the gateway.

// Initialise variables
EthernetClient client;
char *msg;

void setup() {
 Serial.begin(9600);
 while (!Serial) {
  ; // wait for serial port to connect. Needed for Leonardo only
 }
 
 Serial.println("Booting Ethernet chip...");
 
 Ethernet.begin(mac, ip, dnss, gw);
 
 // Wait for the ethernet chip to initialise
 delay(750);
 
 Serial.print("Connecting to the server... ");
 if (client.connect(server, 5555)) {
  Serial.println("connected!");
 } 
 else {
  Serial.println("connection failed.");
 }
 
 if (client.connected()) {
  client.print("NICK ");
  client.print(nick);
  client.print("\r\n");
  client.print("USER ");
  client.print(nick); client.print(" ");
  client.print(nick); client.print(" ");
  client.print(nick); client.print(" ");
  client.print(":AVRIRCCLI\r\n");
 }
 
 msg = (char *)malloc(256);
 
 // Turn light off.
 analogWrite(A0, 0);
}

void startup() {
 client.print("JOIN #oc\r\n");
 client.print("PRIVMSG #oc :Hi, my name is ");
 #if defined(__AVR_ATmega328P__)
 client.print("ATmega328P");
 #else
 client.print("Atmel chip");
 #endif  
 client.print("\r\n");
}

void privmsg(char *channel, char *text) {
 client.print("PRIVMSG #");
 client.print(channel);
 client.print(" :");
 client.print(text);
 client.print("\r\n");
}

void parsemsg(char *nick, char *user, char *server, char *channel, char *text) {
 if (strstr(text, "ping") != NULL) privmsg(channel, "pong");
 if (strstr(text, "turn light off") != NULL) {
  analogWrite(A0, 0);
 }
 if (strstr(text, "turn light on") != NULL) {
  analogWrite(A0, 1023);
 }
 if (strstr(text, "show light status") != NULL) {
  if (analogRead(A0) == 0) privmsg(channel, "The light is off.");
  else privmsg(channel, "The light is on.");
 }
 if (strstr(text, "show pin status") != NULL) {
  client.print("PRIVMSG #");
  client.print(channel);
  client.print(" :Status: ");
  client.print("A0="); client.print(analogRead(A0) * (5.0 / 1023.0)); client.print("V,");
  client.print("A1="); client.print(analogRead(A1) * (5.0 / 1023.0)); client.print("V,");
  client.print("A2="); client.print(analogRead(A2) * (5.0 / 1023.0)); client.print("V,");
  client.print("A3="); client.print(analogRead(A3) * (5.0 / 1023.0)); client.print("V,");
  client.print("A4="); client.print(analogRead(A4) * (5.0 / 1023.0)); client.print("V,");
  client.print("A5="); client.print(analogRead(A5) * (5.0 / 1023.0)); client.print("V,");
  client.print("\r\n");
 }
}

void loop() {
 if (client.available()) {
  char c = client.read();
  if (c != (char)10) { // character 10 = \n
   size_t cur_len = strlen(msg);
   if(cur_len < 254) {
    msg[cur_len] = c;
    msg[cur_len+1] = '\0';
   }
  }
  else {
   // This is a full command
   if (strstr(msg, "/MOTD") != NULL) {
    startup();
   } else if (strstr(msg, "PING") != NULL) {
    char *pch;
    pch = strtok(msg, " ");
    if ((pch != NULL) && (strcmp(pch, "PING") == 0)) {
     pch = strtok(NULL, " ");
     if (pch != NULL) {
      client.print("PONG ");
      client.print(pch);
      client.print("\r\n");
     }
    }
   } else if (strstr(msg, "PRIVMSG") != NULL) {
    char nick[32], user[33], server[64], channel[64], body[256];
    sscanf(msg, ":%31[^!]!%32[^@]@%63s PRIVMSG %63s :%255[^\n]", nick, user, server, channel, body);
    parsemsg(nick, user, server, channel, body);
   }
   msg[0] = '\0';
  }
  Serial.print(c);
 }
 
 // as long as there are bytes in the serial queue,
 // read them and send them out the socket if it's open:
 /*while (Serial.available() > 0) {
  char inChar = Serial.read();
  if (client.connected()) {
   client.print(inChar); 
  }
 }*/
 
 // if the server's disconnected, stop the client:
 if (!client.connected()) {
  Serial.println();
  Serial.println("disconnecting.");
  client.stop();
  while(true); // TODO: Retry
 }
}