12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284 |
- #!/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
- elif c == "\0": # \0 = NUL, ignore
- pass
- 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
- elif c == "\0": # \0 = NUL, ignore
- pass
- 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] # pylint: disable=unbalanced-tuple-unpacking
- 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()
|