/* -*- 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 = NULL;
  atf_reply_t *reply = NULL;
  atf_result_t *result = NULL;
  atf_result_node_t *root = NULL;
  atf_status_t status;
  time_t timestamp;
  atf_id id;

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

  atft = atf_new_transaction(atfc, cpr.c_str());
  if(!atft) goto aaarg;

  id = atf_add_query(atft, query.attributes["class"].c_str(),
                     FILTER_LATEST, USE_NONE, 0, 0);
  if(!atft) goto aaarg;

  reply = atf_commit(atft);
  if(!reply) goto aaarg;

  if(atf_get_num_results(reply, id) != 1) goto aaarg;

  result = atf_get_result(reply, id, 0);
  if(!result) goto aaarg;

  status = atf_get_result_status(result, NULL, 0);
  if(status != ATF_STATUS_OK) goto aaarg;
 
  timestamp = atf_get_result_timestamp(result);

  root = atf_get_result_node(result);
  if(!root) goto aaarg;

  {
    QueryResult qresult = node2result(root, timestamp);
    rroot.groups[query.attributes["class"]] = qresult;
  }

 aaarg:
  if(root) atf_free_result_node(root);
  if(reply) atf_free_reply(reply);
  if(atft) atf_free_transaction(atft);

  return rroot;
}

#ifdef TEST_QUERYHANDLERPENTOMINOS

int main()
{
  return 1;
}

#endif/*TEST_QUERYHANDLERPENTOMINOS*/