/*
 * This file is a part of Winix
 * and is not publicly distributed
 * 
 * Copyright (c) 2010-2011, Tomasz Sowa
 * All rights reserved.
 * 
 */

#include "functions.h"
#include "core/log.h"
#include "core/misc.h"
#include "core/plugin.h"
#include "templates/templates.h"



void Functions::SetConfig(Config * pconfig)
{
	config = pconfig;
}


void Functions::SetCur(Cur * pcur)
{
	cur = pcur;
}



void Functions::SetDb(Db * pdb)
{
	db = pdb;
}


void Functions::SetSystem(System * psystem)
{
	system = psystem;
}


void Functions::SetTemplates(Templates * ptemplates)
{
	templates = ptemplates;
}


void Functions::SetSynchro(Synchro * psynchro)
{
	synchro = psynchro;
}




size_t Functions::FunctionsSize()
{
	return table.size();
}


Functions::Iterator Functions::Begin()
{
	return table.begin();
}


Functions::Iterator Functions::End()
{
	return table.end();
}


FunctionBase * Functions::Find(const std::wstring & function_name)
{
	Table::iterator i = table.find(function_name);

	if( i == table.end() )
		return 0;

return i->second;
}


void Functions::PrepareUrl(Item & item)
{
	TrimWhite(item.url);

	if( item.url.empty() )
		item.url = item.subject; // if the subject is empty then the url will be corrected by CorrectUrlOnlyAllowedChar()
		
	CorrectUrlOnlyAllowedChar(item.url);
	
	if( Find(item.url) )
	{
		// the name provided by an user is the same as a name of a function
		// we add one underscore character at the beginning
		
		// names of functions should not begin with an underscore '_'
		// and we can simply add one '_' at the beginning
		// and the name will be unique
		item.url.insert(item.url.begin(), '_');
	}
}



Error Functions::CheckSpecialFile(const Item & item)
{
	static std::wstring fstab = L"fstab";

	Item * etc = system->dirs.GetEtcDir();

	if( !etc )
		return WINIX_NOTHING_TO_DO;

	if( item.parent_id != etc->id )
		return WINIX_NOTHING_TO_DO;

	if( item.url == fstab )
	{
		log << log3 << "Functions: reloading mount points" << logend;

		system->mounts.ReadMounts(item.content);
		templates->ReadNewIndexTemplates();
		templates->ReadNewChangeTemplates();

		return WINIX_ERR_OK;
	}

return WINIX_NOTHING_TO_DO;
}



void Functions::SetObjects(FunctionBase * fun)
{
	fun->SetConfig(config);
	fun->SetCur(cur);
	fun->SetDb(db);
	fun->SetSystem(system);
	fun->SetFunctions(this);
	fun->SetTemplates(templates);
	fun->SetSynchro(synchro);
}


void Functions::Add(FunctionBase * fun)
{
	if( Find(fun->fun.url) )
	{
		log << log1 << "Functions: function " << fun->fun.url
			<< " already exists (skipped)" << logend;
		return;
	}

	SetObjects(fun);
	table[fun->fun.url] = fun;
}


void Functions::Add(FunctionBase & fun)
{
	Add(&fun);
}





void Functions::CreateFunctions()
{
	Add(fun_adduser);
	Add(fun_cat);
	Add(fun_chmod);
	Add(fun_chown);
	Add(fun_ckeditor);
	Add(fun_cp);
	Add(fun_default);
	Add(fun_download);
	Add(fun_emacs);
	Add(fun_last);
	Add(fun_login);
	Add(fun_logout);
	Add(fun_ln);
	Add(fun_ls);
	Add(fun_meta);
	Add(fun_mkdir);
	Add(fun_mount);
	Add(fun_mv);
	Add(fun_nicedit);
	Add(fun_node);
	Add(fun_passwd);
	Add(fun_priv);
	Add(fun_reload);
	Add(fun_rm);
	Add(fun_run);
	Add(fun_sort);
	Add(fun_special_default);
	Add(fun_stat);
	Add(fun_subject);
	Add(fun_template);
	Add(fun_tinymce);
	Add(fun_uname);
	Add(fun_upload);
	Add(fun_uptime);
	Add(fun_who);
	Add(fun_vim);

	plugin.Call(WINIX_CREATE_FUNCTIONS);
}


void Functions::InitFunctions()
{
	Table::iterator i = table.begin();

	for( ; i!=table.end() ; ++i)
		i->second->Init();
}


void Functions::Init()
{
	CreateFunctions();
	InitFunctions();
}


void Functions::Parse()
{
	function_parser.UTF8(config->utf8);
	function_parser.Parse(cur, db, this, system);
}



void Functions::SetDefaultFunctionForFile()
{
	if( cur->request->item.file_type != WINIX_ITEM_FILETYPE_NONE )
		cur->request->function = &fun_download;
	else
	if( system->HasReadExecAccess(cur->request->item) )
		cur->request->function = &fun_run;
	else
		cur->request->function = &fun_cat;

	log << log3 << "Functions: default function: " << cur->request->function->fun.url << logend;
}




void Functions::SetDefaultFunctionForDir()
{
// !! nie potrzebne
//	if( system->mounts.pmount->type == system->mounts.MountTypeThread() )
//		cur->request->function = &fun_thread;
//	else

	cur->request->function = &fun_ls;

	log << log3 << "Functions: default function: " << cur->request->function->fun.url << logend;
}


void Functions::SetDefaultFunction()
{
	cur->request->function = 0;

	plugin.Call(WINIX_SELECT_DEFAULT_FUNCTION);

	if( cur->request->function )
	{
		log << log3 << "Functions: default function: " << cur->request->function->fun.url 
			<< " (set by a plugin)" << logend;

		return;
	}

	if( cur->request->is_item )
		SetDefaultFunctionForFile();
	else
		SetDefaultFunctionForDir();
}





void Functions::CheckFunctionFollowDir(bool was_default_function)
{
	// directory with 'default' flag

	if( was_default_function )
	{
		if( cur->request->dir_tab.back()->link_redirect == 1 )
		{
			system->RedirectTo(cur->request->dir_tab.back()->link_to);
		}
		else
		{
			if( system->FollowAllLinks(cur->request->dir_tab.back()->link_to, true, true) )
				SetDefaultFunction();
		}
	}
}



void Functions::CheckFunctionFollowSymlink(bool was_default_function)
{
	if( cur->request->item.link_redirect == 1 )
	{
		if( was_default_function )
			system->RedirectTo(cur->request->item.link_to);
		else
			system->RedirectWithFunctionAndParamsTo(cur->request->item.link_to);
	}
	else
	if( system->FollowAllLinks(cur->request->item.link_to, true, true) )
	{
		if( was_default_function )
			SetDefaultFunction();

		if( cur->request->status == WINIX_ERR_OK && !cur->request->redirect_to.empty() && !was_default_function && cur->request->function )
		{
			// !! nie jestem pewny dodania tej nowej funkcji do redirecta... (sprawdzic to)
			cur->request->redirect_to += '/';
			cur->request->redirect_to += cur->request->function->fun.url;
			system->AddParams(cur->request->param_tab, cur->request->redirect_to, false);
		}
	}
}


// making a proper redirection from a directory with 'default' flag
// or from a symlink (or just loading it if there is no redirection flag set)
void Functions::CheckFunctionAndSymlink()
{
bool was_default_function = false;

	if( !cur->request->function || cur->request->function == &fun_special_default )
	{
		was_default_function = true;
		SetDefaultFunction();
	}
	
	if( cur->request->status != WINIX_ERR_OK || !cur->request->redirect_to.empty() )
		return;

	if( !cur->request->is_item && !cur->request->dir_tab.back()->link_to.empty() )
		CheckFunctionFollowDir(was_default_function);
	else
	if( cur->request->is_item && cur->request->item.type == Item::symlink && cur->request->function && cur->request->function->follow_symlinks )
		CheckFunctionFollowSymlink(was_default_function);
}



void Functions::MakeFunction()
{
	if( !cur->request->function )	
	{
		cur->request->status = WINIX_ERR_NO_FUNCTION;
		log << log1 << "Functions: no function (neither cat nor ls)" << logend;
		return;
	}
	
	if( !system->DirsHaveReadExecPerm()						||
		!system->HasReadExecAccess(cur->request->function->fun)	||
		!cur->request->function->HasAccess() )
	{
		cur->request->status = WINIX_ERR_PERMISSION_DENIED;
		return;
	}

	if( cur->request->method == Request::get )
	{
		if( cur->request->redirect_to.empty() )
			cur->request->function->MakeGet();
	}
	else
	if( cur->request->method == Request::post )
	{
		// we don't use post with redirecting (the post variables would be lost)

		if( cur->request->redirect_to.empty() )
			cur->request->function->MakePost();
		else
			cur->request->status = WINIX_ERR_PERMISSION_DENIED;
	}
	else
	if( cur->request->method == Request::head )
	{
		// do nothing
	}
	else
		log << log1 << "Functions: unknown request method (skipping)" << logend;
}





void Functions::CheckGetPostTimes(time_t difference)
{
	time_t now = std::time(0);

	if( cur->session->puser )
		return;
	
	if( cur->request->method != Request::post )
		return;

	if( now - cur->session->last_time_get >= (time_t)difference )
		return;

	if( cur->request->AllPostVarEmpty() )
		return;

	cur->session->spam_score += 1;
	log << log1 << "Functions: spam +1: POST after GET sent too fast" << logend;
}



// !!uwaga zwracana wartość zmieniona (true/false)
bool Functions::CheckAbuse()
{
	if( !system->rebus.CheckRebus() )
	{
		cur->request->status = WINIX_ERR_INCORRECT_REBUS;
		return true;
	}

	CheckGetPostTimes();

	if( cur->session->spam_score > 0 )
	{
		cur->request->status = WINIX_ERR_SPAM;
		log << log1 << "Functions: ignoring due to suspected spamming" << logend;
		return true;
	}

return false;
}




// returning true if the 'url' has to be changed
void Functions::ReadItemUrlSubject(Item & item, Item::Type item_type)
{
	std::wstring * new_subject = cur->request->PostVarp(L"subject");
	std::wstring * new_url     = cur->request->PostVarp(L"url");

	if( new_subject )
		item.subject = *new_subject;

	if( item.subject.empty() )
	{
		item.subject = cur->request->dir_tab.back()->subject;
		item.subject += L"_msg_";
		Toa(db->Size(cur->request->dir_tab.back()->id, Item::file), item.subject, 10, false);
	}

	if( new_url )
		item.url = *new_url;

	// if item.url is empty then it will be set from item.subject
	PrepareUrl(item);
}


void Functions::ReadItemFilterHtml(Item & item)
{
	html_filter.BreakWord(0);
	html_filter.WrapLine(0);
	html_filter.TrimWhite(false);
	html_filter.InsertTabs(0);
	html_filter.SafeMode(true);
	html_filter.ClearOrphans();

	html_filter.Filter(cur->request->PostVar(L"itemcontent"), item.content);
}


void Functions::ReadItemContent(Item & item, const std::wstring & content_type)
{
	bool is_root = cur->session->puser && cur->session->puser->super_user;
	bool filter_html = (content_type == L"2") && config->editors_html_safe_mode;

	if( filter_html && is_root && config->editors_html_safe_mode_skip_root )
		filter_html = false;

	if( filter_html )
		ReadItemFilterHtml(item);
	else
		cur->request->PostVar(L"itemcontent", item.content);
}


void Functions::ReadItemContentWithType(Item & item)
{
	item.content_type = Item::ct_formatted_text; // default is formatted text
	cur->request->PostVar(L"contenttype", temp);

	ReadItemContent(item, temp);


	// ct_text and ct_formatted_text can use everyone

	if( temp == L"0" )
		item.content_type = Item::ct_text;
	else
	if( temp == L"1" )
		item.content_type = Item::ct_formatted_text;


	// those below need special privileges

	if( !cur->session->puser )
		return;

	long user_id = cur->session->puser->id;


	if( temp == L"2" )
	{
		if( system->CanUseHtml(user_id) )
			item.content_type = Item::ct_html;
	}
	else
	if( temp == L"3" )
	{
		if( system->CanUseBBCode(user_id) )
			item.content_type = Item::ct_bbcode;
	}
	else
	if( temp == L"4" )
	{
		if( system->CanUseRaw(user_id) )
			item.content_type = Item::ct_raw;
	}
}




// item_type - the type of an item you are expecting to read
// returns true if the url has to be changed
// at the moment this is only checked for Item::file - for Item::dir it returns always true
// !! zmienic nazwe na ReadUrlSubjectContent
void Functions::ReadItem(Item & item, Item::Type item_type)
{
	if( item_type == Item::none )
		return;

	item.type      = item_type;
	item.parent_id = cur->request->dir_tab.back()->id; // !! moze to dac jako parametr?

	ReadItemUrlSubject(item, item_type);

	if( item_type == Item::file )
		ReadItemContentWithType(item);
}


void Functions::SetUser(Item & item)
{
	if( cur->session && cur->session->puser )
	{
		item.user_id = cur->session->puser->id;
		item.guest_name.clear();
	}
	else
	{
		item.user_id = -1;
		cur->request->PostVar(L"guestname", item.guest_name);
	}

	item.group_id = cur->request->dir_tab.back()->group_id;
}