import os
import subprocess
from openpilot.common.basedir import BASEDIR


class Spinner:
  def __init__(self):
    self.spinner_proc = None

    # Prefer the legacy compiled spinner on device.
    legacy_spinner = os.path.join(BASEDIR, "selfdrive", "ui", "spinner")
    if os.path.isfile("/TICI") and os.path.isfile(legacy_spinner) and os.access(legacy_spinner, os.X_OK):
      try:
        self.spinner_proc = subprocess.Popen([legacy_spinner],
                                             stdin=subprocess.PIPE,
                                             cwd=os.path.join(BASEDIR, "selfdrive", "ui"),
                                             close_fds=True)
        return
      except OSError:
        self.spinner_proc = None

    # Raylib spinner requires Python deps from the repo virtualenv.
    spinner_cwd = os.path.join(BASEDIR, "system", "ui")
    venv_python = os.path.join(BASEDIR, ".venv", "bin", "python")
    python_exec = venv_python if os.path.isfile(venv_python) else "python3"
    try:
      self.spinner_proc = subprocess.Popen([python_exec, "./spinner.py"],
                                           stdin=subprocess.PIPE,
                                           cwd=spinner_cwd,
                                           close_fds=True)
    except OSError:
      self.spinner_proc = None

  def __enter__(self):
    return self

  def update(self, spinner_text: str):
    if self.spinner_proc is not None:
      self.spinner_proc.stdin.write(spinner_text.encode('utf8') + b"\n")
      try:
        self.spinner_proc.stdin.flush()
      except BrokenPipeError:
        pass

  def update_progress(self, cur: float, total: float):
    self.update(str(round(100 * cur / total)))

  def close(self):
    if self.spinner_proc is not None:
      self.spinner_proc.kill()
      try:
        self.spinner_proc.communicate(timeout=2.)
      except subprocess.TimeoutExpired:
        print("WARNING: failed to kill spinner")
      self.spinner_proc = None

  def __del__(self):
    self.close()

  def __exit__(self, exc_type, exc_value, traceback):
    self.close()


if __name__ == "__main__":
  import time
  with Spinner() as s:
    s.update("Spinner text")
    time.sleep(5.0)
  print("gone")
  time.sleep(5.0)