openclaw/extensions/acpx/src/pi-session-paths.ts

import { readFileSync, statSync } from "node:fs";
import os from "node:os";
import path from "node:path";
import { isRecord } from "openclaw/plugin-sdk/string-coerce-runtime";

function optionalString(value: unknown, maxLength: number): string | undefined {
  if (typeof value !== "string") {
    return undefined;
  }
  const trimmed = value.trim();
  return trimmed && trimmed.length <= maxLength ? trimmed : undefined;
}

function piHome(env: NodeJS.ProcessEnv): string {
  const configured = process.platform === "win32" ? env.USERPROFILE?.trim() : env.HOME?.trim();
  return configured || os.homedir();
}

function isPiSessionCatalogPathAbsolute(
  value: string,
  platform: NodeJS.Platform = process.platform,
): boolean {
  if (platform !== "win32") {
    return path.posix.isAbsolute(value);
  }
  const root = path.win32.parse(value).root;
  return path.win32.isAbsolute(value) && root !== "\\" && root !== "/";
}

function resolveConfiguredPath(
  value: string,
  env: NodeJS.ProcessEnv,
  relativeBase?: string,
): string {
  const home = piHome(env);
  let resolved = value;
  if (value === "~") {
    resolved = home;
  }
  if (value.startsWith("~/") || value.startsWith("~\\")) {
    resolved = path.join(home, value.slice(2));
  }
  if (!isPiSessionCatalogPathAbsolute(resolved)) {
    if (relativeBase) {
      return path.resolve(relativeBase, resolved);
    }
    throw new Error("Pi session catalog requires absolute or home-relative storage paths");
  }
  return path.resolve(resolved);
}

function settingsSessionDir(file: string): string | undefined {
  try {
    const value = JSON.parse(readFileSync(file, "utf8")) as unknown;
    return isRecord(value) ? optionalString(value.sessionDir, 4_096) : undefined;
  } catch {
    return undefined;
  }
}

export function piSessionStore(
  env: NodeJS.ProcessEnv,
  cwd = process.cwd(),
): { root: string; flat: boolean } {
  const customSessionDir = env.PI_CODING_AGENT_SESSION_DIR?.trim();
  if (customSessionDir) {
    return { root: resolveConfiguredPath(customSessionDir, env), flat: true };
  }
  const home = piHome(env);
  const customAgentDir = env.PI_CODING_AGENT_DIR?.trim();
  const agentDir = customAgentDir
    ? resolveConfiguredPath(customAgentDir, env)
    : path.join(home, ".pi", "agent");
  const projectSessionDir = settingsSessionDir(path.join(cwd, ".pi", "settings.json"));
  if (projectSessionDir) {
    return {
      root: resolveConfiguredPath(projectSessionDir, env, path.join(cwd, ".pi")),
      flat: true,
    };
  }
  const globalSessionDir = settingsSessionDir(path.join(agentDir, "settings.json"));
  if (globalSessionDir) {
    return { root: resolveConfiguredPath(globalSessionDir, env, agentDir), flat: true };
  }
  return {
    root: path.join(agentDir, "sessions"),
    flat: false,
  };
}

/** Store root scanned by pi-acp@0.0.26 when resolving a native session id. */
export function piAcpSessionStoreRoot(env: NodeJS.ProcessEnv): string | undefined {
  const configuredAgentDir = env.PI_CODING_AGENT_DIR?.trim();
  // Deliberately stricter than piSessionStore(): pi-acp's session lookup reads
  // PI_CODING_AGENT_DIR raw, with no `~` expansion (getPiAgentDir, dist/index.js:1044,
  // reached via findPiSessionFile -> listPiSessions). Expanding `~` here would advertise
  // Continue on sessions pi-acp then fails to resolve with "Unknown sessionId".
  if (configuredAgentDir && !isPiSessionCatalogPathAbsolute(configuredAgentDir)) {
    return undefined;
  }
  const agentDir = configuredAgentDir
    ? path.resolve(configuredAgentDir)
    : path.join(piHome(env), ".pi", "agent");
  return path.join(agentDir, "sessions");
}

export function piSessionStoreAvailable(env: NodeJS.ProcessEnv): boolean {
  try {
    return statSync(piSessionStore(env).root).isDirectory();
  } catch {
    return false;
  }
}