mirror of
https://github.com/keanuplayz/dotfiles.git
synced 2024-08-15 02:33:12 +00:00
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
import sys
|
|
import os
|
|
import subprocess
|
|
from pathlib import Path
|
|
from typing import Iterable, NoReturn
|
|
|
|
|
|
if os.name == "posix":
|
|
DOTFILES_CONFIG_DIR: Path = Path.home() / ".config" / "dotfiles"
|
|
DOTFILES_CACHE_DIR: Path = Path.home() / ".cache" / "dotfiles"
|
|
|
|
|
|
def platform_not_supported_error() -> NoReturn:
|
|
raise Exception("platform '{}' is not supported!".format(sys.platform))
|
|
|
|
|
|
def run_chooser(
|
|
choices: Iterable[str], prompt: str = None, async_read: bool = False
|
|
) -> int:
|
|
supports_result_index = True
|
|
if os.isatty(sys.stderr.fileno()):
|
|
process_args = [
|
|
"fzf",
|
|
"--with-nth=2..",
|
|
"--height=50%",
|
|
"--reverse",
|
|
"--tiebreak=index",
|
|
]
|
|
supports_result_index = False
|
|
elif sys.platform == "darwin":
|
|
process_args = ["choose", "-i"]
|
|
elif os.name == "posix":
|
|
process_args = ["rofi", "-dmenu", "-i", "-format", "i"]
|
|
if prompt is not None:
|
|
process_args += ["-p", prompt]
|
|
if async_read:
|
|
process_args += ["-async-pre-read", "0"]
|
|
else:
|
|
platform_not_supported_error()
|
|
|
|
chooser_process = subprocess.Popen(
|
|
process_args, stdin=subprocess.PIPE, stdout=subprocess.PIPE
|
|
)
|
|
|
|
with chooser_process.stdin as pipe:
|
|
for index, choice in enumerate(choices):
|
|
assert "\n" not in choice
|
|
if not supports_result_index:
|
|
pipe.write(str(index).encode())
|
|
pipe.write(b" ")
|
|
pipe.write(choice.encode())
|
|
pipe.write(b"\n")
|
|
|
|
exit_code: int = chooser_process.wait()
|
|
if exit_code != 0:
|
|
raise Exception("chooser process failed with exit code {}".format(exit_code))
|
|
|
|
chosen_index = int(chooser_process.stdout.read().strip().split()[0])
|
|
return chosen_index
|
|
|
|
|
|
def send_notification(title: str, message: str, url: str = None) -> None:
|
|
if sys.platform == "darwin":
|
|
process_args = [
|
|
"terminal-notifier",
|
|
"-title",
|
|
title,
|
|
"-message",
|
|
message,
|
|
"-open",
|
|
]
|
|
if url is not None:
|
|
process_args += [url]
|
|
elif os.name == "posix":
|
|
process_args = [
|
|
"notify-send",
|
|
"--icon=utilities-terminal",
|
|
"--expire-time=3000",
|
|
title,
|
|
message,
|
|
]
|
|
else:
|
|
platform_not_supported_error()
|
|
|
|
subprocess.run(process_args, check=True)
|
|
|
|
|
|
def set_clipboard(text: str) -> None:
|
|
# TODO: somehow merge program selection with the logic in `zsh/functions.zsh`
|
|
if sys.platform == "darwin":
|
|
process_args = ["pbcopy"]
|
|
elif os.name == "posix":
|
|
process_args = ["xsel", "--clipboard", "--input"]
|
|
# process_args = ["xclip", "-in", "-selection", "clipboard"]
|
|
else:
|
|
platform_not_supported_error()
|
|
|
|
subprocess.run(process_args, input=text.encode(), check=True)
|