""" Script to process Ant <ccm> task commands. """

import ccm

import ccm.extra

import sys

import logging

import ant

import string

logging.basicConfig(level=logging.INFO)

logging.getLogger("ccm").setLevel(logging.INFO)

antlogger = ant.AntHandler(java_ccmtask)

logger = logging.getLogger("ccm.ant")

logger.addHandler(antlogger)

session = None

def execute_update(command):

    java_ccmtask.log(str('Updating project: ' + command.getProject()))

    project = session.create(command.getProject())

    project.update()

def execute_synchronize(command):

    if command.getProject() is not None:

        java_ccmtask.log(str('Synchronizing project: ' + command.getProject()))

        java_ccmtask.log(str("Recursive: %s" % command.getRecursive()))

        project = session.create(command.getProject())

        project.sync(command.getRecursive())    

    else:

        raise Exception("Error during synchronization. Project is not defined.")

def execute_reconcile(command):

    java_ccmtask.log(str('Reconciling project: ' + command.getProject()))

    project = session.create(command.getProject())

    project.reconcile()

def execute_snapshot(command):

    java_ccmtask.log(str('Snapshot of project: ' + command.getProject() + ' to ' + command.getDir()))

    project = session.create(command.getProject())

    if command.getFast()== True:

        ccm.extra.FastSnapshot(project, command.getDir())        

    else:

        project.snapshot(command.getDir(), command.getRecursive())

def execute_changereleasetag(command):

    java_ccmtask.log(str('Changing release tag for all tasks in the folder : ' + command.getFolder() + ' as ' + command.getReleaseTag()))

    #Search all task from the folder    

    folder = session.create("Folder " + str(command.getFolder()))

    for task in folder.tasks:

        if task.release != str(command.getReleaseTag()):

            task.release = str(command.getReleaseTag())

def execute_checkout(command):

    java_ccmtask.log(str('Checking out project: ' + command.getProject() + ' with release tag ' + command.getRelease()))

    project = session.create(command.getProject())

    if (command.getRelease()):

        if command.getWa():

            wa = command.getWa()

        else:

            wa = None

        if command.getRecursive():

            recursive = command.getRecursive()

        else:

            recursive = None

        if command.getRelative():

            relative = command.getRelative()

        else:

            relative = None

        if command.getVersion():

            version = command.getVersion()

        else:

            version = None

        if command.getPurpose():

            purpose = command.getPurpose()

        else:

            purpose = None

        project.checkout(session.create(command.getRelease()), version, purpose, recursive)

def execute_workarea(command):

    java_ccmtask.log(str('Modifying work area for the project : ' + command.getProject()))

    project = session.create(command.getProject())

    if command.getMaintain():

        maintain = command.getMaintain()

    else:

        maintain = None

    if command.getRecursive():

        recursive = command.getRecursive()

    else:

        recursive = None

    if command.getRelative():

        relative = command.getRelative()

    else:

        relative = None

    if command.getPath():

        path = command.getPath()

    else:

        path = None

    if command.getPst():

        pst = command.getPst()

    else:

        pst = None

    if command.getWat():

        wat = command.getWat()

    else:

        wat = None

    project.work_area(maintain, recursive, relative, path, pst, wat)

def execute_addtask(command):

    if command.getFolder() != None:

        tasks = []        

        ccmfolder = session.create("Folder " + command.getFolder())

        role = find_folder_information(command)

        if role == "build_mgr":

            java_ccmtask.log("Changing role to %s" % role)

            session.role = "build_mgr"

        folder = session.create("Folder %s" % str(command.getFolder()))

        for task in folder.tasks:

            folder.remove(task)

        java_ccmtask.log("Adding task to folder %s" % ccmfolder)

        tasks.extend(map(lambda task: session.create("Task " + task.getName()), command.getTasks()))

        for task in tasks:

            ccmfolder.append(task)

def find_folder_information(command):

    cmdline = "folder -sh i " + command.getFolder()    

    result = session.execute(cmdline)

    result_array = str(result).split('\r')

    for ldata in result_array :        

        if ldata.find('Writable By:') > 0:

            wb,wd = ldata.split(':',1)

            if wd.lstrip() == "Build Manager":

                return "build_mgr"

            else:

                return "developer"

def execute_exists(command):

    fpn = None

    if command.getObject() != None:

        fpn = command.getObject()

    elif command.getFolder() != None:

		fpn = "Folder " + command.getFolder()

    elif command.getTask() != None:

		fpn = "Task " + command.getTask()

    if fpn == None:

        raise Exception("You need to define either a 'task'/'folder'/'object' attribute.")

    ccmo = session.create(fpn)

    if ccmo.exists():

        java_ccmtask.log("'%s' exists." % ccmo)

    else:

        raise Exception("Could not find '%s'." % ccmo)

def execute_close(command):

    global session

    java_ccmtask.log(str("Closing session %s." % session))

    session.close_on_exit = True

    session.close()

    session = None

def referenceToObject(obj):

    if obj.isReference() == 1:

        ref = project.getReference(str(obj.getRefid().getRefId()))

        if ref == None:

            raise Exception("Could not find reference '%s'" % str(obj.getRefid().getRefId()))

        return ref

    else:

        return obj        

sessionIds = [] 

for session_set in java_ccmtask.getSessionSets():

    sessionIds.extend(map(lambda session: session.getAddr(), referenceToObject(session_set).getSessions()))

print "Session list: ", sessionIds

if len(sessionIds) > 0:

    sessionid = sessionIds.pop()

    session = ccm.Session(username=None, engine=None, dbpath=None, ccm_addr=sessionid, close_on_exit=False)

else:

   username = java_ccmtask.getUsername()

   password = java_ccmtask.getPassword()

   session = ccm.open_session(username=username, password=password)

if java_ccmtask.getVerbose() == 1:

    logging.getLogger("ccm").setLevel(logging.DEBUG)

#print dir(sys.modules['__main__'] )

ccm_commands = java_ccmtask.getCommands()

for command in ccm_commands:

    print "Running command '%s'" % command.getName()

    method_name = 'execute_' + command.getName()

    method = sys.modules['__main__'].__dict__[method_name]

    try:

        method(command)

    except Exception, e:

        import traceback

        logger.error(traceback.print_exc(file=sys.stdout))

        logger.error(e.result)

        logger.error(e)

        raise e