Subversion Repositories HelenOS

Rev

Rev 2585 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  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]
  460.    
  461.     # Get COMPILER from command line if specified
  462.     if len(sys.argv) >= 5:
  463.         defaults['COMPILER'] = sys.argv[4]
  464.    
  465.     # Get CONFIG_DEBUG from command line if specified
  466.     if len(sys.argv) >= 6:
  467.         defaults['CONFIG_DEBUG'] = sys.argv[5]
  468.    
  469.     # Get MACHINE/IMAGE from command line if specified
  470.     if len(sys.argv) >= 7:
  471.         defaults['MACHINE'] = sys.argv[6]
  472.         defaults['IMAGE'] = sys.argv[6]
  473.  
  474.     # Dry run only with defaults
  475.     varnames = parse_config(INPUT, TMPOUTPUT, DefaultDialog(dlg), defaults)
  476.     # If not in default mode, present selection of all possibilities
  477.     if not defmode:
  478.         defopt = 0
  479.         while 1:
  480.             # varnames contains variable names that were in the
  481.             # last question set
  482.             choices = [ (x[1],defaults[x[0]]) for x in varnames ]
  483.             res = dlg.menu('Configuration',choices,('save','Save'),defopt)
  484.             if res == 'save':
  485.                 parse_config(INPUT, TMPOUTPUT, DefaultDialog(dlg), defaults)
  486.                 break
  487.             # transfer description back to varname
  488.             for i,(vname,descr) in enumerate(varnames):
  489.                 if res == descr:
  490.                     defopt = i
  491.                     break
  492.             # Ask the user a simple question, produce output
  493.             # as if the user answered all the other questions
  494.             # with default answer
  495.             varnames = parse_config(INPUT, TMPOUTPUT, dlg, defaults,
  496.                                     askonly=varnames[i][0])
  497.        
  498.    
  499.     if os.path.exists(OUTPUT):
  500.         os.unlink(OUTPUT)
  501.     os.rename(TMPOUTPUT, OUTPUT)
  502.    
  503.     if not defmode and dlg.yesno('Rebuild everything?') == 'y':
  504.         os.execlp('make','make','clean','build')
  505.  
  506. if __name__ == '__main__':
  507.     main()
  508.