@vite-hub/workflow 0.0.2 → 0.0.3

package/dist/vite.js

@@ -2,8 +2,8 @@ import { t as normalizeWorkflowOptions } from "./config-BDqB6I4E.js";
 import { r as getCloudflareWorkflowClassName, t as createCloudflareWorkflowBindings } from "./cloudflare-Cds1VIWa.js";
 import { builtinModules, createRequire } from "node:module";
 import { cp, mkdir, readFile, readdir, rm, rmdir, stat, symlink, writeFile } from "node:fs/promises";
-import { basename, dirname, isAbsolute, join, relative, resolve } from "node:path";
-import { existsSync, readFileSync, readdirSync, realpathSync, statSync } from "node:fs";
+import { basename, dirname, isAbsolute, join, relative, resolve, sep } from "node:path";
+import { existsSync, lstatSync, readFileSync, readdirSync, realpathSync, statSync } from "node:fs";
 import { basename as basename$1, dirname as dirname$1, normalize, relative as relative$1, resolve as resolve$1 } from "pathe";
 import { fileURLToPath } from "node:url";
 import { build } from "esbuild";
@@ -124,6 +124,7 @@ function deleteOwnedFields(value, ownership) {
 	return next;
 }
 function getKeyedArrayEntryValue(value, key) {
+	if (key === void 0) return value === null || typeof value === "boolean" || typeof value === "number" || typeof value === "string" ? value : void 0;
 	if (!isProviderJsonRecord(value)) return;
 	const entryValue = value[key];
 	return entryValue === null || typeof entryValue === "boolean" || typeof entryValue === "number" || typeof entryValue === "string" ? entryValue : void 0;
@@ -228,6 +229,7 @@ async function writeCloudflareWranglerConfig(options) {
 //#region ../internal/src/build/vite.ts
 const projectRootDirectoryMarkers = [
 	["server", "agents"],
+	["server", "browsers"],
 	["server", "schedules"],
 	["server", "templates"],
 	["server", "workspaces"]
@@ -257,12 +259,12 @@ function includesNitroVitePlugin(value) {
 	if (!value || typeof value !== "object") return false;
 	return "name" in value && value.name === "nitro:main";
 }
-function hasNitroVitePlugin(config) {
-	return includesNitroVitePlugin(config.plugins);
+function hasNitroConfigContext(config) {
+	return config["__vitehubNitroConfigContext"] === true || includesNitroVitePlugin(config.plugins);
 }
 function resolveNitroVercelFunctionName(config, product, env = process.env) {
 	const preset = config.nitro?.preset || env.NITRO_PRESET || env.SERVER_PRESET;
-	const nitro = Boolean(preset) || hasNitroVitePlugin(config);
+	const nitro = Boolean(preset) || hasNitroConfigContext(config);
 	const clientOutDir = config.environments?.client?.build?.outDir;
 	const vercel = preset ? preset.startsWith("vercel") : env.VITEHUB_HOSTING === "vercel" || Boolean(env.VERCEL) || Boolean(clientOutDir && /(^|[/\\])\.vercel[/\\]output([/\\]|$)/.test(clientOutDir));
 	return nitro && vercel ? `__${product}.func` : void 0;
@@ -625,6 +627,23 @@ async function writeCloudflareDeploymentOutput(options) {
 	}
 	await Promise.all(writes);
 }
+async function copyVercelClientOutput(rootDir, clientDir, staticOutputDir) {
+	const cloudflareOutputRoot = createDefaultCloudflareOutputRoot(rootDir);
+	const outputRelativePath = relative(clientDir, cloudflareOutputRoot);
+	const excludesCloudflareOutput = outputRelativePath && outputRelativePath !== ".." && !outputRelativePath.startsWith(`..${sep}`) && !isAbsolute(outputRelativePath);
+	await copyClientOutput(clientDir, staticOutputDir);
+	if (resolve(clientDir) === resolve(staticOutputDir) || !excludesCloudflareOutput) return;
+	try {
+		await readFile(resolve(cloudflareOutputRoot, "wrangler.json"));
+	} catch (error) {
+		if (error.code === "ENOENT") return;
+		throw error;
+	}
+	await rm(resolve(staticOutputDir, outputRelativePath), {
+		force: true,
+		recursive: true
+	});
+}
 async function writeVercelDeploymentOutput(options) {
 	const { clientDir, staticIndex } = resolveClientOutput(options.rootDir, options.clientOutDir);
 	const outputRoot = options.outputRoot ?? createDefaultVercelOutputRoot(options.rootDir);
@@ -646,7 +665,7 @@ async function writeVercelDeploymentOutput(options) {
 		}),
 		writeFile(resolve(serverDir, ".vc-config.json"), `${stringifyProviderOutputConfig(options.functionConfig ?? createNodeFunctionConfig())}\n`, "utf8"),
 		writeProviderOutputConfig(resolve(outputRoot, "config.json"), config, { keys: options.configKeys }),
-		staticIndex ? copyClientOutput(clientDir, options.staticOutputDir ?? resolve(outputRoot, "static")) : Promise.resolve()
+		staticIndex ? copyVercelClientOutput(options.rootDir, clientDir, options.staticOutputDir ?? resolve(outputRoot, "static")) : Promise.resolve()
 	]);
 }
 async function writeNetlifyDeploymentOutput(options) {
@@ -749,6 +768,60 @@ async function writeProviderDeploymentOutputs(options) {
 	}
 }
 //#endregion
+//#region ../internal/src/build/colocated-agent-files.ts
+function formatSize(bytes) {
+	const mebibyte = 1024 * 1024;
+	return bytes % mebibyte === 0 ? `${bytes / mebibyte} MiB` : `${bytes} bytes`;
+}
+function isFolderAgentEntry(file) {
+	if (!/^(?:agent|index)\.(?:c|m)?[jt]s$/i.test(basename(file))) return false;
+	return !(basename(file).toLowerCase().startsWith("agent.") && basename(dirname(file)) === "agents" && basename(dirname(dirname(file))) === "server");
+}
+function readColocatedAgentFiles(handler, directoryName, options = {}) {
+	if (!isFolderAgentEntry(handler)) return;
+	const root = join(dirname(handler), directoryName);
+	if (!existsSync(root)) return;
+	if (!(options.rejectUnsupportedEntries ? lstatSync(root) : statSync(root)).isDirectory()) {
+		if (options.rejectUnsupportedEntries) throw new Error(`[vitehub] ${options.label || "Colocated Agent files"} supports regular files and directories only: ${directoryName}`);
+		return;
+	}
+	const files = {};
+	let fileCount = 0;
+	let totalSize = 0;
+	const visit = (directory) => {
+		for (const entry of readdirSync(directory, { withFileTypes: true }).sort((left, right) => left.name.localeCompare(right.name))) {
+			const file = join(directory, entry.name);
+			const target = relative(root, file).replace(/\\/g, "/");
+			if (entry.isDirectory()) {
+				visit(file);
+				continue;
+			}
+			if (!entry.isFile()) {
+				if (options.rejectUnsupportedEntries) throw new Error(`[vitehub] ${options.label || "Colocated Agent files"} supports regular files and directories only: ${target}`);
+				continue;
+			}
+			const size = statSync(file).size;
+			fileCount += 1;
+			if (options.fileCountLimit !== void 0 && fileCount > options.fileCountLimit) throw new Error(`[vitehub] ${options.label || "Colocated Agent files"} exceeds ${options.fileCountLimit} files.`);
+			if (options.fileSizeLimit !== void 0 && size > options.fileSizeLimit) throw new Error(`[vitehub] ${options.label || "Colocated Agent file"} file exceeds ${formatSize(options.fileSizeLimit)}: ${target}`);
+			totalSize += size;
+			if (options.totalSizeLimit !== void 0 && totalSize > options.totalSizeLimit) throw new Error(`[vitehub] ${options.label || "Colocated Agent files"} exceeds ${formatSize(options.totalSizeLimit)}.`);
+			const content = readFileSync(file);
+			if (content.byteLength !== size) {
+				if (options.fileSizeLimit !== void 0 && content.byteLength > options.fileSizeLimit) throw new Error(`[vitehub] ${options.label || "Colocated Agent file"} file exceeds ${formatSize(options.fileSizeLimit)}: ${target}`);
+				totalSize += content.byteLength - size;
+				if (options.totalSizeLimit !== void 0 && totalSize > options.totalSizeLimit) throw new Error(`[vitehub] ${options.label || "Colocated Agent files"} exceeds ${formatSize(options.totalSizeLimit)}.`);
+			}
+			files[target] = {
+				content: content.toString("base64"),
+				encoding: "base64"
+			};
+		}
+	};
+	visit(root);
+	return Object.keys(files).length ? files : void 0;
+}
+//#endregion
 //#region ../internal/src/build/paths.ts
 function generatedDirSegments(productName) {
 	return [".vitehub", productName];
@@ -1467,7 +1540,7 @@ function discoverFlatServerWorkflowDefinitions(scanDirs, source) {
 function discoverWorkflowDefinitions(options) {
 	if (options.mode === "server-workflows") return mergeDefinitions("workflow", discoverFlatServerWorkflowDefinitions(options.scanDirs, "server-workflows"), discoverWorkflowFolders(options.scanDirs, "server-workflows"));
 	const roots = resolveDefinitionScanRoots(options.rootDir, options.scanDirs);
-	const serverScanDirs = roots.map((root) => resolve$1(root, "server"));
+	const serverScanDirs = options.serverDirs ?? roots.map((root) => resolve$1(root, "server"));
 	const folderDefinitions = discoverWorkflowFolders(serverScanDirs, "server-workflows");
 	return mergeDefinitions("workflow", discoverDefinitions("workflow", [createSuffixDefinitionSource("vite-suffix", roots, workflowSuffixPattern, normalizeSuffixWorkflowName, { createDefinition: ({ file, name }) => ({
 		handler: file,
@@ -1512,7 +1585,6 @@ const WORKFLOW_PRIORITY_NAMES = [
 	"server-workflow.js",
 	"server-workflow.mjs"
 ];
-const folderAgentEntryPattern = /^(?:agent|index)\.(?:c|m)?[jt]s$/i;
 async function loadVercelWorkflowBuilders() {
 	try {
 		const require = createRequire(import.meta.url);
@@ -1621,10 +1693,6 @@ async function buildVercelNativeWorkflowOutput(rootDir, definitions, aliases = {
 		routes: [...workflowConfig.routes ?? [], ...viteHubConfig.routes ?? []]
 	}, null, 2)}\n`, "utf8");
 }
-function isFolderAgentEntry(file) {
-	if (!folderAgentEntryPattern.test(basename(file))) return false;
-	return !(basename(file).toLowerCase().startsWith("agent.") && basename(dirname(file)) === "agents" && basename(dirname(dirname(file))) === "server");
-}
 function resolveWorkflowUserAppEntry(rootDir) {
 	return resolveUserAppEntry(rootDir, { names: getViteMode() === VITEHUB_MODES.workflow ? [...WORKFLOW_PRIORITY_NAMES, ...WORKFLOW_ENTRY_BASE_NAMES] : [...WORKFLOW_ENTRY_BASE_NAMES] });
 }
@@ -1680,29 +1748,31 @@ function readAgentInstructions(file) {
 	return existsSync(instructions) && statSync(instructions).isFile() ? resolveInstructionFile(instructions, /* @__PURE__ */ new Set()) : void 0;
 }
 function readAgentSkills(file) {
-	if (!isFolderAgentEntry(file)) return void 0;
-	const root = dirname(file);
-	const skillsRoot = join(root, "skills");
-	if (!existsSync(skillsRoot) || !statSync(skillsRoot).isDirectory()) return void 0;
-	const sources = {};
-	const visit = (directory) => {
-		for (const entry of readdirSync(directory, { withFileTypes: true }).sort((left, right) => left.name.localeCompare(right.name))) {
-			const target = join(directory, entry.name);
-			if (entry.isDirectory()) visit(target);
-			else if (entry.isFile()) {
-				const workspacePath = relative(root, target).replace(/\\/g, "/");
-				sources[`__vitehubAgentSkill:${workspacePath}`] = {
-					content: readFileSync(target).toString("base64"),
-					encoding: "base64",
-					materialize: "build",
-					mount: "",
-					workspacePath
-				};
-			}
-		}
-	};
-	visit(skillsRoot);
-	return Object.keys(sources).length ? sources : void 0;
+	const files = readColocatedAgentFiles(file, "skills");
+	if (!files) return;
+	return Object.fromEntries(Object.entries(files).map(([path, source]) => {
+		const workspacePath = `skills/${path}`;
+		return [`__vitehubAgentSkill:${workspacePath}`, {
+			...source,
+			materialize: "build",
+			mount: "",
+			workspacePath
+		}];
+	}));
+}
+function readAgentHome(file) {
+	const files = readColocatedAgentFiles(file, "home", {
+		fileCountLimit: 1024,
+		fileSizeLimit: 1024 * 1024,
+		label: "Colocated Agent Home",
+		rejectUnsupportedEntries: true,
+		totalSizeLimit: 4 * 1024 * 1024
+	});
+	if (!files) return;
+	return Object.fromEntries(Object.entries(files).map(([target, source]) => [target, {
+		contents: source.content,
+		encoding: source.encoding
+	}]));
 }
 function renderAgentWorkflowRegistryEntry(registryFile, definition) {
 	return [
@@ -1710,7 +1780,7 @@ function renderAgentWorkflowRegistryEntry(registryFile, definition) {
 		`    const cached = registryEntryCache.get(${JSON.stringify(definition.name)})`,
 		"    if (cached) return cached",
 		`    const loaded = await ${renderRegistryImport(registryFile, definition.handler)}`,
-		`    const agent = agentWithColocatedSkills(workspaceAgentWithSourceRoot("default" in loaded ? loaded.default : loaded, ${JSON.stringify(resolveAgentWorkspaceSourceRoot(definition.handler))}, ${JSON.stringify(readAgentInstructions(definition.handler))}), ${JSON.stringify(readAgentSkills(definition.handler))})`,
+		`    const agent = agentWithColocatedHome(agentWithColocatedSkills(workspaceAgentWithSourceRoot("default" in loaded ? loaded.default : loaded, ${JSON.stringify(resolveAgentWorkspaceSourceRoot(definition.handler))}, ${JSON.stringify(readAgentInstructions(definition.handler))}), ${JSON.stringify(readAgentSkills(definition.handler))}), ${JSON.stringify(readAgentHome(definition.handler))})`,
 		`    const entry = { handler: async (context) => await runAgentWorkflowDefinition(agent, { ...context, payload: { ...context.payload, agentIdentity: context.payload?.agentIdentity || { name: ${JSON.stringify(definition.agentIdentity || definition.name)} } } }, runAgentInline) }`,
 		`    registryEntryCache.set(${JSON.stringify(definition.name)}, entry)`,
 		"    return entry",
@@ -1759,7 +1829,7 @@ function createWorkflowRegistryContents(registryFile, definitions, importBases =
 	const imports = [
 		...needsAgentRuntime ? [
 			`import { runAgentInline } from ${JSON.stringify(agentImportBase)}`,
-			`import { agentWithColocatedSkills, runAgentWorkflowDefinition, workspaceAgentWithSourceRoot } from ${JSON.stringify(`${agentImportBase}/runtime/workflow`)}`,
+			`import { agentWithColocatedHome, agentWithColocatedSkills, runAgentWorkflowDefinition, workspaceAgentWithSourceRoot } from ${JSON.stringify(`${agentImportBase}/runtime/workflow`)}`,
 			...installAgentWorkflowRuntime ? [`import { setAgentWorkflowRuntimeLoaders } from ${JSON.stringify(`${agentImportBase}/server/internal`)}`] : []
 		] : [],
 		...needsWorkflowRuntime ? [`import { createWorkflowSteps } from ${JSON.stringify(`${workflowImportBase}/runtime/execute`)}`, `import { takeInlineWorkflowDefinitionForModule } from ${JSON.stringify(`${workflowImportBase}/runtime/state`)}`] : [],
@@ -1848,11 +1918,14 @@ function renderProviderEntry(spec, entryFile, userAppEntry, serializedWorkflowCo
 		""
 	].filter(Boolean).join("\n");
 }
-async function writeProviderEntries(rootDir, workflow, importBases = {}) {
+async function writeProviderEntries(rootDir, workflow, importBases = {}, serverDirs) {
 	const generatedDir = ensureGeneratedDir(rootDir, productName);
 	await mkdir(generatedDir, { recursive: true });
 	const registryFile = resolve(generatedDir, generatedRegistryFileName);
-	const definitions = discoverWorkflowDefinitions({ rootDir });
+	const definitions = discoverWorkflowDefinitions({
+		rootDir,
+		serverDirs
+	});
 	const providerDefinitions = definitions;
 	const userAppEntry = resolveWorkflowUserAppEntry(rootDir);
 	const cloudflareWorkflowConfig = resolveWorkflowConfig(workflow, "cloudflare");
@@ -1969,7 +2042,7 @@ async function generateProviderOutputs(options) {
 		workflow: options.importBase,
 		workspace: options.workspaceImportBase,
 		workspaceDependencies: options.workspaceDependencyRuntimeImports
-	});
+	}, options.serverDirs);
 	const cloudflareWorkflowConfig = resolveWorkflowConfig(options.workflow, "cloudflare");
 	const vercelWorkflowConfig = resolveWorkflowConfig(options.workflow, "vercel");
 	const cloudflareOutput = cloudflareWorkflowConfig && cloudflareWorkflowConfig.provider === "cloudflare" ? createCloudflareOutput(options.rootDir, artifacts, options.providerImportAliases) : void 0;
@@ -2001,10 +2074,12 @@ function hubWorkflow(options) {
 	const internalOptions = options;
 	let resolved;
 	let workflow = options;
+	let serverDirs;
 	return {
 		name: "@vite-hub/workflow/vite",
 		config(config) {
 			workflow = config.workflow ?? workflow;
+			serverDirs = config["__vitehubServerDirs"] ?? serverDirs;
 		},
 		configResolved(config) {
 			resolved = config;
@@ -2025,6 +2100,7 @@ function hubWorkflow(options) {
 					...internalOptions?.providerImportAliases
 				},
 				rootDir: resolved.root,
+				serverDirs,
 				serverFunctionName: resolveNitroVercelFunctionName(resolved, "workflow"),
 				workflow,
 				workspaceDependencyRuntimeImports: internalOptions?.workspaceDependencyRuntimeImports,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vite-hub/workflow",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "Workflow primitives and Vite integration for ViteHub.",
   "keywords": [
     "cloudflare",
@@ -46,7 +46,7 @@
     "esbuild": "^0.28.1",
     "h3": "2.0.1-rc.25",
     "pathe": "^2.0.3",
-    "@vite-hub/runtime": "0.0.2"
+    "@vite-hub/runtime": "0.0.3"
   },
   "devDependencies": {
     "@types/node": "^24.13.3",
@@ -59,11 +59,11 @@
     "@vite-hub/internal": "0.0.1"
   },
   "peerDependencies": {
-    "@workflow/builders": "4.0.5",
+    "@workflow/builders": "5.0.0-beta.35",
     "openworkflow": "^0.9.0",
     "vite": ">=8.0.0",
-    "workflow": "^4.6.0",
-    "@vite-hub/database": "0.0.2"
+    "workflow": "5.0.0-beta.35",
+    "@vite-hub/database": "0.0.3"
   },
   "peerDependenciesMeta": {
     "@vite-hub/database": {