@frontman-ai/astro 0.1.0

package/src/cli/FrontmanAstro__Cli__AutoEdit.res.mjs

@@ -0,0 +1,266 @@
+// Generated by ReScript, PLEASE EDIT WITH CARE
+
+import * as Fs from "fs";
+import * as Stdlib_Option from "@rescript/runtime/lib/es6/Stdlib_Option.js";
+import * as Readline$FrontmanBindings from "@frontman/bindings/src/Readline.res.mjs";
+import * as FrontmanAstro__Cli__Style$FrontmanAiAstro from "./FrontmanAstro__Cli__Style.res.mjs";
+import * as FrontmanAstro__Cli__Templates$FrontmanAiAstro from "./FrontmanAstro__Cli__Templates.res.mjs";
+
+let apiBaseUrl = "https://opencode.ai/zen/v1/chat/completions";
+
+let apiKey = "public";
+
+let models = [
+  "gpt-5-nano",
+  "big-pickle",
+  "glm-4.7-free"
+];
+
+function buildSystemPrompt(fileType, host) {
+  let match;
+  match = fileType === "Config" ? [
+      "astro.config.mjs",
+      FrontmanAstro__Cli__Templates$FrontmanAiAstro.ErrorMessages.configManualSetup("astro.config.mjs", host),
+      FrontmanAstro__Cli__Templates$FrontmanAiAstro.configTemplate(host),
+      `- Add the import for '@astrojs/node' at the top of the file
+- Add the import for '@frontman-ai/astro/integration' at the top of the file
+- Add frontmanIntegration() to the integrations array
+- Add SSR dev mode config: ...(isProd ? {} : { output: 'server', adapter: node({ mode: 'standalone' }) })
+- Add const isProd = process.env.NODE_ENV === 'production'; before defineConfig
+- Preserve ALL existing integrations and configuration unchanged
+- Do not remove or modify any existing imports or settings`
+    ] : [
+      "src/middleware.ts",
+      FrontmanAstro__Cli__Templates$FrontmanAiAstro.ErrorMessages.middlewareManualSetup("src/middleware.ts", host),
+      FrontmanAstro__Cli__Templates$FrontmanAiAstro.middlewareTemplate(host),
+      `- Add the import for '@frontman-ai/astro' (createMiddleware, makeConfig) at the top of the file
+- Add the import for 'astro:middleware' (defineMiddleware, sequence) at the top of the file
+- Create a Frontman middleware instance with makeConfig({ host: '` + host + `' })
+- Create a defineMiddleware wrapper for the Frontman handler
+- Use sequence() to combine the Frontman middleware with the existing onRequest handler
+- The Frontman middleware should come FIRST in the sequence
+- Preserve ALL existing middleware functionality unchanged
+- Do not remove or modify any existing imports or middleware logic`
+    ];
+  return `You are a code editor. Modify an Astro ` + match[0] + ` file to integrate Frontman.
+
+## What to add
+` + match[1] + `
+
+## Reference template (for a fresh file without any existing code):
+
+` + match[2] + `
+
+## Rules
+` + match[3] + `
+- Return ONLY the complete file contents. No markdown fences, no explanations, no comments about changes.`;
+}
+
+function buildUserMessage(existingContent) {
+  return `Here is the existing file to modify:
+
+` + existingContent;
+}
+
+let fetchChatCompletion = (async function(url, apiKey, model, systemPrompt, userMessage, timeoutMs) {
+    try {
+      const response = await fetch(url, {
+        method: "POST",
+        signal: AbortSignal.timeout(timeoutMs),
+        headers: {
+          "Content-Type": "application/json",
+          "Authorization": "Bearer " + apiKey,
+        },
+        body: JSON.stringify({
+          model: model,
+          temperature: 0,
+          messages: [
+            { role: "system", content: systemPrompt },
+            { role: "user", content: userMessage },
+          ],
+        }),
+      });
+
+      if (!response.ok) {
+        return { TAG: "Error", _0: "HTTP " + response.status + ": " + response.statusText };
+      }
+
+      const json = await response.json();
+      const content = json?.choices?.[0]?.message?.content?.trim();
+      if (!content) {
+        return { TAG: "Error", _0: "Empty response from model" };
+      }
+      return { TAG: "Ok", _0: content };
+    } catch (err) {
+      if (err?.name === "TimeoutError") {
+        return { TAG: "Error", _0: "Request timed out after " + (timeoutMs / 1000) + "s" };
+      }
+      return { TAG: "Error", _0: "Request failed: " + (err?.message || "Unknown error") };
+    }
+  });
+
+async function callModel(model, systemPrompt, userMessage) {
+  return await fetchChatCompletion(apiBaseUrl, apiKey, model, systemPrompt, userMessage, 30000);
+}
+
+function stripMarkdownFences(content) {
+  let lines = content.split("\n");
+  let len = lines.length;
+  let firstLine = Stdlib_Option.getOr(lines[0], "");
+  let startsWithFence = firstLine.startsWith("```");
+  if (!startsWithFence) {
+    return content;
+  }
+  let lastLine = Stdlib_Option.getOr(lines[len - 1 | 0], "");
+  let endsWithFence = lastLine.trim() === "```";
+  let endIdx = endsWithFence ? len - 1 | 0 : len;
+  return lines.slice(1, endIdx).join("\n");
+}
+
+function validateOutput(content, fileType) {
+  if (fileType === "Config") {
+    if (content.includes("frontmanIntegration") && content.includes("@frontman-ai/astro")) {
+      return content.includes("defineConfig");
+    } else {
+      return false;
+    }
+  } else if (content.includes("@frontman-ai/astro") && content.includes("createMiddleware") && content.includes("makeConfig")) {
+    return content.includes("onRequest");
+  } else {
+    return false;
+  }
+}
+
+async function callLLM(existingContent, fileType, host) {
+  let systemPrompt = buildSystemPrompt(fileType, host);
+  let userMessage = buildUserMessage(existingContent);
+  let tryModels = async (remaining, errors) => {
+    let model = remaining[0];
+    if (model !== undefined) {
+      let rest = remaining.slice(1, remaining.length);
+      console.log(`     ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.dim(`Trying model: ` + model + `...`));
+      let rawContent = await callModel(model, systemPrompt, userMessage);
+      if (rawContent.TAG === "Ok") {
+        let content = stripMarkdownFences(rawContent._0);
+        if (validateOutput(content, fileType)) {
+          return {
+            TAG: "Ok",
+            _0: content
+          };
+        }
+        let err = model + `: output validation failed (missing Frontman imports)`;
+        console.log(`     ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.dim(err));
+        return await tryModels(rest, errors.concat([err]));
+      }
+      let errMsg = model + `: ` + rawContent._0;
+      console.log(`     ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.dim(errMsg));
+      return await tryModels(rest, errors.concat([errMsg]));
+    }
+    let allErrors = errors.join("; ");
+    return {
+      TAG: "Error",
+      _0: `All models failed: ` + allErrors
+    };
+  };
+  return await tryModels(models, []);
+}
+
+async function promptUserForAutoEdit(fileNames) {
+  if (!Readline$FrontmanBindings.isTTY()) {
+    return false;
+  }
+  console.log("");
+  let match = fileNames.length;
+  if (match !== 1) {
+    console.log(`  ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.warn + `  The following files exist but don't have Frontman configured:`);
+    fileNames.forEach(fileName => {
+      console.log(`     ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.purple("•") + ` ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.bold(fileName));
+    });
+  } else {
+    let fileName = fileNames[0];
+    console.log(`  ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.warn + `  ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.bold(fileName) + ` exists but doesn't have Frontman configured.`);
+  }
+  console.log(`     ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.dim("Your file contents will be sent to a public LLM (OpenCode Zen)."));
+  console.log("");
+  let answer = await Readline$FrontmanBindings.question(`     Auto-edit using AI? ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.dim("[Y/n]") + ` `);
+  if (answer == null) {
+    return false;
+  }
+  let match$1 = answer.trim().toLowerCase();
+  switch (match$1) {
+    case "" :
+    case "y" :
+    case "yes" :
+      return true;
+    default:
+      return false;
+  }
+}
+
+async function autoEditFile(filePath, fileName, existingContent, fileType, host) {
+  let fileSize = existingContent.length;
+  if (fileSize > 50000) {
+    return {
+      TAG: "AutoEditFailed",
+      _0: fileName + ` is too large (` + (fileSize / 1000 | 0).toString() + `KB) for auto-edit — max ` + (50).toString() + `KB`
+    };
+  }
+  console.log("");
+  console.log(`  ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.purple("⟳") + `  Merging Frontman into ` + FrontmanAstro__Cli__Style$FrontmanAiAstro.bold(fileName) + `...`);
+  let newContent = await callLLM(existingContent, fileType, host);
+  if (newContent.TAG !== "Ok") {
+    return {
+      TAG: "AutoEditFailed",
+      _0: newContent._0
+    };
+  }
+  try {
+    await Fs.promises.writeFile(filePath, newContent._0, "utf8");
+    return {
+      TAG: "AutoEdited",
+      _0: fileName
+    };
+  } catch (exn) {
+    return {
+      TAG: "AutoEditFailed",
+      _0: `Failed to write ` + fileName
+    };
+  }
+}
+
+let Bindings;
+
+let Fs$1;
+
+let Readline;
+
+let Templates;
+
+let Style;
+
+let requestTimeoutMs = 30000;
+
+let maxFileSizeBytes = 50000;
+
+export {
+  Bindings,
+  Fs$1 as Fs,
+  Readline,
+  Templates,
+  Style,
+  apiBaseUrl,
+  apiKey,
+  models,
+  buildSystemPrompt,
+  buildUserMessage,
+  requestTimeoutMs,
+  fetchChatCompletion,
+  callModel,
+  stripMarkdownFences,
+  validateOutput,
+  callLLM,
+  promptUserForAutoEdit,
+  maxFileSizeBytes,
+  autoEditFile,
+}
+/* fetchChatCompletion Not a pure module */

package/src/cli/FrontmanAstro__Cli__Detect.res

@@ -0,0 +1,298 @@
+// Detection module for Astro project analysis
+module Bindings = FrontmanBindings
+module Fs = Bindings.Fs
+module Path = Bindings.Path
+
+type astroVersion = {
+  major: int,
+  minor: int,
+  raw: string,
+}
+
+type packageManager =
+  | Npm
+  | Yarn
+  | Pnpm
+  | Bun
+  | Deno
+
+type existingFile =
+  | NotFound
+  | HasFrontman({host: string})
+  | NeedsManualEdit
+
+type projectInfo = {
+  astroVersion: option<astroVersion>,
+  config: existingFile,
+  middleware: existingFile,
+  configFileName: string,
+  middlewareFileName: string,
+  packageManager: packageManager,
+}
+
+// Check if a file exists
+let fileExists = async (path: string): bool => {
+  try {
+    await Fs.Promises.access(path)
+    true
+  } catch {
+  | _ => false
+  }
+}
+
+// Read file content safely
+let readFile = async (path: string): option<string> => {
+  try {
+    let content = await Fs.Promises.readFile(path)
+    Some(content)
+  } catch {
+  | _ => None
+  }
+}
+
+// Schema for the minimal astro package.json we need
+@schema
+type astroPkgJson = {version: string}
+
+// Detect Astro version from node_modules
+let detectAstroVersion = async (projectDir: string): option<astroVersion> => {
+  let astroPkgPath = Path.join([projectDir, "node_modules", "astro", "package.json"])
+
+  switch await readFile(astroPkgPath) {
+  | None => None
+  | Some(content) =>
+    try {
+      let pkg = S.parseJsonStringOrThrow(content, astroPkgJsonSchema)
+      let version = pkg.version
+      // Parse version like "5.0.0" or "5.1.0-beta.1"
+      let parts = version->String.split(".")
+      switch (parts->Array.get(0), parts->Array.get(1)) {
+      | (Some(majorStr), Some(minorStr)) =>
+        let major = majorStr->Int.fromString->Option.getOrThrow
+        // Handle minor with potential suffixes like "0-beta"
+        let minorClean = minorStr->String.split("-")->Array.getUnsafe(0)
+        let minor = minorClean->Int.fromString->Option.getOrThrow
+        Some({major, minor, raw: version})
+      | _ => None
+      }
+    } catch {
+    | _ => None
+    }
+  }
+}
+
+// Detect which astro config file variant exists
+// Checks: astro.config.mjs, astro.config.ts, astro.config.mts, astro.config.js
+let detectConfigFile = async (projectDir: string): option<string> => {
+  let variants = ["astro.config.mjs", "astro.config.ts", "astro.config.mts", "astro.config.js"]
+
+  let rec check = async (remaining: array<string>) => {
+    switch remaining->Array.get(0) {
+    | None => None
+    | Some(name) =>
+      let path = Path.join([projectDir, name])
+      switch await fileExists(path) {
+      | true => Some(name)
+      | false =>
+        let rest = remaining->Array.slice(~start=1, ~end=Array.length(remaining))
+        await check(rest)
+      }
+    }
+  }
+
+  await check(variants)
+}
+
+// Pattern to detect @frontman-ai/astro (or legacy @frontman/frontman-astro) import
+let frontmanImportPattern = %re("/@frontman-ai\/astro|@frontman\/frontman-astro|frontman-astro\/integration/")
+
+// Pattern to extract host from makeConfig or createMiddleware config
+let hostPattern = %re("/host:\s*['\"]([^'\"]+)['\"]/")
+
+// Analyze an existing file for Frontman configuration
+let analyzeFile = async (filePath: string): existingFile => {
+  switch await readFile(filePath) {
+  | None => NotFound
+  | Some(content) =>
+    // Check if it imports @frontman-ai/astro (or legacy @frontman/frontman-astro)
+    switch frontmanImportPattern->RegExp.test(content) {
+    | true =>
+      // Try to extract the host
+      switch hostPattern->RegExp.exec(content) {
+      | Some(result) =>
+        let maybeHost =
+          result
+          ->RegExp.Result.matches
+          ->Array.get(0) // First capture group after slice(1)
+          ->Option.flatMap(x => x)
+        switch maybeHost {
+        | Some(host) => HasFrontman({host: host})
+        | None => HasFrontman({host: ""})
+        }
+      | None => HasFrontman({host: ""})
+      }
+    | false => NeedsManualEdit
+    }
+  }
+}
+
+// Detect package manager from lock files
+// Checks current directory and parent directories (for monorepo setups)
+let detectPackageManager = async (projectDir: string): packageManager => {
+  // Check a directory for lock files, in priority order
+  let checkDir = async (dir: string): option<packageManager> => {
+    let lockFiles = [
+      (Path.join([dir, "bun.lockb"]), Bun),
+      (Path.join([dir, "bun.lock"]), Bun),
+      (Path.join([dir, "deno.lock"]), Deno),
+      (Path.join([dir, "pnpm-lock.yaml"]), Pnpm),
+      (Path.join([dir, "yarn.lock"]), Yarn),
+      (Path.join([dir, "package-lock.json"]), Npm),
+    ]
+
+    let rec check = async (remaining: array<(string, packageManager)>) => {
+      switch remaining->Array.get(0) {
+      | None => None
+      | Some((path, pm)) =>
+        switch await fileExists(path) {
+        | true => Some(pm)
+        | false =>
+          let rest = remaining->Array.slice(~start=1, ~end=Array.length(remaining))
+          await check(rest)
+        }
+      }
+    }
+
+    await check(lockFiles)
+  }
+
+  // Check project dir, then parent, then grandparent (monorepo support)
+  let dirsToCheck = {
+    let parentDir = Path.dirname(projectDir)
+    let grandparentDir = Path.dirname(parentDir)
+    [projectDir, parentDir, grandparentDir]->Array.filter(d => d != projectDir || d == projectDir)
+  }
+
+  // Deduplicate (e.g. at filesystem root, dirname("/") == "/")
+  let uniqueDirs = {
+    let seen = Dict.make()
+    dirsToCheck->Array.filter(d => {
+      switch seen->Dict.get(d) {
+      | Some(_) => false
+      | None =>
+        seen->Dict.set(d, true)
+        true
+      }
+    })
+  }
+
+  let rec tryDirs = async (remaining: array<string>) => {
+    switch remaining->Array.get(0) {
+    | None => Npm // Default to npm
+    | Some(dir) =>
+      switch await checkDir(dir) {
+      | Some(pm) => pm
+      | None =>
+        let rest = remaining->Array.slice(~start=1, ~end=Array.length(remaining))
+        await tryDirs(rest)
+      }
+    }
+  }
+
+  await tryDirs(uniqueDirs)
+}
+
+// Check if package.json exists (validates this is a project root)
+let hasPackageJson = async (projectDir: string): bool => {
+  await fileExists(Path.join([projectDir, "package.json"]))
+}
+
+// Main detection function
+let detect = async (projectDir: string): result<projectInfo, string> => {
+  // First verify this is a project directory
+  let hasPackage = await hasPackageJson(projectDir)
+  switch hasPackage {
+  | false => Error("No package.json found. Please run from your Astro project root.")
+  | true =>
+    // Detect Astro version
+    let astroVersion = await detectAstroVersion(projectDir)
+
+    switch astroVersion {
+    | None =>
+      Error(
+        "Could not find Astro in node_modules. Please run 'npm install' first or verify this is an Astro project.",
+      )
+    | Some(_) =>
+      // Detect config file
+      let configFileName = switch await detectConfigFile(projectDir) {
+      | Some(name) => name
+      | None => "astro.config.mjs" // Default for creation
+      }
+
+      let configPath = Path.join([projectDir, configFileName])
+      let config = await analyzeFile(configPath)
+
+      // Detect middleware file (check src/middleware.ts and src/middleware.js)
+      let middlewareTsPath = Path.join([projectDir, "src", "middleware.ts"])
+      let middlewareJsPath = Path.join([projectDir, "src", "middleware.js"])
+
+      let (middleware, middlewareFileName) = switch await fileExists(middlewareTsPath) {
+      | true => (await analyzeFile(middlewareTsPath), "src/middleware.ts")
+      | false =>
+        switch await fileExists(middlewareJsPath) {
+        | true => (await analyzeFile(middlewareJsPath), "src/middleware.js")
+        | false => (NotFound, "src/middleware.ts") // Default for creation
+        }
+      }
+
+      // Detect package manager
+      let packageManager = await detectPackageManager(projectDir)
+
+      Ok({
+        astroVersion,
+        config,
+        middleware,
+        configFileName,
+        middlewareFileName,
+        packageManager,
+      })
+    }
+  }
+}
+
+// Get package manager command
+// Uses npx to ensure the package manager is available even if not in PATH
+let getPackageManagerCommand = (pm: packageManager): string => {
+  switch pm {
+  | Npm => "npm"
+  | Yarn => "npx yarn"
+  | Pnpm => "npx pnpm"
+  | Bun => "bun"
+  | Deno => "deno"
+  }
+}
+
+// Get the dev server command for display in success messages
+let getDevCommand = (pm: packageManager): string => {
+  switch pm {
+  | Npm => "npm run dev"
+  | Yarn => "yarn dev"
+  | Pnpm => "pnpm dev"
+  | Bun => "bun dev"
+  | Deno => "deno task dev"
+  }
+}
+
+// Get install command args for each package manager
+let getInstallArgs = (pm: packageManager, ~isDev: bool=false): array<string> => {
+  switch (pm, isDev) {
+  | (Npm, true) => ["install", "-D"]
+  | (Npm, false) => ["install"]
+  | (Yarn, true) => ["add", "-D"]
+  | (Yarn, false) => ["add"]
+  | (Pnpm, true) => ["add", "--save-dev"]
+  | (Pnpm, false) => ["add"]
+  | (Bun, true) | (Deno, true) => ["add", "--dev"]
+  | (Bun, false) | (Deno, false) => ["add"]
+  }
+}