Subversion Repositories HelenOS-historic

Rev

Rev 568 | Rev 604 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

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