@standardagents/builder 0.11.6 → 0.11.8

package/dist/plugin.js

@@ -17,6 +17,74 @@ var TSCONFIG_CONTENT = `{
   "include": ["types.d.ts", "virtual-module.d.ts"]
 }
 `;
+function extractSchemaFields(content) {
+  const fields = [];
+  const schemaMatch = content.match(/requiredSchema:\s*z\.object\(\s*\{([^}]+)\}/s);
+  if (!schemaMatch) {
+    return fields;
+  }
+  const schemaContent = schemaMatch[1];
+  const fieldMatches = schemaContent.matchAll(/(?:['"]([^'"]+)['"]|(\w+))\s*:/g);
+  for (const match of fieldMatches) {
+    const fieldName = match[1] || match[2];
+    if (fieldName && !fieldName.startsWith("z.")) {
+      fields.push(fieldName);
+    }
+  }
+  return fields;
+}
+function scanPromptsWithSchemas(dir) {
+  const results = [];
+  if (!fs2.existsSync(dir)) {
+    return results;
+  }
+  try {
+    const entries = fs2.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.isFile() && entry.name.endsWith(".ts")) {
+        const filePath = path3.join(dir, entry.name);
+        const content = fs2.readFileSync(filePath, "utf-8");
+        const nameMatch = content.match(/name:\s*['"]([^'"]+)['"]/);
+        if (nameMatch) {
+          const name = nameMatch[1];
+          const schemaFields = extractSchemaFields(content);
+          results.push({ name, schemaFields });
+        }
+      }
+    }
+  } catch (error) {
+    console.error(`Error scanning prompts directory ${dir}:`, error);
+  }
+  return results;
+}
+function scanAgentsWithSideA(dir) {
+  const results = [];
+  if (!fs2.existsSync(dir)) {
+    return results;
+  }
+  try {
+    const entries = fs2.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.isFile() && entry.name.endsWith(".ts")) {
+        const filePath = path3.join(dir, entry.name);
+        const content = fs2.readFileSync(filePath, "utf-8");
+        const nameMatch = content.match(/name:\s*['"]([^'"]+)['"]/);
+        if (nameMatch) {
+          const name = nameMatch[1];
+          let sideAPrompt = null;
+          const sideAMatch = content.match(/sideA:\s*\{[^}]*prompt:\s*['"]([^'"]+)['"]/);
+          if (sideAMatch) {
+            sideAPrompt = sideAMatch[1];
+          }
+          results.push({ name, sideAPrompt });
+        }
+      }
+    }
+  } catch (error) {
+    console.error(`Error scanning agents directory ${dir}:`, error);
+  }
+  return results;
+}
 function scanForNames(dir, useFilename = false) {
   const names = [];
   if (!fs2.existsSync(dir)) {
@@ -52,11 +120,38 @@ function generateRegistryProperties(names) {
   }
   return names.map((n) => `'${n}': true;`).join("\n      ");
 }
+function generateSchemaRegistryProperties(promptResults, agentResults) {
+  const entries = [];
+  const promptSchemaMap = /* @__PURE__ */ new Map();
+  for (const prompt of promptResults) {
+    if (prompt.schemaFields.length > 0) {
+      promptSchemaMap.set(prompt.name, prompt.schemaFields);
+    }
+  }
+  for (const prompt of promptResults) {
+    if (prompt.schemaFields.length > 0) {
+      const fieldsUnion = prompt.schemaFields.map((f) => `'${f}'`).join(" | ");
+      entries.push(`'${prompt.name}': ${fieldsUnion};`);
+    }
+  }
+  for (const agent of agentResults) {
+    if (agent.sideAPrompt) {
+      const schemaFields = promptSchemaMap.get(agent.sideAPrompt);
+      if (schemaFields && schemaFields.length > 0) {
+        const fieldsUnion = schemaFields.map((f) => `'${f}'`).join(" | ");
+        entries.push(`'${agent.name}': ${fieldsUnion};`);
+      }
+    }
+  }
+  return entries.join("\n      ");
+}
 function generateTypesContent(config) {
   const models = scanForNames(config.modelsDir);
-  const prompts = scanForNames(config.promptsDir);
-  const agents = scanForNames(config.agentsDir);
+  const promptResults = scanPromptsWithSchemas(config.promptsDir);
+  const agentResults = scanAgentsWithSideA(config.agentsDir);
   const tools = scanForNames(config.toolsDir, true);
+  const prompts = promptResults.map((p) => p.name);
+  const agents = agentResults.map((a) => a.name);
   const callables = [...prompts, ...agents, ...tools];
   return `// Auto-generated by @standardagents/builder - DO NOT EDIT
 // Generated at: ${(/* @__PURE__ */ new Date()).toISOString()}
@@ -97,6 +192,11 @@ declare global {
     interface CallableRegistry {
       ${generateRegistryProperties(callables)}
     }
+
+    /** Schema field names for prompts and agents (used for initUserMessageProperty autocomplete) */
+    interface PromptSchemaRegistry {
+      ${generateSchemaRegistryProperties(promptResults, agentResults)}
+    }
   }
 }