/*
 * Copyright 2008, Haiku.
 * Distributed under the terms of the MIT license.
 *
 * Authors:
 *		Michael Pfeiffer <laplace@users.sourceforge.net>
 */

#include "Statement.h"

#include <stdio.h>


Statement::Statement()
	: fType(kUnknown)
	, fKeyword(NULL)
	, fOption(NULL)
	, fValue(NULL)
	, fChildren(NULL)
{
}

Statement::~Statement()
{
	delete fKeyword;
	delete fOption;
	delete fValue;
	delete fChildren;
}

void Statement::SetType(Type type)
{
	fType = type;
}

Statement::Type Statement::GetType()
{
	return fType;
}

void Statement::SetKeyword(BString* keyword)
{
	fKeyword = keyword;
}

BString* Statement::GetKeyword()
{
	return fKeyword;
}

void Statement::SetOption(Value* option)
{
	fOption = option;
}

Value* Statement::GetOption()
{
	return fOption;
}


void Statement::SetValue(Value* value)
{
	fValue = value;
}

Value* Statement::GetValue()
{
	return fValue;
}

StatementList* Statement::GetChildren()
{
	return fChildren;
}

void Statement::AddChild(Statement* statement)
{
	if (fChildren == NULL) {
		fChildren = new StatementList(true);
	}
	fChildren->Add(statement);
}

const char* Statement::GetKeywordString() 
{
	if (fKeyword != NULL) {
		return fKeyword->String();
	}
	return NULL;
}

const char* Statement::GetOptionString()
{
	Value* option = GetOption();
	if (option != NULL) {
		return option->GetValueString();
	}
	return NULL;
}

const char* Statement::GetTranslationString()
{
	Value* option = GetOption();
	if (option != NULL) {
		return option->GetTranslationString();
	}
	return NULL;
}

const char* Statement::GetValueString()
{
	Value* value = GetValue();
	if (value != NULL) {
		return value->GetValueString();
	}
	return NULL;
}

const char* Statement::GetValueTranslationString()
{
	Value* value = GetValue();
	if (value != NULL) {
		return value->GetTranslationString();
	}
	return NULL;
}


const char* Statement::ElementForType() {
	switch (fType) {
		case kDefault: return "Default";
			break;
		case kParam:   return "Param";
			break;
		case kQuery:   return "Query";
			break;
		case kValue:   return "Value";
			break;
		case kUnknown: return "Unknown";
			break;
	}
	return NULL;
}

void Statement::Print()
{
	bool hasValue = fValue != NULL;
	bool hasOption = fOption != NULL;
	
	printf("<%s", ElementForType());
	
	if (fKeyword != NULL) {
		printf(" keyword=\"%s\"", fKeyword->String());
	}
	
	if (hasValue || hasOption) {
		printf(">\n");
	} else {
		printf("/>\n");
	}


	if (hasOption) {
		printf("\t<option>\n");
		fOption->Print();
		printf("\t</option>\n");
	}			
	
	
	if (hasValue) {
		printf("\t<value>\n");
		fValue->Print();
		printf("\t</value>\n");
	}
	
	if (GetChildren() != NULL) {
		printf("\t<children>\n");
		GetChildren()->Print();
		printf("\t</children>\n");
	}

	if (hasValue || hasOption) {
		printf("</%s>\n\n", ElementForType());
	}	
}