// UNSTABLE IRC BOT
// TODO: Integrate with Messenger.h for stability
#include <Ethernet2.h>
#include <WString.h>
 
#define STRING_BUFFER_SIZE 128
String buffer = String(STRING_BUFFER_SIZE);
 
// string variable to hold server name detection
String hostString = String(50);
 
byte mac[] = { 0xDE, 0xAD, 0xBE, 0xEF, 0xFE, 0xED };
 
// CHANGE ME
byte ip[] = { 1, 2, 3, 4 };  // arduino IP: 1.2.3.4
byte gateway[] = { 1, 2, 3, 4 };  // your router IP: 1.2.3.4
byte subnet[] = { 255, 255, 255, 0 };  // your network mask
 
// irc.freenode.net
byte ircServer[] = { 74, 208, 174, 239 };  // IP of freenode.net
 
int responseWait = 100;         // default: 1000
int joinWait = 100;             // default: 1000
int pingCheckThrottle = 10000;  // check for PING event every 10 seconds once connected
int reconnectWait = 5000;
 
// CHANGE ME
char ircNick[20] = "DuinoFace";
char channel[20] = "#freebsToTheDee";
boolean isLoggedIn = false;
 
Client client(ircServer, 6667);
 
// turn on when debugging
boolean isDebugOn = true;
 
// logged in (in channel) status light
int ledPin = 2;
 
void setup()
{
  //router time?
  delay(500);
  Serial.begin(9600);	// opens serial port, sets data rate to 9600 bps
 
  // Say hello:
  debugNoLine("String Library version: ");
  debug(buffer.version());
 
  Ethernet.begin(mac, ip, gateway, subnet);
  delay(responseWait);
 
  debug("client.connect()");
  if (!client.connect()) {
    debug("client.connect() failed.");
  }
 
  // status light
  pinMode(ledPin, OUTPUT);
 
}
 
 
void loop()
{
  if (client.available() && client.connected()) {
    // parse the incoming buffer
    readIRCRequest(client);
  }
 
  if (!client.connected()) {
    debug("Disconnected");
    client.stop();
    delay(reconnectWait);
 
    debug("Reconnecting...");
    isLoggedIn = false;
    setStatusLight();
    client.connect();
  }
}
 
 
int readIRCRequest(Client client) {
  char inChar = client.read();
 
  if (inChar != '\n') {
    buffer.append(inChar);
  } 
  else {
    debugNoLine("Buffer: ");
    debug(buffer);
 
    if (!isLoggedIn) {
      if (buffer.contains("NOTICE * :*** No Ident response")) {
        login();
        buffer = "";
      } 
      else if (buffer.contains(":End of /MOTD command.")) {
        join();
      } 
      else if (buffer.contains(":End of /NAMES list.")) {
        say();
        isLoggedIn = true;
        setStatusLight();
      } 
    } 
    else {
      checkCommand();        
      checkPing();
      //delay(pingCheckThrottle);
    }
 
    // clear out buffer for next line read
    buffer = "";
 
  }
}
 
 
/* We need to send something that looks like this
 *   NICK yaytofu
 *   USER yaytofu 0 * yaytofu
 * Where yaytofu is our nickname
 */
 
void login() {
  String msg = String(50);
  msg = "NICK ";
  msg.append(ircNick);
  //if (debug) Serial.println(msg);
  debug(msg);
  client.println(msg);
  delay(responseWait); /* wait for a response */
 
  msg = "USER ";
  msg.append(ircNick);
  msg.append(" 0 * ");
  msg.append(ircNick);
  debug(msg);
  client.println(msg);
 
  delay(responseWait); /* wait for a response */
}
 
 
void join() {
  delay(joinWait);
  debug("JOIN...");
  client.println(strcat("JOIN ", channel));
  delay(responseWait);
}
 
 
void say() {
  //if (debug) Serial.println("Sending something to the channel.");
  debug("Sending something to channel.");
  client.println("PRIVMSG #freebsToTheDee :Arduino with Ethernet Shield is here.");
}
 
void say(char *msg) {
  debug("Sending something to channel.");
  client.print("PRIVMSG #freebsToTheDee :");
  client.println(msg);
}
 
 
void checkPing() {
  if (buffer.startsWith("PING :")) {
    debug("GOT PING!");
 
    // parse and store hostname in string
    if (hostString.length() == 0) {
      hostString = buffer.substring(5, buffer.length()-1);
    }
 
    // respond to ping
    debug("PONG!");
    client.print("PONG ");
    client.println(hostString);
    setStatusLight();
 
  }
}
 
void checkCommand() {
  // :nick!~nick@static-1-13-5-1.domain.com PRIVMSG #freebsToTheDee :wut
  // state codes 0=DNS   1=PRIVMSG   2=channel  3=message
 
  if (buffer.startsWith(":") && buffer.contains("PRIVMSG")) {
 
    int state = 0;
    int spaces[3] = { 0, 0, 0};
 
    // split on spaces, TODO: state check
    char *bufferArray = buffer.getChars();
    for (int i=0; i < buffer.length(); i++) {
      char test = bufferArray[i];
      if (test == ' ') {
        debugNoLine("found space at:");
        debug(i);
        spaces[state] = i;
        state++;
      }
    }
 
 
    String dnsName = buffer.substring(0, spaces[0]);
    String privMsg = buffer.substring(spaces[0]+1, spaces[1]);
    String channel = buffer.substring(spaces[1]+1, spaces[2]);
    String cmd = buffer.substring(spaces[2]+2, buffer.length());
 
    debugNoLine("dnsName:");
    debug(dnsName);
    debugNoLine("privMsg:");
    debug(privMsg);
    debugNoLine("channel:");
    debug(channel);
    debugNoLine("cmd:");
    debug(cmd);
  }
}
 
void setStatusLight() {
  if (isLoggedIn) {
    debug("Setting status light to ON.");
    digitalWrite(ledPin, HIGH);
  } 
  else {
    debug("Setting status light to OFF.");
    digitalWrite(ledPin, LOW);
  }
}
 
// No carriage return
void debugNoLine(char *msg) {
  if (isDebugOn) Serial.print(msg);
}
 
// No carriage return overloaded
void debugNoLine(String *msg) {
  if (isDebugOn) Serial.print(*msg);
}
 
void debugNoLine(int *msg) {
  if (isDebugOn) Serial.print(*msg);
}
 
// Carriage return
void debug(char *msg) {
  if (isDebugOn) Serial.println(msg);
}
 
// Carriage return overloaded
void debug(String *msg) {
  if (isDebugOn) Serial.println(*msg);
}
 
void debug(int msg) {
  if (isDebugOn) Serial.println(msg);
}