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)