#include "Usr.ns.h"

#include <Core/Log.ns.h>
#include <UserManager/User.class.h>
#include <SimpleList.class.h>
#include <TaskManager/Task.ns.h>
#include <TextFile.class.h>
#include <VFS/VFS.ns.h>

/*
 * Syntax for Users and Groups configuration files : one entry per line
 * syntax for Users : <uid>:<username>:<basegroup>:<extragroup>,<extragroup>:<completename>:<password>
 * syntax for Groups : <gid>:<name>
 */

namespace Usr {

SimpleList <User> *m_users = 0;
SimpleList <Group> *m_groups = 0;

void load() {
	if (m_users != 0) delete m_users;
	if (m_groups != 0) delete m_groups;
	m_users = 0, m_groups = 0;

	if (VFS::find("/System/Configuration/Groups")) VFS::find("/System/Configuration/Groups")->setPermissions(0600);
	if (VFS::find("/System/Configuration/Users")) VFS::find("/System/Configuration/Users")->setPermissions(0600);
	TextFile groups("/System/Configuration/Groups", FM_READ);
	if (groups.valid()) {
		while (!groups.eof()) {
			String s = groups.readLine();
			Vector<String> data = s.split(":");
			if (data.size() == 2 and !(s[0] == WChar("#"))) {
				m_groups = m_groups->cons(Group(data[1], data[0].toInt()));
			}
		}
	}
	if (m_groups == 0) {
		m_groups = m_groups->cons(Group("root", 0));	//In case, add a default group
		Log::log(KL_WARNING, "Usr.ns : group file invalid, had to add default group.");
	}

	TextFile users("/System/Configuration/Users", FM_READ);
	if (users.valid()) {
		while (!users.eof()) {
			String s = users.readLine();
			if (s == "" or s[0] == WChar("#")) continue;
			Vector<String> data = s.split(":");
			if (data.size() == 6) {
				m_users = m_users->cons(User(data[1], data[4], data[5], group(data[2].toInt()), data[3], data[0].toInt()));
			}
		}
	}
	if (m_users == 0) {
		m_users = m_users->cons(User("melon", "MelOS", "", group(0), "", 0));
		Log::log(KL_WARNING, "Usr.ns : users file invalid, had to add default users.");
	}
}

void save() {
	TextFile groups("/System/Configuration/Groups", FM_TRUNCATE);
	for (SimpleList<Group> *iter = m_groups; iter != 0; iter = iter->next()) {
		groups.write(String::number(iter->v().getGid()) + ":" + iter->v().getName(), true);
	}
	TextFile users("/System/Configuration/Users", FM_TRUNCATE);
	for (SimpleList<User> *iter = m_users; iter != 0; iter = iter->next()) {
		users.write(String::number(iter->v().getUid()) + ":" + iter->v().getUserName() + ":"
				+ String::number(iter->v().getGroup()->getGid()) + ":" 
				+ iter->v().getGroups() + ":" + iter->v().getCompleteName() + ":" +
				iter->v().getPassword(), true);
	}
}

u32int uid() {
	return Task::currProcess()->getUid();
}

User* user(u32int uid) {
	for (SimpleList<User> *iter = m_users; iter != 0; iter = iter->next()) {
		if (iter->v().getUid() == uid) return &iter->v();
	}
	return 0;
}

User* user(String username) {
	for (SimpleList<User> *iter = m_users; iter != 0; iter = iter->next()) {
		if (iter->v().getUserName() == username) return &iter->v();
	}
	return 0;
}

User* user() { return user(uid()); }

Group* group(u32int gid) {
	for (SimpleList<Group> *iter = m_groups; iter != 0; iter = iter->next()) {
		if (iter->v().getGid() == gid) return &iter->v();
	}
	return 0;
}

Group* group(String name) {
	for (SimpleList<Group> *iter = m_groups; iter != 0; iter = iter->next()) {
		if (iter->v().getName() == name) return &iter->v();
	}
	return 0;
}

u32int uid(String username) {
	User* x = user(username);
	return (x != 0 ? x->getUid() : (u32int) - 1);
}

};