Subversion Repositories HelenOS

Rev

Rev 3843 | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
3804 decky 1
#
2
# Copyright (c) 2009 Martin Decky
3
# All rights reserved.
4
#
5
# Redistribution and use in source and binary forms, with or without
6
# modification, are permitted provided that the following conditions
7
# are met:
8
#
9
# - Redistributions of source code must retain the above copyright
10
#   notice, this list of conditions and the following disclaimer.
11
# - Redistributions in binary form must reproduce the above copyright
12
#   notice, this list of conditions and the following disclaimer in the
13
#   documentation and/or other materials provided with the distribution.
14
# - The name of the author may not be used to endorse or promote products
15
#   derived from this software without specific prior written permission.
16
#
17
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27
#
28
"""
29
Text User Interface wrapper
30
"""
31
 
32
import sys
33
import os
34
 
35
def call_dlg(dlgcmd, *args, **kw):
36
    "Wrapper for calling 'dialog' program"
37
 
38
    indesc, outdesc = os.pipe()
39
    pid = os.fork()
40
    if (not pid):
41
        os.close(2)
42
        os.dup(outdesc)
43
        os.close(indesc)
44
 
45
        dlgargs = [dlgcmd]
46
        for key, val in kw.items():
47
            dlgargs.append('--' + key)
48
            dlgargs.append(val)
49
 
50
        dlgargs += args
51
        os.execlp(dlgcmd, *dlgargs)
52
 
53
    os.close(outdesc)
54
 
55
    try:
56
        errout = os.fdopen(indesc, 'r')
57
        data = errout.read()
58
        errout.close()
59
        pid, status = os.wait()
60
    except:
61
        # Reset terminal
62
        os.system('reset')
63
        raise
64
 
65
    if (not os.WIFEXITED(status)):
66
        # Reset terminal
67
        os.system('reset')
68
        raise EOFError
69
 
70
    status = os.WEXITSTATUS(status)
71
    if (status == 255):
72
        raise EOFError
73
 
74
    return (status, data)
75
 
76
try:
77
    import snack
78
    newt = True
79
    dialog = False
80
except ImportError:
81
    newt = False
82
 
83
    dlgcmd = os.environ.get('DIALOG', 'dialog')
84
    if (call_dlg(dlgcmd, '--print-maxsize')[0] != 0):
85
        dialog = False
86
    else:
87
        dialog = True
88
 
89
width_extra = 13
90
height_extra = 11
91
 
92
def width_fix(screen, width):
93
    "Correct width to screen size"
94
 
95
    if (width + width_extra > screen.width):
96
        width = screen.width - width_extra
97
 
98
    if (width <= 0):
99
        width = screen.width
100
 
101
    return width
102
 
103
def height_fix(screen, height):
104
    "Correct height to screen size"
105
 
106
    if (height + height_extra > screen.height):
107
        height = screen.height - height_extra
108
 
109
    if (height <= 0):
110
        height = screen.height
111
 
112
    return height
113
 
114
def screen_init():
115
    "Initialize the screen"
116
 
117
    if (newt):
118
        return snack.SnackScreen()
119
 
120
    return None
121
 
122
def screen_done(screen):
123
    "Cleanup the screen"
124
 
125
    if (newt):
126
        screen.finish()
127
 
128
def choice_window(screen, title, text, options, position):
129
    "Create options menu"
130
 
131
    maxopt = 0
132
    for option in options:
133
        length = len(option)
134
        if (length > maxopt):
135
            maxopt = length
136
 
137
    width = maxopt
138
    height = len(options)
139
 
140
    if (newt):
141
        width = width_fix(screen, width + width_extra)
142
        height = height_fix(screen, height)
143
 
3805 decky 144
        if (height > 3):
145
            large = True
146
        else:
147
            large = False
148
 
3804 decky 149
        buttonbar = snack.ButtonBar(screen, ('Done', 'Cancel'))
150
        textbox = snack.TextboxReflowed(width, text)
3805 decky 151
        listbox = snack.Listbox(height, scroll = large, returnExit = 1)
3804 decky 152
 
153
        cnt = 0
154
        for option in options:
155
            listbox.append(option, cnt)
156
            cnt += 1
157
 
158
        if (position != None):
159
            listbox.setCurrent(position)
160
 
161
        grid = snack.GridForm(screen, title, 1, 3)
162
        grid.add(textbox, 0, 0)
163
        grid.add(listbox, 0, 1, padding = (0, 1, 0, 1))
164
        grid.add(buttonbar, 0, 2, growx = 1)
165
 
166
        retval = grid.runOnce()
167
 
168
        return (buttonbar.buttonPressed(retval), listbox.current())
169
    elif (dialog):
3843 decky 170
        if (width < 35):
171
            width = 35
172
 
3804 decky 173
        args = []
174
        cnt = 0
175
        for option in options:
176
            args.append(str(cnt + 1))
177
            args.append(option)
178
 
179
            cnt += 1
180
 
181
        kw = {}
182
        if (position != None):
183
            kw['default-item'] = str(position + 1)
184
 
185
        status, data = call_dlg(dlgcmd, '--title', title, '--extra-button', '--extra-label', 'Done', '--menu', text, str(height + height_extra), str(width + width_extra), str(cnt), *args, **kw)
186
 
187
        if (status == 1):
188
            return ('cancel', None)
189
 
190
        try:
191
            choice = int(data) - 1
192
        except ValueError:
193
            return ('cancel', None)
194
 
195
        if (status == 0):
196
            return (None, choice)
197
 
198
        return ('done', choice)
199
 
200
    sys.stdout.write("\n *** %s *** \n%s\n\n" % (title, text))
201
 
202
    maxcnt = len(str(len(options)))
203
    cnt = 0
204
    for option in options:
205
        sys.stdout.write("%*s. %s\n" % (maxcnt, cnt + 1, option))
206
        cnt += 1
207
 
208
    sys.stdout.write("\n%*s. Done\n" % (maxcnt, '0'))
209
    sys.stdout.write("%*s. Quit\n\n" % (maxcnt, 'q'))
210
 
211
    while True:
212
        if (position != None):
213
            sys.stdout.write("Selection[%s]: " % str(position + 1))
214
        else:
215
            sys.stdout.write("Selection: ")
216
        inp = sys.stdin.readline()
217
 
218
        if (not inp):
219
            raise EOFError
220
 
221
        if (not inp.strip()):
222
            if (position != None):
223
                return (None, position)
224
            continue
225
 
226
        if (inp.strip() == 'q'):
227
            return ('cancel', None)
228
 
229
        try:
230
            choice = int(inp.strip())
231
        except ValueError:
232
            continue
233
 
234
        if (choice == 0):
235
            return ('done', 0)
236
 
237
        if (choice < 1) or (choice > len(options)):
238
            continue
239
 
240
        return (None, choice - 1)
241
 
242
def error_dialog(screen, title, msg):
243
    "Print error dialog"
244
 
245
    width = len(msg)
246
 
247
    if (newt):
248
        width = width_fix(screen, width)
249
 
250
        buttonbar = snack.ButtonBar(screen, ['Ok'])
251
        textbox = snack.TextboxReflowed(width, msg)
252
 
253
        grid = snack.GridForm(screen, title, 1, 2)
254
        grid.add(textbox, 0, 0, padding = (0, 0, 0, 1))
255
        grid.add(buttonbar, 0, 1, growx = 1)
256
        grid.runOnce()
257
    elif (dialog):
258
        call_dlg(dlgcmd, '--title', title, '--msgbox', msg, '6', str(width + width_extra))
4054 decky 259
    else:
260
        sys.stdout.write("\n%s: %s\n" % (title, msg))