# fancyopts.py - better command line parsing

#

#  Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

import getopt

def gnugetopt(args, options, longoptions):

    """Parse options mostly like getopt.gnu_getopt.

    This is different from getopt.gnu_getopt in that an argument of - will

    become an argument of - instead of vanishing completely.

    """

    extraargs = []

    if '--' in args:

        stopindex = args.index('--')

        extraargs = args[stopindex + 1:]

        args = args[:stopindex]

    opts, parseargs = getopt.getopt(args, options, longoptions)

    args = []

    while parseargs:

        arg = parseargs.pop(0)

        if arg and arg[0] == '-' and len(arg) > 1:

            parseargs.insert(0, arg)

            topts, newparseargs = getopt.getopt(parseargs, options, longoptions)

            opts = opts + topts

            parseargs = newparseargs

        else:

            args.append(arg)

    args.extend(extraargs)

    return opts, args

def fancyopts(args, options, state, gnu=False):

    """

    read args, parse options, and store options in state

    each option is a tuple of:

      short option or ''

      long option

      default value

      description

      option value label(optional)

    option types include:

      boolean or none - option sets variable in state to true

      string - parameter string is stored in state

      list - parameter string is added to a list

      integer - parameter strings is stored as int

      function - call function with parameter

    non-option args are returned

    """

    namelist = []

    shortlist = ''

    argmap = {}

    defmap = {}

    for option in options:

        if len(option) == 5:

            short, name, default, comment, dummy = option

        else:

            short, name, default, comment = option

        # convert opts to getopt format

        oname = name

        name = name.replace('-', '_')

        argmap['-' + short] = argmap['--' + oname] = name

        defmap[name] = default

        # copy defaults to state

        if isinstance(default, list):

            state[name] = default[:]

        elif hasattr(default, '__call__'):

            state[name] = None

        else:

            state[name] = default

        # does it take a parameter?

        if not (default is None or default is True or default is False):

            if short:

                short += ':'

            if oname:

                oname += '='

        if short:

            shortlist += short

        if name:

            namelist.append(oname)

    # parse arguments

    if gnu:

        parse = gnugetopt

    else:

        parse = getopt.getopt

    opts, args = parse(args, shortlist, namelist)

    # transfer result to state

    for opt, val in opts:

        name = argmap[opt]

        t = type(defmap[name])

        if t is type(fancyopts):

            state[name] = defmap[name](val)

        elif t is type(1):

            state[name] = int(val)

        elif t is type(''):

            state[name] = val

        elif t is type([]):

            state[name].append(val)

        elif t is type(None) or t is type(False):

            state[name] = True

    # return unparsed args

    return args