# Part of the A-A-P recipe executive: process the command line arguments.

# Copyright (C) 2002 Stichting NLnet Labs
# Permission to copy and use this file is specified in the file COPYING.
# If this file is missing you can find it here: http://www.a-a-p.org/COPYING

# The arguments come in three catogeries
# - options
#   start with a dash: in the form "-v", "--version", "-f foo".
# - variable assignments
#   contain an equal sign: "foo=asdf"
# - targets
#   the rest

# The code for handing options originally comes from SCons.



import sys
import string
import getopt

from Args import Args
from Error import *
from Version import *

#
# function called for -v and --version
#
def opt_version(opt, arg):
    print _("A-A-P recipe executive version %s; released %s %s GMT") \
		% (version_string, version_date, version_time)
	    
    sys.exit(0)

#
# function called for -h and --help
#
def opt_help(opt, arg):
    global option_list

    help_opts = filter(lambda x: x.helpline, option_list)
    print _("Usage: aap [OPTION] [VAR=VALUE] [TARGET] ...\n") \
	+ _("Options:\n") \
	+ string.join(map(lambda x: x.helpline, help_opts), "\n") + "\n"
    sys.exit(0)


#
# The following variables are filled with all the possible options.
#
option_list = []	# list of all Option objects
short_opts = ""		# string of short (single-character) options
long_opts = []		# array of long (--) options
option_dict = {}	# mapping of each option string to its Option object

def options_init():
    """Initialize command-line arguments processing.
    
    This is in a function mainly so we can easily single-step over
    it in the debugger.
    """

    class Option:
	"""Class for command-line option information.

	This exists to provide a central location for everything
	describing a command-line option, so that we can change
	options without having to update the code to handle the
	option in one place, the -h help message in another place,
	etc.  There are no methods here, only attributes.

	You can initialize an Option with the following:

	func	The function that will be called when this
		option is processed on the command line.
		Calling sequence is:

			func(opt, arg)

		If there is no func, then this Option probably
		stores an optstring to be printed.

	name	The generic option name.  Usually either func or name
		is used.  When a name is specified, the option is stored for
		later use.

	helpline
		The string to be printed in -h output.  If no
		helpline is specified but a help string is
		specified (the usual case), a helpline will be
		constructed automatically from the short, long,
		arg, and help attributes.  (In practice, then,
		setting helpline without setting func allows you
		to print arbitrary lines of text in the -h
		output.)

	short	The string for short, single-hyphen
		command-line options.
		Do not include the hyphen:

			'a' for -a, 'xy' for -x and -y, etc.

	long	An array of strings for long, double-hyphen
		command-line options.  Do not include
		the hyphens:

			['my-option', 'verbose']

	arg	If this option takes an argument, this string
		specifies how you want it to appear in the
		-h output ('DIRECTORY', 'FILE', etc.).

	help	The help string that will be printed for
		this option in the -h output.  Must be
		49 characters or fewer.

	future	If non-zero, this indicates that this feature
		will be supported in a future release, not
		the currently planned one.  SCons will
		recognize the option, but it won't show up
		in the -h output.

	The following attribute is derived from the supplied attributes:

	optstring
		A string, with hyphens, describing the flags
		for this option, as constructed from the
		specified short, long and arg attributes.

	All Option objects are stored in the global option_list list,
	in the order in which they're created.  This is the list
	that's used to generate -h output, so the order in which the
	objects are created is the order in which they're printed.
	"""

	def __init__(self,
			func = None,
			name = None,
			helpline = None,
			short = None,
			long = None,
			arg = None,
			append = 0,
			help = None,
			future = None):
	    self.func = func
	    self.name = name
	    self.short = short
	    self.long = long
	    self.arg = arg
	    self.append = append
	    self.help = help

	    # process the short and long option names to create a help line.
	    opts = []
	    if self.short:
		for c in self.short:
		    if arg:
			c = c + " " + arg
		    opts = opts + ['-' + c]
	    if self.long:
		l = self.long
		if arg:
		    l = map(lambda x,a=arg: x + "=" + a, self.long)
		opts = opts + map(lambda x: '--' + x, l)

	    self.optstring = string.join(opts, ', ')
	    if helpline:
		self.helpline = helpline
	    elif help and not future:
		if len(self.optstring) <= 26:
		    sep = " " * (28 - len(self.optstring))
		else:
		    sep = self.helpstring = "\n" + " " * 30
		self.helpline = "  " + self.optstring + sep + self.help
	    else:
		self.helpline = None

	    # Add the newly created Option object to the list of objects.
	    global option_list
	    option_list.append(self)

    # In the following instantiations, the help string should be no
    # longer than 49 characters.  Use the following as a guide:
    #	help = _("1234567890123456789012345678901234567890123456789")

    Option(func = opt_version,
	short = 'V', long = ['version'],
	help = _("Print version information and exit"))

    Option(func = opt_help,
	short = 'h', long = ['help'],
	help = _("Print help message (this one) and exit"))

    Option(name = "verbose",
	short = 'v', long = ['verbose'],
	help = _("Print more information"))

    Option(name = "silent",
	short = 's', long = ['silent'],
	help = _("Print less information"))

    Option(name = "debug",
	short = 'd', long = ['debug'], arg = 'FLAGS',
	help = _("Debug the specified items"))

    Option(name = "nobuild",
	short = 'n', long = ['nobuild'],
	help = _("Print the build commands but do not execute them"))

    Option(name = "command",
	short = 'c', long = ['command'], arg = 'CMD', append = 1,
	help = _("Execute a command after reading the recipe"))

    Option(name = "continue",
	short = 'k', long = ['continue'],
	help = _("Continue building after encountering an error"))

    Option(name = "stop",
	short = 'S', long = ['stop'],
	help = _("Stop building after encountering an error"))

    Option(name = "refresh-recipe",
	short = 'R', long = ['refresh-recipe'],
	help = _("Refresh recipe file and child recipes"))

    Option(name = "nocache",
	short = 'a', long = ['nocache'],
	help = _("Always download files, don't use the cache"))

    Option(name = "jobs",
	short = 'j', long = ['jobs'], arg = 'N',
	help = _("Maximum number of parallel jobs"))

    Option(name = "search-up",
	short = 'u', long = ['search-up', 'up'],
	help = _("Search directory tree upwards for main.aap recipe"))

    Option(name = "include",
	short = 'I', long = ['include'], arg = 'DIR', append = 1,
	help = _("Directory to search for included recipes"))

    Option(name = "recipe",
	short = 'f', long = ['recipe'], arg = 'FILE',
	help = _("Recipe file to be executed"))


    # Now that all Option objects are created, make the list of short and long
    # option names for use with getopt().
    # This also makes a dictonary to be able to find each Option object by the
    # option name.

    global short_opts
    global long_opts
    global option_dict

    for o in option_list:
	if o.short:
	    for c in o.short:
		option_dict['-' + c] = o
	    short_opts = short_opts + o.short
	    if o.arg:
		short_opts = short_opts + ":"
	if o.long:
	    for l in o.long:
		option_dict['--' + l] = o
	    if o.arg:
		long_opts = long_opts + map(lambda a: a + "=", o.long)
	    else:
		long_opts = long_opts + o.long


#
# The main function to process the command line arguments.
#
def doargs():
    """
    Process the command line arguments.
    Returns an Args object with the resuting args and targets.
    """

    global short_opts
    global long_opts
    global option_dict

    options_init()

    #
    # Start with an empty Args object.
    #
    args = Args()

    #
    # First extract the options.
    #

    # It looks like Python 2.0 changed the name of the exception class raised
    # by getopt.
    try:
	getopt_err = getopt.GetoptError
    except:
	getopt_err = getopt.error

    try:
        cmd_opts, argv = getopt.getopt(sys.argv[1:], short_opts, long_opts)
    except getopt_err, x:
	raise UserError, _("Invalid argument: ") + x
    else:
	# For the recognized options call a function and/or add the argument to
	# the argument dictionary for later use.
	# Note that a function may cause us to exit!
	for opt, arg in cmd_opts:
	    o = option_dict[opt]
	    if o.func:
		o.func(opt, arg)
	    if o.name:
		if o.append:
		    if args.options.has_key(o.name):
			args.options[o.name].append(arg)
		    else:
			args.options[o.name] = [ arg ]
		else:
		    args.options[o.name] = arg

	#
	# The remaining arguments are either variable assignments or targets.
	#
        for a in argv:
            if '=' in a:
		name, value = string.split(a, '=', 2)
		args.values[name] = value
            else:
		args.targets.append(a)
		# Using a "refresh" or "update" target implies the
		# --refresh-recipe option.
		if a == "refresh" or a == "update":
		    args.options["refresh-recipe"] = 1

    return args


# vim: set sw=4 sts=4 tw=79 fo+=l:

.