import pathlib
import subprocess

from openpilot.common.utils import json_dump_dataclass
from openpilot.system.version import BUILD_METADATA_FILENAME, BuildMetadata


CASYNC_ARGS = ["--with=symlinks", "--with=permissions"]
CASYNC_FILES = [BUILD_METADATA_FILENAME, ".caexclude"]


def run(cmd):
  return subprocess.check_output(cmd)


def get_exclude_set(path) -> set[str]:
  exclude_set = set(CASYNC_FILES)

  for file in path.rglob("*"):
    if file.is_file() or file.is_symlink():

      while file.resolve() != path.resolve():
        exclude_set.add(str(file.relative_to(path)))

        file = file.parent

  return exclude_set


def create_caexclude_file(path: pathlib.Path):
  with open(path / ".caexclude", "w") as f:
    # exclude everything except the paths already in the release
    f.write("*\n")
    f.write(".*\n")

    for file in sorted(get_exclude_set(path)):
      f.write(f"!{file}\n")


def create_build_metadata_file(path: pathlib.Path, build_metadata: BuildMetadata):
  with open(path / BUILD_METADATA_FILENAME, "w") as f:
    f.write(json_dump_dataclass(build_metadata))


def create_casync_release(target_dir: pathlib.Path, output_dir: pathlib.Path, channel: str):
  caidx_file = output_dir / f"{channel}.caidx"
  run(["casync", "make", *CASYNC_ARGS, caidx_file, target_dir])
  digest = run(["casync", "digest", *CASYNC_ARGS, target_dir]).decode("utf-8").strip()
  return digest, caidx_file