path: root/mconfig.py

import re, argparse, sys, os
from collections import OrderedDict, namedtuple
import curses.ascii

def indentify(s, indent='    '):
    return s.replace('\n', '\n' + indent)

def argv_to_shell(argv):
    quoteds = []
    for arg in argv:
        if re.match('^[a-zA-Z0-9_\.@/+-]+$', arg):
            quoteds.append(arg)
        else:
            quoted = ''
            for c in arg:
                if c == '\n':
                    quoted += r'\n'
                elif c in r'$`\"':
                    quoted += '\\' + c
                elif not curses.ascii.isprint(c):
                    quoted += r'\x%02x' % ord(c)
                else:
                    quoted += c
            quoteds.append('"' + quoted + '"')
    return ' '.join(quoteds)

class PendingOption(namedtuple('PendingOption', 'opt')):
    def need(self):
        self.opt.need()
    def __str__(self):
        return 'PendingOption(%s)' % (self.opt.name,)

class SettingsGroup(object):
    def __init__(self, group_parent=None, inherit_parent=None, name=None):
        object.__setattr__(self, 'group_parent', inherit_parent)
        object.__setattr__(self, 'inherit_parent', inherit_parent)
        object.__setattr__(self, 'vals', OrderedDict())
        if name is None:
            name = '<0x%x>' % (id(self),)
        object.__setattr__(self, 'name', name)
    @staticmethod
    def get_meat(self, attr, exctype=KeyError, allow_pending=False):
        try:
            obj = object.__getattribute__(self, 'vals')[attr]
        except KeyError:
            inherit_parent = object.__getattribute__(self, 'inherit_parent')
            if inherit_parent is not None:
                ret = SettingsGroup.get_meat(inherit_parent, attr, exctype)
                if isinstance(ret, SettingsGroup):
                    ret = self[attr] = ret.new_inheritor(name='%s.%s' % (object.__getattribute__(self, 'name'), attr))
                return ret
            raise exctype(attr)
        else:
            if not allow_pending and isinstance(obj, PendingOption):
                raise Exception("setting %r is pending a command line option, probably because you didn't 'need' it" % (attr,))
            return obj
    def __getattribute__(self, attr):
        try:
            return object.__getattribute__(self, attr)
        except AttributeError:
            return SettingsGroup.get_meat(self, attr, AttributeError)
    def __setattr__(self, attr, val):
        try:
            object.__getattribute__(self, attr)
        except:
            self[attr] = val
        else:
            object.__setattribute__(self, attr, val)
    def __getitem__(self, attr):
        return self.__getattribute__(attr)
    def __setitem__(self, attr, val):
        self.vals[attr] = val

    def __iter__(self):
        return self.vals.__iter__()
    def items(self):
        return self.vals.items()

    def __str__(self):
        s = 'SettingsGroup %s {\n' % (self.name,)
        o = self
        while True:
            for attr, val in o.vals.items():
                s += '    %s: %s\n' % (attr, indentify(str(val)))
            if o.inherit_parent is None:
                break
            o = o.inherit_parent
            s += '  [inherited from %s:]\n' % (o.name,)
        s += '}'
        return s

    def relative_lookup(self, name):
        if name.startswith('..'):
            return self.group_parent.relative_lookup(name)
        else:
            bits = name.split('.', 1)
            if len(bits) == 1:
                return (self, bits[0])
            else:
                return self[bits[0]].relative_lookup(bits[1])
    def need(self, name):
        if name not in self.vals:
            raise KeyError('need setting that has not been set')
        obj = self.vals[name]
        if hasattr(obj, 'need'):
            obj.need()

    def add_setting_option(self, name, optname, optdesc, default, **kwargs):
        def f(value):
            self[name] = value
        default = Expansion(default, self) if isinstance(default, str) else default
        opt = Option(optname, optdesc, default, f, **kwargs)
        self[name] = PendingOption(opt)

    def new_inheritor(self, *args, **kwargs):
        return SettingsGroup(inherit_parent=self, *args, **kwargs)

    def new_child(self, name, *args, **kwargs):
        sg = SettingsGroup(group_parent=self, name='%s.%s' % (self.name, name), *args, **kwargs)
        self[name] = sg
        return sg

class OptSection(object):
    def __init__(self, desc):
        self.desc = desc
        self.opts = []
        all_opt_sections.append(self)
    def move_to_end(self):
        all_opt_sections.remove(self)
        all_opt_sections.append(self)

class Option(object):
    def __init__(self, name, help, default, on_set, bool=False, show=False, section=None, **kwargs):
        self.name = name
        self.help = help
        self.default = default
        self.on_set = on_set
        self.show = show
        self.bool = bool
        self.section = section if section is not None else default_opt_section
        self.section.opts.append(self)
        assert set(kwargs).issubset({'nargs', 'type', 'choices', 'required', 'metavar'})
        self.argparse_kw = kwargs.copy()
        all_options.append(self)
        if name in all_options_by_name:
            raise KeyError('trying to create Option with duplicate name %r; old is:\n%r' % (name, all_options_by_name[name]))
        all_options_by_name[name] = self
    def __repr__(self):
        return 'Option(name=%r, help=%r, value=%r, default=%r)' % (self.name, self.help, self.value, self.default)
    def set(self, value):
        if value is None:
            value = self.default
            if callable(value):
                value = value()
        self.value = value
        if self.on_set is not None:
            self.on_set(value)
    def need(self):
        self.show = True
        if hasattr(self.default, 'need'):
            self.default.need()

class Expansion(object):
    def __init__(self, fmt, base):
        self.fmt = fmt
        def lookup(path):
            target, arg = base.relative_lookup(path)
            return SettingsGroup.get_meat(target, arg, allow_pending=True)
        self.deps = list(map(lookup, re.findall('\((.*?)\)', fmt)))
    def __repr__(self):
        return 'Expansion(%r)' % (self.fmt,)
    def need(self):
        for dep in self.deps:
            if hasattr(dep, 'need'):
                dep.need()
    def __call__(self):
        deps = self.deps[:]
        return re.sub('\((.*?)\)', lambda m: deps.pop(0), self.fmt)

def installation_dirs_group(sg):
    section = OptSection('Fine tuning of the installation directories:')
    for name, optname, optdesc, default in [
        ('prefix', 'prefix', '', '/usr/local'),
        ('exec_prefix', 'exec-prefix', '', '(prefix)'),
        ('bin', 'bindir', '', '(exec_prefix)/bin'),
        ('sbin', 'sbindir', '', '(exec_prefix)/sbin'),
        ('libexec', 'libexecdir', '', '(exec_prefix)/libexec'),
        ('etc', 'sysconfdir', '', '(prefix)/etc'),
        ('var', 'localstatedir', '', '(prefix)/var'),
        ('lib', 'libdir', '', '(prefix)/lib'),
        ('include', 'includedir', '', '(prefix)/include'),
        ('datarootdir', 'datarootdir', '', '(prefix)/share'),
        ('share', 'datadir', '', '(datarootdir)'),
        ('locale', 'localedir', '', '(datarootdir)/locale'),
        ('man', 'mandir', '', '(datarootdir)/man'),
        ('doc', 'docdir', '', '(datarootdir)/doc/(..package_unix_name)'),
        ('html', 'htmldir', '', '(doc)'),
        ('pdf', 'pdfdir', '', '(doc)'),
    ]:
        sg.add_setting_option(name, optname, optdesc, default, section=section)
    for ignored in ['sharedstatedir', 'oldincludedir', 'infodir', 'dvidir', 'psdir']:
        Option(ignored, 'Ignored autotools compatibility setting', '', None, section=section)

def _make_argparse(include_all):
    parser = argparse.ArgumentParser(add_help=False, usage='configure [OPTION]... [VAR=VALUE]...')
    parser.add_argument('--help', action='store_true', help='Show this help', dest='__help')
    parser.add_argument('--help-all', action='help', help='Show this help, including unused options')
    for sect in all_opt_sections:
        if not include_all and not any(opt.show for opt in sect.opts):
            continue
        ag = parser.add_argument_group(description=sect.desc)
        for opt in sect.opts:
            if not include_all and not opt.show: continue
            ag.add_argument('--' + opt.name,
                            action='store_true' if opt.bool else 'store',
                            dest=opt.name,
                            help=opt.help,
                            **opt.argparse_kw)
    return parser

def parse_args():
    default_opt_section.move_to_end()
    parser = _make_argparse(include_all=True)
    args, argv = parser.parse_known_args()
    if args.__help:
        parser = _make_argparse(include_all=False)
        parser.print_help()
        sys.exit(0)
    def do_env_arg(arg):
        m = re.match('([^ ]+)=(.*)', arg)
        if m:
            os.environ[m.group(1)] = m.group(2)
            return False
        return True
    argv = list(filter(do_env_arg, argv))
    if argv:
        parser = _make_argparse(include_all=False)
        parser.error('unrecognized arguments: %s' % (argv_to_shell(argv),))

    for opt in all_options:
        if opt.show:
            opt.set(getattr(args, opt.name))


#class 

all_options = []
all_options_by_name = {}
all_opt_sections = []
default_opt_section = OptSection('Uncategorized options:')

settings_root = SettingsGroup(name='root')
installation_dirs_group(settings_root.new_child('idirs'))

# --