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

#include "StatementWrapper.h"

static const char* kOpenUIStatement = "OpenUI";
static const char* kCloseUIStatement = "CloseUI";

static const char* kOpenGroupStatement = "OpenGroup";
static const char* kCloseGroupStatement = "CloseGroup";
static const char* kOpenSubGroupStatement = "OpenSubGroup";
static const char* kCloseSubGroupStatement = "CloseSubGroup";

// JCL
static const char* kJCL = "JCL";             
static const char* kJCLOpenUIStatement = "JCLOpenUI";
static const char* kJCLCloseUIStatement = "JCLCloseUI";


StatementWrapper::StatementWrapper(Statement* statement)
	: fStatement(statement)
{
	// nothing to do	
}

GroupStatement::GroupStatement(Statement* statement)
	: StatementWrapper(statement)
{
	// nothing to do	
}

bool GroupStatement::IsUIGroup()
{
	return strcmp(GetKeyword(), kOpenUIStatement) == 0;
}

bool GroupStatement::IsGroup()
{
	return strcmp(GetKeyword(), kOpenGroupStatement) == 0;
}

bool GroupStatement::IsSubGroup()
{
	return strcmp(GetKeyword(), kOpenSubGroupStatement) == 0;
}

bool GroupStatement::IsJCL()
{
	return strstr(GetKeyword(), kJCL) == GetKeyword();
}

bool GroupStatement::IsOpenGroup()
{
	const char* keyword = GetKeyword();
	
	return strcmp(keyword, kOpenUIStatement) == 0 || 
		strcmp(keyword, kOpenGroupStatement) == 0 ||
		strcmp(keyword, kOpenSubGroupStatement) == 0 ||
		strcmp(keyword, kJCLOpenUIStatement) == 0;
}

bool GroupStatement::IsCloseGroup()
{
	const char* keyword = GetKeyword();

	return strcmp(keyword, kCloseUIStatement) == 0 || 
		strcmp(keyword, kCloseGroupStatement) == 0 ||
		strcmp(keyword, kCloseSubGroupStatement) == 0 ||
		strcmp(keyword, kJCLCloseUIStatement) == 0;	
}

Value* GroupStatement::GetValue() 
{
	if (strcmp(GetKeyword(), kOpenUIStatement) == 0 ||
		strcmp(GetKeyword(), kJCLOpenUIStatement) == 0) {
		return GetStatement()->GetOption();
	} else {
		return GetStatement()->GetValue();
	}
}

const char* GroupStatement::GetGroupName()
{
	Value* value = GetValue();
	if (value == NULL) return NULL;
	BString* string = value->GetValue();
	if (string == NULL) return NULL;
	const char* name = string->String();	
	if (name != NULL && *name == '*') {
		// skip '*'
		name ++;
	}
	return name;
}

const char* GroupStatement::GetGroupTranslation()
{
	Value* value = GetValue();
	if (value == NULL) return NULL;
	BString* string = value->GetTranslation();
	if (string == NULL) return NULL;
	return string->String();	
}

GroupStatement::Type GroupStatement::GetType()
{
	const char* type = GetStatement()->GetValueString();
	if (type == NULL) return kUnknown;
	
	if (strstr(type, "PickOne") != NULL) return kPickOne;
	if (strstr(type, "PickMany") != NULL) return kPickMany;
	if (strstr(type, "Boolean") != NULL) return kBoolean;
	
	return kUnknown;
}