dotfiles/script-resources/common_script_utils.py

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)