/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            queryhandlerpentominos.cc
 *
 *  Thu Jan 15 11:35:31 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 "queryhandlerpentominos.h"

#include "debug.h"

#include "configuration.h"

QueryHandlerPentominos::QueryHandlerPentominos(std::string cpr)
  : QueryHandler()
{
  this->cpr = cpr;

  atfh = atf_init();

  bool use_https = false;
  atfc = atf_connect(atfh, Conf::pentominos_addr.c_str(),
                     Conf::pentominos_port, use_https);

}

QueryHandlerPentominos::~QueryHandlerPentominos()
{
  atf_disconnect(atfc);
  atf_close(atfh);
}

static QueryResult node2result(atf_result_node_t *node, time_t timestamp)
{
  QueryResult rnode;
  rnode.timestamp = timestamp;
  rnode.source = "pentominos";

  if(!node) return rnode;

  struct _atf_result_node_t *child = node->child;
  while(child) {
    if(child->value == NULL) {
      rnode.groups[child->name] = node2result(child, timestamp);
    } else {
      rnode.values[child->name] = child->value;
    }
    child = child->next;
  }

  return rnode;
}

QueryResult QueryHandlerPentominos::exec(Query &query)
{
  atf_transaction_t* atft = atf_new_transaction(atfc, cpr.c_str());

  atf_id id = atf_add_query(atft, query.attributes["class"].c_str(), 0, "xml");

  atf_reply_t *reply = atf_commit(atft);

  atf_status_t status = atf_get_reply_status(reply, id);
  if(status != ATF_STATUS_OK) { return QueryResult(); }
 
  time_t timestamp = atf_get_reply_timestamp(reply, id);

  atf_result_node_t *root = atf_get_result_node(reply, id);

  QueryResult rroot;
  rroot.timestamp = timestamp;
  rroot.source = "pentominos";

  QueryResult result = node2result(root, timestamp);
  if(root) rroot.groups[query.attributes["class"]] = result;

  atf_free_result_node(root);
  atf_free_reply(reply);
  atf_free_transaction(atft);

  return rroot;
}

#ifdef TEST_QUERYHANDLERPENTOMINOS

int main()
{
  return 1;
}

#endif/*TEST_QUERYHANDLERPENTOMINOS*/