winix/core/mounts.cpp

/*
 * This file is a part of Winix
 * and is not publicly distributed
 * 
 * Copyright (c) 2009-2010, Tomasz Sowa
 * All rights reserved.
 * 
 */
 
#include "mounts.h"
#include "request.h"
#include "log.h"
#include "db/db.h"
#include "plugin.h"




Mounts::Mounts()
{
	pmount = 0;
}


void Mounts::CreateMountType()
{
	mount_type_cms    = AddMountType("cms");
	mount_type_thread = AddMountType("thread");
	mount_type_ticket = AddMountType("ticket");
}


void Mounts::CreateMountFs()
{
	mount_fs_simplefs = AddMountFs("simplefs");
	mount_fs_hashfs   = AddMountFs("hashfs");
}


void Mounts::CreateMountPar()
{
	mount_par_page						= AddMountPar("page");
	mount_par_thread					= AddMountPar("thread");
	mount_par_ticket					= AddMountPar("ticket");
	mount_par_ticket_type				= AddMountPar("ticket_type");
	mount_par_ticket_type_default		= AddMountPar("ticket_type_default");
	mount_par_ticket_status				= AddMountPar("ticket_status");
	mount_par_ticket_status_default		= AddMountPar("ticket_status_default");
	mount_par_ticket_priority			= AddMountPar("ticket_priority");
	mount_par_ticket_priority_default	= AddMountPar("ticket_priority_default");
	mount_par_ticket_category			= AddMountPar("ticket_category");
	mount_par_ticket_category_default	= AddMountPar("ticket_category_default");
	mount_par_ticket_expected			= AddMountPar("ticket_expected");
	mount_par_ticket_expected_default	= AddMountPar("ticket_expected_default");
	mount_par_createthread_on			= AddMountPar("createthread_on");
	mount_par_createticket_on			= AddMountPar("createticket_on");
	mount_par_only_root_remove			= AddMountPar("only_root_remove");
	mount_par_emacs_on					= AddMountPar("emacs_on");
	mount_par_mkdir_on					= AddMountPar("mkdir_on");
	mount_par_app						= AddMountPar("app");
	mount_par_html_template				= AddMountPar("html_template");
}


void Mounts::CreateMounts()
{
	CreateMountType();
	CreateMountFs();
	CreateMountPar();

	plugin.Call(WINIX_ADD_MOUNTS);
}



void Mounts::SetDirs(Dirs * pdirs)
{
	dirs = pdirs;
}


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

void Mounts::SetRequest(Request * prequest)
{
	request = prequest;
}



int Mounts::AddMountType(const char * type)
{
	mount_type_tab.push_back(type);

return static_cast<int>(mount_type_tab.size()) - 1;
}


int Mounts::AddMountType(const std::string & type)
{
	return AddMountType(type.c_str());
}


int Mounts::AddMountFs(const char * fs)
{
	mount_fs_tab.push_back(fs);

return static_cast<int>(mount_fs_tab.size()) - 1;
}


const std::string & Mounts::GetMountType(int id)
{
	if( id < 0 || id >= (int)mount_type_tab.size() )
		return empty_str;

	return mount_type_tab[id];
}


int Mounts::AddMountFs(const std::string & fs)
{
	return AddMountFs(fs.c_str());
}


const std::string & Mounts::GetMountFs(int id)
{
	if( id < 0 || id >= (int)mount_fs_tab.size() )
		return empty_str;

	return mount_fs_tab[id];
}


int Mounts::AddMountPar(const char * par)
{
	mount_par_tab.push_back(par);

return static_cast<int>(mount_par_tab.size()) - 1;
}



int Mounts::AddMountPar(const std::string & par)
{
	return AddMountPar(par.c_str());
}


const std::string & Mounts::GetMountPar(int id)
{
	if( id < 0 || id >= (int)mount_par_tab.size() )
		return empty_str;

	return mount_par_tab[id];
}



// reading from 'mounts'
Error Mounts::ReadMounts(const std::string & mounts)
{
	mount_parser.SetDirs(dirs);
	mount_parser.SetMountTypeTab(mount_type_tab);
	mount_parser.SetMountFsTab(mount_fs_tab);
	mount_parser.SetMountParTab(mount_par_tab);

	Error err = mount_parser.Parse(mounts, mount_tab);

	if( err != WINIX_ERR_OK )
	{
		log << log1 << "M: some problems with mountpoints (mountpoints table will be empty)" << logend;	
		mount_tab.clear();
	}

	CalcCurMount();

return err;
}



// reading from /etc/fstab
Error Mounts::ReadMounts()
{
	static std::string file = "fstab";

	Item * etc = dirs->GetEtcDir();

	if( !etc )
	{
		log << log1 << "M: there is no /etc directory" << logend;
		return WINIX_ERR_NO_ITEM;
	}

	Item fstab;
	Error err = db->GetItem(etc->id, file, fstab);

	if( err == WINIX_ERR_NO_ITEM )
	{
		log << log1 << "M: there is no /etc/fstab file" << logend;
		return err;
	}

	if( err != WINIX_ERR_OK )
	{
		log << log1 << "M: cannot read /etc/fstab" << logend;
		return err;
	}

	return ReadMounts(fstab.content);
}


void Mounts::MountCmsForRoot()
{
	Mount mount;
	mount.type = MountTypeCms();
	mount.fs   = MountFsSimplefs();

	Item * proot = dirs->GetRootDir();
	
	if( proot )
		mount.dir_id = proot->id;
	else
	{
		mount.dir_id = -1;
		log << log1 << "M: there is no a root dir" << logend;	
	}

	std::pair<MountTab::iterator, bool> res = mount_tab.insert( std::make_pair(mount.dir_id, mount) );
	pmount = &(res.first->second);
}




void Mounts::CalcCurMount()
{
std::vector<Item*>::reverse_iterator i;

	// when the program starts (when the dir_tab is empty() 
	// we don't want to call MountCmsForRoot()
	if( request->dir_tab.empty() )
		return;
	
	for(i = request->dir_tab.rbegin() ; i!=request->dir_tab.rend() ; ++i)
	{
		std::map<long, Mount>::iterator m = mount_tab.find( (*i)->id );
		
		if( m != mount_tab.end() )
		{
			pmount = &(m->second);
			log << log2 << "M: current mount point is: " << GetMountType(pmount->type)
						<< ", fs: " << GetMountFs(pmount->fs) << logend;
			return;	
		}
	}

	// if nothing was found
	// we assume that 'cms' mount point is used
	MountCmsForRoot();
	log << log2 << "M: current mount point is: " << GetMountType(pmount->type) << " (default)"
		<< ", fs: " << GetMountFs(pmount->fs) << logend;
}


// can return null pointer
// and we don't assume cms as a default mount point if nothing is found
Mount * Mounts::CalcMount(long dir_id)
{
	while( true )
	{
		Item * pdir = dirs->GetDir(dir_id);

		if( !pdir )
			return 0;

		std::map<long, Mount>::iterator m = mount_tab.find( pdir->id );
		
		if( m != mount_tab.end() )
			return &(m->second);

		dir_id = pdir->parent_id;
	}
}


const Mounts::MountTab * Mounts::GetMountTab()
{
	return &mount_tab;
}