//
//  MCAccountValidator.cpp
//  mailcore2
//
//  Created by Christopher Hockley on 22/01/15.
//  Copyright (c) 2015 MailCore. All rights reserved.
//

#include "MCAccountValidator.h"
#include "MCMailProvider.h"
#include "MCMailProvidersManager.h"
#include "MCIMAPAsyncSession.h"
#include "MCPOPAsyncSession.h"
#include "MCSMTPAsyncSession.h"
#include "MCNetService.h"
#include "MCAddress.h"
#include "MCIMAPOperation.h"
#include "MCPOPOperation.h"
#include "MCSMTPOperation.h"
#include "MCMXRecordResolverOperation.h"

using namespace mailcore;

/* this is the service being tested */

enum {
    SERVICE_IMAP,  /* IMAP service */
    SERVICE_POP,   /* POP  service */
    SERVICE_SMTP,  /* SMTP service */
};

void AccountValidator::init()
{
    mEmail = NULL;
    mUsername = NULL;
    mPassword = NULL;
    mOAuth2Token = NULL;
    
    mImapServices = new Array();
    mSmtpServices = new Array();
    mPopServices = new Array();
    
    mIdentifier = NULL;
    mImapServer = NULL;
    mPopServer = NULL;
    mSmtpServer = NULL;
    mImapError = ErrorNone;
    mPopError = ErrorNone;
    mSmtpError = ErrorNone;
    
    mCurrentServiceIndex = 0;
    mCurrentServiceTested = 0;
    
    mProvider = NULL;
    
    mOperation = NULL;
    mQueue = NULL;
    mResolveMX = NULL;
    
    mImapSession = NULL;
    mPopSession = NULL;
    mSmtpSession = NULL;
}

AccountValidator::AccountValidator()
{
    init();
}

AccountValidator::~AccountValidator()
{
    MC_SAFE_RELEASE(mEmail);
    MC_SAFE_RELEASE(mUsername);
    MC_SAFE_RELEASE(mPassword);
    MC_SAFE_RELEASE(mOAuth2Token);
    MC_SAFE_RELEASE(mImapServices);
    MC_SAFE_RELEASE(mSmtpServices);
    MC_SAFE_RELEASE(mPopServices);
    MC_SAFE_RELEASE(mIdentifier);
    MC_SAFE_RELEASE(mProvider);
    MC_SAFE_RELEASE(mOperation);
    MC_SAFE_RELEASE(mQueue);
    MC_SAFE_RELEASE(mResolveMX);
    MC_SAFE_RELEASE(mImapSession);
    MC_SAFE_RELEASE(mPopSession);
    MC_SAFE_RELEASE(mSmtpSession);
}

void AccountValidator::start()
{
    if (mEmail == NULL) {
        if (mUsername == NULL) {
            return;
        }
        else {
            mEmail = mUsername;
        }
    }
    else if (mUsername == NULL){
        mUsername = mEmail;
    }
    
    mProvider = MailProvidersManager::sharedManager()->providerForEmail(mUsername);
    
    if (mProvider == NULL) {
        resolveMX();
    }
    else{
        startCheckingHosts();
    }
}

void AccountValidator::cancel()
{
    if(mOperation != NULL)
        mOperation->cancel();
    
    if(mResolveMX != NULL)
        mResolveMX->cancel();
    
    if (mQueue != NULL)
        mQueue->cancelAllOperations();
    
    MC_SAFE_RELEASE(mImapSession);
    MC_SAFE_RELEASE(mPopSession);
    MC_SAFE_RELEASE(mSmtpSession);
        
    Operation::cancel();
}

void AccountValidator::operationFinished(Operation * op)
{
    if (op == mResolveMX) {
        resolveMXDone();
    }
    else {
        checkNextHostDone();
    }
}

void AccountValidator::resolveMX()
{
    Array * components;
    String * domain;
    
    components = mUsername->componentsSeparatedByString(MCSTR("@"));
    if (components->count() >= 2) {
        domain = (String *) components->lastObject();
        mResolveMX = new MXRecordResolverOperation();
        mResolveMX->setHostname(domain);
        mResolveMX->setCallback((OperationCallback *)this);
        
        mQueue = new OperationQueue();
        mQueue->addOperation(mResolveMX);
    }
}

void AccountValidator::resolveMXDone()
{
    Array * mxRecords = mResolveMX->mxRecords();
    
    CFURLRef imageURL = CFBundleCopyResourceURL(CFBundleGetMainBundle(), CFSTR("providers"), CFSTR("json"), NULL);
    CFStringRef imagePath = CFURLCopyFileSystemPath(imageURL, kCFURLPOSIXPathStyle);
    const char *path = CFStringGetCStringPtr(imagePath, CFStringGetSystemEncoding());
    String * sPath = String::stringWithUTF8Characters(path);
    
    MailProvidersManager::sharedManager()->registerProvidersWithFilename(sPath);
    
    mc_foreacharray(String, mxRecord, mxRecords) {
        MailProvider *provider = MailProvidersManager::sharedManager()->providerForMX(mxRecord);
        if (provider){
            mProvider = provider;
            break;
        }
    }
    
    startCheckingHosts();
}

void AccountValidator::startCheckingHosts()
{
    if (mProvider != NULL) {
        mIdentifier = mProvider->identifier();
        
        if (mImapServices->count() == 0 and mProvider->imapServices()->count() > 0)
            mImapServices = mProvider->imapServices();
        
        if (mPopServices->count() == 0 and mProvider->popServices()->count() > 0)
            mPopServices = mProvider->popServices();
        
        if (mSmtpServices->count() == 0 and mProvider->smtpServices()->count() > 0)
            mSmtpServices = mProvider->smtpServices();
    }
    
    if (mImapServices->count() == 0)
        mImapError = ErrorNoValidServerFound;
    
    if (mPopServices->count() == 0)
        mPopError = ErrorNoValidServerFound;
    
    if (mSmtpServices->count() == 0)
        mSmtpError = ErrorNoValidServerFound;
    
    checkNextHost();
}

/**
 Each service(IMAP/POP/SMTP) is tested one after the other.
 For each service we test each server details (NetService),
 Until either:  
    we find on that works and returns ErrorNone in checkNextHostDone().
    we have gone trough the Array of NetService for that service and checkNextHost() is then called for the next service.
 */
void AccountValidator::checkNextHost()
{
    if (mCurrentServiceTested == SERVICE_IMAP) {
        if (mCurrentServiceIndex < mImapServices->count()) {
            
            if (mImapSession == NULL) {
                mImapSession = new IMAPAsyncSession();
                mImapSession->setUsername(mUsername);
                mImapSession->setPassword(mPassword);
            }
            
            mImapServer = (NetService *) mImapServices->objectAtIndex(mCurrentServiceIndex);
            mImapSession->setHostname(mImapServer->hostname());
            mImapSession->setPort(mImapServer->port());
            mImapSession->setConnectionType(mImapServer->connectionType());
            
            mOperation = (IMAPOperation *)mImapSession->checkAccountOperation();
            mOperation->setCallback((OperationCallback *)this);
            mOperation->start();
        
        }
        else {
            mCurrentServiceTested ++;
            mCurrentServiceIndex = 0;
            checkNextHost();
        }
    }
    else if (mCurrentServiceTested == SERVICE_POP){
        if (mCurrentServiceIndex < mPopServices->count()) {
            POPAsyncSession *popSession = new POPAsyncSession();
            popSession->setUsername(mUsername);
            popSession->setPassword(mPassword);
            
            mPopServer = (NetService *) mPopServices->objectAtIndex(mCurrentServiceIndex);
            popSession->setHostname(mPopServer->hostname());
            popSession->setPort(mPopServer->port());
            popSession->setConnectionType(mPopServer->connectionType());
            
            mOperation = (POPOperation *)popSession->checkAccountOperation();
            mOperation->setCallback((OperationCallback *)this);
            mOperation->start();
        }
        else {
            mCurrentServiceTested ++;
            mCurrentServiceIndex = 0;
            checkNextHost();
        }
    }
    else if (mCurrentServiceTested == SERVICE_SMTP){
        if (mCurrentServiceIndex < mSmtpServices->count()) {
            SMTPAsyncSession *smtpSession = new SMTPAsyncSession();
            smtpSession->setUsername(mUsername);
            smtpSession->setPassword(mPassword);
            
            mSmtpServer = (NetService *) mSmtpServices->objectAtIndex(mCurrentServiceIndex);
            smtpSession->setHostname(mSmtpServer->hostname());
            smtpSession->setPort(mSmtpServer->port());
            smtpSession->setConnectionType(mSmtpServer->connectionType());
            
            mOperation =  (SMTPOperation *)smtpSession->checkAccountOperation(Address::addressWithMailbox(mEmail));
            mOperation->setCallback((OperationCallback *)this);
            mOperation->start();

        }
        else {
            mCurrentServiceTested ++;
            mCurrentServiceIndex = 0;
            checkNextHost();
        }
    }
    else {
        MC_SAFE_RELEASE(mPopSession);
        MC_SAFE_RELEASE(mSmtpSession);
        callback()->operationFinished(this);
    }
}

void AccountValidator::checkNextHostDone()
{
    ErrorCode error = ErrorNone;
    
    if (mCurrentServiceTested == SERVICE_IMAP) {
        mImapError = ((IMAPOperation *)mOperation)->error();
        error = mImapError;
    }
    else if (mCurrentServiceTested == SERVICE_POP) {
        mPopError = ((POPOperation *)mOperation)->error();
        error = mPopError;
    }
    else if (mCurrentServiceTested == SERVICE_SMTP) {
        mSmtpError = ((SMTPOperation *)mOperation)->error();
        error = mSmtpError;
    }
    
    if (error == ErrorNone) {
        mCurrentServiceTested ++;
    }
    else {
        mCurrentServiceIndex ++;
    }
    
    checkNextHost();
}

void AccountValidator::setEmail(String * email)
{
    MC_SAFE_REPLACE_COPY(String, mEmail, email);
}

String * AccountValidator::email()
{
    return mEmail;
}

void AccountValidator::setUsername(String * username)
{
    MC_SAFE_REPLACE_COPY(String, mUsername, username);
}

String * AccountValidator::username()
{
    return mUsername;
}

void AccountValidator::setPassword(String * password)
{
    MC_SAFE_REPLACE_COPY(String, mPassword, password);
}

String * AccountValidator::password()
{
    return mPassword;
}

void AccountValidator::setOAuth2Token(String * OAuth2Token)
{
    MC_SAFE_REPLACE_COPY(String, mOAuth2Token, OAuth2Token);
}

String * AccountValidator::OAuth2Token()
{
    return mOAuth2Token;
}

void AccountValidator::setImapServices(Array * imapServices)
{
    MC_SAFE_REPLACE_COPY(Array, mImapServices, imapServices);
}

Array * AccountValidator::imapServices()
{
    return mImapServices;
}

void AccountValidator::setSmtpServices(Array * smtpServices)
{
    MC_SAFE_REPLACE_COPY(Array, mSmtpServices, smtpServices);
}

Array * AccountValidator::smtpServices()
{
    return mSmtpServices;
}

void AccountValidator::setPopServices(Array * popServices)
{
    MC_SAFE_REPLACE_COPY(Array, mPopServices, popServices);
}

Array * AccountValidator::popServices()
{
    return mPopServices;
}

String * AccountValidator::identifier()
{
    return mIdentifier;
}

NetService * AccountValidator::imapServer()
{
    return mImapServer;
}

NetService * AccountValidator::smtpServer()
{
    return mSmtpServer;
}

NetService * AccountValidator::popServer()
{
    return mPopServer;
}

ErrorCode AccountValidator::imapError()
{
    return mImapError;
}

ErrorCode AccountValidator::popError()
{
    return mPopError;
}

ErrorCode AccountValidator::smtpError()
{
    return mSmtpError;
}