@pajecawav/yamf 0.0.3 → 0.0.5

package/src/virtual.d.ts

@@ -11,3 +11,10 @@ declare module "virtual:yamf:pages" {
 declare module "virtual:yamf:template" {
 	export const template: string;
 }
+
+declare module "virtual:yamf:root" {
+	const Root: import("hono/jsx").FC | null;
+	const rootAssets: import("./shared/assets").ImportAssetsResult | null;
+
+	export { Root, rootAssets };
+}

package/src/vite/index.ts

@@ -5,6 +5,7 @@ import type { EnvironmentOptions, PluginOption } from "vite";
 import { islands } from "./islands";
 import { virtualAssets } from "./virtual-assets";
 import { virtualPages } from "./virtual-pages";
+import { virtualRoot } from "./virtual-root";
 import { virtualTemplate } from "./virtual-template";
 
 export interface YamfOptions {
@@ -61,6 +62,7 @@ const yamf = (options?: YamfOptions): PluginOption[] => {
 	plugins.push(virtualAssets());
 	plugins.push(virtualPages());
 	plugins.push(virtualTemplate());
+	plugins.push(virtualRoot());
 
 	plugins.push(
 		nitro({

package/src/vite/virtual-assets.ts

@@ -20,7 +20,7 @@ export const virtualAssets = (): Plugin => {
 			}
 
 			return js`
-import clientAssets from "./src/client/index.ts?assets=client";
+import clientAssets from "./src/client?assets=client";
 
 export { clientAssets };
 `.trim();

package/src/vite/virtual-pages.ts

@@ -20,11 +20,12 @@ export const virtualPages = (): Plugin => {
 			}
 
 			return js`
-const pages = import.meta.glob("/src/pages/**/*.{js,mjs,cjs,ts,mts,cts,tsx,jsx}", {
+const pages = import.meta.glob("/src/pages/**/*.page.{js,mjs,cjs,ts,mts,cts,tsx,jsx}", {
 	import: "default",
 });
+
 const assets = import.meta.glob(
-	"/src/pages/**/*.{js,mjs,cjs,ts,mts,cts,tsx,jsx}",
+	"/src/pages/**/*.page.{js,mjs,cjs,ts,mts,cts,tsx,jsx}",
 	{
 		import: "default",
 		query: "?assets=ssr",

package/src/vite/virtual-root.ts

@@ -0,0 +1,48 @@
+import { resolveModulePath } from "exsolve";
+import type { Plugin } from "vite";
+import { js } from "../shared/utils";
+
+const ROOT_PATH = "./src/root";
+
+export const virtualRoot = (): Plugin => {
+	const virtualModuleId = "virtual:yamf:root";
+	const resolvedVirtualModuleId = "\0" + virtualModuleId;
+
+	return {
+		name: "yamf:virtual-root",
+		resolveId(id) {
+			if (id === virtualModuleId) {
+				return resolvedVirtualModuleId;
+			}
+
+			return undefined;
+		},
+		load(id) {
+			if (id !== resolvedVirtualModuleId) {
+				return;
+			}
+
+			const root = resolveModulePath(ROOT_PATH, {
+				try: true,
+				suffixes: ["", "/index"],
+				extensions: [".tsx", ".jsx", ".ts", ".js", ".mjs", ".cjs", ".mts", ".cts"],
+			});
+
+			if (!root) {
+				return js`
+const Root = null;
+const rootAssets = null;
+
+export { Root, rootAssets };
+        `;
+			}
+
+			return js`
+import Root from "${ROOT_PATH}";
+import rootAssets from "${ROOT_PATH}?assets=ssr";
+
+export { Root, rootAssets };
+`.trim();
+		},
+	};
+};