/*

CIRCDDB - ircDDB client library in C++

Copyright (C) 2010-2011   Michael Dirska, DL1BFF (dl1bff@mdx.de)

This program 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 2 of the License, or
(at your option) any later version.

This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.

*/


#include "IRCProtocol.h"

#include <wx/regex.h>

#define CIRCDDB_VERSION	  "1.2.4"

IRCProtocol::IRCProtocol ( IRCApplication * app,
    const wxString& callsign, const wxString& password, const wxString& channel,
    const wxString& versionInfo )
{
  this -> password = password;
  this -> channel = channel;
  this -> app = app;

  this->versionInfo = wxT("CIRCDDB:");
  this->versionInfo.Append(wxT(CIRCDDB_VERSION));

  if (versionInfo.Len() > 0)
  {
    this->versionInfo.Append(wxT(" "));
    this->versionInfo.Append(versionInfo);
  }


  int hyphenPos = callsign.find(wxT('-'));

  if (hyphenPos == wxNOT_FOUND)
  {
    wxString n;

    n = callsign + wxT("-1");
    nicks.Add(n);
    n = callsign + wxT("-2");
    nicks.Add(n);
    n = callsign + wxT("-3");
    nicks.Add(n);
    n = callsign + wxT("-4");
    nicks.Add(n);
  }
  else
  {
    nicks.Add(callsign);
  }

  name = callsign;

  pingTimer = 60; // 30 seconds
  state = 0;
  timer = 0;


  chooseNewNick();
}

IRCProtocol::~IRCProtocol()
{
}

void IRCProtocol::chooseNewNick()
{
  int r = rand() % nicks.GetCount();

  currentNick = nicks[r];
}

void IRCProtocol::setNetworkReady( bool b )
{
  if (b == true)
  {
    if (state != 0)
    {
      wxLogError(wxT("IRCProtocol::setNetworkReady: unexpected state"));
    }

    state = 1;
    chooseNewNick();
  }
  else
  {
    state = 0;
  }
}


bool IRCProtocol::processQueues ( IRCMessageQueue * recvQ, IRCMessageQueue * sendQ )
{
  if (timer > 0)
  {
    timer --;
  }

  while (recvQ->messageAvailable())
  {
    IRCMessage * m = recvQ -> getMessage();

#if defined(DEBUG_IRC)
    wxString d = wxT("R [") + m->prefix + wxT("] [") + m->command + wxT("]");
    for (int i=0; i < m->numParams; i++)
    {
      d.Append(wxT(" [") + m->params[i] + wxT("]") );
    }
    d.Replace(wxT("%"), wxT("%%"), true);
    d.Replace(wxT("\\"), wxT("\\\\"), true);
    wxLogVerbose(d);
#endif

    if (m->command.IsSameAs(wxT("004")))
    {
      if (state == 4)
      {
	if (m->params.GetCount() > 1)
	{
	  wxRegEx serverNamePattern(wxT("^grp[1-9]s[1-9].ircDDB$"));

	  if (serverNamePattern.Matches( m->params[1] ))
	  {
	    app->setBestServer(wxT("s-") + m->params[1].Mid(0,6));
	  }
	}
	state = 5;  // next: JOIN
	app->setCurrentNick(currentNick);
      }
    }
    else if (m->command.IsSameAs(wxT("PING")))
    {
      IRCMessage * m2 = new IRCMessage();
      m2->command = wxT("PONG");
      if (m->params.GetCount() > 0)
      {
	m2->numParams = 1;
	m2->params.Add( m->params[0] );
      }
      sendQ -> putMessage(m2);
    }
    else if (m->command.IsSameAs(wxT("JOIN")))
    {
      if ((m->numParams >= 1) && m->params[0].IsSameAs(channel))
      {
	if (m->getPrefixNick().IsSameAs(currentNick) && (state == 6))
	{
	  if (debugChannel.Len() > 0)
	  {
	    state = 7;  // next: join debug_channel
	  }
	  else
	  {
	    state = 10; // next: WHO *
	  }
	}
	else if (app != NULL)
	{
		app->userJoin( m->getPrefixNick(), m->getPrefixName(), m->getPrefixHost());
	}
      }

      if ((m->numParams >= 1) && m->params[0].IsSameAs(debugChannel))
      {
	if (m->getPrefixNick().IsSameAs(currentNick) && (state == 8))
	{
	  state = 10; // next: WHO *
	}
      }
    }
    else if (m->command.IsSameAs(wxT("PONG")))
    {
      if (state == 12)
      {
	timer = pingTimer;
	state = 11;
      }
    }
    else if (m->command.IsSameAs(wxT("PART")))
    {
      if ((m->numParams >= 1) && m->params[0].IsSameAs(channel))
      {
	if (app != NULL)
	{
	  app->userLeave( m->getPrefixNick() );
	}
      }
    }
    else if (m->command.IsSameAs(wxT("KICK")))
    {
      if ((m->numParams >= 2) && m->params[0].IsSameAs(channel))
      {
	if (m->params[1].IsSameAs(currentNick))
	{
		// i was kicked!!
		delete m;
		return false;
	}
	else if (app != NULL)
	{
	  app->userLeave( m->params[1] );
	}
      }
    }
    else if (m->command.IsSameAs(wxT("QUIT")))
    {
      if (app != NULL)
      {
	app->userLeave( m->getPrefixNick() );
      }
    }
    else if (m->command.IsSameAs(wxT("MODE")))
    {
      if ((m->numParams >= 3) && m->params[0].IsSameAs(channel))
      {
	if (app != NULL)
	{
	  size_t i;
	  wxString mode = m->params[1];

	  for (i = 1; (i < mode.Len()) && ((size_t) m->numParams >= (i+2)); i++)
	  {
	    if ( mode[i] == wxT('o') )
	    {
	      if ( mode[0] == wxT('+') )
	      {
		      app->userChanOp(m->params[i+1], true);
	      }
	      else if ( mode[0] == wxT('-') )
	      {
		      app->userChanOp(m->params[i+1], false);
	      }
	    }
	  } // for
	}
      }
    }
    else if (m->command.IsSameAs(wxT("PRIVMSG")))
    {
      if ((m->numParams == 2) && (app != NULL))
      {
	if (m->params[0].IsSameAs(channel))
	{
	  app->msgChannel(m);
	}
	else if (m->params[0].IsSameAs(currentNick))
	{
	  app->msgQuery(m);
	}
      }
    }
    else if (m->command.IsSameAs(wxT("352")))  // WHO list
    {
      if ((m->numParams >= 7) && m->params[0].IsSameAs(currentNick)
	      && m->params[1].IsSameAs(channel))
      {
	if (app != NULL)
	{
	  app->userJoin( m->params[5], m->params[2], m->params[3]);
	  app->userChanOp ( m->params[5], m->params[6].IsSameAs(wxT("H@")));
	}
      }
    }
    else if (m->command.IsSameAs(wxT("433")))  // nick collision
    {
      if (state == 2)
      {
	state = 3;  // nick collision, choose new nick
	timer = 10; // wait 5 seconds..
      }
    }
    else if (m->command.IsSameAs(wxT("332")) ||
                  m->command.IsSameAs(wxT("TOPIC")))  // topic
    {
      if ((m->numParams == 2) && (app != NULL) &&
	      m->params[0].IsSameAs(channel) )
      {
	      app->setTopic(m->params[1]);
      }
    }

    delete m;
  }

  IRCMessage * m;

  switch (state)
  {
  case 1:
    m = new IRCMessage();
    m->command = wxT("PASS");
    m->numParams = 1;
    m->params.Add(password);
    sendQ->putMessage(m);

    m = new IRCMessage();
    m->command = wxT("NICK");
    m->numParams = 1;
    m->params.Add(currentNick);
    sendQ->putMessage(m);

    timer = 10;  // wait for possible nick collision message
    state = 2;
    break;

  case 2:
    if (timer == 0)
    {
      m = new IRCMessage();
      m->command = wxT("USER");
      m->numParams = 4;
      m->params.Add(name);
      m->params.Add(wxT("0"));
      m->params.Add(wxT("*"));
      m->params.Add(versionInfo);
      sendQ->putMessage(m);

      timer = 30;
      state = 4; // wait for login message
    }
    break;

  case 3:
    if (timer == 0)
    {
      chooseNewNick();
      m = new IRCMessage();
      m->command = wxT("NICK");
      m->numParams = 1;
      m->params.Add(currentNick);
      sendQ->putMessage(m);

      timer = 10;  // wait for possible nick collision message
      state = 2;
    }
    break;

  case 4:
    if (timer == 0)
    {
      // no login message received -> disconnect
      return false;
    }
    break;

  case 5:
    m = new IRCMessage();
    m->command = wxT("JOIN");
    m->numParams = 1;
    m->params.Add(channel);
    sendQ->putMessage(m);

    timer = 30;
    state = 6; // wait for join message
    break;

  case 6:
    if (timer == 0)
    {
      // no join message received -> disconnect
      return false;
    }
    break;

  case 7:
    if (debugChannel.Len() == 0)
    {
      return false; // this state cannot be processed if there is no debug_channel
    }

    m = new IRCMessage();
    m->command = wxT("JOIN");
    m->numParams = 1;
    m->params.Add(debugChannel);
    sendQ->putMessage(m);

    timer = 30;
    state = 8; // wait for join message
    break;

  case 8:
    if (timer == 0)
    {
      // no join message received -> disconnect
      return false;
    }
    break;

  case 10:
    m = new IRCMessage();
    m->command = wxT("WHO");
    m->numParams = 2;
    m->params.Add(channel);
    m->params.Add(wxT("*"));
    sendQ->putMessage(m);

    timer = pingTimer;
    state = 11; // wait for timer and then send ping

    if (app != NULL)
    {
      app->setSendQ(sendQ);  // this switches the application on
    }
    break;

  case 11:
    if (timer == 0)
    {
      m = new IRCMessage();
      m->command = wxT("PING");
      m->numParams = 1;
      m->params.Add(currentNick);
      sendQ->putMessage(m);

      timer = pingTimer;
      state = 12; // wait for pong
    }
    break;

  case 12:
    if (timer == 0)
    {
      // no pong message received -> disconnect
      return false;
    }
    break;
  }

  return true;
}