/***************************************************************************
                          SIMPLY POWERFUL TOOLKIT (SPTK)
                          CSmtpConnect.h  -  description
                             -------------------
    begin                : Jun 25 2003
    copyright            : (C) 2003-2006 by Alexey Parshin
    email                : alexeyp@gmail.com
 ***************************************************************************/

/***************************************************************************
   This library is free software; you can redistribute it and/or modify it 
   under the terms of the GNU Library General Public License as published by
   the Free Software Foundation; either version 2 of the License, or (at 
   your option) any later version.

   This library 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 Library
   General Public License for more details.

   You should have received a copy of the GNU Library General Public License
   along with this library; if not, write to the Free Software Foundation, 
   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.

   Please report all bugs and problems to "alexeyp@gmail.com"
 ***************************************************************************/

#include <stdio.h>
#include <sptk3/CBase64.h>
#include <sptk3/CSmtpConnect.h>
#include <sptk3/CException.h>

#include <sptk3/CFileLog.h>

using namespace std;
using namespace sptk;

CSmtpConnect::CSmtpConnect() {
   m_port = 25;
}

CSmtpConnect::~CSmtpConnect() {
   close();
}

#define RSP_BLOCK_SIZE 1024
int CSmtpConnect::getResponse(bool decode) {
   char     readBuffer[RSP_BLOCK_SIZE+1];
   string  longLine;
   bool     readCompleted = false;
   int      rc = 0;

   while (!readCompleted) {
      int len = readLine(readBuffer,RSP_BLOCK_SIZE);
      longLine = readBuffer;
      if (len == RSP_BLOCK_SIZE && readBuffer[RSP_BLOCK_SIZE]!='\n') {
         do {
            len = readLine(readBuffer,RSP_BLOCK_SIZE);
            longLine += readBuffer;
         } while(len == RSP_BLOCK_SIZE);
      }

      if (longLine[3] == ' ') {
         readCompleted = true;
         longLine[3] = 0;
         rc = atoi(longLine.c_str());
      }

      const char * text = longLine.c_str()+4;
      if (decode) {
         longLine = unmime(text);
         m_response.push_back(longLine);
      } else {
         m_response.push_back(text);
      }
   }
   return rc;
}

void CSmtpConnect::sendCommand(string cmd,bool encode) {
   if (!active())
      throw CException("Socket isn't open");
   if (encode)
      cmd = mime(cmd);
   cmd += "\n";
   write(cmd.c_str(),cmd.length());
}

int CSmtpConnect::command(string cmd,bool encodeCommand,bool decodeResponse) {
   m_response.clear();
   if (!active())
      throw CException("Socket isn't open");
   sendCommand(cmd,encodeCommand);
   return getResponse(decodeResponse);
}

string CSmtpConnect::mime(string s) {
   string result;
   CBuffer src;
   src.set(s.c_str(),s.length());
   CBase64::encode(result,src);
   return result;
}

string CSmtpConnect::unmime(string s) {
   CBuffer dest;
   CBase64::decode(dest,s);
   string result(dest.data(),dest.bytes());
   return result;
}

void CSmtpConnect::cmd_login(string user,string password) {
   close();
   open();
   m_response.clear();

   getResponse();

   if (command("HELO") > 251) 
      throw CException(m_response.asString("\n"));

   if (trim(user).length()) {
      int rc = command("AUTH LOGIN",false,true);
      if (rc > 432)
         throw CException(m_response.asString("\n"));

      rc = command(user,true,true);
      if (rc > 432)
         throw CException(m_response.asString("\n"));

      rc = command(password,true);
      if (rc > 432)
         throw CException(m_response.asString("\n"));
   }
}

void CSmtpConnect::cmd_send() {
   sendMessage();
}

void CSmtpConnect::cmd_quit() {
   command("QUIT");
   close();
}

string parseAddress(string fullAddress) {
   int p1 = fullAddress.find("<");
   int p2 = fullAddress.find(">");
   if (p1 < 0 || p2 < 0 || p2 < p1) 
      return fullAddress;
   return trim(fullAddress.substr(p1+1,p2-p1-1));
}

void CSmtpConnect::sendMessage() {
   int rc = command("MAIL FROM:<"+parseAddress(m_from)+">");
   if (rc > 251)
      throw CException("Can't send message:\n"+m_response.asString("\n"));

   string rcpts = m_to + ";" + m_cc + ";" + m_bcc;
   rcpts = replaceAll(rcpts,",",";");
   rcpts = replaceAll(rcpts," ",";");
   CStrings recepients(rcpts,";");
   unsigned cnt = recepients.size();
   for (unsigned i = 0; i < cnt; i++) {
      string address = trim(recepients[i]);
      if (address[0] == 0) continue;
      rc = command("RCPT TO:<"+parseAddress(recepients[i])+">");
      if (rc > 251)
         throw CException("Recepient "+recepients[i]+" is not accepted.\n"+m_response.asString("\n"));
   }

   mimeMessage(m_messageBuffer);
   if (command("DATA") != 354)
      throw CException("DATA command is not accepted.\n"+m_response.asString("\n"));

   sendCommand(m_messageBuffer.data());
   rc = command("\n.");
   if (rc > 251)
      throw CException("Message body is not accepted.\n"+m_response.asString("\n"));
}
