#! /usr/bin/python
# Copyright (c) 2009 Symbian Foundation Ltd
# 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:
# Symbian Foundation Ltd - initial contribution.
#
# Contributors:
#
# applytag.py - apply local mercurial tag based on one or more csv files
#                which contain a list of repos and changesets
#                for any <name>.csv file use the <name> is used as the tag name
#

repo_root='c:/sf'
debugfile = "debug.out"

import re
import StringIO
import os
import sys

from mercurial import util
from mercurial.i18n import _

import subprocess

sourceline = re.compile('(.*?)(oss|sfl)/(FCL|MCL)/(.*)',re.I)
MCL = re.compile('^MCL$',re.I)
re_csvfile = re.compile(r"(.*?)\.csv$",re.I)
csvfields = re.compile(r"(.*?),(.*?),(.*?),(.*?),.*")
verbose = True


def debug_info(str):
    hgui.debug(_(str))
    dbgfile.write(str)

def debug_warn(str):
    hgui.write(_(str))
    dbgfile.write(str)

def debug_verbose(str):
    if verbose:
        hgui.debug(_(str))
        dbgfile.write(str)

def commit_hook(ui, repo, hooktype, node=None, source=None, **kwargs):
    # iterate over all the files in added changeset
    global hgui
    global dbgfile
    hgui = ui
    ctx = repo[node]
    dbgfile = file(debugfile,'w')  #open file f for write
    for f in ctx.files():
        debug_info(('file %s \n') % (f))
        csvname = re_csvfile.match(f)
        if (csvname):
            debug_info(('csv file %s\n') % f)
            fctx = ctx.filectx(f)
            filecontent = fctx.data()
            tagname = csvname.group(1).upper()
            debug_info(('tag to use : %s\n')% tagname)
            csvfp = StringIO.StringIO(filecontent)
            parse_csv(ui,csvfp,tagname)
            csvfp.close()
    dbgfile.close()
    # return False meand "don't block"
    return False

def walk_repos(ui,rootpath):
    p = util.pconvert(rootpath)
    for path in util.walkrepos(p, followsym=True):
        debug_info(('repo %s\n') % path)




def parse_csv(ui,fp,tagname):
    '''
        Take a list of repositories (from csv file)
        For each repository apply a local tag
        to the local copy of the repo using the
        specified changeset
    '''
    global verbose
    for line in fp.readlines():
        # check it is basic csv and extract repo url and changeset
        cline = csvfields.match(line)
        if(cline):
            reposrc = cline.group(1)
            changeset = cline.group(4)
        else:
            debug_warn(('ignoring : %s\n' %line))

        # now check the repo url looks valid
        m = sourceline.match(reposrc)
        if (m):
            # construct local path
            applytag(ui,repo_root,m.group(2),m.group(3),m.group(4),changeset,tagname)
            #if we have just tagged the MCL then tag the FCL as well
            if(MCL.match(m.group(3))):
                applytag(ui,repo_root,m.group(2),"FCL",m.group(4),changeset,tagname)
        else:
            debug_warn(('Ignoring Source Line: %s\n' %reposrc))

def applytag(ui,path_root,p_license,p_branch,p_repo,rev,tagname):
    repoabs = '/'.join([path_root,p_license,p_branch,p_repo])
    repoabs.replace("\\","/")
    tagcmd = ['hg','-R',repoabs,'tag','--local','--force','-r',rev,tagname]
    #print tagcmd
    run(ui,tagcmd)

def run(ui,cmd):
    debug_verbose(('cmd %s\n' % ' '.join(cmd)))
    try:
        p = subprocess.Popen(args=cmd, bufsize=1024, shell = False, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
	cmdout, errtxt = p.communicate()
        debug_warn((cmdout))
    except IOError as err:
        debug_warn(("IOError : ",err))
    except OSError as err:
        debug_warn(("OSError : ",err))
    except:
        debug_warn(("Error %s : Could not run %s\n" % (sys.exc_info()[0],cmd)))
        return -1