/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
 *            transactionhandler.cc
 *
 *  Fri Dec 18 10:00:50 CET 2009
 *  Copyright 2009 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of Pracro.
 *
 *  Pracro 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.
 *
 *  Pracro 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 Pracro; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "transactionhandler.h"

#include "macroparser.h"
#include "resumeparser.h"
#include "templateparser.h"
#include "configuration.h"
#include "luaquerymapper.h"
#include "queryhandlerpentominos.h"
#include "queryhandlerpracro.h"
#include "xml_encode_decode.h"
#include "widgetgenerator.h"

static std::string error_box(std::string message)
{
  std::string errorbox =
    "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
    "<pracro version=\"1.0\">\n"
    "  <error>" + message + "</error>\n"
    "</pracro>\n";
  return errorbox;
}

static std::string handleCommits(Transaction *transaction, Database &db, 
                                 JournalWriter &journalwriter, MacroList &macrolist,
                                 TemplateList &templatelist)
{
  std::string answer;

  Commits::iterator i = transaction->commits.begin();
  while(i != transaction->commits.end()) {
    Commit &commit = *i;
    
    MacroParser mp(macrolist.getLatestVersion(commit.macro));
    mp.parse();
    Macro *macro = mp.getMacro();
    
    std::string resume = resume_parser(macro->resume, commit);
    commit.fields["journal.resume"] = resume;
    db.commitTransaction(transaction->user, transaction->cpr, *macro, commit.fields);
    
    if(resume != "") {

      TemplateParser tp(templatelist.getLatestVersion(commit.templ));
      tp.parse();
      Template *templ = tp.getTemplate();

      journalwriter.addEntry(*transaction, commit, resume, templ);
    }
    
    i++;
  }

  return answer;
}


static std::string handleRequest(Transaction *transaction,
                                 TCPSocket &pentominos_socket,
                                 Database &db,
                                 MacroList &macrolist,
                                 TemplateList &templatelist)
{
  std::string answer;

  Requests::iterator i = transaction->requests.begin();
  while(i != transaction->requests.end()) {
    Request &request = *i;
    
    PRACRO_DEBUG(server, "Handling request - macro: %s, template: %s\n",
                 request.macro.c_str(), request.templ.c_str());
    
    // Read and parse the template file.
    TemplateParser tp(templatelist.getLatestVersion(request.templ));
    tp.parse();
    
    Template *templ = tp.getTemplate();
    
    answer += "  <template name=\"";
    answer += templ->attributes["name"];
    answer += "\" title=\"";
    answer += templ->attributes["title"];
    answer += "\">\n";
    
    bool foundmacro = false;
    
    // Generate the macro and return it to the client
    std::vector< Macro >::iterator mi2 = templ->macros.begin();
    while(mi2 != templ->macros.end()) {
      Macro &macro = (*mi2);

      if(macro.isHeader) {
        answer += "    <header caption=\"" + macro.attributes["caption"] + "\"/>\n";
        mi2++;
        continue;
      }

      bool completed = db.checkMacro(transaction->cpr,
                                     macro.attributes["name"],
                                     time(NULL)-Conf::db_max_ttl);

      answer += "    <macro uid=\"42\" completed=";
      if(completed) answer += "\"true\"";
      else answer += "\"false\"";

      std::map< std::string, std::string >::iterator ai = macro.attributes.begin();
      while(ai != macro.attributes.end()) {
        std::string name = ai->first;
        std::string value = ai->second;
        answer += " "+name+"=\"" + value + "\"";
        ai++;
      }

      if(macro.attributes["name"] == request.macro || 
         (macro.attributes.find("static") != macro.attributes.end() &&
          macro.attributes["static"] == "true")
         ) {
        foundmacro = true;

        MacroParser mp(macrolist.getLatestVersion(macro.attributes["name"]));
        mp.parse();
        Macro *m = mp.getMacro();
        answer += " caption=\"" + m->widgets.attributes["caption"] + "\"";
        answer += ">\n";

        LUAQueryMapper lqm;

        ////////////////////////
        std::vector< Query >::iterator qi = m->queries.begin();
        while(qi != m->queries.end()) {

          Query &query = *qi;
          std::string service = query.attributes["service"];

          if(service == "pentominos") {
            // Send the queries to Pentominos (if any)
            QueryHandlerPentominos qh(pentominos_socket, transaction->cpr);

            QueryResult queryresult = qh.exec(*qi);
            lqm.addQueryResult(queryresult);
          }

          if(service == "pracro") {
            // Send the queries to Pentominos (if any)
            QueryHandlerPracro qh(db, transaction->cpr);

            QueryResult queryresult = qh.exec(*qi);
            lqm.addQueryResult(queryresult);
          }

          qi++;
        }

        // Handle scripts
        if(m->scripts.size()) {
          answer += "      <scripts>\n";
            
          std::vector< Script >::iterator spi = m->scripts.begin();
          while(spi != m->scripts.end()) {
            answer += "        <script language=\"" + spi->attributes["language"] 
              + "\" name=\"" + spi->attributes["name"] + "\">\n";
            answer += xml_encode(spi->attributes["code"]);
            answer += "\n        </script>\n";
            spi++;
          }
          answer += "      </scripts>\n";
        }

        answer += widgetgenerator(transaction->cpr, *m, lqm, db);
      } else {
        // only find macro title
        MacroParser mp(macrolist.getLatestVersion(macro.attributes["name"]));
        mp.parse();
        Macro *m = mp.getMacro();
        answer += " caption=\"" + m->widgets.attributes["caption"] + "\"";
        answer += ">\n";

      }

      if(completed) {
        answer += "      <resume>";
        answer += db.getResume(transaction->cpr, macro, time(NULL) - Conf::db_max_ttl);
        answer += "</resume>\n";
      }

      answer += "    </macro>\n";
      mi2++;

    }
      
    if(foundmacro == false && request.macro != "")
      throw NotFoundException(request);

    answer += "  </template>\n";

    i++;
  }

  return answer;
}

std::string handleTransaction(Transaction *transaction,
                              TCPSocket &pentominos_socket,
                              Database &db,
                              JournalWriter &journalwriter,
                              MacroList &macrolist,
                              TemplateList &templatelist)
{
  std::string answer;
  answer += "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n";
  answer += "<pracro version=\"1.0\">\n";

  try {
    answer += handleCommits(transaction, db,
                            journalwriter, macrolist, templatelist);
  } catch( std::exception &e ) {
    PRACRO_ERR(server, "Commit error: %s\n", e.what());
    return error_box(xml_encode(e.what()));
  }

  try {
    answer += handleRequest(transaction, pentominos_socket, db, macrolist, templatelist);
  } catch( std::exception &e ) {
    PRACRO_ERR(server, "Request error: %s\n", e.what());
    return error_box(xml_encode(e.what()));
  }

  answer += "</pracro>\n";

  PRACRO_DEBUG(server, "Done handling transaction\n");
  PRACRO_DEBUG(serverxml, "%s\n", answer.c_str());
  return answer;
}

#ifdef TEST_TRANSACTIONHANDLER
//Additional dependency files
//deps:
//Required cflags (autoconf vars may be used)
//cflags: -I..
//Required link options (autoconf vars may be used)
//libs:
#include "test.h"

TEST_BEGIN;

// TODO: Put some testcode here (see test.h for usable macros).

TEST_END;

#endif/*TEST_TRANSACTIONHANDLER*/