Subversion Repositories HelenOS

Rev

Rev 1963 | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

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