Subversion Repositories HelenOS-historic

Rev

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

  1. #!/usr/bin/env python
  2. """
  3. Kernel configuration script
  4. """
  5. import sys
  6. import os
  7. import re
  8. import commands
  9.  
  10. INPUT = 'kernel.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 Kernel 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.             raise RuntimeError("Variable %s not defined before being asked." %\
  280.                                condname)
  281.         if not defaults.has_key(condname):
  282.             raise RuntimeError("Condition var %s does not exist: %s" % \
  283.                                (condname,text))
  284.  
  285.         if ctype == 'cnf':
  286.             if oper == '=' and  condval == defaults[condname]:
  287.                 return True
  288.             if oper == '!=' and condval != defaults[condname]:
  289.                 return True
  290.         else:
  291.             if oper== '=' and condval != defaults[condname]:
  292.                 return False
  293.             if oper== '!=' and condval == defaults[condname]:
  294.                 return False
  295.     if ctype=='cnf':
  296.         return False
  297.     return True
  298.  
  299. def parse_config(input, output, dlg, defaults={}, askonly=None):
  300.     "Parse configuration file and create Makefile.config on the fly"
  301.     def ask_the_question(dialog):
  302.         "Ask question based on the type of variables to ask"
  303.         # This is quite a hack, this thingy is written just to
  304.         # have access to local variables..
  305.         if vartype == 'y/n':
  306.             return dialog.yesno(comment, default)
  307.         elif vartype == 'n/y':
  308.             return dialog.noyes(comment, default)
  309.         elif vartype == 'choice':
  310.             defopt = None
  311.             if default is not None:
  312.                 for i,(key,val) in enumerate(choices):
  313.                     if key == default:
  314.                         defopt = i
  315.                         break
  316.             return dialog.choice(comment, choices, defopt)
  317.         else:
  318.             raise RuntimeError("Bad method: %s" % vartype)
  319.  
  320.    
  321.     f = file(input, 'r')
  322.     outf = file(output, 'w')
  323.  
  324.     outf.write('#########################################\n')
  325.     outf.write('## AUTO-GENERATED FILE, DO NOT EDIT!!! ##\n')
  326.     outf.write('#########################################\n\n')
  327.  
  328.     asked_names = []
  329.  
  330.     comment = ''
  331.     default = None
  332.     choices = []
  333.     for line in f:
  334.         if line.startswith('%'):
  335.             res = re.match(r'^%\s*(?:\[(.*?)\])?\s*(.*)$', line)
  336.             if not res:
  337.                 raise RuntimeError('Invalid command: %s' % line)
  338.             if res.group(1):
  339.                 if not check_condition(res.group(1), defaults,
  340.                                        asked_names):
  341.                     continue
  342.             args = res.group(2).strip().split(' ')
  343.             cmd = args[0].lower()
  344.             args = args[1:]
  345.             if cmd == 'saveas':
  346.                 outf.write('%s = %s\n' % (args[1],defaults[args[0]]))
  347.             elif cmd == 'shellcmd':
  348.                 varname = args[0]
  349.                 args = args[1:]
  350.                 for i,arg in enumerate(args):
  351.                     if arg.startswith('$'):
  352.                         args[i] = defaults[arg[1:]]
  353.                 data,status = commands.getstatusoutput(' '.join(args))
  354.                 if status:
  355.                     raise RuntimeError('Error running: %s' % ' '.join(args))
  356.                 outf.write('%s = %s\n' % (varname,data.strip()))
  357.             continue
  358.            
  359.         if line.startswith('!'):
  360.             # Ask a question
  361.             res = re.search(r'!\s*(?:\[(.*?)\])?\s*([^\s]+)\s*\((.*)\)\s*$', line)
  362.             if not res:
  363.                 raise RuntimeError("Weird line: %s" % line)
  364.             varname = res.group(2)
  365.             vartype = res.group(3)
  366.  
  367.             default = defaults.get(varname,None)
  368.            
  369.             if res.group(1):
  370.                 if not check_condition(res.group(1), defaults,
  371.                                        asked_names):
  372.                     if default is not None:
  373.                         outf.write('#!# %s = %s\n' % (varname, default))
  374.                     # Clear cumulated values
  375.                     comment = ''
  376.                     default = None
  377.                     choices = []
  378.                     continue
  379.                
  380.             asked_names.append((varname,comment))
  381.  
  382.             if default is None or not askonly or askonly == varname:
  383.                 default = ask_the_question(dlg)
  384.             else:
  385.                 default = ask_the_question(DefaultDialog(dlg))
  386.  
  387.             outf.write('%s = %s\n' % (varname, default))
  388.             # Remeber the selected value
  389.             defaults[varname] = default
  390.             # Clear cumulated values
  391.             comment = ''
  392.             default = None
  393.             choices = []
  394.             continue
  395.        
  396.         if line.startswith('@'):
  397.             # Add new line into the 'choice array'
  398.             res = re.match(r'@\s*(?:\[(.*?)\])?\s*"(.*?)"\s*(.*)$', line)
  399.             if not res:
  400.                 raise RuntimeError("Bad line: %s" % line)
  401.             if res.group(1):
  402.                 if not check_condition(res.group(1),defaults,
  403.                                        asked_names):
  404.                     continue
  405.             choices.append((res.group(2), res.group(3)))
  406.             continue
  407.  
  408.         # All other things print to output file
  409.         outf.write(line)
  410.         if re.match(r'^#[^#]', line):
  411.             # Last comment before question will be displayed to the user
  412.             comment = line[1:].strip()
  413.         elif line.startswith('## '):
  414.             # Set title of the dialog window
  415.             dlg.set_title(line[2:].strip())
  416.  
  417.     outf.write('\n')
  418.     outf.write('REVISION=%s\n' % commands.getoutput('svnversion . 2> /dev/null'))
  419.     outf.write('TIMESTAMP=%s\n' % commands.getoutput('date "+%Y-%m-%d %H:%M:%S"'))
  420.     outf.close()
  421.     f.close()
  422.     return asked_names
  423.  
  424. def main():
  425.     defaults = {}
  426.     try:
  427.         dlg = Dialog()
  428.     except NotImplementedError:
  429.         dlg = NoDialog()
  430.  
  431.     if len(sys.argv) == 2 and sys.argv[1]=='default':
  432.         defmode = True
  433.     else:
  434.         defmode = False
  435.  
  436.     # Default run will update the configuration file
  437.     # with newest options
  438.     if os.path.exists(OUTPUT):
  439.         read_defaults(OUTPUT, defaults)
  440.  
  441.     # Dry run only with defaults
  442.     varnames = parse_config(INPUT, TMPOUTPUT, DefaultDialog(dlg), defaults)
  443.     # If not in default mode, present selection of all possibilities
  444.     if not defmode:
  445.         defopt = 0
  446.         while 1:
  447.             # varnames contains variable names that were in the
  448.             # last question set
  449.             choices = [ (x[1],defaults[x[0]]) for x in varnames ]
  450.             res = dlg.menu('Configuration',choices,('save','Save'),defopt)
  451.             if res == 'save':
  452.                 parse_config(INPUT, TMPOUTPUT, DefaultDialog(dlg), defaults)
  453.                 break
  454.             # transfer description back to varname
  455.             for i,(vname,descr) in enumerate(varnames):
  456.                 if res == descr:
  457.                     defopt = i
  458.                     break
  459.             # Ask the user a simple question, produce output
  460.             # as if the user answered all the other questions
  461.             # with default answer
  462.             varnames = parse_config(INPUT, TMPOUTPUT, dlg, defaults,
  463.                                     askonly=varnames[i][0])
  464.        
  465.    
  466.     if os.path.exists(OUTPUT):
  467.         os.unlink(OUTPUT)
  468.     os.rename(TMPOUTPUT, OUTPUT)
  469.    
  470.     if not defmode and dlg.yesno('Rebuild kernel?') == 'y':
  471.         os.execlp('make','make','clean','build')
  472.  
  473. if __name__ == '__main__':
  474.     main()
  475.