libretto 0.4.4 → 0.5.1

package/src/cli/core/context.ts

@@ -0,0 +1,127 @@
+import { Logger, createFileLogSink } from "../../shared/logger/index.js";
+import type { LLMClient } from "../../shared/llm/index.js";
+import type { LoggerApi } from "../../shared/logger/index.js";
+import { existsSync, mkdirSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+import { resolveLibrettoRepoRoot } from "../../shared/paths/repo-root.js";
+import { validateSessionName } from "./session.js";
+
+export const REPO_ROOT = resolveLibrettoRepoRoot();
+export const LIBRETTO_CONFIG_DIR = join(REPO_ROOT, ".libretto");
+export const LIBRETTO_CONFIG_PATH = join(LIBRETTO_CONFIG_DIR, "config.json");
+export const PROFILES_DIR = join(LIBRETTO_CONFIG_DIR, "profiles");
+export const LIBRETTO_SESSIONS_DIR = join(LIBRETTO_CONFIG_DIR, "sessions");
+export const LIBRETTO_GITIGNORE_PATH = join(LIBRETTO_CONFIG_DIR, ".gitignore");
+
+const LIBRETTO_GITIGNORE_CONTENT = [
+  "# Local libretto runtime state",
+  "sessions/",
+  "profiles/",
+  "",
+].join("\n");
+
+export function getSessionDir(session: string): string {
+  return join(LIBRETTO_SESSIONS_DIR, session);
+}
+
+export function getSessionStatePath(session: string): string {
+  return join(getSessionDir(session), "state.json");
+}
+
+export function getSessionLogsPath(session: string): string {
+  return join(getSessionDir(session), "logs.jsonl");
+}
+
+export function getSessionNetworkLogPath(session: string): string {
+  return join(getSessionDir(session), "network.jsonl");
+}
+
+export function getSessionActionsLogPath(session: string): string {
+  return join(getSessionDir(session), "actions.jsonl");
+}
+
+export function getSessionSnapshotsDir(session: string): string {
+  return join(getSessionDir(session), "snapshots");
+}
+
+export function getSessionSnapshotRunDir(
+  session: string,
+  snapshotRunId: string,
+): string {
+  return join(getSessionSnapshotsDir(session), snapshotRunId);
+}
+
+export function ensureLibrettoSetup(): void {
+  mkdirSync(LIBRETTO_CONFIG_DIR, { recursive: true });
+  mkdirSync(LIBRETTO_SESSIONS_DIR, { recursive: true });
+  mkdirSync(PROFILES_DIR, { recursive: true });
+
+  if (!existsSync(LIBRETTO_GITIGNORE_PATH)) {
+    writeFileSync(LIBRETTO_GITIGNORE_PATH, LIBRETTO_GITIGNORE_CONTENT, "utf-8");
+  }
+}
+
+export function createLoggerForSession(session: string): Logger {
+  validateSessionName(session);
+  const sessionDir = getSessionDir(session);
+  mkdirSync(sessionDir, { recursive: true });
+  const logFilePath = getSessionLogsPath(session);
+  return new Logger(
+    ["libretto"],
+    [createFileLogSink({ filePath: logFilePath })],
+  );
+}
+
+export async function closeLogger(
+  logger: Logger | null | undefined,
+): Promise<void> {
+  if (!logger) return;
+  await logger.close();
+}
+
+export async function withSessionLogger<T>(
+  session: string,
+  run: (logger: Logger) => Promise<T>,
+): Promise<T> {
+  const logger = createLoggerForSession(session);
+  try {
+    return await run(logger);
+  } finally {
+    await closeLogger(logger);
+  }
+}
+
+let llmClientFactory:
+  | ((logger: LoggerApi, model: string) => Promise<LLMClient>)
+  | null = null;
+
+export function setLLMClientFactory(
+  factory: (logger: LoggerApi, model: string) => Promise<LLMClient>,
+): void {
+  llmClientFactory = factory;
+}
+
+export function getLLMClientFactory():
+  | ((logger: LoggerApi, model: string) => Promise<LLMClient>)
+  | null {
+  return llmClientFactory;
+}
+
+export function maybeConfigureLLMClientFactoryFromEnv(): void {
+  if (llmClientFactory) return;
+
+  const hasAnyCreds =
+    process.env.GOOGLE_CLOUD_PROJECT ||
+    process.env.GCLOUD_PROJECT ||
+    process.env.ANTHROPIC_API_KEY ||
+    process.env.OPENAI_API_KEY ||
+    process.env.GEMINI_API_KEY ||
+    process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+
+  if (!hasAnyCreds) return;
+
+  setLLMClientFactory(async (_logger, model) => {
+    const { createLLMClient } = await import("../../shared/llm/index.js");
+    return createLLMClient(model);
+  });
+}

package/src/cli/core/pause-signals.ts

@@ -0,0 +1,35 @@
+import { existsSync } from "node:fs";
+import { unlink } from "node:fs/promises";
+import { join } from "node:path";
+import { getSessionDir } from "./context.js";
+
+export type PauseSignalPaths = {
+  pausedSignalPath: string;
+  resumeSignalPath: string;
+  completedSignalPath: string;
+  failedSignalPath: string;
+  outputSignalPath: string;
+};
+
+export function getPauseSignalPaths(session: string): PauseSignalPaths {
+  const sessionDir = getSessionDir(session);
+  return {
+    pausedSignalPath: join(sessionDir, `${session}.paused`),
+    resumeSignalPath: join(sessionDir, `${session}.resume`),
+    completedSignalPath: join(sessionDir, `${session}.completed`),
+    failedSignalPath: join(sessionDir, `${session}.failed`),
+    outputSignalPath: join(sessionDir, `${session}.output`),
+  };
+}
+
+export async function removeSignalIfExists(path: string): Promise<void> {
+  if (!existsSync(path)) return;
+  try {
+    await unlink(path);
+  } catch (error) {
+    const code = (error as NodeJS.ErrnoException).code;
+    if (code !== "ENOENT") {
+      throw error;
+    }
+  }
+}