zach
4ee9bb3f12
C++ / fmt (pull_request) Successful in 3s
JS/TS / fmt (pull_request) Successful in 17s
JS/TS / lint (pull_request) Successful in 16s
Python / fmt (pull_request) Successful in 34s
Python / lint (pull_request) Successful in 35s
Python / typecheck (pull_request) Failing after 1m5s
C++ / build (pull_request) Successful in 1m36s
Python / test (pull_request) Successful in 1m26s
Rust / fmt (pull_request) Successful in 48s
Rust / build (pull_request) Successful in 1m41s
Rust / clippy (pull_request) Successful in 1m37s
Python / buildcheck (pull_request) Successful in 2m41s
C++ / clang-tidy (pull_request) Successful in 3m32s
84 lines
1.9 KiB
Python
84 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import typer
|
|
from typer._completion_classes import completion_init
|
|
from typer._completion_shared import _get_shell_name, install
|
|
|
|
from zshell.subcommands import (
|
|
battery,
|
|
record,
|
|
scheme,
|
|
screenshot,
|
|
shell,
|
|
wallpaper,
|
|
)
|
|
|
|
app = typer.Typer(name="zshell-cli", add_completion=False)
|
|
|
|
app.add_typer(shell.app, name="shell")
|
|
app.add_typer(scheme.app, name="scheme")
|
|
app.add_typer(screenshot.app, name="screenshot")
|
|
app.add_typer(wallpaper.app, name="wallpaper")
|
|
app.add_typer(record.app, name="record")
|
|
app.add_typer(battery.app, name="battery")
|
|
|
|
|
|
def _completion_installed() -> bool:
|
|
shell = _get_shell_name()
|
|
match shell:
|
|
case "zsh":
|
|
return (Path.home() / ".zfunc" / "_zshell-cli").exists()
|
|
case "bash":
|
|
return (
|
|
Path.home() / ".bash_completions" / "zshell-cli.sh"
|
|
).exists()
|
|
case "fish":
|
|
return (
|
|
Path.home()
|
|
/ ".config"
|
|
/ "fish"
|
|
/ "completions"
|
|
/ "zshell-cli.fish"
|
|
).exists()
|
|
return False
|
|
|
|
|
|
def _install_completion() -> None:
|
|
if _completion_installed():
|
|
print("zshell-cli: Shell completion already installed.")
|
|
sys.exit(0)
|
|
shell = _get_shell_name()
|
|
if shell is None:
|
|
print("zshell-cli: Unable to detect shell type.", file=sys.stderr)
|
|
sys.exit(1)
|
|
try:
|
|
_, path = install(prog_name="zshell-cli")
|
|
print(f"zshell-cli: Shell completion installed ({shell}: {path})")
|
|
print(
|
|
"zshell-cli: Restart your shell or source the file to enable tab-completion."
|
|
)
|
|
except Exception as e:
|
|
print(
|
|
f"zshell-cli: Failed to install shell completion: {e}",
|
|
file=sys.stderr,
|
|
)
|
|
raise typer.Exit(code=1) from None
|
|
|
|
|
|
def main() -> None:
|
|
if "--install-autocomplete" in sys.argv:
|
|
_install_completion()
|
|
return
|
|
if "_ZSHELL_CLI_COMPLETE" in os.environ:
|
|
completion_init()
|
|
if sys.stdout.isatty() and not _completion_installed():
|
|
print(
|
|
"zshell-cli: Tip: run with --install-autocomplete for tab completion.",
|
|
file=sys.stderr,
|
|
)
|
|
app()
|