Chat-O-Matic/application/ProtocolManager.cpp

/*
 * Copyright 2009-2011, Andrea Anzani. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Andrea Anzani, andrea.anzani@gmail.com
 */

#include <stdio.h>
#include <image.h>

#include <Bitmap.h>
#include <Directory.h>
#include <Entry.h>
#include <Handler.h>

#include "Account.h"
#include "AppMessages.h"
#include "ProtocolManager.h"
#include "ChatProtocol.h"
#include "MainWindow.h"
#include "Server.h"
#include "TheApp.h"
#include "Utils.h"

static ProtocolManager*	fInstance = NULL;


bool
ProtocolManager::Init(BDirectory dir, BHandler* target)
{
	BEntry entry;
	BPath path;
	bool ret = false;

	dir.Rewind();

	while (dir.GetNextEntry(&entry) == B_OK) {
		path = BPath(&entry);

		// Load protocol addon
		image_id id = load_add_on(path.Path());
		if (id < 0)
			continue;

		// If add-on's API version fits then load accounts…
		ChatProtocolAddOn* addOn = new ChatProtocolAddOn(id, path.Path());
		if (addOn->Version() != APP_VERSION)
			continue;
		ret = true;

		// If add-on has multiple protocols, also load them
		for (int32 i = 0; i < addOn->CountProtocols(); i++) {
			ChatProtocolAddOn* subAddOn = addOn;
			if (i > 0)
				subAddOn = new ChatProtocolAddOn(id, path.Path(), i);

			ChatProtocol* proto = subAddOn->Protocol();

			fAddOnMap.AddItem(proto->Signature(), subAddOn);
			_LoadAccounts(path.Path(), subAddOn, i, target);
			delete proto;
		}
	}
	return ret;
}


ProtocolManager::ProtocolManager()
{
}


ProtocolManager*
ProtocolManager::Get()
{
	if (fInstance == NULL)
		fInstance = new ProtocolManager();
	return fInstance;
}


uint32
ProtocolManager::CountProtocolAddOns() const
{
	return fAddOnMap.CountItems();
}


ChatProtocolAddOn*
ProtocolManager::ProtocolAddOnAt(uint32 i) const
{
	return fAddOnMap.ValueAt(i);
}


ChatProtocolAddOn*
ProtocolManager::ProtocolAddOn(const char* signature)
{
	return fAddOnMap.ValueFor(signature);
}


uint32
ProtocolManager::CountProtocolInstances() const
{
	return fProtocolMap.CountItems();
}


ChatProtocol*
ProtocolManager::ProtocolInstanceAt(uint32 i) const
{
	return fProtocolMap.ValueAt(i);
}


ChatProtocol*	
ProtocolManager::ProtocolInstance(bigtime_t identifier)
{
	return fProtocolMap.ValueFor(identifier);
}


void
ProtocolManager::AddAccount(ChatProtocolAddOn* addOn, const char* account,
							BHandler* target)
{
	TheApp* theApp = reinterpret_cast<TheApp*>(be_app);
	bigtime_t instanceId = system_time();
	ChatProtocol* cayap = addOn->Protocol();
	Account* acc =
		new Account(instanceId, cayap, account, addOn->Signature(), target);

	// Send a "whoops" notification if hits a failure
	if (acc->InitCheck() != B_OK) {
		BMessage error(APP_ACCOUNT_FAILED);
		cayap->Icon()->Archive(&error);
		error.AddString("name", account);
		theApp->GetMainWindow()->MessageReceived(&error);
		return;
	}

	fProtocolMap.AddItem(instanceId, cayap);

	theApp->GetMainWindow()->GetServer()->AddProtocolLooper(
		instanceId, cayap);
}


void
ProtocolManager::_LoadAccounts(const char* image_path, ChatProtocolAddOn* addOn,
							   int protoIndex, BHandler* target)
{
	// Find accounts path for this protocol
	BPath path(AccountPath(addOn->Signature(), addOn->Protocol()->Signature()));
	if (path.InitCheck() != B_OK)
		return;

	BDirectory dir(path.Path());
	BEntry entry;
	bool firstDone = false;

	while (dir.GetNextEntry(&entry) == B_OK)
		_LoadAccount(addOn, entry, target);
}


void
ProtocolManager::_LoadAccount(const char* imagePath, BEntry accountEntry,
							  int protoIndex, BHandler* target)
{
	image_id id = load_add_on(imagePath);
	if (id < 0)
		return;

	// If add-on's API version fits then load accounts...
	ChatProtocolAddOn* addOn = new ChatProtocolAddOn(id, imagePath, protoIndex);
	if (addOn->Version() != APP_VERSION)
		return;

	_LoadAccount(addOn, accountEntry, target);
}


void
ProtocolManager::_LoadAccount(ChatProtocolAddOn* addOn, BEntry accountEntry,
							  BHandler* target)
{
	BFile file(&accountEntry, B_READ_ONLY);
	BMessage msg;

	if (msg.Unflatten(&file) == B_OK) {
		char buffer[B_PATH_NAME_LENGTH];
		if (accountEntry.GetName(buffer) == B_OK) {
			printf("Found %s for protocol %s!\n", buffer, addOn->Protocol()->Signature());
			AddAccount(addOn, buffer, target);
		}
	}
}