Subversion Repositories HelenOS

#!/usr/bin/env python

#

# Copyright (c) 2006 Ondrej Palkovsky

# Copyright (c) 2009 Martin Decky

# All rights reserved.

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions

# are met:

#

# - Redistributions of source code must retain the above copyright

#   notice, this list of conditions and the following disclaimer.

# - Redistributions in binary form must reproduce the above copyright

#   notice, this list of conditions and the following disclaimer in the

#   documentation and/or other materials provided with the distribution.

# - The name of the author may not be used to endorse or promote products

#   derived from this software without specific prior written permission.

#

# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#

"""

HelenOS configuration system

"""

import sys

import os

import re

import commands

import xtui

INPUT = sys.argv[1]

OUTPUT = 'Makefile.config'

def read_defaults(fname, defaults):

	"Read saved values from last configuration run"

	inf = file(fname,'r')

	for line in inf:

		res = re.match(r'^(?:#!# )?([^#]\w*)\s*=\s*(.*?)\s*$', line)

		if (res):

			defaults[res.group(1)] = res.group(2)

	inf.close()

def check_condition(text, defaults, ask_names):

	"Check for condition"

	ctype = 'cnf'

	if ((')|' in text) or ('|(' in text)):

		ctype = 'dnf'

	if (ctype == 'cnf'):

		conds = text.split('&')

	else:

		conds = text.split('|')

	for cond in conds:

		if (cond.startswith('(')) and (cond.endswith(')')):

			cond = cond[1:-1]

		inside = check_inside(cond, defaults, ctype)

		if (ctype == 'cnf') and (not inside):

			return False

		if (ctype == 'dnf') and (inside):

			return True

	if (ctype == 'cnf'):

		return True

	return False

def check_inside(text, defaults, ctype):

	"Check that the condition specified on input line is True (only CNF is supported)"

	if (ctype == 'cnf'):

		conds = text.split('|')

	else:

		conds = text.split('&')

	for cond in conds:

		res = re.match(r'^(.*?)(!?=)(.*)$', cond)

		if (not res):

			raise RuntimeError("Invalid condition: %s" % cond)

		condname = res.group(1)

		oper = res.group(2)

		condval = res.group(3)

		if (not defaults.has_key(condname)):

			varval = ''

		else:

			varval = defaults[condname]

		if (ctype == 'cnf'):

			if (oper == '=') and (condval == varval):

				return True

			if (oper == '!=') and (condval != varval):

				return True

		else:

			if (oper == '=') and (condval != varval):

				return False

			if (oper == '!=') and (condval == varval):

				return False

	if (ctype == 'cnf'):

		return False

	return True

def parse_config(fname, ask_names):

	"Parse configuration file"

	inf = file(fname, 'r')

	name = ''

	choices = []

	for line in inf:

		if (line.startswith('!')):

			# Ask a question

			res = re.search(r'!\s*(?:\[(.*?)\])?\s*([^\s]+)\s*\((.*)\)\s*$', line)

			if (not res):

				raise RuntimeError("Weird line: %s" % line)

			cond = res.group(1)

			varname = res.group(2)

			vartype = res.group(3)

			ask_names.append((varname, vartype, name, choices, cond))

			name = ''

			choices = []

			continue

		if (line.startswith('@')):

			# Add new line into the 'choices' array

			res = re.match(r'@\s*(?:\[(.*?)\])?\s*"(.*?)"\s*(.*)$', line)

			if not res:

				raise RuntimeError("Bad line: %s" % line)

			choices.append((res.group(2), res.group(3)))

			continue

		if (line.startswith('%')):

			# Name of the option

			name = line[1:].strip()

			continue

		if ((line.startswith('#')) or (line == '\n')):

			# Comment or empty line

			continue

		raise RuntimeError("Unknown syntax: %s" % line)

	inf.close()

def yes_no(default):

	"Return '*' if yes, ' ' if no"

	if (default == 'y'):

		return '*'

	return ' '

def subchoice(screen, name, choices, default):

	"Return choice of choices"

	maxkey = 0

	for key, val in choices:

		length = len(key)

		if (length > maxkey):

			maxkey = length

	options = []

	position = None

	cnt = 0

	for key, val in choices:

		if ((default) and (key == default)):

			position = cnt

		options.append(" %-*s  %s " % (maxkey, key, val))

		cnt += 1

	(button, value) = xtui.choice_window(screen, name, 'Choose value', options, position)

	if (button == 'cancel'):

		return None

	return choices[value][0]

def check_choices(defaults, ask_names):

	"Check whether all accessible variables have a default"

	for varname, vartype, name, choices, cond in ask_names:

		if ((cond) and (not check_condition(cond, defaults, ask_names))):

			continue

		if (not defaults.has_key(varname)):

			return False

	return True

def create_output(fname, defaults, ask_names):

	"Create output configuration"

	outf = file(fname, 'w')

	outf.write('#########################################\n')

	outf.write('## AUTO-GENERATED FILE, DO NOT EDIT!!! ##\n')

	outf.write('#########################################\n\n')

	for varname, vartype, name, choices, cond in ask_names:

		if ((cond) and (not check_condition(cond, defaults, ask_names))):

			continue

		if (not defaults.has_key(varname)):

			default = ''

		else:

			default = defaults[varname]

		outf.write('# %s\n%s = %s\n\n' % (name, varname, default))

	outf.write('REVISION = %s\n' % commands.getoutput('svnversion . 2> /dev/null'))

	outf.write('TIMESTAMP = %s\n' % commands.getoutput('date "+%Y-%m-%d %H:%M:%S"'))

	outf.close()

def main():

	defaults = {}

	ask_names = []

	# Parse configuration file

	parse_config(INPUT, ask_names)

	# Read defaults from previous run

	if os.path.exists(OUTPUT):

		read_defaults(OUTPUT, defaults)

	# Default mode: only check defaults and regenerate configuration

	if ((len(sys.argv) >= 3) and (sys.argv[2] == 'default')):

		if (check_choices(defaults, ask_names)):

			create_output(OUTPUT, defaults, ask_names)

			return 0

	# Check mode: only check defaults

	if ((len(sys.argv) >= 3) and (sys.argv[2] == 'check')):

		if (check_choices(defaults, ask_names)):

			return 0

		return 1

	screen = xtui.screen_init()

	try:

		selname = None

		while True:

			options = []

			opt2row = {}

			position = None

			cnt = 0

			for varname, vartype, name, choices, cond in ask_names:

				if ((cond) and (not check_condition(cond, defaults, ask_names))):

					continue

				if (varname == selname):

					position = cnt

				if (not defaults.has_key(varname)):

					default = None

				else:

					default = defaults[varname]

				if (vartype == 'choice'):

					# Check if the default is an acceptable value

					if ((default) and (not default in [choice[0] for choice in choices])):

						default = None

						defaults.pop(varname)

					# If there is just one option, use it

					if (len(choices) == 1):

						default = choices[0][0]

						defaults[varname] = default

					options.append("     %s [%s] --> " % (name, default))

				elif (vartype == 'y/n'):

					if (default == None):

						default = 'y'

						defaults[varname] = default

					options.append(" <%s> %s " % (yes_no(default), name))

				elif (vartype == 'n/y'):

					if (default == None):

						default = 'n'

						defaults[varname] = default

					options.append(" <%s> %s " % (yes_no(default), name))

				else:

					raise RuntimeError("Unknown variable type: %s" % vartype)

				opt2row[cnt] = (varname, vartype, name, choices)

				cnt += 1

			(button, value) = xtui.choice_window(screen, 'HelenOS configuration', 'Choose configuration option', options, position)

			if (button == 'cancel'):

				return 'Configuration canceled'

			if (not opt2row.has_key(value)):

				raise RuntimeError("Error selecting value: %s" % value)

			(selname, seltype, name, choices) = opt2row[value]

			if (not defaults.has_key(selname)):

					default = None

			else:

				default = defaults[selname]

			if (button == 'done'):

				if (check_choices(defaults, ask_names)):

					break

				else:

					xtui.error_dialog(screen, 'Error', 'Some options have still undefined values.')

					continue

			if (seltype == 'choice'):

				defaults[selname] = subchoice(screen, name, choices, default)

			elif ((seltype == 'y/n') or (seltype == 'n/y')):

				if (defaults[selname] == 'y'):

					defaults[selname] = 'n'

				else:

					defaults[selname] = 'y'

	finally:

		xtui.screen_done(screen)

	create_output(OUTPUT, defaults, ask_names)

	return 0

if __name__ == '__main__':

	sys.exit(main())