237 lines
8.3 KiB
Python
237 lines
8.3 KiB
Python
#!/usr/bin/env python
|
|
# License: GPLv3 Copyright: 2021, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
import os
|
|
import re
|
|
import socket
|
|
import sys
|
|
import termios
|
|
import time
|
|
from contextlib import suppress
|
|
from functools import partial
|
|
from pprint import pformat
|
|
from typing import IO, Callable, Dict, Iterator, Optional, Set, TypeVar
|
|
|
|
from kittens.tui.operations import colored, styled
|
|
|
|
from .cli import version
|
|
from .constants import (
|
|
extensions_dir, is_macos, is_wayland, kitty_base_dir, kitty_exe, shell_path
|
|
)
|
|
from .fast_data_types import Color, num_users
|
|
from .options.types import Options as KittyOpts, defaults
|
|
from .options.utils import SequenceMap
|
|
from .rgb import color_as_sharp
|
|
from .types import MouseEvent, Shortcut, mod_to_names
|
|
|
|
AnyEvent = TypeVar('AnyEvent', MouseEvent, Shortcut)
|
|
Print = Callable[..., None]
|
|
ShortcutMap = Dict[Shortcut, str]
|
|
|
|
|
|
def green(x: str) -> str:
|
|
return colored(x, 'green')
|
|
|
|
|
|
def yellow(x: str) -> str:
|
|
return colored(x, 'yellow')
|
|
|
|
|
|
def title(x: str) -> str:
|
|
return colored(x, 'blue', intense=True)
|
|
|
|
|
|
def print_event(ev: str, defn: str, print: Print) -> None:
|
|
print(f'\t{ev} → {defn}')
|
|
|
|
|
|
def print_mapping_changes(defns: Dict[str, str], changes: Set[str], text: str, print: Print) -> None:
|
|
if changes:
|
|
print(title(text))
|
|
for k in sorted(changes):
|
|
print_event(k, defns[k], print)
|
|
|
|
|
|
def compare_maps(final: Dict[AnyEvent, str], final_kitty_mod: int, initial: Dict[AnyEvent, str], initial_kitty_mod: int, print: Print) -> None:
|
|
ei = {k.human_repr(initial_kitty_mod): v for k, v in initial.items()}
|
|
ef = {k.human_repr(final_kitty_mod): v for k, v in final.items()}
|
|
added = set(ef) - set(ei)
|
|
removed = set(ei) - set(ef)
|
|
changed = {k for k in set(ef) & set(ei) if ef[k] != ei[k]}
|
|
which = 'shortcuts' if isinstance(next(iter(initial)), Shortcut) else 'mouse actions'
|
|
print_mapping_changes(ef, added, f'Added {which}:', print)
|
|
print_mapping_changes(ei, removed, f'Removed {which}:', print)
|
|
print_mapping_changes(ef, changed, f'Changed {which}:', print)
|
|
|
|
|
|
def flatten_sequence_map(m: SequenceMap) -> ShortcutMap:
|
|
ans = {}
|
|
for key_spec, rest_map in m.items():
|
|
for r, action in rest_map.items():
|
|
ans[Shortcut((key_spec,) + (r))] = action
|
|
return ans
|
|
|
|
|
|
def compare_opts(opts: KittyOpts, print: Print) -> None:
|
|
from .config import load_config
|
|
print()
|
|
print('Config options different from defaults:')
|
|
default_opts = load_config()
|
|
ignored = ('keymap', 'sequence_map', 'mousemap', 'map', 'mouse_map')
|
|
changed_opts = [
|
|
f for f in sorted(defaults._fields)
|
|
if f not in ignored and getattr(opts, f) != getattr(defaults, f)
|
|
]
|
|
field_len = max(map(len, changed_opts)) if changed_opts else 20
|
|
fmt = f'{{:{field_len:d}s}}'
|
|
colors = []
|
|
for f in changed_opts:
|
|
val = getattr(opts, f)
|
|
if isinstance(val, dict):
|
|
print(title(f'{f}:'))
|
|
if f == 'symbol_map':
|
|
for k in sorted(val):
|
|
print(f'\tU+{k[0]:04x} - U+{k[1]:04x} → {val[k]}')
|
|
elif f == 'modify_font':
|
|
for k in sorted(val):
|
|
print(' ', val[k])
|
|
else:
|
|
print(pformat(val))
|
|
else:
|
|
val = getattr(opts, f)
|
|
if isinstance(val, Color):
|
|
colors.append(fmt.format(f) + ' ' + color_as_sharp(val) + ' ' + styled(' ', bg=val))
|
|
else:
|
|
if f == 'kitty_mod':
|
|
print(fmt.format(f), '+'.join(mod_to_names(getattr(opts, f))))
|
|
else:
|
|
print(fmt.format(f), str(getattr(opts, f)))
|
|
|
|
compare_maps(opts.mousemap, opts.kitty_mod, default_opts.mousemap, default_opts.kitty_mod, print)
|
|
final_, initial_ = opts.keymap, default_opts.keymap
|
|
final: ShortcutMap = {Shortcut((k,)): v for k, v in final_.items()}
|
|
initial: ShortcutMap = {Shortcut((k,)): v for k, v in initial_.items()}
|
|
final_s, initial_s = map(flatten_sequence_map, (opts.sequence_map, default_opts.sequence_map))
|
|
final.update(final_s)
|
|
initial.update(initial_s)
|
|
compare_maps(final, opts.kitty_mod, initial, default_opts.kitty_mod, print)
|
|
if colors:
|
|
print(f'{title("Colors")}:', end='\n\t')
|
|
print('\n\t'.join(sorted(colors)))
|
|
|
|
|
|
class IssueData:
|
|
|
|
def __init__(self) -> None:
|
|
self.uname = os.uname()
|
|
self.s, self.n, self.r, self.v, self.m = self.uname
|
|
try:
|
|
self.hostname = self.o = socket.gethostname()
|
|
except Exception:
|
|
self.hostname = self.o = 'localhost'
|
|
_time = time.localtime()
|
|
self.formatted_time = self.d = time.strftime('%a %b %d %Y', _time)
|
|
self.formatted_date = self.t = time.strftime('%H:%M:%S', _time)
|
|
try:
|
|
self.tty_name = format_tty_name(os.ctermid())
|
|
except OSError:
|
|
self.tty_name = '(none)'
|
|
self.l = self.tty_name # noqa
|
|
self.baud_rate = 0
|
|
if sys.stdin.isatty():
|
|
with suppress(OSError):
|
|
self.baud_rate = termios.tcgetattr(sys.stdin.fileno())[5]
|
|
self.b = str(self.baud_rate)
|
|
try:
|
|
self.num_users = num_users()
|
|
except RuntimeError:
|
|
self.num_users = -1
|
|
self.u = str(self.num_users)
|
|
self.U = self.u + ' user' + ('' if self.num_users == 1 else 's')
|
|
|
|
def translate_issue_char(self, char: str) -> str:
|
|
try:
|
|
return str(getattr(self, char)) if len(char) == 1 else char
|
|
except AttributeError:
|
|
return char
|
|
|
|
def parse_issue_file(self, issue_file: IO[str]) -> Iterator[str]:
|
|
last_char: Optional[str] = None
|
|
while True:
|
|
this_char = issue_file.read(1)
|
|
if not this_char:
|
|
break
|
|
if last_char == '\\':
|
|
yield self.translate_issue_char(this_char)
|
|
elif last_char is not None:
|
|
yield last_char
|
|
# `\\\a` should not match the last two slashes,
|
|
# so make it look like it was `\?\a` where `?`
|
|
# is some character other than `\`.
|
|
last_char = None if last_char == '\\' else this_char
|
|
if last_char is not None:
|
|
yield last_char
|
|
|
|
|
|
def format_tty_name(raw: str) -> str:
|
|
return re.sub(r'^/dev/([^/]+)/([^/]+)$', r'\1\2', raw)
|
|
|
|
|
|
def debug_config(opts: KittyOpts) -> str:
|
|
from io import StringIO
|
|
out = StringIO()
|
|
p = partial(print, file=out)
|
|
p(version(add_rev=True))
|
|
p(' '.join(os.uname()))
|
|
if is_macos:
|
|
import subprocess
|
|
p(' '.join(subprocess.check_output(['sw_vers']).decode('utf-8').splitlines()).strip())
|
|
if os.path.exists('/etc/issue'):
|
|
try:
|
|
idata = IssueData()
|
|
except Exception:
|
|
pass
|
|
else:
|
|
with open('/etc/issue', encoding='utf-8', errors='replace') as f:
|
|
try:
|
|
datums = idata.parse_issue_file(f)
|
|
except Exception:
|
|
pass
|
|
else:
|
|
p(end=''.join(datums))
|
|
if os.path.exists('/etc/lsb-release'):
|
|
with open('/etc/lsb-release', encoding='utf-8', errors='replace') as f:
|
|
p(f.read().strip())
|
|
if not is_macos:
|
|
p('Running under:', green('Wayland' if is_wayland() else 'X11'))
|
|
p(green('Frozen:'), 'True' if getattr(sys, 'frozen', False) else 'False')
|
|
p(green('Paths:'))
|
|
p(yellow(' kitty:'), os.path.realpath(kitty_exe()))
|
|
p(yellow(' base dir:'), kitty_base_dir)
|
|
p(yellow(' extensions dir:'), extensions_dir)
|
|
p(yellow(' system shell:'), shell_path)
|
|
if opts.config_paths:
|
|
p(green('Loaded config files:'))
|
|
p(' ', '\n '.join(opts.config_paths))
|
|
if opts.config_overrides:
|
|
p(green('Loaded config overrides:'))
|
|
p(' ', '\n '.join(opts.config_overrides))
|
|
compare_opts(opts, p)
|
|
p()
|
|
p(green('Important environment variables seen by the kitty process:'))
|
|
|
|
def penv(k: str) -> None:
|
|
v = os.environ.get(k)
|
|
if v is not None:
|
|
p('\t' + k.ljust(35), styled(v, dim=True))
|
|
|
|
for k in (
|
|
'PATH LANG KITTY_CONFIG_DIRECTORY KITTY_CACHE_DIRECTORY VISUAL EDITOR SHELL'
|
|
' GLFW_IM_MODULE KITTY_WAYLAND_DETECT_MODIFIERS DISPLAY WAYLAND_DISPLAY USER XCURSOR_SIZE'
|
|
).split():
|
|
penv(k)
|
|
for k in os.environ:
|
|
if k.startswith('LC_') or k.startswith('XDG_'):
|
|
penv(k)
|
|
return out.getvalue()
|