agents.yaml 0.2.0 → 0.2.2

package/src/run.ts

@@ -1,20 +1,35 @@
-import * as clack from '@clack/prompts'
+import { MultiSelectPrompt } from "@clack/core"
+import * as clack from "@clack/prompts"
+import { styleText } from "node:util"
 import {
-  addDocuments,
-  initProject,
-  loadAgentsFile,
-  removeDocuments,
-  validateAgentsFile,
-} from './agents-file.ts'
-import { describeAgentDocument, discoverAgentDocuments } from './discover.ts'
-import { cwd, formatProjectPath, resolveFromRoot } from './paths.ts'
-
-type Command = 'add' | 'discover' | 'help' | 'init' | 'remove' | 'validate' | 'version'
+	addDocuments,
+	initProject,
+	loadAgentsFile,
+	removeDocuments,
+	validateAgentsFile,
+} from "./agents-file.ts"
+import { describeAgentDocument, discoverAgentDocuments } from "./discover.ts"
+import { cwd, formatProjectPath, resolveFromRoot } from "./paths.ts"
+
+type Command =
+	| "add"
+	| "discover"
+	| "help"
+	| "init"
+	| "remove"
+	| "validate"
+	| "version"
 
 type ParsedArgs = {
-  command: Command | undefined
-  values: string[]
-  flags: Map<string, string | boolean>
+	command: Command | undefined
+	values: string[]
+	flags: Map<string, string | boolean>
+}
+
+type DocumentOption = {
+	value: string
+	label: string
+	disabled?: boolean
 }
 
 const helpText = `agents
@@ -22,7 +37,7 @@ const helpText = `agents
 Usage:
   agents
   agents init [--force]
-  agents discover [--json]
+  agents discover [--json] [--include-dot-directories]
   agents add <path...>
   agents remove <path...>
   agents validate [--json]
@@ -30,214 +45,297 @@ Usage:
 agents.yaml is a curated table of contents for promoted AGENTS.md guidance.`
 
 export async function run(argv: string[]): Promise<void> {
-  const parsed = parseArgs(argv)
-  const root = cwd()
-
-  switch (parsed.command) {
-    case undefined:
-      await interactive(root)
-      return
-    case 'help':
-      console.log(helpText)
-      return
-    case 'version':
-      console.log('0.1.0')
-      return
-    case 'init':
-      await commandInit(root, parsed.flags.get('force') === true)
-      return
-    case 'discover':
-      await commandDiscover(root, parsed.flags.get('json') === true)
-      return
-    case 'add':
-      await commandAdd(root, parsed.values)
-      return
-    case 'remove':
-      await commandRemove(root, parsed.values)
-      return
-    case 'validate':
-      await commandValidate(root, parsed.flags.get('json') === true)
-      return
-  }
+	const parsed = parseArgs(argv)
+	const root = cwd()
+
+	switch (parsed.command) {
+		case undefined:
+			await interactive(root)
+			return
+		case "help":
+			console.log(helpText)
+			return
+		case "version":
+			console.log("0.1.0")
+			return
+		case "init":
+			await commandInit(root, parsed.flags.get("force") === true)
+			return
+		case "discover":
+			await commandDiscover(root, {
+				json: parsed.flags.get("json") === true,
+				includeDotDirectories:
+					parsed.flags.get("include-dot-directories") === true,
+			})
+			return
+		case "add":
+			await commandAdd(root, parsed.values)
+			return
+		case "remove":
+			await commandRemove(root, parsed.values)
+			return
+		case "validate":
+			await commandValidate(root, parsed.flags.get("json") === true)
+			return
+	}
 }
 
 function parseArgs(argv: string[]): ParsedArgs {
-  const flags = new Map<string, string | boolean>()
-  const values: string[] = []
-  let command: Command | undefined
-
-  for (let index = 0; index < argv.length; index += 1) {
-    const arg = argv[index]
-    if (!arg) continue
-
-    if (arg === '--help' || arg === '-h') {
-      command = 'help'
-      continue
-    }
-
-    if (arg === '--version' || arg === '-v') {
-      command = 'version'
-      continue
-    }
-
-    if (arg.startsWith('--')) {
-      const [rawName, inlineValue] = arg.slice(2).split('=', 2)
-      if (!rawName) continue
-      if (inlineValue !== undefined) {
-        flags.set(rawName, inlineValue)
-        continue
-      }
-
-      flags.set(rawName, true)
-      continue
-    }
-
-    if (!command && isCommand(arg)) {
-      command = arg
-      continue
-    }
-
-    values.push(arg)
-  }
-
-  return { command, values, flags }
+	const flags = new Map<string, string | boolean>()
+	const values: string[] = []
+	let command: Command | undefined
+
+	for (let index = 0; index < argv.length; index += 1) {
+		const arg = argv[index]
+		if (!arg) continue
+
+		if (arg === "--help" || arg === "-h") {
+			command = "help"
+			continue
+		}
+
+		if (arg === "--version" || arg === "-v") {
+			command = "version"
+			continue
+		}
+
+		if (arg.startsWith("--")) {
+			const [rawName, inlineValue] = arg.slice(2).split("=", 2)
+			if (!rawName) continue
+			if (inlineValue !== undefined) {
+				flags.set(rawName, inlineValue)
+				continue
+			}
+
+			flags.set(rawName, true)
+			continue
+		}
+
+		if (!command && isCommand(arg)) {
+			command = arg
+			continue
+		}
+
+		values.push(arg)
+	}
+
+	return { command, values, flags }
 }
 
 function isCommand(value: string): value is Command {
-  return ['add', 'discover', 'help', 'init', 'remove', 'validate', 'version'].includes(value)
+	return [
+		"add",
+		"discover",
+		"help",
+		"init",
+		"remove",
+		"validate",
+		"version",
+	].includes(value)
 }
 
 async function commandInit(root: string, force: boolean): Promise<void> {
-  clack.intro('agents init')
-  const result = await initProject(root, { force })
-  clack.note(result.messages.join('\n'), 'Updated')
-  clack.outro('Project breadcrumb is ready.')
+	clack.intro("agents init")
+	const result = await initProject(root, { force })
+	clack.note(result.messages.join("\n"), "Updated")
+	clack.outro("Project breadcrumb is ready.")
 }
 
-async function commandDiscover(root: string, json: boolean): Promise<void> {
-  const documents = await discoverAgentDocuments(root)
-  if (json) {
-    console.log(JSON.stringify(documents, null, 2))
-    return
-  }
-
-  clack.intro('agents discover')
-  if (documents.length === 0) {
-    clack.outro('No supplemental AGENTS.md files found.')
-    return
-  }
-
-  clack.note(formatDocumentList(documents), `Found ${documents.length}`)
-  clack.outro('Use agents add <path> to enable one.')
+async function commandDiscover(
+	root: string,
+	options: { json: boolean; includeDotDirectories: boolean },
+): Promise<void> {
+	const documents = await discoverAgentDocuments(root, {
+		includeDotDirectories: options.includeDotDirectories,
+	})
+	if (options.json) {
+		console.log(JSON.stringify(documents, null, 2))
+		return
+	}
+
+	clack.intro("agents discover")
+	if (documents.length === 0) {
+		clack.outro("No supplemental AGENTS.md files found.")
+		return
+	}
+
+	clack.note(formatDocumentList(documents), `Found ${documents.length}`)
+	clack.outro("Use agents add <path> to enable one.")
 }
 
 async function commandAdd(root: string, paths: string[]): Promise<void> {
-  if (paths.length === 0) {
-    throw new Error('add requires at least one AGENTS.md path')
-  }
-
-  const documents = await Promise.all(
-    paths.map((path) =>
-      describeAgentDocument(root, formatProjectPath(root, resolveFromRoot(root, path))),
-    ),
-  )
-
-  const file = await addDocuments(root, documents)
-  clack.intro('agents add')
-  clack.note(formatDocumentList(file.documents), 'Promoted documents')
-  clack.outro(`Added ${documents.length} document${documents.length === 1 ? '' : 's'}.`)
+	if (paths.length === 0) {
+		throw new Error("add requires at least one AGENTS.md path")
+	}
+
+	const documents = await Promise.all(
+		paths.map((path) =>
+			describeAgentDocument(
+				root,
+				formatProjectPath(root, resolveFromRoot(root, path)),
+			),
+		),
+	)
+
+	const file = await addDocuments(root, documents)
+	clack.intro("agents add")
+	clack.note(formatDocumentList(file.documents), "Promoted documents")
+	clack.outro(
+		`Added ${documents.length} document${documents.length === 1 ? "" : "s"}.`,
+	)
 }
 
 function formatDocumentList(
-  documents: { path: string; description?: string | undefined }[],
+	documents: { path: string; description?: string | undefined }[],
 ): string {
-  return documents
-    .map((doc) => (doc.description ? `${doc.path}\n  ${doc.description}` : doc.path))
-    .join('\n')
+	return documents
+		.map((doc) =>
+			doc.description ? `${doc.path}\n  ${doc.description}` : doc.path,
+		)
+		.join("\n")
 }
 
 async function commandRemove(root: string, paths: string[]): Promise<void> {
-  if (paths.length === 0) {
-    throw new Error('remove requires at least one path')
-  }
-
-  const normalizedPaths = paths.map((path) => formatProjectPath(root, resolveFromRoot(root, path)))
-  const result = await removeDocuments(root, normalizedPaths)
-  clack.intro('agents remove')
-  clack.note(result.removed.join('\n') || 'No matching documents were listed.', 'Removed')
-  clack.outro(
-    `agents.yaml now has ${result.file.documents.length} promoted document${result.file.documents.length === 1 ? '' : 's'}.`,
-  )
+	if (paths.length === 0) {
+		throw new Error("remove requires at least one path")
+	}
+
+	const normalizedPaths = paths.map((path) =>
+		formatProjectPath(root, resolveFromRoot(root, path)),
+	)
+	const result = await removeDocuments(root, normalizedPaths)
+	clack.intro("agents remove")
+	clack.note(
+		result.removed.join("\n") || "No matching documents were listed.",
+		"Removed",
+	)
+	clack.outro(
+		`agents.yaml now has ${result.file.documents.length} promoted document${result.file.documents.length === 1 ? "" : "s"}.`,
+	)
 }
 
 async function commandValidate(root: string, json: boolean): Promise<void> {
-  const result = await validateAgentsFile(root)
-  if (json) {
-    console.log(JSON.stringify(result, null, 2))
-    return
-  }
-
-  clack.intro('agents validate')
-  if (result.errors.length > 0) {
-    clack.note(result.errors.join('\n'), 'Errors')
-  }
-  if (result.warnings.length > 0) {
-    clack.note(result.warnings.join('\n'), 'Warnings')
-  }
-
-  clack.outro(result.ok ? 'agents.yaml is valid.' : 'agents.yaml needs attention.')
-  if (!result.ok) {
-    process.exitCode = 1
-  }
+	const result = await validateAgentsFile(root)
+	if (json) {
+		console.log(JSON.stringify(result, null, 2))
+		return
+	}
+
+	clack.intro("agents validate")
+	if (result.errors.length > 0) {
+		clack.note(result.errors.join("\n"), "Errors")
+	}
+	if (result.warnings.length > 0) {
+		clack.note(result.warnings.join("\n"), "Warnings")
+	}
+
+	clack.outro(
+		result.ok ? "agents.yaml is valid." : "agents.yaml needs attention.",
+	)
+	if (!result.ok) {
+		process.exitCode = 1
+	}
 }
 
 async function interactive(root: string): Promise<void> {
-  clack.intro('agents')
-  const action = await clack.select({
-    message: 'What would you like to do?',
-    options: [
-      { value: 'discover', label: 'Discover and enable AGENTS.md files' },
-      { value: 'validate', label: 'Validate agents.yaml' },
-      { value: 'init', label: 'Initialize breadcrumb files' },
-    ],
-  })
-
-  if (clack.isCancel(action)) {
-    clack.cancel('Cancelled.')
-    return
-  }
-
-  if (action === 'init') {
-    await commandInit(root, false)
-    return
-  }
-
-  if (action === 'validate') {
-    await commandValidate(root, false)
-    return
-  }
-
-  const existing = await loadAgentsFile(root)
-  const discovered = await discoverAgentDocuments(root)
-  const candidates = discovered.filter(
-    (doc) => !existing.documents.some((active) => active.path === doc.path),
-  )
-
-  if (candidates.length === 0) {
-    clack.outro('No unlisted supplemental AGENTS.md files found.')
-    return
-  }
-
-  const selected = await clack.multiselect({
-    message: 'Choose documents to enable',
-    options: candidates.map((doc) => ({ value: doc.path, label: doc.path })),
-    required: false,
-  })
-
-  if (clack.isCancel(selected) || selected.length === 0) {
-    clack.cancel('No documents selected.')
-    return
-  }
-
-  await commandAdd(root, selected)
+	clack.intro("agents")
+	const action = await clack.select({
+		message: "What would you like to do?",
+		options: [
+			{ value: "discover", label: "Discover and enable AGENTS.md files" },
+			{ value: "validate", label: "Validate agents.yaml" },
+			{ value: "init", label: "Initialize breadcrumb files" },
+		],
+	})
+
+	if (clack.isCancel(action)) {
+		clack.cancel("Cancelled.")
+		return
+	}
+
+	if (action === "init") {
+		await commandInit(root, false)
+		return
+	}
+
+	if (action === "validate") {
+		await commandValidate(root, false)
+		return
+	}
+
+	const existing = await loadAgentsFile(root)
+	const discovered = await discoverAgentDocuments(root)
+	const candidates = discovered.filter(
+		(doc) => !existing.documents.some((active) => active.path === doc.path),
+	)
+
+	if (candidates.length === 0) {
+		clack.outro("No unlisted supplemental AGENTS.md files found.")
+		return
+	}
+
+	const selected = await chooseDocumentsToEnable(
+		candidates.map((doc) => ({ value: doc.path, label: doc.path })),
+	)
+
+	if (
+		clack.isCancel(selected) ||
+		selected === undefined ||
+		selected.length === 0
+	) {
+		clack.cancel("No documents selected.")
+		return
+	}
+
+	await commandAdd(root, selected)
+}
+
+function chooseDocumentsToEnable(
+	options: DocumentOption[],
+): Promise<string[] | symbol | undefined> {
+	return new MultiSelectPrompt<DocumentOption>({
+		options,
+		required: false,
+		render() {
+			const prefix = `${styleText("cyan", clack.S_BAR)}  `
+			const selected = this.value ?? []
+
+			return `${styleText("gray", clack.S_BAR)}
+${clack.symbol(this.state)}  Choose documents to enable
+${prefix}${clack.limitOptions({
+				options: this.options,
+				cursor: this.cursor,
+				columnPadding: prefix.length,
+				style: (option, active) =>
+					styleDocumentOption(option, {
+						active,
+						selected: selected.includes(option.value),
+					}),
+			}).join(`
+${prefix}`)}
+${styleText("cyan", clack.S_BAR_END)}
+`
+		},
+	}).prompt()
+}
+
+function styleDocumentOption(
+	option: DocumentOption,
+	state: { active: boolean; selected: boolean },
+): string {
+	if (option.disabled) {
+		return `${styleText("gray", clack.S_CHECKBOX_INACTIVE)} ${styleText(
+			["strikethrough", "gray"],
+			option.label,
+		)}`
+	}
+
+	const checkbox = state.selected
+		? styleText("green", clack.S_CHECKBOX_SELECTED)
+		: state.active
+			? styleText("cyan", clack.S_CHECKBOX_ACTIVE)
+			: styleText("dim", clack.S_CHECKBOX_INACTIVE)
+	const cursor = state.active ? styleText("cyan", ">") : " "
+	const label = state.active ? option.label : styleText("dim", option.label)
+	return `${cursor} ${checkbox} ${label}`
 }