#include "selectcommand.h"
#include "error.h"
#include <boost/multi_index_container.hpp>
#include <boost/multi_index/ordered_index.hpp>
#include <compileTimeFormatter.h>

namespace DB {
	ColumnIndexOutOfRange::ColumnIndexOutOfRange(unsigned int n) : colNo(n) { }

	AdHocFormatter(ColumnIndexOutOfRangeMsg, "Column (%?) index out of range");
	std::string
	ColumnIndexOutOfRange::message() const throw()
	{
		return ColumnIndexOutOfRangeMsg::get(colNo);
	}

	ColumnDoesNotExist::ColumnDoesNotExist(const Glib::ustring & n) : colName(n) { }

	AdHocFormatter(ColumnDoesNotExistMsg, "Column (%?) does not exist");
	std::string
	ColumnDoesNotExist::message() const throw()
	{
		return ColumnDoesNotExistMsg::get(colName);
	}
};

DB::SelectCommand::SelectCommand(const std::string & sql) :
	DB::Command(sql),
	columns(new Columns)
{
}

DB::SelectCommand::~SelectCommand()
{
	delete columns;
}

const DB::Column&
DB::SelectCommand::operator[](unsigned int n) const
{
	if (n < columns->size()) {
		return **columns->get<0>().find(n);
	}
	throw ColumnIndexOutOfRange(n);
}

const DB::Column&
DB::SelectCommand::operator[](const Glib::ustring & n) const
{
	typedef Columns::nth_index<1>::type CbyName;
	CbyName::iterator i = columns->get<1>().find(n.collate_key());
	if (i != columns->get<1>().end()) {
		return **i;
	}
	throw ColumnDoesNotExist(n);
}

unsigned int
DB::SelectCommand::getOrdinal(const Glib::ustring & n) const
{
	return operator[](n).colNo;
}

unsigned int
DB::SelectCommand::columnCount() const
{
	return columns->size();
}

DB::ColumnPtr
DB::SelectCommand::insertColumn(ColumnPtr col)
{
	return *columns->insert(col).first;
}

DB::RowBase::RowBase(SelectCommand * s) :
	sel(s)
{
}

const DB::Column&
DB::RowBase::operator[](const Glib::ustring & n) const
{
	return sel->operator[](n);
}

const DB::Column&
DB::RowBase::operator[](unsigned int col) const
{
	return sel->operator[](col);
}