WebSVN – HelenOS – Blame – /trunk/tools/config.py/

Rev	Author	Line No.	Line
1802	decky	1	#!/usr/bin/env python
		2	"""
		3	HelenOS configuration script
		4	"""
		5	import sys
		6	import os
		7	import re
		8	import commands
		9
		10	INPUT = sys.argv[1]
		11	OUTPUT = 'Makefile.config'
		12	TMPOUTPUT = 'Makefile.config.tmp'
		13
		14	class DefaultDialog:
		15	"Wrapper dialog that tries to return default values"
		16	def __init__(self, dlg):
		17	self.dlg = dlg
		18
		19	def set_title(self,text):
		20	self.dlg.set_title(text)
		21
		22	def yesno(self, text, default=None):
		23	if default is not None:
		24	return default
		25	return self.dlg.yesno(text, default)
		26	def noyes(self, text, default=None):
		27	if default is not None:
		28	return default
		29	return self.dlg.noyes(text, default)
		30
		31	def choice(self, text, choices, defopt=None):
		32	if defopt is not None:
		33	return choices[defopt][0]
		34	return self.dlg.choice(text, choices, defopt)
		35
		36	class NoDialog:
		37	def __init__(self):
		38	self.printed = None
		39	self.title = 'HelenOS Configuration'
		40
		41	def print_title(self):
		42	if not self.printed:
		43	sys.stdout.write("\n* %s *\n" % self.title)
		44	self.printed = True
		45
		46	def set_title(self, text):
		47	self.title = text
		48	self.printed = False
		49
		50	def noyes(self, text, default=None):
		51	if not default:
		52	default = 'n'
		53	return self.yesno(text, default)
		54
		55	def yesno(self, text, default=None):
		56	self.print_title()
		57
		58	if default != 'n':
		59	default = 'y'
		60	while 1:
		61	sys.stdout.write("%s (y/n)[%s]: " % (text,default))
		62	inp = sys.stdin.readline()
		63	if not inp:
		64	raise EOFError
		65	inp = inp.strip().lower()
		66	if not inp:
		67	return default
		68	if inp == 'y':
		69	return 'y'
		70	elif inp == 'n':
		71	return 'n'
		72
		73	def _print_choice(self, text, choices, defopt):
		74	sys.stdout.write('%s:\n' % text)
		75	for i,(text,descr) in enumerate(choices):
		76	sys.stdout.write('\t%2d. %s\n' % (i, descr))
		77	if defopt is not None:
		78	sys.stdout.write('Enter choice number[%d]: ' % defopt)
		79	else:
		80	sys.stdout.write('Enter choice number: ')
		81
		82	def menu(self, text, choices, button, defopt=None):
		83	self.title = 'Main menu'
		84	menu = []
		85	for key, descr in choices:
		86	txt = key + (45-len(key))*' ' + ': ' + descr
		87	menu.append((key, txt))
		88
		89	return self.choice(text, [button] + menu)
		90
		91	def choice(self, text, choices, defopt=None):
		92	self.print_title()
		93	while 1:
		94	self._print_choice(text, choices, defopt)
		95	inp = sys.stdin.readline()
		96	if not inp:
		97	raise EOFError
		98	if not inp.strip():
		99	if defopt is not None:
		100	return choices[defopt][0]
		101	continue
		102	try:
		103	number = int(inp.strip())
		104	except ValueError:
		105	continue
		106	if number < 0 or number >= len(choices):
		107	continue
		108	return choices[number][0]
		109
		110
		111	def eof_checker(fnc):
		112	def wrapper(self, args, *kw):
		113	try:
		114	return fnc(self, args, *kw)
		115	except EOFError:
		116	return getattr(self.bckdialog,fnc.func_name)(args, *kw)
		117	return wrapper
		118
		119	class Dialog(NoDialog):
		120	def __init__(self):
		121	NoDialog.__init__(self)
		122	self.dlgcmd = os.environ.get('DIALOG','dialog')
		123	self.title = ''
		124	self.backtitle = 'HelenOS Configuration'
		125
		126	if os.system('%s --print-maxsize >/dev/null 2>&1' % self.dlgcmd) != 0:
		127	raise NotImplementedError
		128
		129	self.bckdialog = NoDialog()
		130
		131	def set_title(self,text):
		132	self.title = text
		133	self.bckdialog.set_title(text)
		134
		135	def calldlg(self,args,*kw):
		136	"Wrapper for calling 'dialog' program"
		137	indesc, outdesc = os.pipe()
		138	pid = os.fork()
		139	if not pid:
		140	os.close(2)
		141	os.dup(outdesc)
		142	os.close(indesc)
		143
		144	dlgargs = [self.dlgcmd,'--title',self.title,
		145	'--backtitle', self.backtitle]
		146	for key,val in kw.items():
		147	dlgargs.append('--'+key)
		148	dlgargs.append(val)
		149	dlgargs += args
		150	os.execlp(self.dlgcmd,*dlgargs)
		151
		152	os.close(outdesc)
		153
		154	try:
		155	errout = os.fdopen(indesc,'r')
		156	data = errout.read()
		157	errout.close()
		158	pid,status = os.wait()
		159	except:
		160	os.system('reset') # Reset terminal
		161	raise
		162
		163	if not os.WIFEXITED(status):
		164	os.system('reset') # Reset terminal
		165	raise EOFError
		166
		167	status = os.WEXITSTATUS(status)
		168	if status == 255:
		169	raise EOFError
		170	return status,data
		171
		172	def yesno(self, text, default=None):
		173	if text[-1] not in ('?',':'):
		174	text = text + ':'
		175	width = '50'
		176	height = '5'
		177	if len(text) < 48:
		178	text = ' '*int(((48-len(text))/2)) + text
		179	else:
		180	width = '0'
		181	height = '0'
		182	if default == 'n':
		183	res,data = self.calldlg('--defaultno','--yesno',text,height,width)
		184	else:
		185	res,data = self.calldlg('--yesno',text,height,width)
		186
		187	if res == 0:
		188	return 'y'
		189	return 'n'
		190	yesno = eof_checker(yesno)
		191
		192	def menu(self, text, choices, button, defopt=None):
		193	self.title = 'Main menu'
		194	text = text + ':'
		195	width = '70'
		196	height = str(8 + len(choices))
		197	args = []
		198	for key,val in choices:
		199	args.append(key)
		200	args.append(val)
		201
		202	kw = {}
		203	if defopt:
		204	kw['default-item'] = choices[defopt][0]
		205	res,data = self.calldlg('--ok-label','Change',
		206	'--extra-label',button[1],
		207	'--extra-button',
		208	'--menu',text,height,width,
		209	str(len(choices)),args,*kw)
		210	if res == 3:
		211	return button[0]
		212	if res == 1: # Cancel
		213	sys.exit(1)
		214	elif res:
		215	print data
		216	raise EOFError
		217	return data
		218	menu = eof_checker(menu)
		219
		220	def choice(self, text, choices, defopt=None):
		221	text = text + ':'
		222	width = '50'
		223	height = str(8 + len(choices))
		224	args = []
		225	for key,val in choices:
		226	args.append(key)
		227	args.append(val)
		228
		229	kw = {}
		230	if defopt:
		231	kw['default-item'] = choices[defopt][0]
		232	res,data = self.calldlg('--nocancel','--menu',text,height,width,
		233	str(len(choices)),args, *kw)
		234	if res:
		235	print data
		236	raise EOFError
		237	return data
		238	choice = eof_checker(choice)
		239
		240	def read_defaults(fname,defaults):
		241	"Read saved values from last configuration run"
		242	f = file(fname,'r')
		243	for line in f:
		244	res = re.match(r'^(?:#!# )?([^#]\w)\s=\s(.?)\s*$', line)
		245	if res:
		246	defaults[res.group(1)] = res.group(2)
		247	f.close()
		248
		249	def check_condition(text, defaults, asked_names):
		250	seen_vars = [ x[0] for x in asked_names ]
		251	ctype = 'cnf'
		252	if ')\|' in text or '\|(' in text:
		253	ctype = 'dnf'
		254
		255	if ctype == 'cnf':
		256	conds = text.split('&')
		257	else:
		258	conds = text.split('\|')
		259
		260	for cond in conds:
		261	if cond.startswith('(') and cond.endswith(')'):
		262	cond = cond[1:-1]
		263
		264	inside = check_inside(cond, defaults, ctype, seen_vars)
		265
		266	if ctype == 'cnf' and not inside:
		267	return False
		268	if ctype == 'dnf' and inside:
		269	return True
		270
		271	if ctype == 'cnf':
		272	return True
		273	return False
		274
		275	def check_inside(text, defaults, ctype, seen_vars):
		276	"""
		277	Check that the condition specified on input line is True
		278
		279	only CNF is supported
		280	"""
		281	if ctype == 'cnf':
		282	conds = text.split('\|')
		283	else:
		284	conds = text.split('&')
		285	for cond in conds:
		286	res = re.match(r'^(.?)(!?=)(.)$', cond)
		287	if not res:
		288	raise RuntimeError("Invalid condition: %s" % cond)
		289	condname = res.group(1)
		290	oper = res.group(2)
		291	condval = res.group(3)
		292	if condname not in seen_vars:
		293	varval = ''
		294	## raise RuntimeError("Variable %s not defined before being asked." %\
		295	## condname)
		296	elif not defaults.has_key(condname):
		297	raise RuntimeError("Condition var %s does not exist: %s" % \
		298	(condname,text))
		299	else:
		300	varval = defaults[condname]
		301	if ctype == 'cnf':
		302	if oper == '=' and condval == varval:
		303	return True
		304	if oper == '!=' and condval != varval:
		305	return True
		306	else:
		307	if oper== '=' and condval != varval:
		308	return False
		309	if oper== '!=' and condval == varval:
		310	return False
		311	if ctype=='cnf':
		312	return False
		313	return True
		314
		315	def parse_config(input, output, dlg, defaults={}, askonly=None):
		316	"Parse configuration file and create Makefile.config on the fly"
		317	def ask_the_question(dialog):
		318	"Ask question based on the type of variables to ask"
		319	# This is quite a hack, this thingy is written just to
		320	# have access to local variables..
		321	if vartype == 'y/n':
		322	return dialog.yesno(comment, default)
		323	elif vartype == 'n/y':
		324	return dialog.noyes(comment, default)
		325	elif vartype == 'choice':
		326	defopt = None
		327	if default is not None:
		328	for i,(key,val) in enumerate(choices):
		329	if key == default:
		330	defopt = i
		331	break
		332	return dialog.choice(comment, choices, defopt)
		333	else:
		334	raise RuntimeError("Bad method: %s" % vartype)
		335
		336
		337	f = file(input, 'r')
		338	outf = file(output, 'w')
		339
		340	outf.write('#########################################\n')
		341	outf.write('## AUTO-GENERATED FILE, DO NOT EDIT!!! ##\n')
		342	outf.write('#########################################\n\n')
		343
		344	asked_names = []
		345
		346	comment = ''
		347	default = None
		348	choices = []
		349	for line in f:
		350	if line.startswith('%'):
		351	res = re.match(r'^%\s(?:\[(.?)\])?\s(.)$', line)
		352	if not res:
		353	raise RuntimeError('Invalid command: %s' % line)
		354	if res.group(1):
		355	if not check_condition(res.group(1), defaults,
		356	asked_names):
		357	continue
		358	args = res.group(2).strip().split(' ')
		359	cmd = args[0].lower()
		360	args = args[1:]
		361	if cmd == 'saveas':
		362	outf.write('%s = %s\n' % (args[1],defaults[args[0]]))
		363	elif cmd == 'shellcmd':
		364	varname = args[0]
		365	args = args[1:]
		366	for i,arg in enumerate(args):
		367	if arg.startswith('$'):
		368	args[i] = defaults[arg[1:]]
		369	data,status = commands.getstatusoutput(' '.join(args))
		370	if status:
		371	raise RuntimeError('Error running: %s' % ' '.join(args))
		372	outf.write('%s = %s\n' % (varname,data.strip()))
		373	continue
		374
		375	if line.startswith('!'):
		376	# Ask a question
		377	res = re.search(r'!\s(?:\[(.?)\])?\s([^\s]+)\s$(.)$\s$', line)
		378	if not res:
		379	raise RuntimeError("Weird line: %s" % line)
		380	varname = res.group(2)
		381	vartype = res.group(3)
		382
		383	default = defaults.get(varname,None)
		384
		385	if res.group(1):
		386	if not check_condition(res.group(1), defaults,
		387	asked_names):
		388	if default is not None:
		389	outf.write('#!# %s = %s\n' % (varname, default))
		390	# Clear cumulated values
		391	comment = ''
		392	default = None
		393	choices = []
		394	continue
		395
		396	asked_names.append((varname,comment))
		397
		398	if default is None or not askonly or askonly == varname:
		399	default = ask_the_question(dlg)
		400	else:
		401	default = ask_the_question(DefaultDialog(dlg))
		402
		403	outf.write('%s = %s\n' % (varname, default))
		404	# Remeber the selected value
		405	defaults[varname] = default
		406	# Clear cumulated values
		407	comment = ''
		408	default = None
		409	choices = []
		410	continue
		411
		412	if line.startswith('@'):
		413	# Add new line into the 'choice array'
		414	res = re.match(r'@\s(?:\[(.?)\])?\s"(.?)"\s(.)$', line)
		415	if not res:
		416	raise RuntimeError("Bad line: %s" % line)
		417	if res.group(1):
		418	if not check_condition(res.group(1),defaults,
		419	asked_names):
		420	continue
		421	choices.append((res.group(2), res.group(3)))
		422	continue
		423
		424	# All other things print to output file
		425	outf.write(line)
		426	if re.match(r'^#[^#]', line):
		427	# Last comment before question will be displayed to the user
		428	comment = line[1:].strip()
		429	elif line.startswith('## '):
		430	# Set title of the dialog window
		431	dlg.set_title(line[2:].strip())
		432
		433	outf.write('\n')
		434	outf.write('REVISION = %s\n' % commands.getoutput('svnversion . 2> /dev/null'))
		435	outf.write('TIMESTAMP = %s\n' % commands.getoutput('date "+%Y-%m-%d %H:%M:%S"'))
		436	outf.close()
		437	f.close()
		438	return asked_names
		439
		440	def main():
		441	defaults = {}
		442	try:
		443	dlg = Dialog()
		444	except NotImplementedError:
		445	dlg = NoDialog()
		446
		447	if len(sys.argv) >= 3 and sys.argv[2]=='default':
		448	defmode = True
		449	else:
		450	defmode = False
		451
		452	# Default run will update the configuration file
		453	# with newest options
		454	if os.path.exists(OUTPUT):
		455	read_defaults(OUTPUT, defaults)
		456
		457	# Get ARCH from command line if specified
		458	if len(sys.argv) >= 4:
		459	defaults['ARCH'] = sys.argv[3]
1846	decky	460	defaults['PLATFORM'] = sys.argv[3]
1802	decky	461
		462	# Get COMPILER from command line if specified
		463	if len(sys.argv) >= 5:
		464	defaults['COMPILER'] = sys.argv[4]
		465
		466	# Get CONFIG_DEBUG from command line if specified
		467	if len(sys.argv) >= 6:
		468	defaults['CONFIG_DEBUG'] = sys.argv[5]
		469
		470	# Get MACHINE/IMAGE from command line if specified
		471	if len(sys.argv) >= 7:
		472	defaults['MACHINE'] = sys.argv[6]
		473	defaults['IMAGE'] = sys.argv[6]
		474
		475	# Dry run only with defaults
		476	varnames = parse_config(INPUT, TMPOUTPUT, DefaultDialog(dlg), defaults)
		477	# If not in default mode, present selection of all possibilities
		478	if not defmode:
		479	defopt = 0
		480	while 1:
		481	# varnames contains variable names that were in the
		482	# last question set
		483	choices = [ (x[1],defaults[x[0]]) for x in varnames ]
		484	res = dlg.menu('Configuration',choices,('save','Save'),defopt)
		485	if res == 'save':
		486	parse_config(INPUT, TMPOUTPUT, DefaultDialog(dlg), defaults)
		487	break
		488	# transfer description back to varname
		489	for i,(vname,descr) in enumerate(varnames):
		490	if res == descr:
		491	defopt = i
		492	break
		493	# Ask the user a simple question, produce output
		494	# as if the user answered all the other questions
		495	# with default answer
		496	varnames = parse_config(INPUT, TMPOUTPUT, dlg, defaults,
		497	askonly=varnames[i][0])
		498
		499
		500	if os.path.exists(OUTPUT):
		501	os.unlink(OUTPUT)
		502	os.rename(TMPOUTPUT, OUTPUT)
		503
		504	if not defmode and dlg.yesno('Rebuild everything?') == 'y':
		505	os.execlp('make','make','clean','build')
		506
		507	if __name__ == '__main__':
		508	main()

Subversion Repositories HelenOS

(root)/trunk/tools/config.py/ – Rev 1846