MailDatabase.cpp

/*
 * Copyright (c) 2009 David Verhasselt <david@crowdway.com>
 * Licensed under The MIT License - http://gmailcc.crowdway.com/license.txt
 */

#include "MailDatabase.h"

#include <sys/types.h>
#include <sys/stat.h>
#include <cstdlib>
#include <iostream>
#include <fstream>
#include <exception>

#include <boost/filesystem.hpp>
namespace bf = boost::filesystem;

#include "LogSink.h"
#include "MailLink.h"
#include "MailBox.h"
#include "MailRecord.h"


MailBox* MailDatabase::add_mailbox(string mailbox)
{
	MailBox* box = get_mailbox(mailbox);
	
	if (box != NULL)
		return box;
		
	box = new MailBox(this, mailbox);

	// Make sure the directory structure is there
	create_maildir(box->get_path());
	
	mailboxes.push_back(box);
	
	return box;
}

bool MailDatabase::mail_exists(unsigned long uid)
{
	return mail_exists("{imap.gmail.com}[Gmail]/All Mail", uid);
	
}

bool MailDatabase::mail_exists(string mailbox, unsigned long uid)
{
	MailBox* mb = get_mailbox(mailbox);
	
	if (mb != NULL)
	{
		for (vector<MailLink*>::iterator iter = mb->mails.begin(); iter < mb->mails.end(); iter++)
		{
				if ((*iter)->uid == uid)
					return true;
		}
	}
	
	return false;
}

/*
 * Find the iterator in the messages list corresponding the mail with given messageid
 */
vector<MailRecord*>::iterator MailDatabase::get_imail(string messageid)	// TODO: This function gets called unnecessary when a new mail is added (maildb->new_mail()). Remove it.
{
	for (vector<MailRecord*>::iterator iter = messages.begin(); iter < messages.end(); iter++)
	{
			if (messageid.compare((*iter)->messageid) == 0)
			{
				return iter;	
			}
	}
	
	return messages.end();	
}

MailRecord* MailDatabase::get_mail(MailBox* mailbox, unsigned long uid)
{
	for (vector<MailLink*>::iterator iter = mailbox->mails.begin(); iter < mailbox->mails.end(); iter++)
	{
		if ((*iter)->uid == uid)
			return (*iter)->mailrecord;			
	}
	
	return NULL;	
}

MailRecord* MailDatabase::get_mail(unsigned long uid)
{
	return get_mail(get_mailbox("[Gmail]/All Mail"), uid);
}

MailRecord* MailDatabase::get_mail(string messageid)
{
	vector<MailRecord*>::iterator iter = get_imail(messageid);
	
	if (iter == messages.end())	
		return NULL;
	else
		return *iter;	
}

MailRecord* MailDatabase::add_mail(string messageid, MailBox* mailbox, unsigned long uid, string path)
{
	MailRecord* mr = get_mail(messageid);
	
	if (mr == NULL)
	{
		mr = new MailRecord;
		
		mr->messageid = messageid;
		messages.push_back(mr);
	}
	
	MailLink* ml = mr->add_to_mailbox(mailbox, uid, path);
	
	if (mailbox->primary)
	{
		mr->mainlink = ml;
		mr->load_md_info();
	}
	
	return mr;
}


MailRecord* MailDatabase::new_mail(string messageid, MailBox* mailbox, unsigned long uid)
{
	MailRecord* mr = get_mail(messageid);
	
	if (mr == NULL)
		return NULL;
		
	mr->add_to_mailbox(mailbox, uid);
	
	return mr;
}


MailRecord* MailDatabase::new_mail(string messageid, unsigned long uid, string header, string content)
{
	return new_mail(messageid, uid, header + '\n' + content);	

}

MailRecord* MailDatabase::new_mail(string messageid, unsigned long uid, string body)
{
	MailRecord* mr = get_mail(messageid);
	
	if (mr != NULL)
		return mr;
	else
		mr = new MailRecord;
	
	mr->messageid = messageid;
	mr->add_to_mailbox(get_mailbox("[Gmail]/All Mail"), uid);
	mr->save_content(body);
	
	messages.push_back(mr);
	
	return mr;
}

void MailDatabase::remove_mail(MailRecord* mr, MailBox* mailbox)
{		
	if (mr == NULL)
		return;
		
	if (mailbox != NULL)
		mr->remove_from_mailbox(mailbox);
	else
		mr->remove();
	
	if (mr->links.size() <= 0)
	{
		vector<MailRecord*>::iterator mr_iter = get_imail(mr->messageid);
		
		if (mr_iter < messages.end())
		{
			messages.erase(mr_iter);
		}
		else
			Log::error << "Could not find mail with message-id '" << mr->messageid << "'." << Log::endl;
	}	
}


void MailDatabase::remove_mail(string messageid, MailBox* mailbox)
{
	remove_mail(get_mail(messageid), mailbox);
}


MailDatabase* MailDatabase::create(string path)
{
	return NULL;
}


MailBox* MailDatabase::get_mailbox(string mailbox)
{
	for (vector<MailBox*>::iterator it = mailboxes.begin(); it < mailboxes.end(); it++)
	{
		if ((*it)->name.compare(mailbox) == 0)
		{
			return *it;
		}
	}
	
	return NULL;
	
}

/**
 * mailbox-name
 * mailbox-uid_validity
 * mailbox-next_uid
 * mailbox-msgcount
 * maillink
 * maillink
 * maillink
 * /n
 */

void MailDatabase::save()
{
	
	Log::info << "Saving database." << Log::endl;
	
	ofstream dbfile;
	
	dbfile.open((maildir + "database").c_str());

	if (dbfile.is_open())
	{
		// Header
		dbfile << "Generated by Gmail Carbon Copy -- http://code.crowdway.com/projects/gmailcc" << endl;
		dbfile << endl;				// Signals end of comments
		dbfile << "1" << endl;		// Format version
		
		for (vector<MailBox*>::iterator mb = mailboxes.begin(); mb < mailboxes.end(); mb++)
		{
			dbfile << (*mb)->name << endl;					// mailbox-name
			dbfile << (*mb)->uid_validity << endl;			// mailbox-uid_validity
			dbfile << (*mb)->next_uid << endl;				// mailbox-next_uid
			dbfile << (*mb)->messagecount << endl;			// mailbox-msgcount
			
			for (vector<MailLink*>::iterator mail = (*mb)->mails.begin(); mail < (*mb)->mails.end(); mail++)
			{
				dbfile << (*mail)->uid << endl;						// uid
				dbfile << (*mail)->mailrecord->messageid << endl;	// messageid
				dbfile << (*mail)->get_path() << endl;				// path				
			}
			
			dbfile << endl;
		}
		
		dbfile.close();		

	}
	else
	{
		Log::error << "Unable to open database for writing (" << maildir + "database). Could not save database." << Log::endl;
	}
}

MailDatabase* MailDatabase::load(string path)
{
	Log::info << "Loading MailDatabase @ " << path << Log::endl;
	
	// Make sure it ends with "/"
	if (path[path.size()-1] != '/')
		path += '/';
	
	// Make sure we have a Maildir-format type directory structure
	create_maildir(path);	

	MailDatabase* mdb = new MailDatabase();
	mdb->maildir = path;
	
	// Load databasefile
	ifstream dbfile;
	
	Log::info << "Opening database-file" << Log::endl;
	dbfile.open((path + "database").c_str());
	
	if (dbfile.is_open())
	{
		string line;
		
		MailBox* currentmb;
		
		// Header
		getline(dbfile, line);	// First line of comment
		while(!line.empty())
		{
			getline(dbfile, line);	// Read next line of comment
		}
		getline(dbfile, line);	// version, "1"
		
		getline(dbfile, line);
		while(!dbfile.eof())
		{
			currentmb = mdb->add_mailbox(line);
			getline(dbfile, line);	currentmb->uid_validity = atoi(line.c_str());
			getline(dbfile, line);	currentmb->next_uid = atoi(line.c_str());
			getline(dbfile, line);	currentmb->messagecount = atoi(line.c_str());
			
			unsigned long uid;
			string path;
			string messageid;
			
			while(!dbfile.eof())
			{
				getline(dbfile, line);

				if (line.empty())
				{
					break;	// Next mailbox
				}
				
				uid = atoi(line.c_str());
				getline(dbfile, messageid);
				getline(dbfile, path);
				
				mdb->add_mail(messageid, currentmb, uid, path);
			}
			
			getline(dbfile, line);
		}
		
		dbfile.close();
	}
	
	else
	{
		Log::info << "Couldn't open database file ( " << path << "database" << " ). Starting from scratch." << Log::endl;
	}

	return mdb;	
}

void MailDatabase::sweep()
{
	for (vector<MailRecord*>::iterator iter = messages.begin(); iter < messages.end(); )
	{
		if (!(*iter)->marked)
		{
			(*iter)->remove();
			delete *iter;
			
			iter = messages.erase(iter);
		}
		else
			iter++;
	}
	
	for (vector<MailBox*>::iterator iter = mailboxes.begin(); iter < mailboxes.end(); iter++)
	{
		(*iter)->sweep();	
	}
}

string MailDatabase::get_maildir()
{
	return maildir;
}

/*
 * Creates a directory structure fit for Maildir format.
 * Example:
 * 		.dir_path/new
 * 		.dir_path/cur
 * 		.dir_path/tmp
 */
void MailDatabase::create_maildir(string dir_path)
{
	bf::path path(dir_path);
	bf::path subpath;
	
	bf::path paths[] = {path, path / "cur", path / "new", path / "tmp"};
		
	for (uint i = 0; i < 4; i++)
	{
		if (!bf::exists(paths[i])) {
			Log::info << "Creating directory " << paths[i] << Log::endl;
			bf::create_directory(paths[i]);
		}
		else if (!bf::is_directory(paths[i])) {
			Log::error << "Directory at " << paths[i] << " already exists and is not a directory." << Log::endl;
			throw exception();
		}
	}
}
	


MailDatabase::MailDatabase()
{
}

MailDatabase::~MailDatabase()
{
	for (vector<MailBox*>::iterator iter = mailboxes.begin(); iter < mailboxes.end(); iter++)
	{
		delete (*iter);		
	}
}