#include "stdafx.h"

static int CompareServices(const CCloudService *p1, const CCloudService *p2)
{
	return mir_strcmp(p1->GetAccountName(), p2->GetAccountName());
}

LIST<CCloudService> Services(10, CompareServices);

static INT_PTR GetService(WPARAM wParam, LPARAM lParam)
{
	ptrA accountName(mir_strdup((char*)wParam));
	if (!accountName || !mir_strlen(accountName))
		accountName = db_get_sa(NULL, MODULE, "DefaultService");
	if (accountName == nullptr)
		return 2;

	CCloudServiceSearch search(accountName);
	CCloudService *service = Services.find(&search);
	if (service == nullptr)
		return 3;

	CFSERVICEINFO *info = (CFSERVICEINFO*)lParam;
	if (info != nullptr) {
		info->accountName = service->GetAccountName();
		info->userName = service->GetUserName();
	}

	return 0;
}

static INT_PTR EnumServices(WPARAM wParam, LPARAM lParam)
{
	CFSERVICEINFO info = {};
	enumCFServiceFunc enumFunc = (enumCFServiceFunc)wParam;
	void *param = (void*)lParam;

	for (auto &service : Services) {
		info.accountName = service->GetAccountName();
		info.userName = service->GetUserName();
		int res = enumFunc(&info, param);
		if (res != 0)
			return res;
	}

	return 0;
}

INT_PTR Upload(WPARAM wParam, LPARAM lParam)
{
	CFUPLOADDATA *uploadData = (CFUPLOADDATA*)wParam;
	if (uploadData == nullptr)
		return 1;

	ptrA accountName(mir_strdup(uploadData->accountName));
	if (!mir_strlen(accountName))
		accountName = db_get_sa(NULL, MODULE, "DefaultService");
	if (accountName == nullptr)
		return 2;

	CCloudServiceSearch search(uploadData->accountName);
	CCloudService *service = Services.find(&search);
	if (service == nullptr)
		return 3;

	FileTransferParam *ftp = new FileTransferParam(0);
	ftp->SetWorkingDirectory(uploadData->localPath);
	ftp->SetServerFolder(uploadData->serverFolder);

	if (PathIsDirectory(uploadData->localPath))
	{
		// temporary unsupported
		Transfers.remove(ftp);
		delete ftp;

		return ACKRESULT_FAILED;
	}
	else
		ftp->AddFile(uploadData->localPath);

	int res = service->Upload(ftp);
	if (res == ACKRESULT_SUCCESS && lParam) {
		CFUPLOADRESULT *result = (CFUPLOADRESULT*)lParam;
		const char **links = ftp->GetSharedLinks(result->linkCount);
		result->links = (char**)mir_calloc(sizeof(char*) * result->linkCount);
		for (size_t i = 0; i < result->linkCount; i++)
			result->links[i] = mir_strdup(links[i]);
		result->description = mir_wstrdup(ftp->GetDescription());
	}

	Transfers.remove(ftp);
	delete ftp;

	return res;
}

void InitializeServices()
{
	PROTOCOLDESCRIPTOR pd = { sizeof(pd) };
	pd.type = PROTOTYPE_PROTOCOL;

	pd.szName = MODULE "/Dropbox";
	pd.fnInit = (pfnInitProto)CDropboxService::Init;
	pd.fnUninit = (pfnUninitProto)CDropboxService::UnInit;
	Proto_RegisterModule(&pd);

	pd.szName = MODULE "/GDrive";
	pd.fnInit = (pfnInitProto)CGDriveService::Init;
	pd.fnUninit = (pfnUninitProto)CGDriveService::UnInit;
	Proto_RegisterModule(&pd);

	pd.szName = MODULE "/OneDrivre";
	pd.fnInit = (pfnInitProto)COneDriveService::Init;
	pd.fnUninit = (pfnUninitProto)COneDriveService::UnInit;
	Proto_RegisterModule(&pd);

	pd.szName = MODULE "/YandexDisk";
	pd.fnInit = (pfnInitProto)CYandexService::Init;
	pd.fnUninit = (pfnUninitProto)CYandexService::UnInit;
	Proto_RegisterModule(&pd);

	pd.szName = MODULE;
	pd.type = PROTOTYPE_FILTER;
	Proto_RegisterModule(&pd);

	CreateServiceFunction(MODULE PSS_FILE, &CCloudService::SendFileInterceptor);

	CreateServiceFunction(MS_CLOUDFILE_GETSERVICE, GetService);
	CreateServiceFunction(MS_CLOUDFILE_ENUMSERVICES, EnumServices);
	CreateServiceFunction(MS_CLOUDFILE_UPLOAD, Upload);
}