yaak/crates/yaak-sync/index.ts

import { Channel, invoke } from "@tauri-apps/api/core";
import { emit } from "@tauri-apps/api/event";
import type { WatchResult } from "@yaakapp-internal/tauri-client";
import { SyncOp } from "./bindings/gen_sync";
import { WatchEvent } from "./bindings/gen_watch";

export * from "./bindings/gen_models";

export async function calculateSync(workspaceId: string, syncDir: string) {
  return invoke<SyncOp[]>("cmd_sync_calculate", {
    workspaceId,
    syncDir,
  });
}

export async function calculateSyncFsOnly(dir: string) {
  return invoke<SyncOp[]>("cmd_sync_calculate_fs", { dir });
}

export async function applySync(
  workspaceId: string,
  syncDir: string,
  syncOps: SyncOp[],
) {
  return invoke<void>("cmd_sync_apply", {
    workspaceId,
    syncDir,
    syncOps: syncOps,
  });
}

export function watchWorkspaceFiles(
  workspaceId: string,
  syncDir: string,
  callback: (e: WatchEvent) => void,
) {
  console.log("Watching workspace files", workspaceId, syncDir);
  const channel = new Channel<WatchEvent>();
  channel.onmessage = callback;
  const unlistenPromise = invoke<WatchResult>("cmd_sync_watch", {
    workspaceId,
    syncDir,
    channel,
  });

  unlistenPromise.then(({ unlistenEvent }) => {
    addWatchKey(unlistenEvent);
  });

  return () =>
    unlistenPromise
      .then(async ({ unlistenEvent }) => {
        console.log("Unwatching workspace files", workspaceId, syncDir);
        unlistenToWatcher(unlistenEvent);
      })
      .catch(console.error);
}

function unlistenToWatcher(unlistenEvent: string) {
  emit(unlistenEvent).then(() => {
    removeWatchKey(unlistenEvent);
  });
}

function getWatchKeys() {
  return (
    sessionStorage
      .getItem("workspace-file-watchers")
      ?.split(",")
      .filter(Boolean) ?? []
  );
}

function setWatchKeys(keys: string[]) {
  sessionStorage.setItem("workspace-file-watchers", keys.join(","));
}

function addWatchKey(key: string) {
  const keys = getWatchKeys();
  setWatchKeys([...keys, key]);
}

function removeWatchKey(key: string) {
  const keys = getWatchKeys();
  setWatchKeys(keys.filter((k) => k !== key));
}

// On page load, unlisten to all zombie watchers
const keys = getWatchKeys();
if (keys.length > 0) {
  console.log("Unsubscribing to zombie file watchers", keys);
  keys.forEach(unlistenToWatcher);
}