Add new derive colors script for terminal, vim, and bat
It's not being used yet
This commit is contained in:
ViViDboarder
parent
a666ccf041
commit
6703812918
|
|
@ -0,0 +1,223 @@
|
|||
#! /usr/bin/env python3
|
||||
import argparse
|
||||
import os
|
||||
import textwrap
|
||||
from subprocess import Popen, PIPE, check_output
|
||||
from typing import Tuple
|
||||
|
||||
|
||||
TERM_VAR = "TERM_PROFILE"
|
||||
VIM_VAR = "VIM_COLOR"
|
||||
BAT_VAR = "BAT_THEME"
|
||||
TERMINAL_SETTINGS_SCRIPT = """
|
||||
set current_tty to "{}"
|
||||
tell application "Terminal"
|
||||
repeat with win in windows
|
||||
repeat with the_tab in tabs of win
|
||||
set tab_settings to the current settings of the_tab
|
||||
if current_tty is equal to the tty of the_tab then
|
||||
return the name of tab_settings
|
||||
end if
|
||||
end repeat
|
||||
end repeat
|
||||
end tell
|
||||
"""
|
||||
|
||||
|
||||
def run_applescript(script: str) -> Tuple[int, str, str]:
|
||||
p = Popen(
|
||||
['osascript', '-'],
|
||||
stdin=PIPE,
|
||||
stdout=PIPE,
|
||||
stderr=PIPE,
|
||||
universal_newlines=True,
|
||||
)
|
||||
stdout, stderr = p.communicate(script)
|
||||
return p.returncode, str(stdout).strip(), str(stderr).strip()
|
||||
|
||||
|
||||
def get_terminal_profile(force=False):
|
||||
"""Returns the terminal profile from TERM_PROFILE or
|
||||
derrives through detecting the profile"""
|
||||
if not force and TERM_VAR in os.environ:
|
||||
return os.environ[TERM_VAR]
|
||||
|
||||
term_program = os.environ.get("TERM_PROGRAM")
|
||||
if term_program == "Apple_Terminal":
|
||||
tty = check_output(["tty"]).strip()
|
||||
tty = str(tty, encoding="utf-8")
|
||||
code, stdout, stderr = run_applescript(
|
||||
TERMINAL_SETTINGS_SCRIPT.format(tty),
|
||||
)
|
||||
if code:
|
||||
raise SystemError("Could not get results from applescript")
|
||||
return stdout
|
||||
elif term_program == "iTerm.app":
|
||||
return os.environ["ITERM_PROFILE"]
|
||||
|
||||
# If we got this far, we don't know what to do
|
||||
raise ValueError(f"Unknown terminal {term_program}")
|
||||
|
||||
|
||||
def get_vim_colorscheme(terminal_profile: str, is_dark=False, force=False):
|
||||
"""Returns the best colorscheme for a given terminal profile"""
|
||||
if not force and VIM_VAR in os.environ:
|
||||
return os.environ[VIM_VAR]
|
||||
|
||||
if "Wombat" in terminal_profile:
|
||||
return "wombat256mod"
|
||||
elif "Solarized" in terminal_profile:
|
||||
return "solarized"
|
||||
elif "Yosemite Light" == terminal_profile:
|
||||
return "morning"
|
||||
elif "Yosemite Dark" == terminal_profile:
|
||||
return "vividchalk"
|
||||
|
||||
# Default
|
||||
return "wombat256mod"
|
||||
|
||||
|
||||
def get_bat_theme(terminal_profile: str, is_dark=False, force=False):
|
||||
if not force and BAT_VAR in os.environ:
|
||||
return os.environ[BAT_VAR]
|
||||
|
||||
if "Wombat" in terminal_profile:
|
||||
return "DarkNeon"
|
||||
elif "Solarized" in terminal_profile:
|
||||
if is_dark:
|
||||
return "Solarized (dark)"
|
||||
else:
|
||||
return "Solarized (light)"
|
||||
|
||||
# Default
|
||||
if is_dark:
|
||||
return "DarkNeon"
|
||||
else:
|
||||
return "github"
|
||||
|
||||
|
||||
def parse_args(**args) -> argparse.Namespace:
|
||||
parser = argparse.ArgumentParser(
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
description=textwrap.dedent("""
|
||||
This tool inspects the current terminal to determine which
|
||||
colorschemes is in use and which themes should should be used by
|
||||
various other tools, like vim and bat.
|
||||
|
||||
When running this command, the variables will be printed such
|
||||
that they can be sourced by your shell.
|
||||
|
||||
If your terminal themes are not derrivable, or you'd like to
|
||||
hard code for performance reasons (detecting Terminal.app profiles
|
||||
is kinda slow), the variables to export are:
|
||||
|
||||
TERM_PROFILE Profile name of your current terminal. Used
|
||||
to derive the other colorschemes.
|
||||
|
||||
VIM_COLOR Colorscheme to set in Vim. This can be used to
|
||||
set your theme per example: (insert link)
|
||||
|
||||
BAT_THEME Theme to be used by bat when printing syntax
|
||||
highlighted files.
|
||||
"""),
|
||||
epilog=textwrap.dedent("""\
|
||||
To source all variable into your shell, run:
|
||||
eval $(derive_colors.py --export)
|
||||
If running fish, you must instead use:
|
||||
eval (derive_colors.py --export --fish)
|
||||
"""))
|
||||
group = parser.add_mutually_exclusive_group()
|
||||
group.add_argument(
|
||||
"--print-term",
|
||||
action="store_true",
|
||||
help="Print only the value of the terminal profile",
|
||||
)
|
||||
group.add_argument(
|
||||
"--print-vim",
|
||||
action="store_true",
|
||||
help="Print only the value of the vim colorscheme",
|
||||
)
|
||||
group.add_argument(
|
||||
"--print-bat",
|
||||
action="store_true",
|
||||
help="Print only the value of the bat theme",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--force",
|
||||
action="store_true",
|
||||
help="Force reparsing all variables",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--dark",
|
||||
action="store_true",
|
||||
help="Get dark versions of anything available",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--fish",
|
||||
action="store_true",
|
||||
help="Print fish compatible env commands",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--export",
|
||||
action="store_true",
|
||||
help="Print env variables to be exported",
|
||||
)
|
||||
return parser.parse_args(**args)
|
||||
|
||||
|
||||
def print_all_env(force=False, is_dark=False, export=False, fish=False):
|
||||
term_profile = get_terminal_profile(force=force)
|
||||
print_env(TERM_VAR, term_profile, export=export, fish=fish)
|
||||
|
||||
vim_colors = get_vim_colorscheme(
|
||||
term_profile,
|
||||
is_dark=is_dark,
|
||||
force=force,
|
||||
)
|
||||
print_env(VIM_VAR, vim_colors, export=export, fish=fish)
|
||||
|
||||
bat_theme = get_bat_theme(term_profile, is_dark=is_dark, force=force)
|
||||
print_env(BAT_VAR, bat_theme, export=export, fish=fish)
|
||||
|
||||
|
||||
def print_env(var: str, val: str, export=False, fish=False):
|
||||
if not fish:
|
||||
if export:
|
||||
print(f'export {var}="{val}"')
|
||||
else:
|
||||
print(f'{var}="{val}"')
|
||||
elif fish:
|
||||
if export:
|
||||
print(f'set -gx {var} "{val}";')
|
||||
else:
|
||||
print(f'set -g {var} "{val}";')
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
args = parse_args()
|
||||
if args.print_term:
|
||||
term_profile = get_terminal_profile(force=args.force)
|
||||
print(term_profile)
|
||||
elif args.print_vim:
|
||||
term_profile = get_terminal_profile(force=args.force)
|
||||
vim_colors = get_vim_colorscheme(
|
||||
term_profile,
|
||||
is_dark=args.dark,
|
||||
force=args.force,
|
||||
)
|
||||
print(vim_colors)
|
||||
elif args.print_bat:
|
||||
term_profile = get_terminal_profile(force=args.force)
|
||||
bat_theme = get_bat_theme(
|
||||
term_profile,
|
||||
is_dark=args.dark,
|
||||
force=args.force,
|
||||
)
|
||||
print(bat_theme)
|
||||
else:
|
||||
print_all_env(
|
||||
force=args.force,
|
||||
is_dark=args.dark,
|
||||
fish=args.fish,
|
||||
export=args.export,
|
||||
)
|
||||
Loading…
Reference in New Issue