@funkai/cli 0.1.4 → 0.3.0

package/src/lib/prompts/pipeline.ts

@@ -1,8 +1,11 @@
 import { existsSync, readFileSync } from "node:fs";
-import { resolve } from "node:path";
+import { relative, resolve } from "node:path";
 
+import type { PromptGroup } from "@funkai/config";
 import { clean, PARTIALS_DIR } from "@funkai/prompts/cli";
+import picomatch from "picomatch";
 
+import { toFileSlug } from "./codegen.js";
 import type { ParsedPrompt } from "./codegen.js";
 import { extractVariables } from "./extract-variables.js";
 import { flattenPartials } from "./flatten.js";
@@ -11,9 +14,58 @@ import { lintPrompt } from "./lint.js";
 import type { LintResult } from "./lint.js";
 import { discoverPrompts } from "./paths.js";
 
+/**
+ * Validate that no two prompts share the same group+name combination.
+ *
+ * @param prompts - Parsed prompts with group and name fields.
+ * @throws If duplicate group+name combinations are found.
+ *
+ * @private
+ */
+function validateUniqueness(prompts: readonly ParsedPrompt[]): void {
+  const bySlug = Map.groupBy(prompts, (p) => toFileSlug(p.name, p.group));
+  const duplicate = [...bySlug.entries()].find(([, entries]) => entries.length > 1);
+
+  if (duplicate) {
+    const [slug, entries] = duplicate;
+    const paths = entries.map((p) => p.sourcePath).join("\n  ");
+    throw new Error(`Duplicate prompt "${slug}" (group+name) found in:\n  ${paths}`);
+  }
+}
+
+/**
+ * Resolve a prompt's group from config-defined group patterns.
+ *
+ * Matches the prompt's file path against each group's `includes`/`excludes`
+ * patterns. First matching group wins.
+ *
+ * @param filePath - Absolute path to the prompt file.
+ * @param groups - Config-defined group definitions.
+ * @returns The matching group name, or undefined if no match.
+ *
+ * @private
+ */
+function resolveGroupFromConfig(
+  filePath: string,
+  groups: readonly PromptGroup[],
+): string | undefined {
+  const matchPath = relative(process.cwd(), filePath).replaceAll("\\", "/");
+
+  for (const group of groups) {
+    const isIncluded = picomatch(group.includes as string[]);
+    const isExcluded = picomatch((group.excludes ?? []) as string[]);
+
+    if (isIncluded(matchPath) && !isExcluded(matchPath)) {
+      return group.name;
+    }
+  }
+  return undefined;
+}
+
 /**
  * Resolve the list of partial directories to search.
  *
+ * @private
  * @param customDir - Custom partials directory path.
  * @returns Array of directories to search for partials.
  */
@@ -29,7 +81,8 @@ function resolvePartialsDirs(customDir: string): readonly string[] {
  * Options for the prompts lint pipeline.
  */
 export interface LintPipelineOptions {
-  readonly roots: readonly string[];
+  readonly includes: readonly string[];
+  readonly excludes?: readonly string[];
   readonly partials?: string;
 }
 
@@ -48,15 +101,22 @@ export interface LintPipelineResult {
  * @returns Lint results for all discovered prompts.
  */
 export function runLintPipeline(options: LintPipelineOptions): LintPipelineResult {
-  const discovered = discoverPrompts([...options.roots]);
+  let excludes: string[] | undefined;
+  if (options.excludes) {
+    excludes = [...options.excludes];
+  }
+  const discovered = discoverPrompts({
+    includes: [...options.includes],
+    excludes,
+  });
   const customPartialsDir = resolve(options.partials ?? ".prompts/partials");
   const partialsDirs = resolvePartialsDirs(customPartialsDir);
 
   const results = discovered.map((d) => {
     // oxlint-disable-next-line security/detect-non-literal-fs-filename -- safe: reading discovered prompt file
     const raw = readFileSync(d.filePath, "utf8");
-    const frontmatter = parseFrontmatter(raw, d.filePath);
-    const template = flattenPartials(clean(raw), partialsDirs);
+    const frontmatter = parseFrontmatter({ content: raw, filePath: d.filePath });
+    const template = flattenPartials({ template: clean(raw), partialsDirs });
     const templateVars = extractVariables(template);
     return lintPrompt(frontmatter.name, d.filePath, frontmatter.schema, templateVars);
   });
@@ -68,9 +128,11 @@ export function runLintPipeline(options: LintPipelineOptions): LintPipelineResul
  * Options for the prompts generate pipeline.
  */
 export interface GeneratePipelineOptions {
-  readonly roots: readonly string[];
+  readonly includes: readonly string[];
+  readonly excludes?: readonly string[];
   readonly out: string;
   readonly partials?: string;
+  readonly groups?: readonly PromptGroup[];
 }
 
 /**
@@ -91,22 +153,33 @@ export interface GeneratePipelineResult {
  * @returns Parsed prompts ready for code generation, along with lint results.
  */
 export function runGeneratePipeline(options: GeneratePipelineOptions): GeneratePipelineResult {
-  const discovered = discoverPrompts([...options.roots]);
+  let excludes: string[] | undefined;
+  if (options.excludes) {
+    excludes = [...options.excludes];
+  }
+  const discovered = discoverPrompts({
+    includes: [...options.includes],
+    excludes,
+  });
   const customPartialsDir = resolve(options.partials ?? resolve(options.out, "../partials"));
   const partialsDirs = resolvePartialsDirs(customPartialsDir);
+  const configGroups = options.groups ?? [];
 
   const processed = discovered.map((d) => {
     // oxlint-disable-next-line security/detect-non-literal-fs-filename -- safe: reading discovered prompt file
     const raw = readFileSync(d.filePath, "utf8");
-    const frontmatter = parseFrontmatter(raw, d.filePath);
-    const template = flattenPartials(clean(raw), partialsDirs);
+    const frontmatter = parseFrontmatter({ content: raw, filePath: d.filePath });
+    const template = flattenPartials({ template: clean(raw), partialsDirs });
     const templateVars = extractVariables(template);
 
+    // Frontmatter group wins; fall back to config-defined group patterns
+    const group = frontmatter.group ?? resolveGroupFromConfig(d.filePath, configGroups);
+
     return {
       lintResult: lintPrompt(frontmatter.name, d.filePath, frontmatter.schema, templateVars),
       prompt: {
         name: frontmatter.name,
-        group: frontmatter.group,
+        group,
         schema: frontmatter.schema,
         template,
         sourcePath: d.filePath,
@@ -114,9 +187,12 @@ export function runGeneratePipeline(options: GeneratePipelineOptions): GenerateP
     };
   });
 
+  const prompts = processed.map((p) => p.prompt);
+  validateUniqueness(prompts);
+
   return {
     discovered: discovered.length,
     lintResults: processed.map((p) => p.lintResult),
-    prompts: processed.map((p) => p.prompt),
+    prompts,
   };
 }