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]

MAKEFILE = 'Makefile.config'

MACROS = 'config.h'

DEFS = 'config.defs'

PRECONF = 'defaults'

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 that the condition specified on input line is True (only CNF and DNF is supported)"

	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 for condition"

	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 (varval == '*'):

				varval = 'y'

		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(mkname, mcname, dfname, defaults, ask_names):

	"Create output configuration"

	revision = commands.getoutput('svnversion . 2> /dev/null')

	timestamp = commands.getoutput('date "+%Y-%m-%d %H:%M:%S"')

	outmk = file(mkname, 'w')

	outmc = file(mcname, 'w')

	outdf = file(dfname, 'w')

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

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

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

	outmc.write('/***************************************\n')

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

	outmc.write(' ***************************************/\n\n')

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

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

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

	outdf.write('CONFIG_DEFS =')

	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]

			if (default == '*'):

				default = 'y'

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

		if ((vartype == "y") or (vartype == "n") or (vartype == "y/n") or (vartype == "n/y")):

			if (default == "y"):

				outmc.write('/* %s */\n#define %s\n\n' % (name, varname))

				outdf.write(' -D%s' % varname)

		else:

			outmc.write('/* %s */\n#define %s %s\n#define %s_%s\n\n' % (name, varname, default, varname, default))

			outdf.write(' -D%s=%s -D%s_%s' % (varname, default, varname, default))

	outmk.write('REVISION = %s\n' % revision)

	outmk.write('TIMESTAMP = %s\n' % timestamp)

	outmc.write('#define REVISION %s\n' % revision)

	outmc.write('#define TIMESTAMP %s\n' % timestamp)

	outdf.write(' "-DREVISION=%s" "-DTIMESTAMP=%s"\n' % (revision, timestamp))

	outmk.close()

	outmc.close()

	outdf.close()

def sorted_dir(root):

	list = os.listdir(root)

	list.sort()

	return list

def read_preconfigured(root, fname, screen, defaults):

	options = []

	opt2path = {}

	cnt = 0

	# Look for profiles

	for name in sorted_dir(root):

		path = os.path.join(root, name)

		canon = os.path.join(path, fname)

		if ((os.path.isdir(path)) and (os.path.exists(canon)) and (os.path.isfile(canon))):

			subprofile = False

			# Look for subprofiles

			for subname in sorted_dir(path):

				subpath = os.path.join(path, subname)

				subcanon = os.path.join(subpath, fname)

				if ((os.path.isdir(subpath)) and (os.path.exists(subcanon)) and (os.path.isfile(subcanon))):

					subprofile = True

					options.append("%s (%s)" % (name, subname))

					opt2path[cnt] = (canon, subcanon)

					cnt += 1

			if (not subprofile):

				options.append(name)

				opt2path[cnt] = (canon, None)

				cnt += 1

	(button, value) = xtui.choice_window(screen, 'Load preconfigured defaults', 'Choose configuration profile', options, None)

	if (button == 'cancel'):

		return None

	read_defaults(opt2path[value][0], defaults)

	if (opt2path[value][1] != None):

		read_defaults(opt2path[value][1], defaults)

def main():

	defaults = {}

	ask_names = []

	# Parse configuration file

	parse_config(INPUT, ask_names)

	# Read defaults from previous run

	if os.path.exists(MAKEFILE):

		read_defaults(MAKEFILE, 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(MAKEFILE, MACROS, DEFS, 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

		position = None

		while True:

			# Cancel out all defaults which have to be deduced

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

				if ((vartype == 'y') and (defaults.has_key(varname)) and (defaults[varname] == '*')):

					defaults[varname] = None

			options = []

			opt2row = {}

			cnt = 1

			options.append("  --- Load preconfigured defaults ... ")

			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):

						defaults[varname] = choices[0][0]

						continue

					if (default == None):

						options.append("?     %s --> " % name)

					else:

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

				elif (vartype == 'y'):

					defaults[varname] = '*'

					continue

				elif (vartype == 'n'):

					defaults[varname] = 'n'

					continue

				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

			if (position >= options):

				position = None

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

			if (button == 'cancel'):

				return 'Configuration canceled'

			if (button == 'done'):

				if (check_choices(defaults, ask_names)):

					break

				else:

					xtui.error_dialog(screen, 'Error', 'Some options have still undefined values. These options are marked with the "?" sign.')

					continue

			if (value == 0):

				read_preconfigured(PRECONF, MAKEFILE, screen, defaults)

				position = 1

				continue

			position = None

			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 (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(MAKEFILE, MACROS, DEFS, defaults, ask_names)

	return 0

if __name__ == '__main__':

	sys.exit(main())