Subversion Repositories HelenOS

#

# Copyright (c) 2009 Martin Decky

# All rights reserved.

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions

# are met:

#

# - Redistributions of source code must retain the above copyright

#   notice, this list of conditions and the following disclaimer.

# - Redistributions in binary form must reproduce the above copyright

#   notice, this list of conditions and the following disclaimer in the

#   documentation and/or other materials provided with the distribution.

# - The name of the author may not be used to endorse or promote products

#   derived from this software without specific prior written permission.

#

# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#

"""

Text User Interface wrapper

"""

import sys

import os

def call_dlg(dlgcmd, *args, **kw):

    "Wrapper for calling 'dialog' program"

    indesc, outdesc = os.pipe()

    pid = os.fork()

    if (not pid):

        os.close(2)

        os.dup(outdesc)

        os.close(indesc)

        dlgargs = [dlgcmd]

        for key, val in kw.items():

            dlgargs.append('--' + key)

            dlgargs.append(val)

        dlgargs += args

        os.execlp(dlgcmd, *dlgargs)

    os.close(outdesc)

    try:

        errout = os.fdopen(indesc, 'r')

        data = errout.read()

        errout.close()

        pid, status = os.wait()

    except:

        # Reset terminal

        os.system('reset')

        raise

    if (not os.WIFEXITED(status)):

        # Reset terminal

        os.system('reset')

        raise EOFError

    status = os.WEXITSTATUS(status)

    if (status == 255):

        raise EOFError

    return (status, data)

try:

    import snack

    newt = True

    dialog = False

except ImportError:

    newt = False

    dlgcmd = os.environ.get('DIALOG', 'dialog')

    if (call_dlg(dlgcmd, '--print-maxsize')[0] != 0):

        dialog = False

    else:

        dialog = True

width_extra = 13

height_extra = 11

def width_fix(screen, width):

    "Correct width to screen size"

    if (width + width_extra > screen.width):

        width = screen.width - width_extra

    if (width <= 0):

        width = screen.width

    return width

def height_fix(screen, height):

    "Correct height to screen size"

    if (height + height_extra > screen.height):

        height = screen.height - height_extra

    if (height <= 0):

        height = screen.height

    return height

def screen_init():

    "Initialize the screen"

    if (newt):

        return snack.SnackScreen()

    return None

def screen_done(screen):

    "Cleanup the screen"

    if (newt):

        screen.finish()

def choice_window(screen, title, text, options, position):

    "Create options menu"

    maxopt = 0

    for option in options:

        length = len(option)

        if (length > maxopt):

            maxopt = length

    width = maxopt

    height = len(options)

    if (newt):

        width = width_fix(screen, width + width_extra)

        height = height_fix(screen, height)

        if (height > 3):

            large = True

        else:

            large = False

        buttonbar = snack.ButtonBar(screen, ('Done', 'Cancel'))

        textbox = snack.TextboxReflowed(width, text)

        listbox = snack.Listbox(height, scroll = large, returnExit = 1)

        cnt = 0

        for option in options:

            listbox.append(option, cnt)

            cnt += 1

        if (position != None):

            listbox.setCurrent(position)

        grid = snack.GridForm(screen, title, 1, 3)

        grid.add(textbox, 0, 0)

        grid.add(listbox, 0, 1, padding = (0, 1, 0, 1))

        grid.add(buttonbar, 0, 2, growx = 1)

        retval = grid.runOnce()

        return (buttonbar.buttonPressed(retval), listbox.current())

    elif (dialog):

        if (width < 35):

            width = 35

        args = []

        cnt = 0

        for option in options:

            args.append(str(cnt + 1))

            args.append(option)

            cnt += 1

        kw = {}

        if (position != None):

            kw['default-item'] = str(position + 1)

        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)

        if (status == 1):

            return ('cancel', None)

        try:

            choice = int(data) - 1

        except ValueError:

            return ('cancel', None)

        if (status == 0):

            return (None, choice)

        return ('done', choice)

    sys.stdout.write("\n *** %s *** \n%s\n\n" % (title, text))

    maxcnt = len(str(len(options)))

    cnt = 0

    for option in options:

        sys.stdout.write("%*s. %s\n" % (maxcnt, cnt + 1, option))

        cnt += 1

    sys.stdout.write("\n%*s. Done\n" % (maxcnt, '0'))

    sys.stdout.write("%*s. Quit\n\n" % (maxcnt, 'q'))

    while True:

        if (position != None):

            sys.stdout.write("Selection[%s]: " % str(position + 1))

        else:

            sys.stdout.write("Selection: ")

        inp = sys.stdin.readline()

        if (not inp):

            raise EOFError

        if (not inp.strip()):

            if (position != None):

                return (None, position)

            continue

        if (inp.strip() == 'q'):

            return ('cancel', None)

        try:

            choice = int(inp.strip())

        except ValueError:

            continue

        if (choice == 0):

            return ('done', 0)

        if (choice < 1) or (choice > len(options)):

            continue

        return (None, choice - 1)

def error_dialog(screen, title, msg):

    "Print error dialog"

    width = len(msg)

    if (newt):

        width = width_fix(screen, width)

        buttonbar = snack.ButtonBar(screen, ['Ok'])

        textbox = snack.TextboxReflowed(width, msg)

        grid = snack.GridForm(screen, title, 1, 2)

        grid.add(textbox, 0, 0, padding = (0, 0, 0, 1))

        grid.add(buttonbar, 0, 1, growx = 1)

        grid.runOnce()

    elif (dialog):

        call_dlg(dlgcmd, '--title', title, '--msgbox', msg, '6', str(width + width_extra))

    sys.stdout.write("\n%s: %s\n" % (title, msg))