#============================================================================ 

#Name        : ccmtask.py 

#Part of     : Helium 

#Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

#All rights reserved.

#This component and the accompanying materials are made available

#under the terms of the License "Eclipse Public License v1.0"

#which accompanies this distribution, and is available

#at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

#Initial Contributors:

#Nokia Corporation - initial contribution.

#

#Contributors:

#

#Description:

#===============================================================================

""" 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)

antlogger = ant.AntHandler(java_ccmtask)

rlogger = logging.getLogger('')

# replacing default handler by Ant one.

if len(rlogger.handlers) > 0:

    rlogger.removeHandler(rlogger.handlers[0])

rlogger.addHandler(antlogger)

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

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

session = None

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

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

    project.update()

    if not command.getRole():

        raise Exception("The 'role' attribute has not been defined.")

    java_ccmtask.log(str('Updating role: ' + command.getRole()))

    session.role = command.getRole()

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

    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())    

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

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

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

    project.reconcile()

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

    if not command.getDir():

        raise Exception("The 'dir' attribute has not been defined.")

    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())

    if not command.getFolder():

        raise Exception("The 'folder' attribute has not been defined.")

    if not command.getReleaseTag():

        raise Exception("The 'releaseTag' attribute has not been defined.")

    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())

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

    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)

    java_ccmtask.log(str('creating a release tag'))

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

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

    if (command.getNewTag()):

        new_release_tag = command.getNewTag();

        project.create_release_tag(command.getRelease(), new_release_tag)

    java_ccmtask.log(str('deleting a release tag'))

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

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

    if (command.getNewTag()):

        new_release_tag = command.getNewTag();

        project.delete_release_tag(command.getRelease(), new_release_tag)

    if not command.getProject():

        raise Exception("The 'project' attribute has not been defined.")

    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)

    if command.getFolder() != None:

        tasks = []        

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

        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)

    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"

    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)

    session.close()

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:

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:

    except Exception, e:

        if java_ccmtask.getVerbose() == 1:

            import traceback

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

            if 'result' in e:

                logger.error(e.result)

        logger.error(e)

        raise e