vividboarder/shoestrap
1
0
Fork 0
mirror of https://github.com/ViViDboarder/shoestrap.git synced 2024-11-24 23:46:31 +00:00
Code Issues Releases Wiki Activity

Update derive colors to support nvim and better standards

Browse Source
This commit is contained in:
ViViDboarder 2021-09-01 10:00:37 -07:00
parent aff0c7aa99
commit a75ef60330
1 changed files with 101 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

140
assets/default/bin/derive_colors.py
View File

@ -1,14 +1,17 @@
#! /usr/bin/env python3
"""Derive colors determines colorschemes for various tools based on the
current terminal"""
import argparse
import os
import sys
import textwrap
from subprocess import Popen, PIPE, check_output
from subprocess import PIPE, Popen, check_output
from typing import Tuple
TERM_VAR = "TERM_PROFILE"
VIM_VAR = "VIM_COLOR"
NVIM_VAR = "NVIM_COLOR"
BAT_VAR = "BAT_THEME"
TERMINAL_SETTINGS_SCRIPT = """
set current_tty to "{}"
@ -26,18 +29,19 @@ end tell
def run_applescript(script: str) -> Tuple[int, str, str]:
p = Popen(
"""Executes an applescript string and return the results"""
with 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()
) as proc:
stdout, stderr = proc.communicate(script)
return proc.returncode, str(stdout).strip(), str(stderr).strip()
def get_terminal_profile(force=False):
def get_terminal_profile(force=False) -> str:
"""Returns the terminal profile from TERM_PROFILE or
derrives through detecting the profile"""
if not force and TERM_VAR in os.environ:
@ -51,26 +55,25 @@ def get_terminal_profile(force=False):
f" Instead forward {TERM_VAR} from your source machine."),
file=sys.stderr,
)
exit(0)
sys.exit(0)
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(
tty_b = check_output(["tty"]).strip()
tty = str(tty_b, encoding="utf-8")
code, stdout, _ = run_applescript(
TERMINAL_SETTINGS_SCRIPT.format(tty),
)
if code:
raise SystemError("Could not get results from applescript")
return stdout
elif term_program == "iTerm.app":
if term_program == "iTerm.app":
if "ITERM_PROFILE" in os.environ:
return os.environ["ITERM_PROFILE"]
else:
raise ValueError("Using iTerm but no profile found")
elif term_program == "Alacritty":
if term_program == "Alacritty":
return "Alacritty"
else:
if os.environ.get("GNOME_TERMINAL_SCREEN") is not None:
return "Gnome Terminal"
@ -78,53 +81,85 @@ def get_terminal_profile(force=False):
raise ValueError(f"Unknown terminal {term_program}")
def get_vim_colorscheme(terminal_profile: str, force_dark=False, force=False):
def get_vim_colorscheme(
terminal_profile: str,
_force_dark=False,
force=False,
) -> str:
"""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 "Alacritty" == terminal_profile:
return "wombat256mod"
colorscheme = "wombat256mod"
if terminal_profile in "Wombat":
colorscheme = "wombat256mod"
elif terminal_profile == "Alacritty":
colorscheme = "wombat256mod"
elif terminal_profile == "Yosemite Light":
colorscheme = "morning"
elif terminal_profile == "Yosemite Dark":
colorscheme = "vividchalk"
elif terminal_profile == "Basic":
colorscheme = "default"
elif "Solarized" in terminal_profile:
return "solarized"
elif "Yosemite Light" == terminal_profile:
return "morning"
elif "Yosemite Dark" == terminal_profile:
return "vividchalk"
elif "Basic" == terminal_profile:
return "default"
colorscheme = "solarized"
# Default
return "wombat256mod"
return colorscheme
def get_bat_theme(terminal_profile: str, force_dark=False, force=False):
def get_nvim_colorscheme(
terminal_profile: str,
_force_dark=False,
force=False,
) -> str:
"""Returns the best colorscheme for a given terminal profile"""
if not force and NVIM_VAR in os.environ:
return os.environ[NVIM_VAR]
colorscheme = "wombat"
if terminal_profile in "Wombat":
colorscheme = "wombat"
elif terminal_profile == "Alacritty":
colorscheme = "wombat"
elif terminal_profile == "Yosemite Light":
colorscheme = "morning"
elif terminal_profile == "Yosemite Dark":
colorscheme = "vividchalk"
elif terminal_profile == "Basic":
colorscheme = "default"
elif "Solarized" in terminal_profile:
colorscheme = "solarized"
return colorscheme
def get_bat_theme(terminal_profile: str, force_dark=False, force=False) -> str:
"""Returns the best matched bat theme for the terminal"""
if not force and BAT_VAR in os.environ:
return os.environ[BAT_VAR]
# Determine if this is a dark theme
is_dark = force_dark or "dark" in terminal_profile.lower()
bat_theme = "DarkNeon" if is_dark else "ansi-light"
if "Wombat" in terminal_profile:
return "DarkNeon"
elif "Alacritty" == terminal_profile:
return "DarkNeon"
bat_theme = "DarkNeon"
elif terminal_profile == "Alacritty":
bat_theme = "DarkNeon"
elif "Solarized" in terminal_profile:
if is_dark:
return "Solarized (dark)"
bat_theme = "Solarized (dark)"
else:
return "Solarized (light)"
bat_theme = "Solarized (light)"
# Default
if is_dark:
return "DarkNeon"
else:
return "ansi-light"
return bat_theme
def parse_args(**args) -> argparse.Namespace:
"""Parse and return args from the terminal"""
parser = argparse.ArgumentParser(
formatter_class=argparse.RawDescriptionHelpFormatter,
description=textwrap.dedent("""
@ -194,21 +229,30 @@ def parse_args(**args) -> argparse.Namespace:
def print_all_env(force=False, force_dark=False, export=False, fish=False):
"""Print all variables in env format"""
term_profile = get_terminal_profile(force=force)
print_env(TERM_VAR, term_profile, export=export, fish=fish)
vim_colors = get_vim_colorscheme(
term_profile,
force_dark=force_dark,
_force_dark=force_dark,
force=force,
)
print_env(VIM_VAR, vim_colors, export=export, fish=fish)
nvim_colors = get_nvim_colorscheme(
term_profile,
_force_dark=force_dark,
force=force,
)
print_env(NVIM_VAR, nvim_colors, export=export, fish=fish)
bat_theme = get_bat_theme(term_profile, force_dark=force_dark, force=force)
print_env(BAT_VAR, bat_theme, export=export, fish=fish)
def print_env(var: str, val: str, export=False, fish=False):
"""Print variable in env format"""
if not fish:
if export:
print(f'export {var}="{val}"')
@ -222,12 +266,14 @@ def print_env(var: str, val: str, export=False, fish=False):
def is_ssh() -> bool:
"""Detect if we're in an SSH session"""
if os.environ.get("SSH_TTY"):
return True
return False
if __name__ == "__main__":
def main():
"""Main function"""
args = parse_args()
if args.print_term:
term_profile = get_terminal_profile(force=args.force)
@ -236,10 +282,16 @@ if __name__ == "__main__":
term_profile = get_terminal_profile(force=args.force)
vim_colors = get_vim_colorscheme(
term_profile,
force_dark=args.dark,
_force_dark=args.dark,
force=args.force,
)
print(vim_colors)
nvim_colors = get_nvim_colorscheme(
term_profile,
_force_dark=args.dark,
force=args.force,
)
print(nvim_colors)
elif args.print_bat:
term_profile = get_terminal_profile(force=args.force)
bat_theme = get_bat_theme(
@ -255,3 +307,7 @@ if __name__ == "__main__":
fish=args.fish,
export=args.export,
)
if __name__ == "__main__":
main()