mirror of
				https://github.com/Klipper3d/klipper.git
				synced 2025-10-31 18:36:09 +01:00 
			
		
		
		
	
		
			
	
	
		
			3279 lines
		
	
	
		
			102 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
		
		
			
		
	
	
			3279 lines
		
	
	
		
			102 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
|  | #!/usr/bin/env python3 | ||
|  | 
 | ||
|  | # Copyright (c) 2018-2019, Nordic Semiconductor ASA and Ulf Magnusson | ||
|  | # SPDX-License-Identifier: ISC | ||
|  | 
 | ||
|  | """
 | ||
|  | Overview | ||
|  | ======== | ||
|  | 
 | ||
|  | A curses-based Python 2/3 menuconfig implementation. The interface should feel | ||
|  | familiar to people used to mconf ('make menuconfig'). | ||
|  | 
 | ||
|  | Supports the same keys as mconf, and also supports a set of keybindings | ||
|  | inspired by Vi: | ||
|  | 
 | ||
|  |   J/K     : Down/Up | ||
|  |   L       : Enter menu/Toggle item | ||
|  |   H       : Leave menu | ||
|  |   Ctrl-D/U: Page Down/Page Up | ||
|  |   G/End   : Jump to end of list | ||
|  |   g/Home  : Jump to beginning of list | ||
|  | 
 | ||
|  | [Space] toggles values if possible, and enters menus otherwise. [Enter] works | ||
|  | the other way around. | ||
|  | 
 | ||
|  | The mconf feature where pressing a key jumps to a menu entry with that | ||
|  | character in it in the current menu isn't supported. A jump-to feature for | ||
|  | jumping directly to any symbol (including invisible symbols), choice, menu or | ||
|  | comment (as in a Kconfig 'comment "Foo"') is available instead. | ||
|  | 
 | ||
|  | A few different modes are available: | ||
|  | 
 | ||
|  |   F: Toggle show-help mode, which shows the help text of the currently selected | ||
|  |   item in the window at the bottom of the menu display. This is handy when | ||
|  |   browsing through options. | ||
|  | 
 | ||
|  |   C: Toggle show-name mode, which shows the symbol name before each symbol menu | ||
|  |   entry | ||
|  | 
 | ||
|  |   A: Toggle show-all mode, which shows all items, including currently invisible | ||
|  |   items and items that lack a prompt. Invisible items are drawn in a different | ||
|  |   style to make them stand out. | ||
|  | 
 | ||
|  | 
 | ||
|  | Running | ||
|  | ======= | ||
|  | 
 | ||
|  | menuconfig.py can be run either as a standalone executable or by calling the | ||
|  | menuconfig() function with an existing Kconfig instance. The second option is a | ||
|  | bit inflexible in that it will still load and save .config, etc. | ||
|  | 
 | ||
|  | When run in standalone mode, the top-level Kconfig file to load can be passed | ||
|  | as a command-line argument. With no argument, it defaults to "Kconfig". | ||
|  | 
 | ||
|  | The KCONFIG_CONFIG environment variable specifies the .config file to load (if | ||
|  | it exists) and save. If KCONFIG_CONFIG is unset, ".config" is used. | ||
|  | 
 | ||
|  | When overwriting a configuration file, the old version is saved to | ||
|  | <filename>.old (e.g. .config.old). | ||
|  | 
 | ||
|  | $srctree is supported through Kconfiglib. | ||
|  | 
 | ||
|  | 
 | ||
|  | Color schemes | ||
|  | ============= | ||
|  | 
 | ||
|  | It is possible to customize the color scheme by setting the MENUCONFIG_STYLE | ||
|  | environment variable. For example, setting it to 'aquatic' will enable an | ||
|  | alternative, less yellow, more 'make menuconfig'-like color scheme, contributed | ||
|  | by Mitja Horvat (pinkfluid). | ||
|  | 
 | ||
|  | This is the current list of built-in styles: | ||
|  |     - default       classic Kconfiglib theme with a yellow accent | ||
|  |     - monochrome    colorless theme (uses only bold and standout) attributes, | ||
|  |                     this style is used if the terminal doesn't support colors | ||
|  |     - aquatic       blue-tinted style loosely resembling the lxdialog theme | ||
|  | 
 | ||
|  | It is possible to customize the current style by changing colors of UI | ||
|  | elements on the screen. This is the list of elements that can be stylized: | ||
|  | 
 | ||
|  |     - path          Top row in the main display, with the menu path | ||
|  |     - separator     Separator lines between windows. Also used for the top line | ||
|  |                     in the symbol information display. | ||
|  |     - list          List of items, e.g. the main display | ||
|  |     - selection     Style for the selected item | ||
|  |     - inv-list      Like list, but for invisible items. Used in show-all mode. | ||
|  |     - inv-selection Like selection, but for invisible items. Used in show-all | ||
|  |                     mode. | ||
|  |     - help          Help text windows at the bottom of various fullscreen | ||
|  |                     dialogs | ||
|  |     - show-help     Window showing the help text in show-help mode | ||
|  |     - frame         Frame around dialog boxes | ||
|  |     - body          Body of dialog boxes | ||
|  |     - edit          Edit box in pop-up dialogs | ||
|  |     - jump-edit     Edit box in jump-to dialog | ||
|  |     - text          Symbol information text | ||
|  | 
 | ||
|  | The color definition is a comma separated list of attributes: | ||
|  | 
 | ||
|  |     - fg:COLOR      Set the foreground/background colors. COLOR can be one of | ||
|  |       * or *        the basic 16 colors (black, red, green, yellow, blue, | ||
|  |     - bg:COLOR      magenta, cyan, white and brighter versions, for example, | ||
|  |                     brightred). On terminals that support more than 8 colors, | ||
|  |                     you can also directly put in a color number, e.g. fg:123 | ||
|  |                     (hexadecimal and octal constants are accepted as well). | ||
|  |                     Colors outside the range -1..curses.COLORS-1 (which is | ||
|  |                     terminal-dependent) are ignored (with a warning). The COLOR | ||
|  |                     can be also specified using a RGB value in the HTML | ||
|  |                     notation, for example #RRGGBB. If the terminal supports | ||
|  |                     color changing, the color is rendered accurately. | ||
|  |                     Otherwise, the visually nearest color is used. | ||
|  | 
 | ||
|  |                     If the background or foreground color of an element is not | ||
|  |                     specified, it defaults to -1, representing the default | ||
|  |                     terminal foreground or background color. | ||
|  | 
 | ||
|  |                     Note: On some terminals a bright version of the color | ||
|  |                     implies bold. | ||
|  |     - bold          Use bold text | ||
|  |     - underline     Use underline text | ||
|  |     - standout      Standout text attribute (reverse color) | ||
|  | 
 | ||
|  | More often than not, some UI elements share the same color definition. In such | ||
|  | cases the right value may specify an UI element from which the color definition | ||
|  | will be copied. For example, "separator=help" will apply the current color | ||
|  | definition for "help" to "separator". | ||
|  | 
 | ||
|  | A keyword without the '=' is assumed to be a style template. The template name | ||
|  | is looked up in the built-in styles list and the style definition is expanded | ||
|  | in-place. With this, built-in styles can be used as basis for new styles. | ||
|  | 
 | ||
|  | For example, take the aquatic theme and give it a red selection bar: | ||
|  | 
 | ||
|  | MENUCONFIG_STYLE="aquatic selection=fg:white,bg:red" | ||
|  | 
 | ||
|  | If there's an error in the style definition or if a missing style is assigned | ||
|  | to, the assignment will be ignored, along with a warning being printed on | ||
|  | stderr. | ||
|  | 
 | ||
|  | The 'default' theme is always implicitly parsed first, so the following two | ||
|  | settings have the same effect: | ||
|  | 
 | ||
|  |     MENUCONFIG_STYLE="selection=fg:white,bg:red" | ||
|  |     MENUCONFIG_STYLE="default selection=fg:white,bg:red" | ||
|  | 
 | ||
|  | If the terminal doesn't support colors, the 'monochrome' theme is used, and | ||
|  | MENUCONFIG_STYLE is ignored. The assumption is that the environment is broken | ||
|  | somehow, and that the important thing is to get something usable. | ||
|  | 
 | ||
|  | 
 | ||
|  | Other features | ||
|  | ============== | ||
|  | 
 | ||
|  |   - Seamless terminal resizing | ||
|  | 
 | ||
|  |   - No dependencies on *nix, as the 'curses' module is in the Python standard | ||
|  |     library | ||
|  | 
 | ||
|  |   - Unicode text entry | ||
|  | 
 | ||
|  |   - Improved information screen compared to mconf: | ||
|  | 
 | ||
|  |       * Expressions are split up by their top-level &&/|| operands to improve | ||
|  |         readability | ||
|  | 
 | ||
|  |       * Undefined symbols in expressions are pointed out | ||
|  | 
 | ||
|  |       * Menus and comments have information displays | ||
|  | 
 | ||
|  |       * Kconfig definitions are printed | ||
|  | 
 | ||
|  |       * The include path is shown, listing the locations of the 'source' | ||
|  |         statements that included the Kconfig file of the symbol (or other | ||
|  |         item) | ||
|  | 
 | ||
|  | 
 | ||
|  | Limitations | ||
|  | =========== | ||
|  | 
 | ||
|  | Doesn't work out of the box on Windows, but can be made to work with | ||
|  | 
 | ||
|  |     pip install windows-curses | ||
|  | 
 | ||
|  | See the https://github.com/zephyrproject-rtos/windows-curses repository. | ||
|  | """
 | ||
|  | from __future__ import print_function | ||
|  | 
 | ||
|  | import os | ||
|  | import sys | ||
|  | 
 | ||
|  | _IS_WINDOWS = os.name == "nt"  # Are we running on Windows? | ||
|  | 
 | ||
|  | try: | ||
|  |     import curses | ||
|  | except ImportError as e: | ||
|  |     if not _IS_WINDOWS: | ||
|  |         raise | ||
|  |     sys.exit("""\
 | ||
|  | menuconfig failed to import the standard Python 'curses' library. Try | ||
|  | installing a package like windows-curses | ||
|  | (https://github.com/zephyrproject-rtos/windows-curses) by running this command | ||
|  | in cmd.exe: | ||
|  | 
 | ||
|  |     pip install windows-curses | ||
|  | 
 | ||
|  | Starting with Kconfiglib 13.0.0, windows-curses is no longer automatically | ||
|  | installed when installing Kconfiglib via pip on Windows (because it breaks | ||
|  | installation on MSYS2). | ||
|  | 
 | ||
|  | Exception: | ||
|  | {}: {}""".format(type(e).__name__, e))
 | ||
|  | 
 | ||
|  | import errno | ||
|  | import locale | ||
|  | import re | ||
|  | import textwrap | ||
|  | 
 | ||
|  | from kconfiglib import Symbol, Choice, MENU, COMMENT, MenuNode, \ | ||
|  |                        BOOL, TRISTATE, STRING, INT, HEX, \ | ||
|  |                        AND, OR, \ | ||
|  |                        expr_str, expr_value, split_expr, \ | ||
|  |                        standard_sc_expr_str, \ | ||
|  |                        TRI_TO_STR, TYPE_TO_STR, \ | ||
|  |                        standard_kconfig, standard_config_filename | ||
|  | 
 | ||
|  | 
 | ||
|  | # | ||
|  | # Configuration variables | ||
|  | # | ||
|  | 
 | ||
|  | # If True, try to change LC_CTYPE to a UTF-8 locale if it is set to the C | ||
|  | # locale (which implies ASCII). This fixes curses Unicode I/O issues on systems | ||
|  | # with bad defaults. ncurses configures itself from the locale settings. | ||
|  | # | ||
|  | # Related PEP: https://www.python.org/dev/peps/pep-0538/ | ||
|  | _CHANGE_C_LC_CTYPE_TO_UTF8 = True | ||
|  | 
 | ||
|  | # How many steps an implicit submenu will be indented. Implicit submenus are | ||
|  | # created when an item depends on the symbol before it. Note that symbols | ||
|  | # defined with 'menuconfig' create a separate menu instead of indenting. | ||
|  | _SUBMENU_INDENT = 4 | ||
|  | 
 | ||
|  | # Number of steps for Page Up/Down to jump | ||
|  | _PG_JUMP = 6 | ||
|  | 
 | ||
|  | # Height of the help window in show-help mode | ||
|  | _SHOW_HELP_HEIGHT = 8 | ||
|  | 
 | ||
|  | # How far the cursor needs to be from the edge of the window before it starts | ||
|  | # to scroll. Used for the main menu display, the information display, the | ||
|  | # search display, and for text boxes. | ||
|  | _SCROLL_OFFSET = 5 | ||
|  | 
 | ||
|  | # Minimum width of dialogs that ask for text input | ||
|  | _INPUT_DIALOG_MIN_WIDTH = 30 | ||
|  | 
 | ||
|  | # Number of arrows pointing up/down to draw when a window is scrolled | ||
|  | _N_SCROLL_ARROWS = 14 | ||
|  | 
 | ||
|  | # Lines of help text shown at the bottom of the "main" display | ||
|  | _MAIN_HELP_LINES = """
 | ||
|  | [Space/Enter] Toggle/enter  [ESC] Leave menu           [S] Save | ||
|  | [O] Load                    [?] Symbol info            [/] Jump to symbol | ||
|  | [F] Toggle show-help mode   [C] Toggle show-name mode  [A] Toggle show-all mode | ||
|  | [Q] Quit (prompts for save) [D] Save minimal config (advanced) | ||
|  | """[1:-1].split("\n")
 | ||
|  | 
 | ||
|  | # Lines of help text shown at the bottom of the information dialog | ||
|  | _INFO_HELP_LINES = """
 | ||
|  | [ESC/q] Return to menu      [/] Jump to symbol | ||
|  | """[1:-1].split("\n")
 | ||
|  | 
 | ||
|  | # Lines of help text shown at the bottom of the search dialog | ||
|  | _JUMP_TO_HELP_LINES = """
 | ||
|  | Type text to narrow the search. Regexes are supported (via Python's 're' | ||
|  | module). The up/down cursor keys step in the list. [Enter] jumps to the | ||
|  | selected symbol. [ESC] aborts the search. Type multiple space-separated | ||
|  | strings/regexes to find entries that match all of them. Type Ctrl-F to | ||
|  | view the help of the selected item without leaving the dialog. | ||
|  | """[1:-1].split("\n")
 | ||
|  | 
 | ||
|  | # | ||
|  | # Styling | ||
|  | # | ||
|  | 
 | ||
|  | _STYLES = { | ||
|  |     "default": """
 | ||
|  |     path=fg:black,bg:white,bold | ||
|  |     separator=fg:black,bg:yellow,bold | ||
|  |     list=fg:black,bg:white | ||
|  |     selection=fg:white,bg:blue,bold | ||
|  |     inv-list=fg:red,bg:white | ||
|  |     inv-selection=fg:red,bg:blue | ||
|  |     help=path | ||
|  |     show-help=list | ||
|  |     frame=fg:black,bg:yellow,bold | ||
|  |     body=fg:white,bg:black | ||
|  |     edit=fg:white,bg:blue | ||
|  |     jump-edit=edit | ||
|  |     text=list | ||
|  |     """,
 | ||
|  | 
 | ||
|  |     # This style is forced on terminals that do no support colors | ||
|  |     "monochrome": """
 | ||
|  |     path=bold | ||
|  |     separator=bold,standout | ||
|  |     list= | ||
|  |     selection=bold,standout | ||
|  |     inv-list=bold | ||
|  |     inv-selection=bold,standout | ||
|  |     help=bold | ||
|  |     show-help= | ||
|  |     frame=bold,standout | ||
|  |     body= | ||
|  |     edit=standout | ||
|  |     jump-edit= | ||
|  |     text= | ||
|  |     """,
 | ||
|  | 
 | ||
|  |     # Blue-tinted style loosely resembling lxdialog | ||
|  |     "aquatic": """
 | ||
|  |     path=fg:white,bg:blue | ||
|  |     separator=fg:white,bg:cyan | ||
|  |     help=path | ||
|  |     frame=fg:white,bg:cyan | ||
|  |     body=fg:white,bg:blue | ||
|  |     edit=fg:black,bg:white | ||
|  |     """
 | ||
|  | } | ||
|  | 
 | ||
|  | _NAMED_COLORS = { | ||
|  |     # Basic colors | ||
|  |     "black":         curses.COLOR_BLACK, | ||
|  |     "red":           curses.COLOR_RED, | ||
|  |     "green":         curses.COLOR_GREEN, | ||
|  |     "yellow":        curses.COLOR_YELLOW, | ||
|  |     "blue":          curses.COLOR_BLUE, | ||
|  |     "magenta":       curses.COLOR_MAGENTA, | ||
|  |     "cyan":          curses.COLOR_CYAN, | ||
|  |     "white":         curses.COLOR_WHITE, | ||
|  | 
 | ||
|  |     # Bright versions | ||
|  |     "brightblack":   curses.COLOR_BLACK + 8, | ||
|  |     "brightred":     curses.COLOR_RED + 8, | ||
|  |     "brightgreen":   curses.COLOR_GREEN + 8, | ||
|  |     "brightyellow":  curses.COLOR_YELLOW + 8, | ||
|  |     "brightblue":    curses.COLOR_BLUE + 8, | ||
|  |     "brightmagenta": curses.COLOR_MAGENTA + 8, | ||
|  |     "brightcyan":    curses.COLOR_CYAN + 8, | ||
|  |     "brightwhite":   curses.COLOR_WHITE + 8, | ||
|  | 
 | ||
|  |     # Aliases | ||
|  |     "purple":        curses.COLOR_MAGENTA, | ||
|  |     "brightpurple":  curses.COLOR_MAGENTA + 8, | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | def _rgb_to_6cube(rgb): | ||
|  |     # Converts an 888 RGB color to a 3-tuple (nice in that it's hashable) | ||
|  |     # representing the closest xterm 256-color 6x6x6 color cube color. | ||
|  |     # | ||
|  |     # The xterm 256-color extension uses a RGB color palette with components in | ||
|  |     # the range 0-5 (a 6x6x6 cube). The catch is that the mapping is nonlinear. | ||
|  |     # Index 0 in the 6x6x6 cube is mapped to 0, index 1 to 95, then 135, 175, | ||
|  |     # etc., in increments of 40. See the links below: | ||
|  |     # | ||
|  |     #   https://commons.wikimedia.org/wiki/File:Xterm_256color_chart.svg | ||
|  |     #   https://github.com/tmux/tmux/blob/master/colour.c | ||
|  | 
 | ||
|  |     # 48 is the middle ground between 0 and 95. | ||
|  |     return tuple(0 if x < 48 else int(round(max(1, (x - 55)/40))) for x in rgb) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _6cube_to_rgb(r6g6b6): | ||
|  |     # Returns the 888 RGB color for a 666 xterm color cube index | ||
|  | 
 | ||
|  |     return tuple(0 if x == 0 else 40*x + 55 for x in r6g6b6) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _rgb_to_gray(rgb): | ||
|  |     # Converts an 888 RGB color to the index of an xterm 256-color grayscale | ||
|  |     # color with approx. the same perceived brightness | ||
|  | 
 | ||
|  |     # Calculate the luminance (gray intensity) of the color. See | ||
|  |     #   https://stackoverflow.com/questions/596216/formula-to-determine-brightness-of-rgb-color | ||
|  |     # and | ||
|  |     #   https://www.w3.org/TR/AERT/#color-contrast | ||
|  |     luma = 0.299*rgb[0] + 0.587*rgb[1] + 0.114*rgb[2] | ||
|  | 
 | ||
|  |     # Closest index in the grayscale palette, which starts at RGB 0x080808, | ||
|  |     # with stepping 0x0A0A0A | ||
|  |     index = int(round((luma - 8)/10)) | ||
|  | 
 | ||
|  |     # Clamp the index to 0-23, corresponding to 232-255 | ||
|  |     return max(0, min(index, 23)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _gray_to_rgb(index): | ||
|  |     # Convert a grayscale index to its closet single RGB component | ||
|  | 
 | ||
|  |     return 3*(10*index + 8,)  # Returns a 3-tuple | ||
|  | 
 | ||
|  | 
 | ||
|  | # Obscure Python: We never pass a value for rgb2index, and it keeps pointing to | ||
|  | # the same dict. This avoids a global. | ||
|  | def _alloc_rgb(rgb, rgb2index={}): | ||
|  |     # Initialize a new entry in the xterm palette to the given RGB color, | ||
|  |     # returning its index. If the color has already been initialized, the index | ||
|  |     # of the existing entry is returned. | ||
|  |     # | ||
|  |     # ncurses is palette-based, so we need to overwrite palette entries to make | ||
|  |     # new colors. | ||
|  |     # | ||
|  |     # The colors from 0 to 15 are user-defined, and there's no way to query | ||
|  |     # their RGB values, so we better leave them untouched. Also leave any | ||
|  |     # hypothetical colors above 255 untouched (though we're unlikely to | ||
|  |     # allocate that many colors anyway). | ||
|  | 
 | ||
|  |     if rgb in rgb2index: | ||
|  |         return rgb2index[rgb] | ||
|  | 
 | ||
|  |     # Many terminals allow the user to customize the first 16 colors. Avoid | ||
|  |     # changing their values. | ||
|  |     color_index = 16 + len(rgb2index) | ||
|  |     if color_index >= 256: | ||
|  |         _warn("Unable to allocate new RGB color ", rgb, ". Too many colors " | ||
|  |               "allocated.") | ||
|  |         return 0 | ||
|  | 
 | ||
|  |     # Map each RGB component from the range 0-255 to the range 0-1000, which is | ||
|  |     # what curses uses | ||
|  |     curses.init_color(color_index, *(int(round(1000*x/255)) for x in rgb)) | ||
|  |     rgb2index[rgb] = color_index | ||
|  | 
 | ||
|  |     return color_index | ||
|  | 
 | ||
|  | 
 | ||
|  | def _color_from_num(num): | ||
|  |     # Returns the index of a color that looks like color 'num' in the xterm | ||
|  |     # 256-color palette (but that might not be 'num', if we're redefining | ||
|  |     # colors) | ||
|  | 
 | ||
|  |     # - _alloc_rgb() won't touch the first 16 colors or any (hypothetical) | ||
|  |     #   colors above 255, so we can always return them as-is | ||
|  |     # | ||
|  |     # - If the terminal doesn't support changing color definitions, or if | ||
|  |     #   curses.COLORS < 256, _alloc_rgb() won't touch any color, and all colors | ||
|  |     #   can be returned as-is | ||
|  |     if num < 16 or num > 255 or not curses.can_change_color() or \ | ||
|  |        curses.COLORS < 256: | ||
|  |         return num | ||
|  | 
 | ||
|  |     # _alloc_rgb() might redefine colors, so emulate the xterm 256-color | ||
|  |     # palette by allocating new colors instead of returning color numbers | ||
|  |     # directly | ||
|  | 
 | ||
|  |     if num < 232: | ||
|  |         num -= 16 | ||
|  |         return _alloc_rgb(_6cube_to_rgb(((num//36)%6, (num//6)%6, num%6))) | ||
|  | 
 | ||
|  |     return _alloc_rgb(_gray_to_rgb(num - 232)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _color_from_rgb(rgb): | ||
|  |     # Returns the index of a color matching the 888 RGB color 'rgb'. The | ||
|  |     # returned color might be an ~exact match or an approximation, depending on | ||
|  |     # terminal capabilities. | ||
|  | 
 | ||
|  |     # Calculates the Euclidean distance between two RGB colors | ||
|  |     def dist(r1, r2): return sum((x - y)**2 for x, y in zip(r1, r2)) | ||
|  | 
 | ||
|  |     if curses.COLORS >= 256: | ||
|  |         # Assume we're dealing with xterm's 256-color extension | ||
|  | 
 | ||
|  |         if curses.can_change_color(): | ||
|  |             # Best case -- the terminal supports changing palette entries via | ||
|  |             # curses.init_color(). Initialize an unused palette entry and | ||
|  |             # return it. | ||
|  |             return _alloc_rgb(rgb) | ||
|  | 
 | ||
|  |         # Second best case -- pick between the xterm 256-color extension colors | ||
|  | 
 | ||
|  |         # Closest 6-cube "color" color | ||
|  |         c6 = _rgb_to_6cube(rgb) | ||
|  |         # Closest gray color | ||
|  |         gray = _rgb_to_gray(rgb) | ||
|  | 
 | ||
|  |         if dist(rgb, _6cube_to_rgb(c6)) < dist(rgb, _gray_to_rgb(gray)): | ||
|  |             # Use the "color" color from the 6x6x6 color palette. Calculate the | ||
|  |             # color number from the 6-cube index triplet. | ||
|  |             return 16 + 36*c6[0] + 6*c6[1] + c6[2] | ||
|  | 
 | ||
|  |         # Use the color from the gray palette | ||
|  |         return 232 + gray | ||
|  | 
 | ||
|  |     # Terminal not in xterm 256-color mode. This is probably the best we can | ||
|  |     # do, or is it? Submit patches. :) | ||
|  |     min_dist = float('inf') | ||
|  |     best = -1 | ||
|  |     for color in range(curses.COLORS): | ||
|  |         # ncurses uses the range 0..1000. Scale that down to 0..255. | ||
|  |         d = dist(rgb, tuple(int(round(255*c/1000)) | ||
|  |                             for c in curses.color_content(color))) | ||
|  |         if d < min_dist: | ||
|  |             min_dist = d | ||
|  |             best = color | ||
|  | 
 | ||
|  |     return best | ||
|  | 
 | ||
|  | 
 | ||
|  | def _parse_style(style_str, parsing_default): | ||
|  |     # Parses a string with '<element>=<style>' assignments. Anything not | ||
|  |     # containing '=' is assumed to be a reference to a built-in style, which is | ||
|  |     # treated as if all the assignments from the style were inserted at that | ||
|  |     # point in the string. | ||
|  |     # | ||
|  |     # The parsing_default flag is set to True when we're implicitly parsing the | ||
|  |     # 'default'/'monochrome' style, to prevent warnings. | ||
|  | 
 | ||
|  |     for sline in style_str.split(): | ||
|  |         # Words without a "=" character represents a style template | ||
|  |         if "=" in sline: | ||
|  |             key, data = sline.split("=", 1) | ||
|  | 
 | ||
|  |             # The 'default' style template is assumed to define all keys. We | ||
|  |             # run _style_to_curses() for non-existing keys as well, so that we | ||
|  |             # print warnings for errors to the right of '=' for those too. | ||
|  |             if key not in _style and not parsing_default: | ||
|  |                 _warn("Ignoring non-existent style", key) | ||
|  | 
 | ||
|  |             # If data is a reference to another key, copy its style | ||
|  |             if data in _style: | ||
|  |                 _style[key] = _style[data] | ||
|  |             else: | ||
|  |                 _style[key] = _style_to_curses(data) | ||
|  | 
 | ||
|  |         elif sline in _STYLES: | ||
|  |             # Recursively parse style template. Ignore styles that don't exist, | ||
|  |             # for backwards/forwards compatibility. | ||
|  |             _parse_style(_STYLES[sline], parsing_default) | ||
|  | 
 | ||
|  |         else: | ||
|  |             _warn("Ignoring non-existent style template", sline) | ||
|  | 
 | ||
|  | # Dictionary mapping element types to the curses attributes used to display | ||
|  | # them | ||
|  | _style = {} | ||
|  | 
 | ||
|  | 
 | ||
|  | def _style_to_curses(style_def): | ||
|  |     # Parses a style definition string (<element>=<style>), returning | ||
|  |     # a (fg_color, bg_color, attributes) tuple. | ||
|  | 
 | ||
|  |     def parse_color(color_def): | ||
|  |         color_def = color_def.split(":", 1)[1] | ||
|  | 
 | ||
|  |         # HTML format, #RRGGBB | ||
|  |         if re.match("#[A-Fa-f0-9]{6}", color_def): | ||
|  |             return _color_from_rgb(( | ||
|  |                 int(color_def[1:3], 16), | ||
|  |                 int(color_def[3:5], 16), | ||
|  |                 int(color_def[5:7], 16))) | ||
|  | 
 | ||
|  |         if color_def in _NAMED_COLORS: | ||
|  |             color_num = _color_from_num(_NAMED_COLORS[color_def]) | ||
|  |         else: | ||
|  |             try: | ||
|  |                 color_num = _color_from_num(int(color_def, 0)) | ||
|  |             except ValueError: | ||
|  |                 _warn("Ignoring color", color_def, "that's neither " | ||
|  |                       "predefined nor a number") | ||
|  |                 return -1 | ||
|  | 
 | ||
|  |         if not -1 <= color_num < curses.COLORS: | ||
|  |             _warn("Ignoring color {}, which is outside the range " | ||
|  |                   "-1..curses.COLORS-1 (-1..{})" | ||
|  |                   .format(color_def, curses.COLORS - 1)) | ||
|  |             return -1 | ||
|  | 
 | ||
|  |         return color_num | ||
|  | 
 | ||
|  |     fg_color = -1 | ||
|  |     bg_color = -1 | ||
|  |     attrs = 0 | ||
|  | 
 | ||
|  |     if style_def: | ||
|  |         for field in style_def.split(","): | ||
|  |             if field.startswith("fg:"): | ||
|  |                 fg_color = parse_color(field) | ||
|  |             elif field.startswith("bg:"): | ||
|  |                 bg_color = parse_color(field) | ||
|  |             elif field == "bold": | ||
|  |                 # A_BOLD tends to produce faint and hard-to-read text on the | ||
|  |                 # Windows console, especially with the old color scheme, before | ||
|  |                 # the introduction of | ||
|  |                 # https://blogs.msdn.microsoft.com/commandline/2017/08/02/updating-the-windows-console-colors/ | ||
|  |                 attrs |= curses.A_NORMAL if _IS_WINDOWS else curses.A_BOLD | ||
|  |             elif field == "standout": | ||
|  |                 attrs |= curses.A_STANDOUT | ||
|  |             elif field == "underline": | ||
|  |                 attrs |= curses.A_UNDERLINE | ||
|  |             else: | ||
|  |                 _warn("Ignoring unknown style attribute", field) | ||
|  | 
 | ||
|  |     return _style_attr(fg_color, bg_color, attrs) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _init_styles(): | ||
|  |     if curses.has_colors(): | ||
|  |         try: | ||
|  |             curses.use_default_colors() | ||
|  |         except curses.error: | ||
|  |             # Ignore errors on funky terminals that support colors but not | ||
|  |             # using default colors. Worst it can do is break transparency and | ||
|  |             # the like. Ran across this with the MSYS2/winpty setup in | ||
|  |             # https://github.com/msys2/MINGW-packages/issues/5823, though there | ||
|  |             # seems to be a lot of general brokenness there. | ||
|  |             pass | ||
|  | 
 | ||
|  |         # Use the 'default' theme as the base, and add any user-defined style | ||
|  |         # settings from the environment | ||
|  |         _parse_style("default", True) | ||
|  |         if "MENUCONFIG_STYLE" in os.environ: | ||
|  |             _parse_style(os.environ["MENUCONFIG_STYLE"], False) | ||
|  |     else: | ||
|  |         # Force the 'monochrome' theme if the terminal doesn't support colors. | ||
|  |         # MENUCONFIG_STYLE is likely to mess things up here (though any colors | ||
|  |         # would be ignored), so ignore it. | ||
|  |         _parse_style("monochrome", True) | ||
|  | 
 | ||
|  | 
 | ||
|  | # color_attribs holds the color pairs we've already created, indexed by a | ||
|  | # (<foreground color>, <background color>) tuple. | ||
|  | # | ||
|  | # Obscure Python: We never pass a value for color_attribs, and it keeps | ||
|  | # pointing to the same dict. This avoids a global. | ||
|  | def _style_attr(fg_color, bg_color, attribs, color_attribs={}): | ||
|  |     # Returns an attribute with the specified foreground and background color | ||
|  |     # and the attributes in 'attribs'. Reuses color pairs already created if | ||
|  |     # possible, and creates a new color pair otherwise. | ||
|  |     # | ||
|  |     # Returns 'attribs' if colors aren't supported. | ||
|  | 
 | ||
|  |     if not curses.has_colors(): | ||
|  |         return attribs | ||
|  | 
 | ||
|  |     if (fg_color, bg_color) not in color_attribs: | ||
|  |         # Create new color pair. Color pair number 0 is hardcoded and cannot be | ||
|  |         # changed, hence the +1s. | ||
|  |         curses.init_pair(len(color_attribs) + 1, fg_color, bg_color) | ||
|  |         color_attribs[(fg_color, bg_color)] = \ | ||
|  |             curses.color_pair(len(color_attribs) + 1) | ||
|  | 
 | ||
|  |     return color_attribs[(fg_color, bg_color)] | attribs | ||
|  | 
 | ||
|  | 
 | ||
|  | # | ||
|  | # Main application | ||
|  | # | ||
|  | 
 | ||
|  | 
 | ||
|  | def _main(): | ||
|  |     menuconfig(standard_kconfig(__doc__)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def menuconfig(kconf): | ||
|  |     """
 | ||
|  |     Launches the configuration interface, returning after the user exits. | ||
|  | 
 | ||
|  |     kconf: | ||
|  |       Kconfig instance to be configured | ||
|  |     """
 | ||
|  |     global _kconf | ||
|  |     global _conf_filename | ||
|  |     global _conf_changed | ||
|  |     global _minconf_filename | ||
|  |     global _show_all | ||
|  | 
 | ||
|  |     _kconf = kconf | ||
|  | 
 | ||
|  |     # Filename to save configuration to | ||
|  |     _conf_filename = standard_config_filename() | ||
|  | 
 | ||
|  |     # Load existing configuration and set _conf_changed True if it is outdated | ||
|  |     _conf_changed = _load_config() | ||
|  | 
 | ||
|  |     # Filename to save minimal configuration to | ||
|  |     _minconf_filename = "defconfig" | ||
|  | 
 | ||
|  |     # Any visible items in the top menu? | ||
|  |     _show_all = False | ||
|  |     if not _shown_nodes(kconf.top_node): | ||
|  |         # Nothing visible. Start in show-all mode and try again. | ||
|  |         _show_all = True | ||
|  |         if not _shown_nodes(kconf.top_node): | ||
|  |             # Give up. The implementation relies on always having a selected | ||
|  |             # node. | ||
|  |             print("Empty configuration -- nothing to configure.\n" | ||
|  |                   "Check that environment variables are set properly.") | ||
|  |             return | ||
|  | 
 | ||
|  |     # Disable warnings. They get mangled in curses mode, and we deal with | ||
|  |     # errors ourselves. | ||
|  |     kconf.warn = False | ||
|  | 
 | ||
|  |     # Make curses use the locale settings specified in the environment | ||
|  |     locale.setlocale(locale.LC_ALL, "") | ||
|  | 
 | ||
|  |     # Try to fix Unicode issues on systems with bad defaults | ||
|  |     if _CHANGE_C_LC_CTYPE_TO_UTF8: | ||
|  |         _change_c_lc_ctype_to_utf8() | ||
|  | 
 | ||
|  |     # Get rid of the delay between pressing ESC and jumping to the parent menu, | ||
|  |     # unless the user has set ESCDELAY (see ncurses(3)). This makes the UI much | ||
|  |     # smoother to work with. | ||
|  |     # | ||
|  |     # Note: This is strictly pretty iffy, since escape codes for e.g. cursor | ||
|  |     # keys start with ESC, but I've never seen it cause problems in practice | ||
|  |     # (probably because it's unlikely that the escape code for a key would get | ||
|  |     # split up across read()s, at least with a terminal emulator). Please | ||
|  |     # report if you run into issues. Some suitable small default value could be | ||
|  |     # used here instead in that case. Maybe it's silly to not put in the | ||
|  |     # smallest imperceptible delay here already, though I don't like guessing. | ||
|  |     # | ||
|  |     # (From a quick glance at the ncurses source code, ESCDELAY might only be | ||
|  |     # relevant for mouse events there, so maybe escapes are assumed to arrive | ||
|  |     # in one piece already...) | ||
|  |     os.environ.setdefault("ESCDELAY", "0") | ||
|  | 
 | ||
|  |     # Enter curses mode. _menuconfig() returns a string to print on exit, after | ||
|  |     # curses has been de-initialized. | ||
|  |     print(curses.wrapper(_menuconfig)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _load_config(): | ||
|  |     # Loads any existing .config file. See the Kconfig.load_config() docstring. | ||
|  |     # | ||
|  |     # Returns True if .config is missing or outdated. We always prompt for | ||
|  |     # saving the configuration in that case. | ||
|  | 
 | ||
|  |     print(_kconf.load_config()) | ||
|  |     if not os.path.exists(_conf_filename): | ||
|  |         # No .config | ||
|  |         return True | ||
|  | 
 | ||
|  |     return _needs_save() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _needs_save(): | ||
|  |     # Returns True if a just-loaded .config file is outdated (would get | ||
|  |     # modified when saving) | ||
|  | 
 | ||
|  |     if _kconf.missing_syms: | ||
|  |         # Assignments to undefined symbols in the .config | ||
|  |         return True | ||
|  | 
 | ||
|  |     for sym in _kconf.unique_defined_syms: | ||
|  |         if sym.user_value is None: | ||
|  |             if sym.config_string: | ||
|  |                 # Unwritten symbol | ||
|  |                 return True | ||
|  |         elif sym.orig_type in (BOOL, TRISTATE): | ||
|  |             if sym.tri_value != sym.user_value: | ||
|  |                 # Written bool/tristate symbol, new value | ||
|  |                 return True | ||
|  |         elif sym.str_value != sym.user_value: | ||
|  |             # Written string/int/hex symbol, new value | ||
|  |             return True | ||
|  | 
 | ||
|  |     # No need to prompt for save | ||
|  |     return False | ||
|  | 
 | ||
|  | 
 | ||
|  | # Global variables used below: | ||
|  | # | ||
|  | #   _stdscr: | ||
|  | #     stdscr from curses | ||
|  | # | ||
|  | #   _cur_menu: | ||
|  | #     Menu node of the menu (or menuconfig symbol, or choice) currently being | ||
|  | #     shown | ||
|  | # | ||
|  | #   _shown: | ||
|  | #     List of items in _cur_menu that are shown (ignoring scrolling). In | ||
|  | #     show-all mode, this list contains all items in _cur_menu. Otherwise, it | ||
|  | #     contains just the visible items. | ||
|  | # | ||
|  | #   _sel_node_i: | ||
|  | #     Index in _shown of the currently selected node | ||
|  | # | ||
|  | #   _menu_scroll: | ||
|  | #     Index in _shown of the top row of the main display | ||
|  | # | ||
|  | #   _parent_screen_rows: | ||
|  | #     List/stack of the row numbers that the selections in the parent menus | ||
|  | #     appeared on. This is used to prevent the scrolling from jumping around | ||
|  | #     when going in and out of menus. | ||
|  | # | ||
|  | #   _show_help/_show_name/_show_all: | ||
|  | #     If True, the corresponding mode is on. See the module docstring. | ||
|  | # | ||
|  | #   _conf_filename: | ||
|  | #     File to save the configuration to | ||
|  | # | ||
|  | #   _minconf_filename: | ||
|  | #     File to save minimal configurations to | ||
|  | # | ||
|  | #   _conf_changed: | ||
|  | #     True if the configuration has been changed. If False, we don't bother | ||
|  | #     showing the save-and-quit dialog. | ||
|  | # | ||
|  | #     We reset this to False whenever the configuration is saved explicitly | ||
|  | #     from the save dialog. | ||
|  | 
 | ||
|  | 
 | ||
|  | def _menuconfig(stdscr): | ||
|  |     # Logic for the main display, with the list of symbols, etc. | ||
|  | 
 | ||
|  |     global _stdscr | ||
|  |     global _conf_filename | ||
|  |     global _conf_changed | ||
|  |     global _minconf_filename | ||
|  |     global _show_help | ||
|  |     global _show_name | ||
|  | 
 | ||
|  |     _stdscr = stdscr | ||
|  | 
 | ||
|  |     _init() | ||
|  | 
 | ||
|  |     while True: | ||
|  |         _draw_main() | ||
|  |         curses.doupdate() | ||
|  | 
 | ||
|  | 
 | ||
|  |         c = _getch_compat(_menu_win) | ||
|  | 
 | ||
|  |         if c == curses.KEY_RESIZE: | ||
|  |             _resize_main() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_DOWN, "j", "J"): | ||
|  |             _select_next_menu_entry() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_UP, "k", "K"): | ||
|  |             _select_prev_menu_entry() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_NPAGE, "\x04"):  # Page Down/Ctrl-D | ||
|  |             # Keep it simple. This way we get sane behavior for small windows, | ||
|  |             # etc., for free. | ||
|  |             for _ in range(_PG_JUMP): | ||
|  |                 _select_next_menu_entry() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_PPAGE, "\x15"):  # Page Up/Ctrl-U | ||
|  |             for _ in range(_PG_JUMP): | ||
|  |                 _select_prev_menu_entry() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_END, "G"): | ||
|  |             _select_last_menu_entry() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_HOME, "g"): | ||
|  |             _select_first_menu_entry() | ||
|  | 
 | ||
|  |         elif c == " ": | ||
|  |             # Toggle the node if possible | ||
|  |             sel_node = _shown[_sel_node_i] | ||
|  |             if not _change_node(sel_node): | ||
|  |                 _enter_menu(sel_node) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_RIGHT, "\n", "l", "L"): | ||
|  |             # Enter the node if possible | ||
|  |             sel_node = _shown[_sel_node_i] | ||
|  |             if not _enter_menu(sel_node): | ||
|  |                 _change_node(sel_node) | ||
|  | 
 | ||
|  |         elif c in ("n", "N"): | ||
|  |             _set_sel_node_tri_val(0) | ||
|  | 
 | ||
|  |         elif c in ("m", "M"): | ||
|  |             _set_sel_node_tri_val(1) | ||
|  | 
 | ||
|  |         elif c in ("y", "Y"): | ||
|  |             _set_sel_node_tri_val(2) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_LEFT, curses.KEY_BACKSPACE, _ERASE_CHAR, | ||
|  |                    "\x1B", "h", "H"):  # \x1B = ESC | ||
|  | 
 | ||
|  |             if c == "\x1B" and _cur_menu is _kconf.top_node: | ||
|  |                 res = _quit_dialog() | ||
|  |                 if res: | ||
|  |                     return res | ||
|  |             else: | ||
|  |                 _leave_menu() | ||
|  | 
 | ||
|  |         elif c in ("o", "O"): | ||
|  |             _load_dialog() | ||
|  | 
 | ||
|  |         elif c in ("s", "S"): | ||
|  |             filename = _save_dialog(_kconf.write_config, _conf_filename, | ||
|  |                                     "configuration") | ||
|  |             if filename: | ||
|  |                 _conf_filename = filename | ||
|  |                 _conf_changed = False | ||
|  | 
 | ||
|  |         elif c in ("d", "D"): | ||
|  |             filename = _save_dialog(_kconf.write_min_config, _minconf_filename, | ||
|  |                                     "minimal configuration") | ||
|  |             if filename: | ||
|  |                 _minconf_filename = filename | ||
|  | 
 | ||
|  |         elif c == "/": | ||
|  |             _jump_to_dialog() | ||
|  |             # The terminal might have been resized while the fullscreen jump-to | ||
|  |             # dialog was open | ||
|  |             _resize_main() | ||
|  | 
 | ||
|  |         elif c == "?": | ||
|  |             _info_dialog(_shown[_sel_node_i], False) | ||
|  |             # The terminal might have been resized while the fullscreen info | ||
|  |             # dialog was open | ||
|  |             _resize_main() | ||
|  | 
 | ||
|  |         elif c in ("f", "F"): | ||
|  |             _show_help = not _show_help | ||
|  |             _set_style(_help_win, "show-help" if _show_help else "help") | ||
|  |             _resize_main() | ||
|  | 
 | ||
|  |         elif c in ("c", "C"): | ||
|  |             _show_name = not _show_name | ||
|  | 
 | ||
|  |         elif c in ("a", "A"): | ||
|  |             _toggle_show_all() | ||
|  | 
 | ||
|  |         elif c in ("q", "Q"): | ||
|  |             res = _quit_dialog() | ||
|  |             if res: | ||
|  |                 return res | ||
|  | 
 | ||
|  | 
 | ||
|  | def _quit_dialog(): | ||
|  |     if not _conf_changed: | ||
|  |         return "No changes to save (for '{}')".format(_conf_filename) | ||
|  | 
 | ||
|  |     while True: | ||
|  |         c = _key_dialog( | ||
|  |             "Quit", | ||
|  |             " Save configuration?\n" | ||
|  |             "\n" | ||
|  |             "(Y)es  (N)o  (C)ancel", | ||
|  |             "ync") | ||
|  | 
 | ||
|  |         if c is None or c == "c": | ||
|  |             return None | ||
|  | 
 | ||
|  |         if c == "y": | ||
|  |             # Returns a message to print | ||
|  |             msg = _try_save(_kconf.write_config, _conf_filename, "configuration") | ||
|  |             if msg: | ||
|  |                 return msg | ||
|  | 
 | ||
|  |         elif c == "n": | ||
|  |             return "Configuration ({}) was not saved".format(_conf_filename) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _init(): | ||
|  |     # Initializes the main display with the list of symbols, etc. Also does | ||
|  |     # misc. global initialization that needs to happen after initializing | ||
|  |     # curses. | ||
|  | 
 | ||
|  |     global _ERASE_CHAR | ||
|  | 
 | ||
|  |     global _path_win | ||
|  |     global _top_sep_win | ||
|  |     global _menu_win | ||
|  |     global _bot_sep_win | ||
|  |     global _help_win | ||
|  | 
 | ||
|  |     global _parent_screen_rows | ||
|  |     global _cur_menu | ||
|  |     global _shown | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     global _show_help | ||
|  |     global _show_name | ||
|  | 
 | ||
|  |     # Looking for this in addition to KEY_BACKSPACE (which is unreliable) makes | ||
|  |     # backspace work with TERM=vt100. That makes it likely to work in sane | ||
|  |     # environments. | ||
|  |     _ERASE_CHAR = curses.erasechar() | ||
|  |     if sys.version_info[0] >= 3: | ||
|  |         # erasechar() returns a one-byte bytes object on Python 3. This sets | ||
|  |         # _ERASE_CHAR to a blank string if it can't be decoded, which should be | ||
|  |         # harmless. | ||
|  |         _ERASE_CHAR = _ERASE_CHAR.decode("utf-8", "ignore") | ||
|  | 
 | ||
|  |     _init_styles() | ||
|  | 
 | ||
|  |     # Hide the cursor | ||
|  |     _safe_curs_set(0) | ||
|  | 
 | ||
|  |     # Initialize windows | ||
|  | 
 | ||
|  |     # Top row, with menu path | ||
|  |     _path_win = _styled_win("path") | ||
|  | 
 | ||
|  |     # Separator below menu path, with title and arrows pointing up | ||
|  |     _top_sep_win = _styled_win("separator") | ||
|  | 
 | ||
|  |     # List of menu entries with symbols, etc. | ||
|  |     _menu_win = _styled_win("list") | ||
|  |     _menu_win.keypad(True) | ||
|  | 
 | ||
|  |     # Row below menu list, with arrows pointing down | ||
|  |     _bot_sep_win = _styled_win("separator") | ||
|  | 
 | ||
|  |     # Help window with keys at the bottom. Shows help texts in show-help mode. | ||
|  |     _help_win = _styled_win("help") | ||
|  | 
 | ||
|  |     # The rows we'd like the nodes in the parent menus to appear on. This | ||
|  |     # prevents the scroll from jumping around when going in and out of menus. | ||
|  |     _parent_screen_rows = [] | ||
|  | 
 | ||
|  |     # Initial state | ||
|  | 
 | ||
|  |     _cur_menu = _kconf.top_node | ||
|  |     _shown = _shown_nodes(_cur_menu) | ||
|  |     _sel_node_i = _menu_scroll = 0 | ||
|  | 
 | ||
|  |     _show_help = _show_name = False | ||
|  | 
 | ||
|  |     # Give windows their initial size | ||
|  |     _resize_main() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _resize_main(): | ||
|  |     # Resizes the main display, with the list of symbols, etc., to fill the | ||
|  |     # terminal | ||
|  | 
 | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     screen_height, screen_width = _stdscr.getmaxyx() | ||
|  | 
 | ||
|  |     _path_win.resize(1, screen_width) | ||
|  |     _top_sep_win.resize(1, screen_width) | ||
|  |     _bot_sep_win.resize(1, screen_width) | ||
|  | 
 | ||
|  |     help_win_height = _SHOW_HELP_HEIGHT if _show_help else \ | ||
|  |         len(_MAIN_HELP_LINES) | ||
|  | 
 | ||
|  |     menu_win_height = screen_height - help_win_height - 3 | ||
|  | 
 | ||
|  |     if menu_win_height >= 1: | ||
|  |         _menu_win.resize(menu_win_height, screen_width) | ||
|  |         _help_win.resize(help_win_height, screen_width) | ||
|  | 
 | ||
|  |         _top_sep_win.mvwin(1, 0) | ||
|  |         _menu_win.mvwin(2, 0) | ||
|  |         _bot_sep_win.mvwin(2 + menu_win_height, 0) | ||
|  |         _help_win.mvwin(2 + menu_win_height + 1, 0) | ||
|  |     else: | ||
|  |         # Degenerate case. Give up on nice rendering and just prevent errors. | ||
|  | 
 | ||
|  |         menu_win_height = 1 | ||
|  | 
 | ||
|  |         _menu_win.resize(1, screen_width) | ||
|  |         _help_win.resize(1, screen_width) | ||
|  | 
 | ||
|  |         for win in _top_sep_win, _menu_win, _bot_sep_win, _help_win: | ||
|  |             win.mvwin(0, 0) | ||
|  | 
 | ||
|  |     # Adjust the scroll so that the selected node is still within the window, | ||
|  |     # if needed | ||
|  |     if _sel_node_i - _menu_scroll >= menu_win_height: | ||
|  |         _menu_scroll = _sel_node_i - menu_win_height + 1 | ||
|  | 
 | ||
|  | 
 | ||
|  | def _height(win): | ||
|  |     # Returns the height of 'win' | ||
|  | 
 | ||
|  |     return win.getmaxyx()[0] | ||
|  | 
 | ||
|  | 
 | ||
|  | def _width(win): | ||
|  |     # Returns the width of 'win' | ||
|  | 
 | ||
|  |     return win.getmaxyx()[1] | ||
|  | 
 | ||
|  | 
 | ||
|  | def _enter_menu(menu): | ||
|  |     # Makes 'menu' the currently displayed menu. In addition to actual 'menu's, | ||
|  |     # "menu" here includes choices and symbols defined with the 'menuconfig' | ||
|  |     # keyword. | ||
|  |     # | ||
|  |     # Returns False if 'menu' can't be entered. | ||
|  | 
 | ||
|  |     global _cur_menu | ||
|  |     global _shown | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     if not menu.is_menuconfig: | ||
|  |         return False  # Not a menu | ||
|  | 
 | ||
|  |     shown_sub = _shown_nodes(menu) | ||
|  |     # Never enter empty menus. We depend on having a current node. | ||
|  |     if not shown_sub: | ||
|  |         return False | ||
|  | 
 | ||
|  |     # Remember where the current node appears on the screen, so we can try | ||
|  |     # to get it to appear in the same place when we leave the menu | ||
|  |     _parent_screen_rows.append(_sel_node_i - _menu_scroll) | ||
|  | 
 | ||
|  |     # Jump into menu | ||
|  |     _cur_menu = menu | ||
|  |     _shown = shown_sub | ||
|  |     _sel_node_i = _menu_scroll = 0 | ||
|  | 
 | ||
|  |     if isinstance(menu.item, Choice): | ||
|  |         _select_selected_choice_sym() | ||
|  | 
 | ||
|  |     return True | ||
|  | 
 | ||
|  | 
 | ||
|  | def _select_selected_choice_sym(): | ||
|  |     # Puts the cursor on the currently selected (y-valued) choice symbol, if | ||
|  |     # any. Does nothing if if the choice has no selection (is not visible/in y | ||
|  |     # mode). | ||
|  | 
 | ||
|  |     global _sel_node_i | ||
|  | 
 | ||
|  |     choice = _cur_menu.item | ||
|  |     if choice.selection: | ||
|  |         # Search through all menu nodes to handle choice symbols being defined | ||
|  |         # in multiple locations | ||
|  |         for node in choice.selection.nodes: | ||
|  |             if node in _shown: | ||
|  |                 _sel_node_i = _shown.index(node) | ||
|  |                 _center_vertically() | ||
|  |                 return | ||
|  | 
 | ||
|  | 
 | ||
|  | def _jump_to(node): | ||
|  |     # Jumps directly to the menu node 'node' | ||
|  | 
 | ||
|  |     global _cur_menu | ||
|  |     global _shown | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  |     global _show_all | ||
|  |     global _parent_screen_rows | ||
|  | 
 | ||
|  |     # Clear remembered menu locations. We might not even have been in the | ||
|  |     # parent menus before. | ||
|  |     _parent_screen_rows = [] | ||
|  | 
 | ||
|  |     old_show_all = _show_all | ||
|  |     jump_into = (isinstance(node.item, Choice) or node.item == MENU) and \ | ||
|  |                 node.list | ||
|  | 
 | ||
|  |     # If we're jumping to a non-empty choice or menu, jump to the first entry | ||
|  |     # in it instead of jumping to its menu node | ||
|  |     if jump_into: | ||
|  |         _cur_menu = node | ||
|  |         node = node.list | ||
|  |     else: | ||
|  |         _cur_menu = _parent_menu(node) | ||
|  | 
 | ||
|  |     _shown = _shown_nodes(_cur_menu) | ||
|  |     if node not in _shown: | ||
|  |         # The node wouldn't be shown. Turn on show-all to show it. | ||
|  |         _show_all = True | ||
|  |         _shown = _shown_nodes(_cur_menu) | ||
|  | 
 | ||
|  |     _sel_node_i = _shown.index(node) | ||
|  | 
 | ||
|  |     if jump_into and not old_show_all and _show_all: | ||
|  |         # If we're jumping into a choice or menu and were forced to turn on | ||
|  |         # show-all because the first entry wasn't visible, try turning it off. | ||
|  |         # That will land us at the first visible node if there are visible | ||
|  |         # nodes, and is a no-op otherwise. | ||
|  |         _toggle_show_all() | ||
|  | 
 | ||
|  |     _center_vertically() | ||
|  | 
 | ||
|  |     # If we're jumping to a non-empty choice, jump to the selected symbol, if | ||
|  |     # any | ||
|  |     if jump_into and isinstance(_cur_menu.item, Choice): | ||
|  |         _select_selected_choice_sym() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _leave_menu(): | ||
|  |     # Jumps to the parent menu of the current menu. Does nothing if we're in | ||
|  |     # the top menu. | ||
|  | 
 | ||
|  |     global _cur_menu | ||
|  |     global _shown | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     if _cur_menu is _kconf.top_node: | ||
|  |         return | ||
|  | 
 | ||
|  |     # Jump to parent menu | ||
|  |     parent = _parent_menu(_cur_menu) | ||
|  |     _shown = _shown_nodes(parent) | ||
|  |     _sel_node_i = _shown.index(_cur_menu) | ||
|  |     _cur_menu = parent | ||
|  | 
 | ||
|  |     # Try to make the menu entry appear on the same row on the screen as it did | ||
|  |     # before we entered the menu. | ||
|  | 
 | ||
|  |     if _parent_screen_rows: | ||
|  |         # The terminal might have shrunk since we were last in the parent menu | ||
|  |         screen_row = min(_parent_screen_rows.pop(), _height(_menu_win) - 1) | ||
|  |         _menu_scroll = max(_sel_node_i - screen_row, 0) | ||
|  |     else: | ||
|  |         # No saved parent menu locations, meaning we jumped directly to some | ||
|  |         # node earlier | ||
|  |         _center_vertically() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _select_next_menu_entry(): | ||
|  |     # Selects the menu entry after the current one, adjusting the scroll if | ||
|  |     # necessary. Does nothing if we're already at the last menu entry. | ||
|  | 
 | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     if _sel_node_i < len(_shown) - 1: | ||
|  |         # Jump to the next node | ||
|  |         _sel_node_i += 1 | ||
|  | 
 | ||
|  |         # If the new node is sufficiently close to the edge of the menu window | ||
|  |         # (as determined by _SCROLL_OFFSET), increase the scroll by one. This | ||
|  |         # gives nice and non-jumpy behavior even when | ||
|  |         # _SCROLL_OFFSET >= _height(_menu_win). | ||
|  |         if _sel_node_i >= _menu_scroll + _height(_menu_win) - _SCROLL_OFFSET \ | ||
|  |            and _menu_scroll < _max_scroll(_shown, _menu_win): | ||
|  | 
 | ||
|  |             _menu_scroll += 1 | ||
|  | 
 | ||
|  | 
 | ||
|  | def _select_prev_menu_entry(): | ||
|  |     # Selects the menu entry before the current one, adjusting the scroll if | ||
|  |     # necessary. Does nothing if we're already at the first menu entry. | ||
|  | 
 | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     if _sel_node_i > 0: | ||
|  |         # Jump to the previous node | ||
|  |         _sel_node_i -= 1 | ||
|  | 
 | ||
|  |         # See _select_next_menu_entry() | ||
|  |         if _sel_node_i < _menu_scroll + _SCROLL_OFFSET: | ||
|  |             _menu_scroll = max(_menu_scroll - 1, 0) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _select_last_menu_entry(): | ||
|  |     # Selects the last menu entry in the current menu | ||
|  | 
 | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     _sel_node_i = len(_shown) - 1 | ||
|  |     _menu_scroll = _max_scroll(_shown, _menu_win) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _select_first_menu_entry(): | ||
|  |     # Selects the first menu entry in the current menu | ||
|  | 
 | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     _sel_node_i = _menu_scroll = 0 | ||
|  | 
 | ||
|  | 
 | ||
|  | def _toggle_show_all(): | ||
|  |     # Toggles show-all mode on/off. If turning it off would give no visible | ||
|  |     # items in the current menu, it is left on. | ||
|  | 
 | ||
|  |     global _show_all | ||
|  |     global _shown | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     # Row on the screen the cursor is on. Preferably we want the same row to | ||
|  |     # stay highlighted. | ||
|  |     old_row = _sel_node_i - _menu_scroll | ||
|  | 
 | ||
|  |     _show_all = not _show_all | ||
|  |     # List of new nodes to be shown after toggling _show_all | ||
|  |     new_shown = _shown_nodes(_cur_menu) | ||
|  | 
 | ||
|  |     # Find a good node to select. The selected node might disappear if show-all | ||
|  |     # mode is turned off. | ||
|  | 
 | ||
|  |     # Select the previously selected node itself if it is still visible. If | ||
|  |     # there are visible nodes before it, select the closest one. | ||
|  |     for node in _shown[_sel_node_i::-1]: | ||
|  |         if node in new_shown: | ||
|  |             _sel_node_i = new_shown.index(node) | ||
|  |             break | ||
|  |     else: | ||
|  |         # No visible nodes before the previously selected node. Select the | ||
|  |         # closest visible node after it instead. | ||
|  |         for node in _shown[_sel_node_i + 1:]: | ||
|  |             if node in new_shown: | ||
|  |                 _sel_node_i = new_shown.index(node) | ||
|  |                 break | ||
|  |         else: | ||
|  |             # No visible nodes at all, meaning show-all was turned off inside | ||
|  |             # an invisible menu. Don't allow that, as the implementation relies | ||
|  |             # on always having a selected node. | ||
|  |             _show_all = True | ||
|  |             return | ||
|  | 
 | ||
|  |     _shown = new_shown | ||
|  | 
 | ||
|  |     # Try to make the cursor stay on the same row in the menu window. This | ||
|  |     # might be impossible if too many nodes have disappeared above the node. | ||
|  |     _menu_scroll = max(_sel_node_i - old_row, 0) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _center_vertically(): | ||
|  |     # Centers the selected node vertically, if possible | ||
|  | 
 | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     _menu_scroll = min(max(_sel_node_i - _height(_menu_win)//2, 0), | ||
|  |                        _max_scroll(_shown, _menu_win)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _draw_main(): | ||
|  |     # Draws the "main" display, with the list of symbols, the header, and the | ||
|  |     # footer. | ||
|  |     # | ||
|  |     # This could be optimized to only update the windows that have actually | ||
|  |     # changed, but keep it simple for now and let curses sort it out. | ||
|  | 
 | ||
|  |     term_width = _width(_stdscr) | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update the separator row below the menu path | ||
|  |     # | ||
|  | 
 | ||
|  |     _top_sep_win.erase() | ||
|  | 
 | ||
|  |     # Draw arrows pointing up if the symbol window is scrolled down. Draw them | ||
|  |     # before drawing the title, so the title ends up on top for small windows. | ||
|  |     if _menu_scroll > 0: | ||
|  |         _safe_hline(_top_sep_win, 0, 4, curses.ACS_UARROW, _N_SCROLL_ARROWS) | ||
|  | 
 | ||
|  |     # Add the 'mainmenu' text as the title, centered at the top | ||
|  |     _safe_addstr(_top_sep_win, | ||
|  |                  0, max((term_width - len(_kconf.mainmenu_text))//2, 0), | ||
|  |                  _kconf.mainmenu_text) | ||
|  | 
 | ||
|  |     _top_sep_win.noutrefresh() | ||
|  | 
 | ||
|  |     # Note: The menu path at the top is deliberately updated last. See below. | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update the symbol window | ||
|  |     # | ||
|  | 
 | ||
|  |     _menu_win.erase() | ||
|  | 
 | ||
|  |     # Draw the _shown nodes starting from index _menu_scroll up to either as | ||
|  |     # many as fit in the window, or to the end of _shown | ||
|  |     for i in range(_menu_scroll, | ||
|  |                    min(_menu_scroll + _height(_menu_win), len(_shown))): | ||
|  | 
 | ||
|  |         node = _shown[i] | ||
|  | 
 | ||
|  |         # The 'not _show_all' test avoids showing invisible items in red | ||
|  |         # outside show-all mode, which could look confusing/broken. Invisible | ||
|  |         # symbols show up outside show-all mode if an invisible symbol has | ||
|  |         # visible children in an implicit (indented) menu. | ||
|  |         if _visible(node) or not _show_all: | ||
|  |             style = _style["selection" if i == _sel_node_i else "list"] | ||
|  |         else: | ||
|  |             style = _style["inv-selection" if i == _sel_node_i else "inv-list"] | ||
|  | 
 | ||
|  |         _safe_addstr(_menu_win, i - _menu_scroll, 0, _node_str(node), style) | ||
|  | 
 | ||
|  |     _menu_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update the bottom separator window | ||
|  |     # | ||
|  | 
 | ||
|  |     _bot_sep_win.erase() | ||
|  | 
 | ||
|  |     # Draw arrows pointing down if the symbol window is scrolled up | ||
|  |     if _menu_scroll < _max_scroll(_shown, _menu_win): | ||
|  |         _safe_hline(_bot_sep_win, 0, 4, curses.ACS_DARROW, _N_SCROLL_ARROWS) | ||
|  | 
 | ||
|  |     # Indicate when show-name/show-help/show-all mode is enabled | ||
|  |     enabled_modes = [] | ||
|  |     if _show_help: | ||
|  |         enabled_modes.append("show-help (toggle with [F])") | ||
|  |     if _show_name: | ||
|  |         enabled_modes.append("show-name") | ||
|  |     if _show_all: | ||
|  |         enabled_modes.append("show-all") | ||
|  |     if enabled_modes: | ||
|  |         s = " and ".join(enabled_modes) + " mode enabled" | ||
|  |         _safe_addstr(_bot_sep_win, 0, max(term_width - len(s) - 2, 0), s) | ||
|  | 
 | ||
|  |     _bot_sep_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update the help window, which shows either key bindings or help texts | ||
|  |     # | ||
|  | 
 | ||
|  |     _help_win.erase() | ||
|  | 
 | ||
|  |     if _show_help: | ||
|  |         node = _shown[_sel_node_i] | ||
|  |         if isinstance(node.item, (Symbol, Choice)) and node.help: | ||
|  |             help_lines = textwrap.wrap(node.help, _width(_help_win)) | ||
|  |             for i in range(min(_height(_help_win), len(help_lines))): | ||
|  |                 _safe_addstr(_help_win, i, 0, help_lines[i]) | ||
|  |         else: | ||
|  |             _safe_addstr(_help_win, 0, 0, "(no help)") | ||
|  |     else: | ||
|  |         for i, line in enumerate(_MAIN_HELP_LINES): | ||
|  |             _safe_addstr(_help_win, i, 0, line) | ||
|  | 
 | ||
|  |     _help_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update the top row with the menu path. | ||
|  |     # | ||
|  |     # Doing this last leaves the cursor on the top row, which avoids some minor | ||
|  |     # annoying jumpiness in gnome-terminal when reducing the height of the | ||
|  |     # terminal. It seems to happen whenever the row with the cursor on it | ||
|  |     # disappears. | ||
|  |     # | ||
|  | 
 | ||
|  |     _path_win.erase() | ||
|  | 
 | ||
|  |     # Draw the menu path ("(Top) -> Menu -> Submenu -> ...") | ||
|  | 
 | ||
|  |     menu_prompts = [] | ||
|  | 
 | ||
|  |     menu = _cur_menu | ||
|  |     while menu is not _kconf.top_node: | ||
|  |         # Promptless choices can be entered in show-all mode. Use | ||
|  |         # standard_sc_expr_str() for them, so they show up as | ||
|  |         # '<choice (name if any)>'. | ||
|  |         menu_prompts.append(menu.prompt[0] if menu.prompt else | ||
|  |                             standard_sc_expr_str(menu.item)) | ||
|  |         menu = menu.parent | ||
|  |     menu_prompts.append("(Top)") | ||
|  |     menu_prompts.reverse() | ||
|  | 
 | ||
|  |     # Hack: We can't put ACS_RARROW directly in the string. Temporarily | ||
|  |     # represent it with NULL. | ||
|  |     menu_path_str = " \0 ".join(menu_prompts) | ||
|  | 
 | ||
|  |     # Scroll the menu path to the right if needed to make the current menu's | ||
|  |     # title visible | ||
|  |     if len(menu_path_str) > term_width: | ||
|  |         menu_path_str = menu_path_str[len(menu_path_str) - term_width:] | ||
|  | 
 | ||
|  |     # Print the path with the arrows reinserted | ||
|  |     split_path = menu_path_str.split("\0") | ||
|  |     _safe_addstr(_path_win, split_path[0]) | ||
|  |     for s in split_path[1:]: | ||
|  |         _safe_addch(_path_win, curses.ACS_RARROW) | ||
|  |         _safe_addstr(_path_win, s) | ||
|  | 
 | ||
|  |     _path_win.noutrefresh() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _parent_menu(node): | ||
|  |     # Returns the menu node of the menu that contains 'node'. In addition to | ||
|  |     # proper 'menu's, this might also be a 'menuconfig' symbol or a 'choice'. | ||
|  |     # "Menu" here means a menu in the interface. | ||
|  | 
 | ||
|  |     menu = node.parent | ||
|  |     while not menu.is_menuconfig: | ||
|  |         menu = menu.parent | ||
|  |     return menu | ||
|  | 
 | ||
|  | 
 | ||
|  | def _shown_nodes(menu): | ||
|  |     # Returns the list of menu nodes from 'menu' (see _parent_menu()) that | ||
|  |     # would be shown when entering it | ||
|  | 
 | ||
|  |     def rec(node): | ||
|  |         res = [] | ||
|  | 
 | ||
|  |         while node: | ||
|  |             if _visible(node) or _show_all: | ||
|  |                 res.append(node) | ||
|  |                 if node.list and not node.is_menuconfig: | ||
|  |                     # Nodes from implicit menu created from dependencies. Will | ||
|  |                     # be shown indented. Note that is_menuconfig is True for | ||
|  |                     # menus and choices as well as 'menuconfig' symbols. | ||
|  |                     res += rec(node.list) | ||
|  | 
 | ||
|  |             elif node.list and isinstance(node.item, Symbol): | ||
|  |                 # Show invisible symbols if they have visible children. This | ||
|  |                 # can happen for an m/y-valued symbol with an optional prompt | ||
|  |                 # ('prompt "foo" is COND') that is currently disabled. Note | ||
|  |                 # that it applies to both 'config' and 'menuconfig' symbols. | ||
|  |                 shown_children = rec(node.list) | ||
|  |                 if shown_children: | ||
|  |                     res.append(node) | ||
|  |                     if not node.is_menuconfig: | ||
|  |                         res += shown_children | ||
|  | 
 | ||
|  |             node = node.next | ||
|  | 
 | ||
|  |         return res | ||
|  | 
 | ||
|  |     if isinstance(menu.item, Choice): | ||
|  |         # For named choices defined in multiple locations, entering the choice | ||
|  |         # at a particular menu node would normally only show the choice symbols | ||
|  |         # defined there (because that's what the MenuNode tree looks like). | ||
|  |         # | ||
|  |         # That might look confusing, and makes extending choices by defining | ||
|  |         # them in multiple locations less useful. Instead, gather all the child | ||
|  |         # menu nodes for all the choices whenever a choice is entered. That | ||
|  |         # makes all choice symbols visible at all locations. | ||
|  |         # | ||
|  |         # Choices can contain non-symbol items (people do all sorts of weird | ||
|  |         # stuff with them), hence the generality here. We really need to | ||
|  |         # preserve the menu tree at each choice location. | ||
|  |         # | ||
|  |         # Note: Named choices are pretty broken in the C tools, and this is | ||
|  |         # super obscure, so you probably won't find much that relies on this. | ||
|  |         # This whole 'if' could be deleted if you don't care about defining | ||
|  |         # choices in multiple locations to add symbols (which will still work, | ||
|  |         # just with things being displayed in a way that might be unexpected). | ||
|  | 
 | ||
|  |         # Do some additional work to avoid listing choice symbols twice if all | ||
|  |         # or part of the choice is copied in multiple locations (e.g. by | ||
|  |         # including some Kconfig file multiple times). We give the prompts at | ||
|  |         # the current location precedence. | ||
|  |         seen_syms = {node.item for node in rec(menu.list) | ||
|  |                      if isinstance(node.item, Symbol)} | ||
|  |         res = [] | ||
|  |         for choice_node in menu.item.nodes: | ||
|  |             for node in rec(choice_node.list): | ||
|  |                 # 'choice_node is menu' checks if we're dealing with the | ||
|  |                 # current location | ||
|  |                 if node.item not in seen_syms or choice_node is menu: | ||
|  |                     res.append(node) | ||
|  |                     if isinstance(node.item, Symbol): | ||
|  |                         seen_syms.add(node.item) | ||
|  |         return res | ||
|  | 
 | ||
|  |     return rec(menu.list) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _visible(node): | ||
|  |     # Returns True if the node should appear in the menu (outside show-all | ||
|  |     # mode) | ||
|  | 
 | ||
|  |     return node.prompt and expr_value(node.prompt[1]) and not \ | ||
|  |         (node.item == MENU and not expr_value(node.visibility)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _change_node(node): | ||
|  |     # Changes the value of the menu node 'node' if it is a symbol. Bools and | ||
|  |     # tristates are toggled, while other symbol types pop up a text entry | ||
|  |     # dialog. | ||
|  |     # | ||
|  |     # Returns False if the value of 'node' can't be changed. | ||
|  | 
 | ||
|  |     if not _changeable(node): | ||
|  |         return False | ||
|  | 
 | ||
|  |     # sc = symbol/choice | ||
|  |     sc = node.item | ||
|  | 
 | ||
|  |     if sc.orig_type in (INT, HEX, STRING): | ||
|  |         s = sc.str_value | ||
|  | 
 | ||
|  |         while True: | ||
|  |             s = _input_dialog( | ||
|  |                 "{} ({})".format(node.prompt[0], TYPE_TO_STR[sc.orig_type]), | ||
|  |                 s, _range_info(sc)) | ||
|  | 
 | ||
|  |             if s is None: | ||
|  |                 break | ||
|  | 
 | ||
|  |             if sc.orig_type in (INT, HEX): | ||
|  |                 s = s.strip() | ||
|  | 
 | ||
|  |                 # 'make menuconfig' does this too. Hex values not starting with | ||
|  |                 # '0x' are accepted when loading .config files though. | ||
|  |                 if sc.orig_type == HEX and not s.startswith(("0x", "0X")): | ||
|  |                     s = "0x" + s | ||
|  | 
 | ||
|  |             if _check_valid(sc, s): | ||
|  |                 _set_val(sc, s) | ||
|  |                 break | ||
|  | 
 | ||
|  |     elif len(sc.assignable) == 1: | ||
|  |         # Handles choice symbols for choices in y mode, which are a special | ||
|  |         # case: .assignable can be (2,) while .tri_value is 0. | ||
|  |         _set_val(sc, sc.assignable[0]) | ||
|  | 
 | ||
|  |     else: | ||
|  |         # Set the symbol to the value after the current value in | ||
|  |         # sc.assignable, with wrapping | ||
|  |         val_index = sc.assignable.index(sc.tri_value) | ||
|  |         _set_val(sc, sc.assignable[(val_index + 1) % len(sc.assignable)]) | ||
|  | 
 | ||
|  | 
 | ||
|  |     if _is_y_mode_choice_sym(sc) and not node.list: | ||
|  |         # Immediately jump to the parent menu after making a choice selection, | ||
|  |         # like 'make menuconfig' does, except if the menu node has children | ||
|  |         # (which can happen if a symbol 'depends on' a choice symbol that | ||
|  |         # immediately precedes it). | ||
|  |         _leave_menu() | ||
|  | 
 | ||
|  | 
 | ||
|  |     return True | ||
|  | 
 | ||
|  | 
 | ||
|  | def _changeable(node): | ||
|  |     # Returns True if the value if 'node' can be changed | ||
|  | 
 | ||
|  |     sc = node.item | ||
|  | 
 | ||
|  |     if not isinstance(sc, (Symbol, Choice)): | ||
|  |         return False | ||
|  | 
 | ||
|  |     # This will hit for invisible symbols, which appear in show-all mode and | ||
|  |     # when an invisible symbol has visible children (which can happen e.g. for | ||
|  |     # symbols with optional prompts) | ||
|  |     if not (node.prompt and expr_value(node.prompt[1])): | ||
|  |         return False | ||
|  | 
 | ||
|  |     return sc.orig_type in (STRING, INT, HEX) or len(sc.assignable) > 1 \ | ||
|  |         or _is_y_mode_choice_sym(sc) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _set_sel_node_tri_val(tri_val): | ||
|  |     # Sets the value of the currently selected menu entry to 'tri_val', if that | ||
|  |     # value can be assigned | ||
|  | 
 | ||
|  |     sc = _shown[_sel_node_i].item | ||
|  |     if isinstance(sc, (Symbol, Choice)) and tri_val in sc.assignable: | ||
|  |         _set_val(sc, tri_val) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _set_val(sc, val): | ||
|  |     # Wrapper around Symbol/Choice.set_value() for updating the menu state and | ||
|  |     # _conf_changed | ||
|  | 
 | ||
|  |     global _conf_changed | ||
|  | 
 | ||
|  |     # Use the string representation of tristate values. This makes the format | ||
|  |     # consistent for all symbol types. | ||
|  |     if val in TRI_TO_STR: | ||
|  |         val = TRI_TO_STR[val] | ||
|  | 
 | ||
|  |     if val != sc.str_value: | ||
|  |         sc.set_value(val) | ||
|  |         _conf_changed = True | ||
|  | 
 | ||
|  |         # Changing the value of the symbol might have changed what items in the | ||
|  |         # current menu are visible. Recalculate the state. | ||
|  |         _update_menu() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _update_menu(): | ||
|  |     # Updates the current menu after the value of a symbol or choice has been | ||
|  |     # changed. Changing a value might change which items in the menu are | ||
|  |     # visible. | ||
|  |     # | ||
|  |     # If possible, preserves the location of the cursor on the screen when | ||
|  |     # items are added/removed above the selected item. | ||
|  | 
 | ||
|  |     global _shown | ||
|  |     global _sel_node_i | ||
|  |     global _menu_scroll | ||
|  | 
 | ||
|  |     # Row on the screen the cursor was on | ||
|  |     old_row = _sel_node_i - _menu_scroll | ||
|  | 
 | ||
|  |     sel_node = _shown[_sel_node_i] | ||
|  | 
 | ||
|  |     # New visible nodes | ||
|  |     _shown = _shown_nodes(_cur_menu) | ||
|  | 
 | ||
|  |     # New index of selected node | ||
|  |     _sel_node_i = _shown.index(sel_node) | ||
|  | 
 | ||
|  |     # Try to make the cursor stay on the same row in the menu window. This | ||
|  |     # might be impossible if too many nodes have disappeared above the node. | ||
|  |     _menu_scroll = max(_sel_node_i - old_row, 0) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _input_dialog(title, initial_text, info_text=None): | ||
|  |     # Pops up a dialog that prompts the user for a string | ||
|  |     # | ||
|  |     # title: | ||
|  |     #   Title to display at the top of the dialog window's border | ||
|  |     # | ||
|  |     # initial_text: | ||
|  |     #   Initial text to prefill the input field with | ||
|  |     # | ||
|  |     # info_text: | ||
|  |     #   String to show next to the input field. If None, just the input field | ||
|  |     #   is shown. | ||
|  | 
 | ||
|  |     win = _styled_win("body") | ||
|  |     win.keypad(True) | ||
|  | 
 | ||
|  |     info_lines = info_text.split("\n") if info_text else [] | ||
|  | 
 | ||
|  |     # Give the input dialog its initial size | ||
|  |     _resize_input_dialog(win, title, info_lines) | ||
|  | 
 | ||
|  |     _safe_curs_set(2) | ||
|  | 
 | ||
|  |     # Input field text | ||
|  |     s = initial_text | ||
|  | 
 | ||
|  |     # Cursor position | ||
|  |     i = len(initial_text) | ||
|  | 
 | ||
|  |     def edit_width(): | ||
|  |         return _width(win) - 4 | ||
|  | 
 | ||
|  |     # Horizontal scroll offset | ||
|  |     hscroll = max(i - edit_width() + 1, 0) | ||
|  | 
 | ||
|  |     while True: | ||
|  |         # Draw the "main" display with the menu, etc., so that resizing still | ||
|  |         # works properly. This is like a stack of windows, only hardcoded for | ||
|  |         # now. | ||
|  |         _draw_main() | ||
|  |         _draw_input_dialog(win, title, info_lines, s, i, hscroll) | ||
|  |         curses.doupdate() | ||
|  | 
 | ||
|  | 
 | ||
|  |         c = _getch_compat(win) | ||
|  | 
 | ||
|  |         if c == curses.KEY_RESIZE: | ||
|  |             # Resize the main display too. The dialog floats above it. | ||
|  |             _resize_main() | ||
|  |             _resize_input_dialog(win, title, info_lines) | ||
|  | 
 | ||
|  |         elif c == "\n": | ||
|  |             _safe_curs_set(0) | ||
|  |             return s | ||
|  | 
 | ||
|  |         elif c == "\x1B":  # \x1B = ESC | ||
|  |             _safe_curs_set(0) | ||
|  |             return None | ||
|  | 
 | ||
|  |         else: | ||
|  |             s, i, hscroll = _edit_text(c, s, i, hscroll, edit_width()) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _resize_input_dialog(win, title, info_lines): | ||
|  |     # Resizes the input dialog to a size appropriate for the terminal size | ||
|  | 
 | ||
|  |     screen_height, screen_width = _stdscr.getmaxyx() | ||
|  | 
 | ||
|  |     win_height = 5 | ||
|  |     if info_lines: | ||
|  |         win_height += len(info_lines) + 1 | ||
|  |     win_height = min(win_height, screen_height) | ||
|  | 
 | ||
|  |     win_width = max(_INPUT_DIALOG_MIN_WIDTH, | ||
|  |                     len(title) + 4, | ||
|  |                     *(len(line) + 4 for line in info_lines)) | ||
|  |     win_width = min(win_width, screen_width) | ||
|  | 
 | ||
|  |     win.resize(win_height, win_width) | ||
|  |     win.mvwin((screen_height - win_height)//2, | ||
|  |               (screen_width - win_width)//2) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _draw_input_dialog(win, title, info_lines, s, i, hscroll): | ||
|  |     edit_width = _width(win) - 4 | ||
|  | 
 | ||
|  |     win.erase() | ||
|  | 
 | ||
|  |     # Note: Perhaps having a separate window for the input field would be nicer | ||
|  |     visible_s = s[hscroll:hscroll + edit_width] | ||
|  |     _safe_addstr(win, 2, 2, visible_s + " "*(edit_width - len(visible_s)), | ||
|  |                  _style["edit"]) | ||
|  | 
 | ||
|  |     for linenr, line in enumerate(info_lines): | ||
|  |         _safe_addstr(win, 4 + linenr, 2, line) | ||
|  | 
 | ||
|  |     # Draw the frame last so that it overwrites the body text for small windows | ||
|  |     _draw_frame(win, title) | ||
|  | 
 | ||
|  |     _safe_move(win, 2, 2 + i - hscroll) | ||
|  | 
 | ||
|  |     win.noutrefresh() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _load_dialog(): | ||
|  |     # Dialog for loading a new configuration | ||
|  | 
 | ||
|  |     global _conf_changed | ||
|  |     global _conf_filename | ||
|  |     global _show_all | ||
|  | 
 | ||
|  |     if _conf_changed: | ||
|  |         c = _key_dialog( | ||
|  |             "Load", | ||
|  |             "You have unsaved changes. Load new\n" | ||
|  |             "configuration anyway?\n" | ||
|  |             "\n" | ||
|  |             "         (O)K  (C)ancel", | ||
|  |             "oc") | ||
|  | 
 | ||
|  |         if c is None or c == "c": | ||
|  |             return | ||
|  | 
 | ||
|  |     filename = _conf_filename | ||
|  |     while True: | ||
|  |         filename = _input_dialog("File to load", filename, _load_save_info()) | ||
|  |         if filename is None: | ||
|  |             return | ||
|  | 
 | ||
|  |         filename = os.path.expanduser(filename) | ||
|  | 
 | ||
|  |         if _try_load(filename): | ||
|  |             _conf_filename = filename | ||
|  |             _conf_changed = _needs_save() | ||
|  | 
 | ||
|  |             # Turn on show-all mode if the selected node is not visible after | ||
|  |             # loading the new configuration. _shown still holds the old state. | ||
|  |             if _shown[_sel_node_i] not in _shown_nodes(_cur_menu): | ||
|  |                 _show_all = True | ||
|  | 
 | ||
|  |             _update_menu() | ||
|  | 
 | ||
|  |             # The message dialog indirectly updates the menu display, so _msg() | ||
|  |             # must be called after the new state has been initialized | ||
|  |             _msg("Success", "Loaded " + filename) | ||
|  |             return | ||
|  | 
 | ||
|  | 
 | ||
|  | def _try_load(filename): | ||
|  |     # Tries to load a configuration file. Pops up an error and returns False on | ||
|  |     # failure. | ||
|  |     # | ||
|  |     # filename: | ||
|  |     #   Configuration file to load | ||
|  | 
 | ||
|  |     try: | ||
|  |         _kconf.load_config(filename) | ||
|  |         return True | ||
|  |     except EnvironmentError as e: | ||
|  |         _error("Error loading '{}'\n\n{} (errno: {})" | ||
|  |                .format(filename, e.strerror, errno.errorcode[e.errno])) | ||
|  |         return False | ||
|  | 
 | ||
|  | 
 | ||
|  | def _save_dialog(save_fn, default_filename, description): | ||
|  |     # Dialog for saving the current configuration | ||
|  |     # | ||
|  |     # save_fn: | ||
|  |     #   Function to call with 'filename' to save the file | ||
|  |     # | ||
|  |     # default_filename: | ||
|  |     #   Prefilled filename in the input field | ||
|  |     # | ||
|  |     # description: | ||
|  |     #   String describing the thing being saved | ||
|  |     # | ||
|  |     # Return value: | ||
|  |     #   The path to the saved file, or None if no file was saved | ||
|  | 
 | ||
|  |     filename = default_filename | ||
|  |     while True: | ||
|  |         filename = _input_dialog("Filename to save {} to".format(description), | ||
|  |                                  filename, _load_save_info()) | ||
|  |         if filename is None: | ||
|  |             return None | ||
|  | 
 | ||
|  |         filename = os.path.expanduser(filename) | ||
|  | 
 | ||
|  |         msg = _try_save(save_fn, filename, description) | ||
|  |         if msg: | ||
|  |             _msg("Success", msg) | ||
|  |             return filename | ||
|  | 
 | ||
|  | 
 | ||
|  | def _try_save(save_fn, filename, description): | ||
|  |     # Tries to save a configuration file. Returns a message to print on | ||
|  |     # success. | ||
|  |     # | ||
|  |     # save_fn: | ||
|  |     #   Function to call with 'filename' to save the file | ||
|  |     # | ||
|  |     # description: | ||
|  |     #   String describing the thing being saved | ||
|  |     # | ||
|  |     # Return value: | ||
|  |     #   A message to print on success, and None on failure | ||
|  | 
 | ||
|  |     try: | ||
|  |         # save_fn() returns a message to print | ||
|  |         return save_fn(filename) | ||
|  |     except EnvironmentError as e: | ||
|  |         _error("Error saving {} to '{}'\n\n{} (errno: {})" | ||
|  |                .format(description, e.filename, e.strerror, | ||
|  |                        errno.errorcode[e.errno])) | ||
|  |         return None | ||
|  | 
 | ||
|  | 
 | ||
|  | def _key_dialog(title, text, keys): | ||
|  |     # Pops up a dialog that can be closed by pressing a key | ||
|  |     # | ||
|  |     # title: | ||
|  |     #   Title to display at the top of the dialog window's border | ||
|  |     # | ||
|  |     # text: | ||
|  |     #   Text to show in the dialog | ||
|  |     # | ||
|  |     # keys: | ||
|  |     #   List of keys that will close the dialog. Other keys (besides ESC) are | ||
|  |     #   ignored. The caller is responsible for providing a hint about which | ||
|  |     #   keys can be pressed in 'text'. | ||
|  |     # | ||
|  |     # Return value: | ||
|  |     #   The key that was pressed to close the dialog. Uppercase characters are | ||
|  |     #   converted to lowercase. ESC will always close the dialog, and returns | ||
|  |     #   None. | ||
|  | 
 | ||
|  |     win = _styled_win("body") | ||
|  |     win.keypad(True) | ||
|  | 
 | ||
|  |     _resize_key_dialog(win, text) | ||
|  | 
 | ||
|  |     while True: | ||
|  |         # See _input_dialog() | ||
|  |         _draw_main() | ||
|  |         _draw_key_dialog(win, title, text) | ||
|  |         curses.doupdate() | ||
|  | 
 | ||
|  | 
 | ||
|  |         c = _getch_compat(win) | ||
|  | 
 | ||
|  |         if c == curses.KEY_RESIZE: | ||
|  |             # Resize the main display too. The dialog floats above it. | ||
|  |             _resize_main() | ||
|  |             _resize_key_dialog(win, text) | ||
|  | 
 | ||
|  |         elif c == "\x1B":  # \x1B = ESC | ||
|  |             return None | ||
|  | 
 | ||
|  |         elif isinstance(c, str): | ||
|  |             c = c.lower() | ||
|  |             if c in keys: | ||
|  |                 return c | ||
|  | 
 | ||
|  | 
 | ||
|  | def _resize_key_dialog(win, text): | ||
|  |     # Resizes the key dialog to a size appropriate for the terminal size | ||
|  | 
 | ||
|  |     screen_height, screen_width = _stdscr.getmaxyx() | ||
|  | 
 | ||
|  |     lines = text.split("\n") | ||
|  | 
 | ||
|  |     win_height = min(len(lines) + 4, screen_height) | ||
|  |     win_width = min(max(len(line) for line in lines) + 4, screen_width) | ||
|  | 
 | ||
|  |     win.resize(win_height, win_width) | ||
|  |     win.mvwin((screen_height - win_height)//2, | ||
|  |               (screen_width - win_width)//2) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _draw_key_dialog(win, title, text): | ||
|  |     win.erase() | ||
|  | 
 | ||
|  |     for i, line in enumerate(text.split("\n")): | ||
|  |         _safe_addstr(win, 2 + i, 2, line) | ||
|  | 
 | ||
|  |     # Draw the frame last so that it overwrites the body text for small windows | ||
|  |     _draw_frame(win, title) | ||
|  | 
 | ||
|  |     win.noutrefresh() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _draw_frame(win, title): | ||
|  |     # Draw a frame around the inner edges of 'win', with 'title' at the top | ||
|  | 
 | ||
|  |     win_height, win_width = win.getmaxyx() | ||
|  | 
 | ||
|  |     win.attron(_style["frame"]) | ||
|  | 
 | ||
|  |     # Draw top/bottom edge | ||
|  |     _safe_hline(win,              0, 0, " ", win_width) | ||
|  |     _safe_hline(win, win_height - 1, 0, " ", win_width) | ||
|  | 
 | ||
|  |     # Draw left/right edge | ||
|  |     _safe_vline(win, 0,             0, " ", win_height) | ||
|  |     _safe_vline(win, 0, win_width - 1, " ", win_height) | ||
|  | 
 | ||
|  |     # Draw title | ||
|  |     _safe_addstr(win, 0, max((win_width - len(title))//2, 0), title) | ||
|  | 
 | ||
|  |     win.attroff(_style["frame"]) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _jump_to_dialog(): | ||
|  |     # Implements the jump-to dialog, where symbols can be looked up via | ||
|  |     # incremental search and jumped to. | ||
|  |     # | ||
|  |     # Returns True if the user jumped to a symbol, and False if the dialog was | ||
|  |     # canceled. | ||
|  | 
 | ||
|  |     s = ""  # Search text | ||
|  |     prev_s = None  # Previous search text | ||
|  |     s_i = 0  # Search text cursor position | ||
|  |     hscroll = 0  # Horizontal scroll offset | ||
|  | 
 | ||
|  |     sel_node_i = 0  # Index of selected row | ||
|  |     scroll = 0  # Index in 'matches' of the top row of the list | ||
|  | 
 | ||
|  |     # Edit box at the top | ||
|  |     edit_box = _styled_win("jump-edit") | ||
|  |     edit_box.keypad(True) | ||
|  | 
 | ||
|  |     # List of matches | ||
|  |     matches_win = _styled_win("list") | ||
|  | 
 | ||
|  |     # Bottom separator, with arrows pointing down | ||
|  |     bot_sep_win = _styled_win("separator") | ||
|  | 
 | ||
|  |     # Help window with instructions at the bottom | ||
|  |     help_win = _styled_win("help") | ||
|  | 
 | ||
|  |     # Give windows their initial size | ||
|  |     _resize_jump_to_dialog(edit_box, matches_win, bot_sep_win, help_win, | ||
|  |                            sel_node_i, scroll) | ||
|  | 
 | ||
|  |     _safe_curs_set(2) | ||
|  | 
 | ||
|  |     # Logic duplication with _select_{next,prev}_menu_entry(), except we do a | ||
|  |     # functional variant that returns the new (sel_node_i, scroll) values to | ||
|  |     # avoid 'nonlocal'. TODO: Can this be factored out in some nice way? | ||
|  | 
 | ||
|  |     def select_next_match(): | ||
|  |         if sel_node_i == len(matches) - 1: | ||
|  |             return sel_node_i, scroll | ||
|  | 
 | ||
|  |         if sel_node_i + 1 >= scroll + _height(matches_win) - _SCROLL_OFFSET \ | ||
|  |            and scroll < _max_scroll(matches, matches_win): | ||
|  | 
 | ||
|  |             return sel_node_i + 1, scroll + 1 | ||
|  | 
 | ||
|  |         return sel_node_i + 1, scroll | ||
|  | 
 | ||
|  |     def select_prev_match(): | ||
|  |         if sel_node_i == 0: | ||
|  |             return sel_node_i, scroll | ||
|  | 
 | ||
|  |         if sel_node_i - 1 < scroll + _SCROLL_OFFSET: | ||
|  |             return sel_node_i - 1, max(scroll - 1, 0) | ||
|  | 
 | ||
|  |         return sel_node_i - 1, scroll | ||
|  | 
 | ||
|  |     while True: | ||
|  |         if s != prev_s: | ||
|  |             # The search text changed. Find new matching nodes. | ||
|  | 
 | ||
|  |             prev_s = s | ||
|  | 
 | ||
|  |             try: | ||
|  |                 # We could use re.IGNORECASE here instead of lower(), but this | ||
|  |                 # is noticeably less jerky while inputting regexes like | ||
|  |                 # '.*debug$' (though the '.*' is redundant there). Those | ||
|  |                 # probably have bad interactions with re.search(), which | ||
|  |                 # matches anywhere in the string. | ||
|  |                 # | ||
|  |                 # It's not horrible either way. Just a bit smoother. | ||
|  |                 regex_searches = [re.compile(regex).search | ||
|  |                                   for regex in s.lower().split()] | ||
|  | 
 | ||
|  |                 # No exception thrown, so the regexes are okay | ||
|  |                 bad_re = None | ||
|  | 
 | ||
|  |                 # List of matching nodes | ||
|  |                 matches = [] | ||
|  |                 add_match = matches.append | ||
|  | 
 | ||
|  |                 # Search symbols and choices | ||
|  | 
 | ||
|  |                 for node in _sorted_sc_nodes(): | ||
|  |                     # Symbol/choice | ||
|  |                     sc = node.item | ||
|  | 
 | ||
|  |                     for search in regex_searches: | ||
|  |                         # Both the name and the prompt might be missing, since | ||
|  |                         # we're searching both symbols and choices | ||
|  | 
 | ||
|  |                         # Does the regex match either the symbol name or the | ||
|  |                         # prompt (if any)? | ||
|  |                         if not (sc.name and search(sc.name.lower()) or | ||
|  |                                 node.prompt and search(node.prompt[0].lower())): | ||
|  | 
 | ||
|  |                             # Give up on the first regex that doesn't match, to | ||
|  |                             # speed things up a bit when multiple regexes are | ||
|  |                             # entered | ||
|  |                             break | ||
|  | 
 | ||
|  |                     else: | ||
|  |                         add_match(node) | ||
|  | 
 | ||
|  |                 # Search menus and comments | ||
|  | 
 | ||
|  |                 for node in _sorted_menu_comment_nodes(): | ||
|  |                     for search in regex_searches: | ||
|  |                         if not search(node.prompt[0].lower()): | ||
|  |                             break | ||
|  |                     else: | ||
|  |                         add_match(node) | ||
|  | 
 | ||
|  |             except re.error as e: | ||
|  |                 # Bad regex. Remember the error message so we can show it. | ||
|  |                 bad_re = "Bad regular expression" | ||
|  |                 # re.error.msg was added in Python 3.5 | ||
|  |                 if hasattr(e, "msg"): | ||
|  |                     bad_re += ": " + e.msg | ||
|  | 
 | ||
|  |                 matches = [] | ||
|  | 
 | ||
|  |             # Reset scroll and jump to the top of the list of matches | ||
|  |             sel_node_i = scroll = 0 | ||
|  | 
 | ||
|  |         _draw_jump_to_dialog(edit_box, matches_win, bot_sep_win, help_win, | ||
|  |                              s, s_i, hscroll, | ||
|  |                              bad_re, matches, sel_node_i, scroll) | ||
|  |         curses.doupdate() | ||
|  | 
 | ||
|  | 
 | ||
|  |         c = _getch_compat(edit_box) | ||
|  | 
 | ||
|  |         if c == "\n": | ||
|  |             if matches: | ||
|  |                 _jump_to(matches[sel_node_i]) | ||
|  |                 _safe_curs_set(0) | ||
|  |                 return True | ||
|  | 
 | ||
|  |         elif c == "\x1B":  # \x1B = ESC | ||
|  |             _safe_curs_set(0) | ||
|  |             return False | ||
|  | 
 | ||
|  |         elif c == curses.KEY_RESIZE: | ||
|  |             # We adjust the scroll so that the selected node stays visible in | ||
|  |             # the list when the terminal is resized, hence the 'scroll' | ||
|  |             # assignment | ||
|  |             scroll = _resize_jump_to_dialog( | ||
|  |                 edit_box, matches_win, bot_sep_win, help_win, | ||
|  |                 sel_node_i, scroll) | ||
|  | 
 | ||
|  |         elif c == "\x06":  # \x06 = Ctrl-F | ||
|  |             if matches: | ||
|  |                 _safe_curs_set(0) | ||
|  |                 _info_dialog(matches[sel_node_i], True) | ||
|  |                 _safe_curs_set(2) | ||
|  | 
 | ||
|  |                 scroll = _resize_jump_to_dialog( | ||
|  |                     edit_box, matches_win, bot_sep_win, help_win, | ||
|  |                     sel_node_i, scroll) | ||
|  | 
 | ||
|  |         elif c == curses.KEY_DOWN: | ||
|  |             sel_node_i, scroll = select_next_match() | ||
|  | 
 | ||
|  |         elif c == curses.KEY_UP: | ||
|  |             sel_node_i, scroll = select_prev_match() | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_NPAGE, "\x04"):  # Page Down/Ctrl-D | ||
|  |             # Keep it simple. This way we get sane behavior for small windows, | ||
|  |             # etc., for free. | ||
|  |             for _ in range(_PG_JUMP): | ||
|  |                 sel_node_i, scroll = select_next_match() | ||
|  | 
 | ||
|  |         # Page Up (no Ctrl-U, as it's already used by the edit box) | ||
|  |         elif c == curses.KEY_PPAGE: | ||
|  |             for _ in range(_PG_JUMP): | ||
|  |                 sel_node_i, scroll = select_prev_match() | ||
|  | 
 | ||
|  |         elif c == curses.KEY_END: | ||
|  |             sel_node_i = len(matches) - 1 | ||
|  |             scroll = _max_scroll(matches, matches_win) | ||
|  | 
 | ||
|  |         elif c == curses.KEY_HOME: | ||
|  |             sel_node_i = scroll = 0 | ||
|  | 
 | ||
|  |         else: | ||
|  |             s, s_i, hscroll = _edit_text(c, s, s_i, hscroll, | ||
|  |                                          _width(edit_box) - 2) | ||
|  | 
 | ||
|  | 
 | ||
|  | # Obscure Python: We never pass a value for cached_nodes, and it keeps pointing | ||
|  | # to the same list. This avoids a global. | ||
|  | def _sorted_sc_nodes(cached_nodes=[]): | ||
|  |     # Returns a sorted list of symbol and choice nodes to search. The symbol | ||
|  |     # nodes appear first, sorted by name, and then the choice nodes, sorted by | ||
|  |     # prompt and (secondarily) name. | ||
|  | 
 | ||
|  |     if not cached_nodes: | ||
|  |         # Add symbol nodes | ||
|  |         for sym in sorted(_kconf.unique_defined_syms, | ||
|  |                           key=lambda sym: sym.name): | ||
|  |             # += is in-place for lists | ||
|  |             cached_nodes += sym.nodes | ||
|  | 
 | ||
|  |         # Add choice nodes | ||
|  | 
 | ||
|  |         choices = sorted(_kconf.unique_choices, | ||
|  |                          key=lambda choice: choice.name or "") | ||
|  | 
 | ||
|  |         cached_nodes += sorted( | ||
|  |             [node for choice in choices for node in choice.nodes], | ||
|  |             key=lambda node: node.prompt[0] if node.prompt else "") | ||
|  | 
 | ||
|  |     return cached_nodes | ||
|  | 
 | ||
|  | 
 | ||
|  | def _sorted_menu_comment_nodes(cached_nodes=[]): | ||
|  |     # Returns a list of menu and comment nodes to search, sorted by prompt, | ||
|  |     # with the menus first | ||
|  | 
 | ||
|  |     if not cached_nodes: | ||
|  |         def prompt_text(mc): | ||
|  |             return mc.prompt[0] | ||
|  | 
 | ||
|  |         cached_nodes += sorted(_kconf.menus, key=prompt_text) | ||
|  |         cached_nodes += sorted(_kconf.comments, key=prompt_text) | ||
|  | 
 | ||
|  |     return cached_nodes | ||
|  | 
 | ||
|  | 
 | ||
|  | def _resize_jump_to_dialog(edit_box, matches_win, bot_sep_win, help_win, | ||
|  |                            sel_node_i, scroll): | ||
|  |     # Resizes the jump-to dialog to fill the terminal. | ||
|  |     # | ||
|  |     # Returns the new scroll index. We adjust the scroll if needed so that the | ||
|  |     # selected node stays visible. | ||
|  | 
 | ||
|  |     screen_height, screen_width = _stdscr.getmaxyx() | ||
|  | 
 | ||
|  |     bot_sep_win.resize(1, screen_width) | ||
|  | 
 | ||
|  |     help_win_height = len(_JUMP_TO_HELP_LINES) | ||
|  |     matches_win_height = screen_height - help_win_height - 4 | ||
|  | 
 | ||
|  |     if matches_win_height >= 1: | ||
|  |         edit_box.resize(3, screen_width) | ||
|  |         matches_win.resize(matches_win_height, screen_width) | ||
|  |         help_win.resize(help_win_height, screen_width) | ||
|  | 
 | ||
|  |         matches_win.mvwin(3, 0) | ||
|  |         bot_sep_win.mvwin(3 + matches_win_height, 0) | ||
|  |         help_win.mvwin(3 + matches_win_height + 1, 0) | ||
|  |     else: | ||
|  |         # Degenerate case. Give up on nice rendering and just prevent errors. | ||
|  | 
 | ||
|  |         matches_win_height = 1 | ||
|  | 
 | ||
|  |         edit_box.resize(screen_height, screen_width) | ||
|  |         matches_win.resize(1, screen_width) | ||
|  |         help_win.resize(1, screen_width) | ||
|  | 
 | ||
|  |         for win in matches_win, bot_sep_win, help_win: | ||
|  |             win.mvwin(0, 0) | ||
|  | 
 | ||
|  |     # Adjust the scroll so that the selected row is still within the window, if | ||
|  |     # needed | ||
|  |     if sel_node_i - scroll >= matches_win_height: | ||
|  |         return sel_node_i - matches_win_height + 1 | ||
|  |     return scroll | ||
|  | 
 | ||
|  | 
 | ||
|  | def _draw_jump_to_dialog(edit_box, matches_win, bot_sep_win, help_win, | ||
|  |                          s, s_i, hscroll, | ||
|  |                          bad_re, matches, sel_node_i, scroll): | ||
|  | 
 | ||
|  |     edit_width = _width(edit_box) - 2 | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update list of matches | ||
|  |     # | ||
|  | 
 | ||
|  |     matches_win.erase() | ||
|  | 
 | ||
|  |     if matches: | ||
|  |         for i in range(scroll, | ||
|  |                        min(scroll + _height(matches_win), len(matches))): | ||
|  | 
 | ||
|  |             node = matches[i] | ||
|  | 
 | ||
|  |             if isinstance(node.item, (Symbol, Choice)): | ||
|  |                 node_str = _name_and_val_str(node.item) | ||
|  |                 if node.prompt: | ||
|  |                     node_str += ' "{}"'.format(node.prompt[0]) | ||
|  |             elif node.item == MENU: | ||
|  |                 node_str = 'menu "{}"'.format(node.prompt[0]) | ||
|  |             else:  # node.item == COMMENT | ||
|  |                 node_str = 'comment "{}"'.format(node.prompt[0]) | ||
|  | 
 | ||
|  |             _safe_addstr(matches_win, i - scroll, 0, node_str, | ||
|  |                          _style["selection" if i == sel_node_i else "list"]) | ||
|  | 
 | ||
|  |     else: | ||
|  |         # bad_re holds the error message from the re.error exception on errors | ||
|  |         _safe_addstr(matches_win, 0, 0, bad_re or "No matches") | ||
|  | 
 | ||
|  |     matches_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update bottom separator line | ||
|  |     # | ||
|  | 
 | ||
|  |     bot_sep_win.erase() | ||
|  | 
 | ||
|  |     # Draw arrows pointing down if the symbol list is scrolled up | ||
|  |     if scroll < _max_scroll(matches, matches_win): | ||
|  |         _safe_hline(bot_sep_win, 0, 4, curses.ACS_DARROW, _N_SCROLL_ARROWS) | ||
|  | 
 | ||
|  |     bot_sep_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update help window at bottom | ||
|  |     # | ||
|  | 
 | ||
|  |     help_win.erase() | ||
|  | 
 | ||
|  |     for i, line in enumerate(_JUMP_TO_HELP_LINES): | ||
|  |         _safe_addstr(help_win, i, 0, line) | ||
|  | 
 | ||
|  |     help_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update edit box. We do this last since it makes it handy to position the | ||
|  |     # cursor. | ||
|  |     # | ||
|  | 
 | ||
|  |     edit_box.erase() | ||
|  | 
 | ||
|  |     _draw_frame(edit_box, "Jump to symbol/choice/menu/comment") | ||
|  | 
 | ||
|  |     # Draw arrows pointing up if the symbol list is scrolled down | ||
|  |     if scroll > 0: | ||
|  |         # TODO: Bit ugly that _style["frame"] is repeated here | ||
|  |         _safe_hline(edit_box, 2, 4, curses.ACS_UARROW, _N_SCROLL_ARROWS, | ||
|  |                     _style["frame"]) | ||
|  | 
 | ||
|  |     visible_s = s[hscroll:hscroll + edit_width] | ||
|  |     _safe_addstr(edit_box, 1, 1, visible_s) | ||
|  | 
 | ||
|  |     _safe_move(edit_box, 1, 1 + s_i - hscroll) | ||
|  | 
 | ||
|  |     edit_box.noutrefresh() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _info_dialog(node, from_jump_to_dialog): | ||
|  |     # Shows a fullscreen window with information about 'node'. | ||
|  |     # | ||
|  |     # If 'from_jump_to_dialog' is True, the information dialog was opened from | ||
|  |     # within the jump-to-dialog. In this case, we make '/' from within the | ||
|  |     # information dialog just return, to avoid a confusing recursive invocation | ||
|  |     # of the jump-to-dialog. | ||
|  | 
 | ||
|  |     # Top row, with title and arrows point up | ||
|  |     top_line_win = _styled_win("separator") | ||
|  | 
 | ||
|  |     # Text display | ||
|  |     text_win = _styled_win("text") | ||
|  |     text_win.keypad(True) | ||
|  | 
 | ||
|  |     # Bottom separator, with arrows pointing down | ||
|  |     bot_sep_win = _styled_win("separator") | ||
|  | 
 | ||
|  |     # Help window with keys at the bottom | ||
|  |     help_win = _styled_win("help") | ||
|  | 
 | ||
|  |     # Give windows their initial size | ||
|  |     _resize_info_dialog(top_line_win, text_win, bot_sep_win, help_win) | ||
|  | 
 | ||
|  | 
 | ||
|  |     # Get lines of help text | ||
|  |     lines = _info_str(node).split("\n") | ||
|  | 
 | ||
|  |     # Index of first row in 'lines' to show | ||
|  |     scroll = 0 | ||
|  | 
 | ||
|  |     while True: | ||
|  |         _draw_info_dialog(node, lines, scroll, top_line_win, text_win, | ||
|  |                           bot_sep_win, help_win) | ||
|  |         curses.doupdate() | ||
|  | 
 | ||
|  | 
 | ||
|  |         c = _getch_compat(text_win) | ||
|  | 
 | ||
|  |         if c == curses.KEY_RESIZE: | ||
|  |             _resize_info_dialog(top_line_win, text_win, bot_sep_win, help_win) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_DOWN, "j", "J"): | ||
|  |             if scroll < _max_scroll(lines, text_win): | ||
|  |                 scroll += 1 | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_NPAGE, "\x04"):  # Page Down/Ctrl-D | ||
|  |             scroll = min(scroll + _PG_JUMP, _max_scroll(lines, text_win)) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_PPAGE, "\x15"):  # Page Up/Ctrl-U | ||
|  |             scroll = max(scroll - _PG_JUMP, 0) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_END, "G"): | ||
|  |             scroll = _max_scroll(lines, text_win) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_HOME, "g"): | ||
|  |             scroll = 0 | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_UP, "k", "K"): | ||
|  |             if scroll > 0: | ||
|  |                 scroll -= 1 | ||
|  | 
 | ||
|  |         elif c == "/": | ||
|  |             # Support starting a search from within the information dialog | ||
|  | 
 | ||
|  |             if from_jump_to_dialog: | ||
|  |                 return  # Avoid recursion | ||
|  | 
 | ||
|  |             if _jump_to_dialog(): | ||
|  |                 return  # Jumped to a symbol. Cancel the information dialog. | ||
|  | 
 | ||
|  |             # Stay in the information dialog if the jump-to dialog was | ||
|  |             # canceled. Resize it in case the terminal was resized while the | ||
|  |             # fullscreen jump-to dialog was open. | ||
|  |             _resize_info_dialog(top_line_win, text_win, bot_sep_win, help_win) | ||
|  | 
 | ||
|  |         elif c in (curses.KEY_LEFT, curses.KEY_BACKSPACE, _ERASE_CHAR, | ||
|  |                    "\x1B",  # \x1B = ESC | ||
|  |                    "q", "Q", "h", "H"): | ||
|  | 
 | ||
|  |             return | ||
|  | 
 | ||
|  | 
 | ||
|  | def _resize_info_dialog(top_line_win, text_win, bot_sep_win, help_win): | ||
|  |     # Resizes the info dialog to fill the terminal | ||
|  | 
 | ||
|  |     screen_height, screen_width = _stdscr.getmaxyx() | ||
|  | 
 | ||
|  |     top_line_win.resize(1, screen_width) | ||
|  |     bot_sep_win.resize(1, screen_width) | ||
|  | 
 | ||
|  |     help_win_height = len(_INFO_HELP_LINES) | ||
|  |     text_win_height = screen_height - help_win_height - 2 | ||
|  | 
 | ||
|  |     if text_win_height >= 1: | ||
|  |         text_win.resize(text_win_height, screen_width) | ||
|  |         help_win.resize(help_win_height, screen_width) | ||
|  | 
 | ||
|  |         text_win.mvwin(1, 0) | ||
|  |         bot_sep_win.mvwin(1 + text_win_height, 0) | ||
|  |         help_win.mvwin(1 + text_win_height + 1, 0) | ||
|  |     else: | ||
|  |         # Degenerate case. Give up on nice rendering and just prevent errors. | ||
|  | 
 | ||
|  |         text_win.resize(1, screen_width) | ||
|  |         help_win.resize(1, screen_width) | ||
|  | 
 | ||
|  |         for win in text_win, bot_sep_win, help_win: | ||
|  |             win.mvwin(0, 0) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _draw_info_dialog(node, lines, scroll, top_line_win, text_win, | ||
|  |                       bot_sep_win, help_win): | ||
|  | 
 | ||
|  |     text_win_height, text_win_width = text_win.getmaxyx() | ||
|  | 
 | ||
|  |     # Note: The top row is deliberately updated last. See _draw_main(). | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update text display | ||
|  |     # | ||
|  | 
 | ||
|  |     text_win.erase() | ||
|  | 
 | ||
|  |     for i, line in enumerate(lines[scroll:scroll + text_win_height]): | ||
|  |         _safe_addstr(text_win, i, 0, line) | ||
|  | 
 | ||
|  |     text_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update bottom separator line | ||
|  |     # | ||
|  | 
 | ||
|  |     bot_sep_win.erase() | ||
|  | 
 | ||
|  |     # Draw arrows pointing down if the symbol window is scrolled up | ||
|  |     if scroll < _max_scroll(lines, text_win): | ||
|  |         _safe_hline(bot_sep_win, 0, 4, curses.ACS_DARROW, _N_SCROLL_ARROWS) | ||
|  | 
 | ||
|  |     bot_sep_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update help window at bottom | ||
|  |     # | ||
|  | 
 | ||
|  |     help_win.erase() | ||
|  | 
 | ||
|  |     for i, line in enumerate(_INFO_HELP_LINES): | ||
|  |         _safe_addstr(help_win, i, 0, line) | ||
|  | 
 | ||
|  |     help_win.noutrefresh() | ||
|  | 
 | ||
|  |     # | ||
|  |     # Update top row | ||
|  |     # | ||
|  | 
 | ||
|  |     top_line_win.erase() | ||
|  | 
 | ||
|  |     # Draw arrows pointing up if the information window is scrolled down. Draw | ||
|  |     # them before drawing the title, so the title ends up on top for small | ||
|  |     # windows. | ||
|  |     if scroll > 0: | ||
|  |         _safe_hline(top_line_win, 0, 4, curses.ACS_UARROW, _N_SCROLL_ARROWS) | ||
|  | 
 | ||
|  |     title = ("Symbol" if isinstance(node.item, Symbol) else | ||
|  |              "Choice" if isinstance(node.item, Choice) else | ||
|  |              "Menu"   if node.item == MENU else | ||
|  |              "Comment") + " information" | ||
|  |     _safe_addstr(top_line_win, 0, max((text_win_width - len(title))//2, 0), | ||
|  |                  title) | ||
|  | 
 | ||
|  |     top_line_win.noutrefresh() | ||
|  | 
 | ||
|  | 
 | ||
|  | def _info_str(node): | ||
|  |     # Returns information about the menu node 'node' as a string. | ||
|  |     # | ||
|  |     # The helper functions are responsible for adding newlines. This allows | ||
|  |     # them to return "" if they don't want to add any output. | ||
|  | 
 | ||
|  |     if isinstance(node.item, Symbol): | ||
|  |         sym = node.item | ||
|  | 
 | ||
|  |         return ( | ||
|  |             _name_info(sym) + | ||
|  |             _prompt_info(sym) + | ||
|  |             "Type: {}\n".format(TYPE_TO_STR[sym.type]) + | ||
|  |             _value_info(sym) + | ||
|  |             _help_info(sym) + | ||
|  |             _direct_dep_info(sym) + | ||
|  |             _defaults_info(sym) + | ||
|  |             _select_imply_info(sym) + | ||
|  |             _kconfig_def_info(sym) | ||
|  |         ) | ||
|  | 
 | ||
|  |     if isinstance(node.item, Choice): | ||
|  |         choice = node.item | ||
|  | 
 | ||
|  |         return ( | ||
|  |             _name_info(choice) + | ||
|  |             _prompt_info(choice) + | ||
|  |             "Type: {}\n".format(TYPE_TO_STR[choice.type]) + | ||
|  |             'Mode: {}\n'.format(choice.str_value) + | ||
|  |             _help_info(choice) + | ||
|  |             _choice_syms_info(choice) + | ||
|  |             _direct_dep_info(choice) + | ||
|  |             _defaults_info(choice) + | ||
|  |             _kconfig_def_info(choice) | ||
|  |         ) | ||
|  | 
 | ||
|  |     return _kconfig_def_info(node)  # node.item in (MENU, COMMENT) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _name_info(sc): | ||
|  |     # Returns a string with the name of the symbol/choice. Names are optional | ||
|  |     # for choices. | ||
|  | 
 | ||
|  |     return "Name: {}\n".format(sc.name) if sc.name else "" | ||
|  | 
 | ||
|  | 
 | ||
|  | def _prompt_info(sc): | ||
|  |     # Returns a string listing the prompts of 'sc' (Symbol or Choice) | ||
|  | 
 | ||
|  |     s = "" | ||
|  | 
 | ||
|  |     for node in sc.nodes: | ||
|  |         if node.prompt: | ||
|  |             s += "Prompt: {}\n".format(node.prompt[0]) | ||
|  | 
 | ||
|  |     return s | ||
|  | 
 | ||
|  | 
 | ||
|  | def _value_info(sym): | ||
|  |     # Returns a string showing 'sym's value | ||
|  | 
 | ||
|  |     # Only put quotes around the value for string symbols | ||
|  |     return "Value: {}\n".format( | ||
|  |         '"{}"'.format(sym.str_value) | ||
|  |         if sym.orig_type == STRING | ||
|  |         else sym.str_value) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _choice_syms_info(choice): | ||
|  |     # Returns a string listing the choice symbols in 'choice'. Adds | ||
|  |     # "(selected)" next to the selected one. | ||
|  | 
 | ||
|  |     s = "Choice symbols:\n" | ||
|  | 
 | ||
|  |     for sym in choice.syms: | ||
|  |         s += "  - " + sym.name | ||
|  |         if sym is choice.selection: | ||
|  |             s += " (selected)" | ||
|  |         s += "\n" | ||
|  | 
 | ||
|  |     return s + "\n" | ||
|  | 
 | ||
|  | 
 | ||
|  | def _help_info(sc): | ||
|  |     # Returns a string with the help text(s) of 'sc' (Symbol or Choice). | ||
|  |     # Symbols and choices defined in multiple locations can have multiple help | ||
|  |     # texts. | ||
|  | 
 | ||
|  |     s = "\n" | ||
|  | 
 | ||
|  |     for node in sc.nodes: | ||
|  |         if node.help is not None: | ||
|  |             s += "Help:\n\n{}\n\n".format(_indent(node.help, 2)) | ||
|  | 
 | ||
|  |     return s | ||
|  | 
 | ||
|  | 
 | ||
|  | def _direct_dep_info(sc): | ||
|  |     # Returns a string describing the direct dependencies of 'sc' (Symbol or | ||
|  |     # Choice). The direct dependencies are the OR of the dependencies from each | ||
|  |     # definition location. The dependencies at each definition location come | ||
|  |     # from 'depends on' and dependencies inherited from parent items. | ||
|  | 
 | ||
|  |     return "" if sc.direct_dep is _kconf.y else \ | ||
|  |         'Direct dependencies (={}):\n{}\n' \ | ||
|  |         .format(TRI_TO_STR[expr_value(sc.direct_dep)], | ||
|  |                 _split_expr_info(sc.direct_dep, 2)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _defaults_info(sc): | ||
|  |     # Returns a string describing the defaults of 'sc' (Symbol or Choice) | ||
|  | 
 | ||
|  |     if not sc.defaults: | ||
|  |         return "" | ||
|  | 
 | ||
|  |     s = "Default" | ||
|  |     if len(sc.defaults) > 1: | ||
|  |         s += "s" | ||
|  |     s += ":\n" | ||
|  | 
 | ||
|  |     for val, cond in sc.orig_defaults: | ||
|  |         s += "  - " | ||
|  |         if isinstance(sc, Symbol): | ||
|  |             s += _expr_str(val) | ||
|  | 
 | ||
|  |             # Skip the tristate value hint if the expression is just a single | ||
|  |             # symbol. _expr_str() already shows its value as a string. | ||
|  |             # | ||
|  |             # This also avoids showing the tristate value for string/int/hex | ||
|  |             # defaults, which wouldn't make any sense. | ||
|  |             if isinstance(val, tuple): | ||
|  |                 s += '  (={})'.format(TRI_TO_STR[expr_value(val)]) | ||
|  |         else: | ||
|  |             # Don't print the value next to the symbol name for choice | ||
|  |             # defaults, as it looks a bit confusing | ||
|  |             s += val.name | ||
|  |         s += "\n" | ||
|  | 
 | ||
|  |         if cond is not _kconf.y: | ||
|  |             s += "    Condition (={}):\n{}" \ | ||
|  |                  .format(TRI_TO_STR[expr_value(cond)], | ||
|  |                          _split_expr_info(cond, 4)) | ||
|  | 
 | ||
|  |     return s + "\n" | ||
|  | 
 | ||
|  | 
 | ||
|  | def _split_expr_info(expr, indent): | ||
|  |     # Returns a string with 'expr' split into its top-level && or || operands, | ||
|  |     # with one operand per line, together with the operand's value. This is | ||
|  |     # usually enough to get something readable for long expressions. A fancier | ||
|  |     # recursive thingy would be possible too. | ||
|  |     # | ||
|  |     # indent: | ||
|  |     #   Number of leading spaces to add before the split expression. | ||
|  | 
 | ||
|  |     if len(split_expr(expr, AND)) > 1: | ||
|  |         split_op = AND | ||
|  |         op_str = "&&" | ||
|  |     else: | ||
|  |         split_op = OR | ||
|  |         op_str = "||" | ||
|  | 
 | ||
|  |     s = "" | ||
|  |     for i, term in enumerate(split_expr(expr, split_op)): | ||
|  |         s += "{}{} {}".format(indent*" ", | ||
|  |                               "  " if i == 0 else op_str, | ||
|  |                               _expr_str(term)) | ||
|  | 
 | ||
|  |         # Don't bother showing the value hint if the expression is just a | ||
|  |         # single symbol. _expr_str() already shows its value. | ||
|  |         if isinstance(term, tuple): | ||
|  |             s += "  (={})".format(TRI_TO_STR[expr_value(term)]) | ||
|  | 
 | ||
|  |         s += "\n" | ||
|  | 
 | ||
|  |     return s | ||
|  | 
 | ||
|  | 
 | ||
|  | def _select_imply_info(sym): | ||
|  |     # Returns a string with information about which symbols 'select' or 'imply' | ||
|  |     # 'sym'. The selecting/implying symbols are grouped according to which | ||
|  |     # value they select/imply 'sym' to (n/m/y). | ||
|  | 
 | ||
|  |     def sis(expr, val, title): | ||
|  |         # sis = selects/implies | ||
|  |         sis = [si for si in split_expr(expr, OR) if expr_value(si) == val] | ||
|  |         if not sis: | ||
|  |             return "" | ||
|  | 
 | ||
|  |         res = title | ||
|  |         for si in sis: | ||
|  |             res += "  - {}\n".format(split_expr(si, AND)[0].name) | ||
|  |         return res + "\n" | ||
|  | 
 | ||
|  |     s = "" | ||
|  | 
 | ||
|  |     if sym.rev_dep is not _kconf.n: | ||
|  |         s += sis(sym.rev_dep, 2, | ||
|  |                  "Symbols currently y-selecting this symbol:\n") | ||
|  |         s += sis(sym.rev_dep, 1, | ||
|  |                  "Symbols currently m-selecting this symbol:\n") | ||
|  |         s += sis(sym.rev_dep, 0, | ||
|  |                  "Symbols currently n-selecting this symbol (no effect):\n") | ||
|  | 
 | ||
|  |     if sym.weak_rev_dep is not _kconf.n: | ||
|  |         s += sis(sym.weak_rev_dep, 2, | ||
|  |                  "Symbols currently y-implying this symbol:\n") | ||
|  |         s += sis(sym.weak_rev_dep, 1, | ||
|  |                  "Symbols currently m-implying this symbol:\n") | ||
|  |         s += sis(sym.weak_rev_dep, 0, | ||
|  |                  "Symbols currently n-implying this symbol (no effect):\n") | ||
|  | 
 | ||
|  |     return s | ||
|  | 
 | ||
|  | 
 | ||
|  | def _kconfig_def_info(item): | ||
|  |     # Returns a string with the definition of 'item' in Kconfig syntax, | ||
|  |     # together with the definition location(s) and their include and menu paths | ||
|  | 
 | ||
|  |     nodes = [item] if isinstance(item, MenuNode) else item.nodes | ||
|  | 
 | ||
|  |     s = "Kconfig definition{}, with parent deps. propagated to 'depends on'\n" \ | ||
|  |         .format("s" if len(nodes) > 1 else "") | ||
|  |     s += (len(s) - 1)*"=" | ||
|  | 
 | ||
|  |     for node in nodes: | ||
|  |         s += "\n\n" \ | ||
|  |              "At {}:{}\n" \ | ||
|  |              "{}" \ | ||
|  |              "Menu path: {}\n\n" \ | ||
|  |              "{}" \ | ||
|  |              .format(node.filename, node.linenr, | ||
|  |                      _include_path_info(node), | ||
|  |                      _menu_path_info(node), | ||
|  |                      _indent(node.custom_str(_name_and_val_str), 2)) | ||
|  | 
 | ||
|  |     return s | ||
|  | 
 | ||
|  | 
 | ||
|  | def _include_path_info(node): | ||
|  |     if not node.include_path: | ||
|  |         # In the top-level Kconfig file | ||
|  |         return "" | ||
|  | 
 | ||
|  |     return "Included via {}\n".format( | ||
|  |         " -> ".join("{}:{}".format(filename, linenr) | ||
|  |                     for filename, linenr in node.include_path)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _menu_path_info(node): | ||
|  |     # Returns a string describing the menu path leading up to 'node' | ||
|  | 
 | ||
|  |     path = "" | ||
|  | 
 | ||
|  |     while node.parent is not _kconf.top_node: | ||
|  |         node = node.parent | ||
|  | 
 | ||
|  |         # Promptless choices might appear among the parents. Use | ||
|  |         # standard_sc_expr_str() for them, so that they show up as | ||
|  |         # '<choice (name if any)>'. | ||
|  |         path = " -> " + (node.prompt[0] if node.prompt else | ||
|  |                          standard_sc_expr_str(node.item)) + path | ||
|  | 
 | ||
|  |     return "(Top)" + path | ||
|  | 
 | ||
|  | 
 | ||
|  | def _indent(s, n): | ||
|  |     # Returns 's' with each line indented 'n' spaces. textwrap.indent() is not | ||
|  |     # available in Python 2 (it's 3.3+). | ||
|  | 
 | ||
|  |     return "\n".join(n*" " + line for line in s.split("\n")) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _name_and_val_str(sc): | ||
|  |     # Custom symbol/choice printer that shows symbol values after symbols | ||
|  | 
 | ||
|  |     # Show the values of non-constant (non-quoted) symbols that don't look like | ||
|  |     # numbers. Things like 123 are actually symbol references, and only work as | ||
|  |     # expected due to undefined symbols getting their name as their value. | ||
|  |     # Showing the symbol value for those isn't helpful though. | ||
|  |     if isinstance(sc, Symbol) and not sc.is_constant and not _is_num(sc.name): | ||
|  |         if not sc.nodes: | ||
|  |             # Undefined symbol reference | ||
|  |             return "{}(undefined/n)".format(sc.name) | ||
|  | 
 | ||
|  |         return '{}(={})'.format(sc.name, sc.str_value) | ||
|  | 
 | ||
|  |     # For other items, use the standard format | ||
|  |     return standard_sc_expr_str(sc) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _expr_str(expr): | ||
|  |     # Custom expression printer that shows symbol values | ||
|  |     return expr_str(expr, _name_and_val_str) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _styled_win(style): | ||
|  |     # Returns a new curses window with style 'style' and space as the fill | ||
|  |     # character. The initial dimensions are (1, 1), so the window needs to be | ||
|  |     # sized and positioned separately. | ||
|  | 
 | ||
|  |     win = curses.newwin(1, 1) | ||
|  |     _set_style(win, style) | ||
|  |     return win | ||
|  | 
 | ||
|  | 
 | ||
|  | def _set_style(win, style): | ||
|  |     # Changes the style of an existing window | ||
|  | 
 | ||
|  |     win.bkgdset(" ", _style[style]) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _max_scroll(lst, win): | ||
|  |     # Assuming 'lst' is a list of items to be displayed in 'win', | ||
|  |     # returns the maximum number of steps 'win' can be scrolled down. | ||
|  |     # We stop scrolling when the bottom item is visible. | ||
|  | 
 | ||
|  |     return max(0, len(lst) - _height(win)) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _edit_text(c, s, i, hscroll, width): | ||
|  |     # Implements text editing commands for edit boxes. Takes a character (which | ||
|  |     # could also be e.g. curses.KEY_LEFT) and the edit box state, and returns | ||
|  |     # the new state after the character has been processed. | ||
|  |     # | ||
|  |     # c: | ||
|  |     #   Character from user | ||
|  |     # | ||
|  |     # s: | ||
|  |     #   Current contents of string | ||
|  |     # | ||
|  |     # i: | ||
|  |     #   Current cursor index in string | ||
|  |     # | ||
|  |     # hscroll: | ||
|  |     #   Index in s of the leftmost character in the edit box, for horizontal | ||
|  |     #   scrolling | ||
|  |     # | ||
|  |     # width: | ||
|  |     #   Width in characters of the edit box | ||
|  |     # | ||
|  |     # Return value: | ||
|  |     #   An (s, i, hscroll) tuple for the new state | ||
|  | 
 | ||
|  |     if c == curses.KEY_LEFT: | ||
|  |         if i > 0: | ||
|  |             i -= 1 | ||
|  | 
 | ||
|  |     elif c == curses.KEY_RIGHT: | ||
|  |         if i < len(s): | ||
|  |             i += 1 | ||
|  | 
 | ||
|  |     elif c in (curses.KEY_HOME, "\x01"):  # \x01 = CTRL-A | ||
|  |         i = 0 | ||
|  | 
 | ||
|  |     elif c in (curses.KEY_END, "\x05"):  # \x05 = CTRL-E | ||
|  |         i = len(s) | ||
|  | 
 | ||
|  |     elif c in (curses.KEY_BACKSPACE, _ERASE_CHAR): | ||
|  |         if i > 0: | ||
|  |             s = s[:i-1] + s[i:] | ||
|  |             i -= 1 | ||
|  | 
 | ||
|  |     elif c == curses.KEY_DC: | ||
|  |         s = s[:i] + s[i+1:] | ||
|  | 
 | ||
|  |     elif c == "\x17":  # \x17 = CTRL-W | ||
|  |         # The \W removes characters like ',' one at a time | ||
|  |         new_i = re.search(r"(?:\w*|\W)\s*$", s[:i]).start() | ||
|  |         s = s[:new_i] + s[i:] | ||
|  |         i = new_i | ||
|  | 
 | ||
|  |     elif c == "\x0B":  # \x0B = CTRL-K | ||
|  |         s = s[:i] | ||
|  | 
 | ||
|  |     elif c == "\x15":  # \x15 = CTRL-U | ||
|  |         s = s[i:] | ||
|  |         i = 0 | ||
|  | 
 | ||
|  |     elif isinstance(c, str): | ||
|  |         # Insert character | ||
|  |         s = s[:i] + c + s[i:] | ||
|  |         i += 1 | ||
|  | 
 | ||
|  |     # Adjust the horizontal scroll so that the cursor never touches the left or | ||
|  |     # right edges of the edit box, except when it's at the beginning or the end | ||
|  |     # of the string | ||
|  |     if i < hscroll + _SCROLL_OFFSET: | ||
|  |         hscroll = max(i - _SCROLL_OFFSET, 0) | ||
|  |     elif i >= hscroll + width - _SCROLL_OFFSET: | ||
|  |         max_scroll = max(len(s) - width + 1, 0) | ||
|  |         hscroll = min(i - width + _SCROLL_OFFSET + 1, max_scroll) | ||
|  | 
 | ||
|  |     return s, i, hscroll | ||
|  | 
 | ||
|  | 
 | ||
|  | def _load_save_info(): | ||
|  |     # Returns an information string for load/save dialog boxes | ||
|  | 
 | ||
|  |     return "(Relative to {})\n\nRefer to your home directory with ~" \ | ||
|  |            .format(os.path.join(os.getcwd(), "")) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _msg(title, text): | ||
|  |     # Pops up a message dialog that can be dismissed with Space/Enter/ESC | ||
|  | 
 | ||
|  |     _key_dialog(title, text, " \n") | ||
|  | 
 | ||
|  | 
 | ||
|  | def _error(text): | ||
|  |     # Pops up an error dialog that can be dismissed with Space/Enter/ESC | ||
|  | 
 | ||
|  |     _msg("Error", text) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _node_str(node): | ||
|  |     # Returns the complete menu entry text for a menu node. | ||
|  |     # | ||
|  |     # Example return value: "[*] Support for X" | ||
|  | 
 | ||
|  |     # Calculate the indent to print the item with by checking how many levels | ||
|  |     # above it the closest 'menuconfig' item is (this includes menus and | ||
|  |     # choices as well as menuconfig symbols) | ||
|  |     indent = 0 | ||
|  |     parent = node.parent | ||
|  |     while not parent.is_menuconfig: | ||
|  |         indent += _SUBMENU_INDENT | ||
|  |         parent = parent.parent | ||
|  | 
 | ||
|  |     # This approach gives nice alignment for empty string symbols ("()  Foo") | ||
|  |     s = "{:{}}".format(_value_str(node), 3 + indent) | ||
|  | 
 | ||
|  |     if _should_show_name(node): | ||
|  |         if isinstance(node.item, Symbol): | ||
|  |             s += " <{}>".format(node.item.name) | ||
|  |         else: | ||
|  |             # For choices, use standard_sc_expr_str(). That way they show up as | ||
|  |             # '<choice (name if any)>'. | ||
|  |             s += " " + standard_sc_expr_str(node.item) | ||
|  | 
 | ||
|  |     if node.prompt: | ||
|  |         if node.item == COMMENT: | ||
|  |             s += " *** {} ***".format(node.prompt[0]) | ||
|  |         else: | ||
|  |             s += " " + node.prompt[0] | ||
|  | 
 | ||
|  |         if isinstance(node.item, Symbol): | ||
|  |             sym = node.item | ||
|  | 
 | ||
|  |             # Print "(NEW)" next to symbols without a user value (from e.g. a | ||
|  |             # .config), but skip it for choice symbols in choices in y mode, | ||
|  |             # and for symbols of UNKNOWN type (which generate a warning though) | ||
|  |             if sym.user_value is None and sym.orig_type and \ | ||
|  |                not (sym.choice and sym.choice.tri_value == 2): | ||
|  | 
 | ||
|  |                 s += " (NEW)" | ||
|  | 
 | ||
|  |     if isinstance(node.item, Choice) and node.item.tri_value == 2: | ||
|  |         # Print the prompt of the selected symbol after the choice for | ||
|  |         # choices in y mode | ||
|  |         sym = node.item.selection | ||
|  |         if sym: | ||
|  |             for sym_node in sym.nodes: | ||
|  |                 # Use the prompt used at this choice location, in case the | ||
|  |                 # choice symbol is defined in multiple locations | ||
|  |                 if sym_node.parent is node and sym_node.prompt: | ||
|  |                     s += " ({})".format(sym_node.prompt[0]) | ||
|  |                     break | ||
|  |             else: | ||
|  |                 # If the symbol isn't defined at this choice location, then | ||
|  |                 # just use whatever prompt we can find for it | ||
|  |                 for sym_node in sym.nodes: | ||
|  |                     if sym_node.prompt: | ||
|  |                         s += " ({})".format(sym_node.prompt[0]) | ||
|  |                         break | ||
|  | 
 | ||
|  |     # Print "--->" next to nodes that have menus that can potentially be | ||
|  |     # entered. Print "----" if the menu is empty. We don't allow those to be | ||
|  |     # entered. | ||
|  |     if node.is_menuconfig: | ||
|  |         s += "  --->" if _shown_nodes(node) else "  ----" | ||
|  | 
 | ||
|  |     return s | ||
|  | 
 | ||
|  | 
 | ||
|  | def _should_show_name(node): | ||
|  |     # Returns True if 'node' is a symbol or choice whose name should shown (if | ||
|  |     # any, as names are optional for choices) | ||
|  | 
 | ||
|  |     # The 'not node.prompt' case only hits in show-all mode, for promptless | ||
|  |     # symbols and choices | ||
|  |     return not node.prompt or \ | ||
|  |            (_show_name and isinstance(node.item, (Symbol, Choice))) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _value_str(node): | ||
|  |     # Returns the value part ("[*]", "<M>", "(foo)" etc.) of a menu node | ||
|  | 
 | ||
|  |     item = node.item | ||
|  | 
 | ||
|  |     if item in (MENU, COMMENT): | ||
|  |         return "" | ||
|  | 
 | ||
|  |     # Wouldn't normally happen, and generates a warning | ||
|  |     if not item.orig_type: | ||
|  |         return "" | ||
|  | 
 | ||
|  |     if item.orig_type in (STRING, INT, HEX): | ||
|  |         return "({})".format(item.str_value) | ||
|  | 
 | ||
|  |     # BOOL or TRISTATE | ||
|  | 
 | ||
|  |     if _is_y_mode_choice_sym(item): | ||
|  |         return "(X)" if item.choice.selection is item else "( )" | ||
|  | 
 | ||
|  |     tri_val_str = (" ", "M", "*")[item.tri_value] | ||
|  | 
 | ||
|  |     if len(item.assignable) <= 1: | ||
|  |         # Pinned to a single value | ||
|  |         return "" if isinstance(item, Choice) else "-{}-".format(tri_val_str) | ||
|  | 
 | ||
|  |     if item.type == BOOL: | ||
|  |         return "[{}]".format(tri_val_str) | ||
|  | 
 | ||
|  |     # item.type == TRISTATE | ||
|  |     if item.assignable == (1, 2): | ||
|  |         return "{{{}}}".format(tri_val_str)  # {M}/{*} | ||
|  |     return "<{}>".format(tri_val_str) | ||
|  | 
 | ||
|  | 
 | ||
|  | def _is_y_mode_choice_sym(item): | ||
|  |     # The choice mode is an upper bound on the visibility of choice symbols, so | ||
|  |     # we can check the choice symbols' own visibility to see if the choice is | ||
|  |     # in y mode | ||
|  |     return isinstance(item, Symbol) and item.choice and item.visibility == 2 | ||
|  | 
 | ||
|  | 
 | ||
|  | def _check_valid(sym, s): | ||
|  |     # Returns True if the string 's' is a well-formed value for 'sym'. | ||
|  |     # Otherwise, displays an error and returns False. | ||
|  | 
 | ||
|  |     if sym.orig_type not in (INT, HEX): | ||
|  |         return True  # Anything goes for non-int/hex symbols | ||
|  | 
 | ||
|  |     base = 10 if sym.orig_type == INT else 16 | ||
|  |     try: | ||
|  |         int(s, base) | ||
|  |     except ValueError: | ||
|  |         _error("'{}' is a malformed {} value" | ||
|  |                .format(s, TYPE_TO_STR[sym.orig_type])) | ||
|  |         return False | ||
|  | 
 | ||
|  |     for low_sym, high_sym, cond in sym.ranges: | ||
|  |         if expr_value(cond): | ||
|  |             low_s = low_sym.str_value | ||
|  |             high_s = high_sym.str_value | ||
|  | 
 | ||
|  |             if not int(low_s, base) <= int(s, base) <= int(high_s, base): | ||
|  |                 _error("{} is outside the range {}-{}" | ||
|  |                        .format(s, low_s, high_s)) | ||
|  |                 return False | ||
|  | 
 | ||
|  |             break | ||
|  | 
 | ||
|  |     return True | ||
|  | 
 | ||
|  | 
 | ||
|  | def _range_info(sym): | ||
|  |     # Returns a string with information about the valid range for the symbol | ||
|  |     # 'sym', or None if 'sym' doesn't have a range | ||
|  | 
 | ||
|  |     if sym.orig_type in (INT, HEX): | ||
|  |         for low, high, cond in sym.ranges: | ||
|  |             if expr_value(cond): | ||
|  |                 return "Range: {}-{}".format(low.str_value, high.str_value) | ||
|  | 
 | ||
|  |     return None | ||
|  | 
 | ||
|  | 
 | ||
|  | def _is_num(name): | ||
|  |     # Heuristic to see if a symbol name looks like a number, for nicer output | ||
|  |     # when printing expressions. Things like 16 are actually symbol names, only | ||
|  |     # they get their name as their value when the symbol is undefined. | ||
|  | 
 | ||
|  |     try: | ||
|  |         int(name) | ||
|  |     except ValueError: | ||
|  |         if not name.startswith(("0x", "0X")): | ||
|  |             return False | ||
|  | 
 | ||
|  |         try: | ||
|  |             int(name, 16) | ||
|  |         except ValueError: | ||
|  |             return False | ||
|  | 
 | ||
|  |     return True | ||
|  | 
 | ||
|  | 
 | ||
|  | def _getch_compat(win): | ||
|  |     # Uses get_wch() if available (Python 3.3+) and getch() otherwise. | ||
|  |     # | ||
|  |     # Also falls back on getch() if get_wch() raises curses.error, to work | ||
|  |     # around an issue when resizing the terminal on at least macOS Catalina. | ||
|  |     # See https://github.com/ulfalizer/Kconfiglib/issues/84. | ||
|  |     # | ||
|  |     # Also handles a PDCurses resizing quirk. | ||
|  | 
 | ||
|  |     try: | ||
|  |         c = win.get_wch() | ||
|  |     except (AttributeError, curses.error): | ||
|  |         c = win.getch() | ||
|  |         if 0 <= c <= 255: | ||
|  |             c = chr(c) | ||
|  | 
 | ||
|  |     # Decent resizing behavior on PDCurses requires calling resize_term(0, 0) | ||
|  |     # after receiving KEY_RESIZE, while ncurses (usually) handles terminal | ||
|  |     # resizing automatically in get(_w)ch() (see the end of the | ||
|  |     # resizeterm(3NCURSES) man page). | ||
|  |     # | ||
|  |     # resize_term(0, 0) reliably fails and does nothing on ncurses, so this | ||
|  |     # hack gives ncurses/PDCurses compatibility for resizing. I don't know | ||
|  |     # whether it would cause trouble for other implementations. | ||
|  |     if c == curses.KEY_RESIZE: | ||
|  |         try: | ||
|  |             curses.resize_term(0, 0) | ||
|  |         except curses.error: | ||
|  |             pass | ||
|  | 
 | ||
|  |     return c | ||
|  | 
 | ||
|  | 
 | ||
|  | def _warn(*args): | ||
|  |     # Temporarily returns from curses to shell mode and prints a warning to | ||
|  |     # stderr. The warning would get lost in curses mode. | ||
|  |     curses.endwin() | ||
|  |     print("menuconfig warning: ", end="", file=sys.stderr) | ||
|  |     print(*args, file=sys.stderr) | ||
|  |     curses.doupdate() | ||
|  | 
 | ||
|  | 
 | ||
|  | # Ignore exceptions from some functions that might fail, e.g. for small | ||
|  | # windows. They usually do reasonable things anyway. | ||
|  | 
 | ||
|  | 
 | ||
|  | def _safe_curs_set(visibility): | ||
|  |     try: | ||
|  |         curses.curs_set(visibility) | ||
|  |     except curses.error: | ||
|  |         pass | ||
|  | 
 | ||
|  | 
 | ||
|  | def _safe_addstr(win, *args): | ||
|  |     # Clip the line to avoid wrapping to the next line, which looks glitchy. | ||
|  |     # addchstr() would do it for us, but it's not available in the 'curses' | ||
|  |     # module. | ||
|  | 
 | ||
|  |     attr = None | ||
|  |     if isinstance(args[0], str): | ||
|  |         y, x = win.getyx() | ||
|  |         s = args[0] | ||
|  |         if len(args) == 2: | ||
|  |             attr = args[1] | ||
|  |     else: | ||
|  |         y, x, s = args[:3] | ||
|  |         if len(args) == 4: | ||
|  |             attr = args[3] | ||
|  | 
 | ||
|  |     maxlen = _width(win) - x | ||
|  |     s = s.expandtabs() | ||
|  | 
 | ||
|  |     try: | ||
|  |         # The 'curses' module uses wattr_set() internally if you pass 'attr', | ||
|  |         # overwriting the background style, so setting 'attr' to 0 in the first | ||
|  |         # case won't do the right thing | ||
|  |         if attr is None: | ||
|  |             win.addnstr(y, x, s, maxlen) | ||
|  |         else: | ||
|  |             win.addnstr(y, x, s, maxlen, attr) | ||
|  |     except curses.error: | ||
|  |         pass | ||
|  | 
 | ||
|  | 
 | ||
|  | def _safe_addch(win, *args): | ||
|  |     try: | ||
|  |         win.addch(*args) | ||
|  |     except curses.error: | ||
|  |         pass | ||
|  | 
 | ||
|  | 
 | ||
|  | def _safe_hline(win, *args): | ||
|  |     try: | ||
|  |         win.hline(*args) | ||
|  |     except curses.error: | ||
|  |         pass | ||
|  | 
 | ||
|  | 
 | ||
|  | def _safe_vline(win, *args): | ||
|  |     try: | ||
|  |         win.vline(*args) | ||
|  |     except curses.error: | ||
|  |         pass | ||
|  | 
 | ||
|  | 
 | ||
|  | def _safe_move(win, *args): | ||
|  |     try: | ||
|  |         win.move(*args) | ||
|  |     except curses.error: | ||
|  |         pass | ||
|  | 
 | ||
|  | 
 | ||
|  | def _change_c_lc_ctype_to_utf8(): | ||
|  |     # See _CHANGE_C_LC_CTYPE_TO_UTF8 | ||
|  | 
 | ||
|  |     if _IS_WINDOWS: | ||
|  |         # Windows rarely has issues here, and the PEP 538 implementation avoids | ||
|  |         # changing the locale on it. None of the UTF-8 locales below were | ||
|  |         # supported from some quick testing either. Play it safe. | ||
|  |         return | ||
|  | 
 | ||
|  |     def try_set_locale(loc): | ||
|  |         try: | ||
|  |             locale.setlocale(locale.LC_CTYPE, loc) | ||
|  |             return True | ||
|  |         except locale.Error: | ||
|  |             return False | ||
|  | 
 | ||
|  |     # Is LC_CTYPE set to the C locale? | ||
|  |     if locale.setlocale(locale.LC_CTYPE) == "C": | ||
|  |         # This list was taken from the PEP 538 implementation in the CPython | ||
|  |         # code, in Python/pylifecycle.c | ||
|  |         for loc in "C.UTF-8", "C.utf8", "UTF-8": | ||
|  |             if try_set_locale(loc): | ||
|  |                 # LC_CTYPE successfully changed | ||
|  |                 return | ||
|  | 
 | ||
|  | 
 | ||
|  | if __name__ == "__main__": | ||
|  |     _main() |