#include "report.h"
#include "transform.h"
#include "util.h"

namespace ledger {

report_t::~report_t()
{
  for (std::list<transform_t *>::const_iterator i = transforms.begin();
       i != transforms.end();
       i++)
    delete *i;
}

void report_t::apply_transforms(xml::document_t * document)
{
  for (std::list<transform_t *>::const_iterator i = transforms.begin();
       i != transforms.end();
       i++)
    (*i)->execute(document);
}

void report_t::abbrev(value_t& result, xml::xpath_t::scope_t * locals)
{
  if (locals->args.size() < 2)
    throw new error("usage: abbrev(STRING, WIDTH [, STYLE, ABBREV_LEN])");

  std::string str = locals->args[0].to_string();
  long	      wid = locals->args[1];

  elision_style_t style = session->elision_style;
  if (locals->args.size() == 3)
    style = (elision_style_t)locals->args[2].to_integer();

  long abbrev_len = session->abbrev_length;
  if (locals->args.size() == 4)
    abbrev_len = locals->args[3].to_integer();

  result.set_string(abbreviate(str, wid, style, true, (int)abbrev_len));
}

void report_t::ftime(value_t& result, xml::xpath_t::scope_t * locals)
{
  if (locals->args.size() < 1)
    throw new error("usage: ftime(DATE [, DATE_FORMAT])");

  datetime_t date = locals->args[0].to_datetime();

  std::string date_format;
  if (locals->args.size() == 2)
    date_format = locals->args[1].to_string();
  else
    date_format = datetime_t::output_format;

  result.set_string(date.to_string(date_format));
}

bool report_t::resolve(const std::string& name, value_t& result,
		       xml::xpath_t::scope_t * locals)
{
  const char * p = name.c_str();
  switch (*p) {
  case 'a':
    if (name == "abbrev") {
      abbrev(result, locals);
      return true;
    }
    break;

  case 'f':
    if (name == "ftime") {
      ftime(result, locals);
      return true;
    }
    break;
  }

  return xml::xpath_t::scope_t::resolve(name, result, locals);
}

xml::xpath_t::op_t * report_t::lookup(const std::string& name)
{
  const char * p = name.c_str();
  switch (*p) {
  case 'o':
    if (std::strncmp(p, "option_", 7) == 0) {
      p = p + 7;
      switch (*p) {
      case 'a':
#if 0
	if (std::strcmp(p, "accounts") == 0)
	  return MAKE_FUNCTOR(report_t, option_accounts);
	else
#endif
	  if (std::strcmp(p, "amount") == 0)
	  return MAKE_FUNCTOR(report_t, option_amount);
	break;

      case 'b':
	if (std::strcmp(p, "bar") == 0)
	  return MAKE_FUNCTOR(report_t, option_bar);
	break;

#if 0
      case 'c':
	if (std::strcmp(p, "clean") == 0)
	  return MAKE_FUNCTOR(report_t, option_clean);
	else if (std::strcmp(p, "compact") == 0)
	  return MAKE_FUNCTOR(report_t, option_compact);
	break;
#endif

      case 'e':
#if 0
	if (std::strcmp(p, "entries") == 0)
	  return MAKE_FUNCTOR(report_t, option_entries);
	else if (std::strcmp(p, "eval") == 0)
	  return MAKE_FUNCTOR(report_t, option_eval);
	else if (std::strcmp(p, "exclude") == 0)
	  return MAKE_FUNCTOR(report_t, option_remove);
#endif
	break;

      case 'f':
	if (std::strcmp(p, "foo") == 0)
	  return MAKE_FUNCTOR(report_t, option_foo);
	else if (std::strcmp(p, "format") == 0)
	  return MAKE_FUNCTOR(report_t, option_format);
	break;

      case 'i':
#if 0
	if (std::strcmp(p, "include") == 0)
	  return MAKE_FUNCTOR(report_t, option_select);
#endif
	break;

      case 'l':
#if 0
	if (! *(p + 1) || std::strcmp(p, "limit") == 0)
	  return MAKE_FUNCTOR(report_t, option_limit);
#endif
	break;

#if 0
      case 'm':
	if (std::strcmp(p, "merge") == 0)
	  return MAKE_FUNCTOR(report_t, option_merge);
	break;
#endif

      case 'r':
#if 0
	if (std::strcmp(p, "remove") == 0)
	  return MAKE_FUNCTOR(report_t, option_remove);
#endif
	break;

#if 0
      case 's':
	if (std::strcmp(p, "select") == 0)
	  return MAKE_FUNCTOR(report_t, option_select);
	else if (std::strcmp(p, "split") == 0)
	  return MAKE_FUNCTOR(report_t, option_split);
	break;
#endif

      case 't':
	if (! *(p + 1))
	  return MAKE_FUNCTOR(report_t, option_amount);
	else if (std::strcmp(p, "total") == 0)
	  return MAKE_FUNCTOR(report_t, option_total);
	break;

      case 'T':
	if (! *(p + 1))
	  return MAKE_FUNCTOR(report_t, option_total);
	break;
      }
    }
    break;
  }

  return xml::xpath_t::scope_t::lookup(name);
}

} // namespace ledger

#ifdef USE_BOOST_PYTHON

#include <boost/python.hpp>

using namespace boost::python;
using namespace ledger;

void export_report()
{
  class_< report_t > ("Report")
    .add_property("session",
		  make_getter(&report_t::session,
			      return_value_policy<reference_existing_object>()))

    .def("apply_transforms", &report_t::apply_transforms)
    ;
}

#endif // USE_BOOST_PYTHON