Port: migrate tmpdir state to CLAUDE_PLUGIN_DATA when available #47

export function stateRoot(workspacePath) {
  const base =
    process.env.CLAUDE_PLUGIN_DATA
      ? path.join(process.env.CLAUDE_PLUGIN_DATA, "state")
      : path.join("/tmp", "opencode-companion");
  const hash = crypto.createHash("sha256").update(workspacePath).digest("hex").slice(0, 16);
  return path.join(base, hash);
}

import os from "node:os";

const FALLBACK_STATE_ROOT_DIR = path.join(os.tmpdir(), "opencode-companion");

export function stateRoot(workspacePath) {
  const hash = crypto.createHash("sha256").update(workspacePath).digest("hex").slice(0, 16);
  const dirName = `${hash}`;  // or include a slug — see below

  const pluginDataDir = process.env.CLAUDE_PLUGIN_DATA;

  if (pluginDataDir) {
    const primaryDir = path.join(pluginDataDir, "state", dirName);
    const fallbackDir = path.join(FALLBACK_STATE_ROOT_DIR, dirName);

    // If state exists only in the tmpdir fallback (written by a Bash command
    // without CLAUDE_PLUGIN_DATA), migrate it to the persistent location so
    // future reads/writes use the plugin data dir and state survives tmp cleanup.
    if (
      !fs.existsSync(path.join(primaryDir, "state.json")) &&
      fs.existsSync(path.join(fallbackDir, "state.json"))
    ) {
      fs.cpSync(fallbackDir, primaryDir, { recursive: true });

      // Rewrite absolute paths in the migrated JSON so logFile / jobs/*.json
      // references point at the new location.
      const rewritePaths = (filePath) => {
        try {
          let txt = fs.readFileSync(filePath, "utf8");
          const orig = txt;
          txt = txt.replaceAll(fallbackDir, primaryDir);
          const escapedFallback = fallbackDir.replaceAll("\\", "\\\\");
          const escapedPrimary = primaryDir.replaceAll("\\", "\\\\");
          if (escapedFallback !== fallbackDir) {
            txt = txt.replaceAll(escapedFallback, escapedPrimary);
          }
          if (txt !== orig) fs.writeFileSync(filePath, txt, "utf8");
        } catch { /* non-fatal */ }
      };

      rewritePaths(path.join(primaryDir, "state.json"));
      const jobsDir = path.join(primaryDir, "jobs");
      if (fs.existsSync(jobsDir)) {
        for (const entry of fs.readdirSync(jobsDir)) {
          if (entry.endsWith(".json")) rewritePaths(path.join(jobsDir, entry));
        }
      }
    }

    return primaryDir;
  }

  return path.join(FALLBACK_STATE_ROOT_DIR, dirName);
}