#! /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:

#

# Description:

# Python script to manipulate the hgrc files

from ConfigParser import *

import optparse

import os

import sys

import re

verbose = False;

credentials= re.compile(r"//.*?@")

def strip_credentials(hgrc):

    """  Remove the user credentials from the default path in hgrc file"""

    # e.g.

    # before http://user:pass@prod.foundationhost.org/sfl/MCL/sf/os/boardsupport/

    # after  http://prod.foundationhost.org/sfl/MCL/sf/os/boardsupport/

    if hgrc.has_section('paths'):

        if (verbose): print hgrc.items('paths')

        defpath = hgrc.get('paths', 'default')

        newpath = credentials.sub(r"//",defpath)

        #print "new path ", newpath

        hgrc.set('paths', 'default',newpath)

    elif (verbose):

        if (verbose): print "No [paths] section\n"

def add_hooks(hgrc):

    if (hgrc.has_section('hooks')):

        # unpdate

        if (verbose) : print 'updating existing hooks section'

    else:

        if (verbose) : print 'adding hooks section'

        hgrc.add_section('hooks')

    # add example (windows only) hook to block local commit to the repo

    hgrc.set('hooks', 'pretxncommit.abort', 'exit /b 1')

    hgrc.set('hooks', 'pretxncommit.message', 'ERROR: This is a read only repo')

def write_hgrcfile(hgrc,fout):

    fnewini = file(fout,'w')

    hgrc.write(fnewini)

    fnewini.close()

def main():

    global verbose

    usage = "usage: %prog [options]"

    try:

        parser = optparse.OptionParser(usage)

        parser.set_defaults(filename=".hg/hgrc")

        parser.add_option("-f","--file", dest="filename", default=".hg/hgrc",metavar="FILE" , help='file to be patched')

        parser.add_option("-v", action="store_true",dest="verbose",default=False, help='Verbose trace information')

        (options, args) = parser.parse_args()

    except:

        parser.print_help()

        sys.exit(1)

    f = os.path.abspath(options.filename)

    if(options.verbose):

        verbose = True

        print f

    if(os.path.isfile(f)):

        try:

            #conff = file(f,'w')  #open file f for read/write

            hgrcfile = RawConfigParser()

            hgrcfile.read(f)

            if (verbose):

                print hgrcfile.sections()

        except:

            print 'Something failed opening the configuration file'

            sys.exit(2)

    else:

        print "Configuration file does not exist? ",f

        sys.exit(2)

    strip_credentials(hgrcfile)

    add_hooks(hgrcfile)

    write_hgrcfile(hgrcfile,f)

if __name__ == "__main__":

    main()