@rainfw/core 0.1.2 → 0.2.1

package/scripts/generate.js

@@ -3,6 +3,13 @@ const path = require("node:path");
 const { execSync } = require("node:child_process");
 const ts = require("typescript");
 
+let esbuild;
+try {
+  esbuild = require("esbuild");
+} catch (_esbuildOptional) {
+  esbuild = null;
+}
+
 const PROJECT_ROOT = process.cwd();
 
 function unwrapExpression(node) {
@@ -166,6 +173,111 @@ function middlewarePathToDir(filePath) {
     .replace(/\/$/, "");
 }
 
+function detectUseClientDirective(content) {
+  const sourceFile = ts.createSourceFile(
+    "file.tsx",
+    content,
+    ts.ScriptTarget.Latest,
+    true,
+  );
+  const firstStatement = sourceFile.statements[0];
+  if (!firstStatement) return false;
+  return (
+    ts.isExpressionStatement(firstStatement) &&
+    ts.isStringLiteral(firstStatement.expression) &&
+    firstStatement.expression.text === "use client"
+  );
+}
+
+function getClientFiles(dir, base = "") {
+  if (!fs.existsSync(dir)) return [];
+  const entries = fs.readdirSync(dir, { withFileTypes: true });
+  const files = [];
+
+  for (const entry of entries) {
+    const fullPath = path.join(dir, entry.name);
+    const relativePath = path.join(base, entry.name);
+
+    if (entry.isDirectory()) {
+      if (entry.name === "node_modules" || entry.name === ".rainjs") continue;
+      files.push(...getClientFiles(fullPath, relativePath));
+    } else if (entry.name.endsWith(".ts") || entry.name.endsWith(".tsx")) {
+      const content = fs.readFileSync(fullPath, "utf-8");
+      if (detectUseClientDirective(content)) {
+        files.push(relativePath);
+      }
+    }
+  }
+
+  return files;
+}
+
+function bundleClientFilesSync(clientFiles, srcDir) {
+  if (clientFiles.length === 0) return [];
+  if (!esbuild) {
+    console.warn(
+      "[Rain] Warning: esbuild not found.\n" +
+        "  → Install esbuild to enable client bundling: npm install -D esbuild\n" +
+        "  → Client-side components will not be bundled.",
+    );
+    return [];
+  }
+
+  const outDir = path.join(PROJECT_ROOT, "public", "_rain");
+  if (!fs.existsSync(outDir)) {
+    fs.mkdirSync(outDir, { recursive: true });
+  }
+
+  for (const file of fs.readdirSync(outDir)) {
+    if (file.startsWith("island-")) {
+      fs.unlinkSync(path.join(outDir, file));
+    }
+  }
+
+  const entryPoints = clientFiles.map((f) => path.join(srcDir, f));
+
+  const result = esbuild.buildSync({
+    entryPoints,
+    outdir: outDir,
+    bundle: true,
+    minify: true,
+    format: "esm",
+    metafile: true,
+    entryNames: "island-[hash]",
+    write: true,
+    treeShaking: true,
+    platform: "browser",
+    target: ["es2022"],
+    jsx: "automatic",
+    jsxImportSource: "@rainfw/core",
+    alias: {
+      "@rainfw/core/jsx-runtime": path.resolve(
+        PROJECT_ROOT,
+        "src/framework/client/jsx-runtime.ts",
+      ),
+    },
+    loader: { ".ts": "ts", ".tsx": "tsx" },
+  });
+
+  const publicDir = path.join(PROJECT_ROOT, "public");
+  const scripts = [];
+  for (const [outPath, meta] of Object.entries(result.metafile.outputs)) {
+    if (meta.entryPoint) {
+      const relPath = path.relative(publicDir, outPath);
+      scripts.push(`/${relPath.replace(/\\/g, "/")}`);
+    }
+  }
+
+  return scripts;
+}
+
+function stripRouteGroupSegments(filePath) {
+  return filePath
+    .split("/")
+    .filter((segment) => !/^\(.+\)$/.test(segment))
+    .join("/");
+}
+
 function routePathToDir(filePath) {
   return filePath
     .replace(/\\/g, "/")
@@ -180,6 +292,7 @@ function middlewareImportName(filePath) {
     .replace(/\//g, "_")
     .replace(/\[/g, "$")
     .replace(/\]/g, "")
+    .replace(/[()]/g, "")
     .replace(/-/g, "_")
     .replace(/_+$/, "");
   return `mw_${base || "root"}`;
@@ -202,6 +315,7 @@ function layoutPathToDir(filePath) {
 function pageFilePathToUrlPath(filePath) {
   let urlPath = filePath.replace(/\.tsx?$/, "");
   urlPath = urlPath.replace(/\\/g, "/");
+  urlPath = stripRouteGroupSegments(urlPath);
   urlPath = urlPath.replace(/\[([^\]]+)\]/g, ":$1");
   urlPath = urlPath.replace(/\/page$/, "");
   if (urlPath === "page") urlPath = "";
@@ -217,6 +331,7 @@ function pageFilePathToImportName(filePath) {
       .replace(/\//g, "_")
       .replace(/\[/g, "$")
       .replace(/\]/g, "")
+      .replace(/[()]/g, "")
       .replace(/-/g, "_")
   );
 }
@@ -228,6 +343,7 @@ function layoutImportName(filePath) {
     .replace(/\//g, "_")
     .replace(/\[/g, "$")
     .replace(/\]/g, "")
+    .replace(/[()]/g, "")
     .replace(/-/g, "_")
     .replace(/_+$/, "");
   return `layout_${base || "root"}`;
@@ -297,6 +413,63 @@ function validateNoPageRouteColocation(routeFiles, pageFiles) {
     }
   }
 
+  const routeUrlMap = new Map();
+  for (const f of routeFiles) {
+    routeUrlMap.set(filePathToUrlPath(f), f);
+  }
+  for (const pageFile of pageFiles) {
+    const url = pageFilePathToUrlPath(pageFile);
+    const conflicting = routeUrlMap.get(url);
+    if (conflicting) {
+      const pageDir = pageFilePathToDir(pageFile);
+      if (!routeDirs.has(pageDir)) {
+        errors.push(
+          `[Rain] Error: page "${pageFile}" and route "${conflicting}" resolve to the same URL path "${url}":\n` +
+            "  → This conflict occurs because route group folders are stripped from URLs.\n" +
+            "  → Move one of them to a different URL path.",
+        );
+      }
+    }
+  }
+
+  return errors;
+}
+
+function validateNoDuplicateUrls(routeFiles, pageFiles) {
+  const errors = [];
+
+  const routeUrlMap = new Map();
+  for (const f of routeFiles) {
+    const url = filePathToUrlPath(f);
+    if (routeUrlMap.has(url)) {
+      errors.push(
+        `[Rain] Error: multiple route files resolve to the same URL path "${url}":\n` +
+          `  → ${routeUrlMap.get(url)}\n` +
+          `  → ${f}\n` +
+          "  → Route group folders are stripped from URLs.\n" +
+          "  → Rename one of the routes to avoid the conflict.",
+      );
+    } else {
+      routeUrlMap.set(url, f);
+    }
+  }
+
+  const pageUrlMap = new Map();
+  for (const f of pageFiles) {
+    const url = pageFilePathToUrlPath(f);
+    if (pageUrlMap.has(url)) {
+      errors.push(
+        `[Rain] Error: multiple page files resolve to the same URL path "${url}":\n` +
+          `  → ${pageUrlMap.get(url)}\n` +
+          `  → ${f}\n` +
+          "  → Route group folders are stripped from URLs.\n" +
+          "  → Rename one of the pages to avoid the conflict.",
+      );
+    } else {
+      pageUrlMap.set(url, f);
+    }
+  }
+
   return errors;
 }
 
@@ -324,6 +497,7 @@ function getMiddlewaresForRoute(routeFile, middlewareFiles) {
 function filePathToUrlPath(filePath) {
   let urlPath = filePath.replace(/\.tsx?$/, "");
   urlPath = urlPath.replace(/\\/g, "/");
+  urlPath = stripRouteGroupSegments(urlPath);
   urlPath = urlPath.replace(/\[([^\]]+)\]/g, ":$1");
   urlPath = urlPath.replace(/\/route$/, "");
   if (urlPath === "route") urlPath = "";
@@ -339,6 +513,7 @@ function filePathToImportName(filePath) {
       .replace(/\//g, "_")
       .replace(/\[/g, "$")
       .replace(/\]/g, "")
+      .replace(/[()]/g, "")
       .replace(/-/g, "_")
   );
 }
@@ -546,13 +721,9 @@ function validateCompatibilityFlags() {
   if (!fs.existsSync(wranglerPath)) return;
 
   const content = fs.readFileSync(wranglerPath, "utf-8");
-  const flagsMatch = content.match(
-    /compatibility_flags\s*=\s*\[([^\]]*)\]/,
-  );
+  const flagsMatch = content.match(/compatibility_flags\s*=\s*\[([^\]]*)\]/);
   const flags = flagsMatch
-    ? (flagsMatch[1].match(/"([^"]+)"/g) || []).map((s) =>
-        s.replace(/"/g, ""),
-      )
+    ? (flagsMatch[1].match(/"([^"]+)"/g) || []).map((s) => s.replace(/"/g, ""))
     : [];
 
   if (!(flags.includes("nodejs_compat") || flags.includes("nodejs_als"))) {
@@ -571,6 +742,37 @@ function validateCompatibilityFlags() {
   }
 }
 
+function buildAppInitLine(clientScripts, hasConfig) {
+  if (hasConfig) {
+    return clientScripts.length > 0
+      ? `const app = new Rain({ ...config, clientScripts: ${JSON.stringify(clientScripts)} });`
+      : "const app = new Rain(config);";
+  }
+  return clientScripts.length > 0
+    ? `const app = new Rain({ clientScripts: ${JSON.stringify(clientScripts)} });`
+    : "const app = new Rain();";
+}
+
+function regenerateClient() {
+  const srcDir = path.join(PROJECT_ROOT, "src");
+  const clientFiles = getClientFiles(srcDir);
+  const clientScripts = bundleClientFilesSync(clientFiles, srcDir);
+
+  if (!fs.existsSync(ENTRY_FILE)) return;
+
+  const content = fs.readFileSync(ENTRY_FILE, "utf-8");
+  const hasConfig = fs.existsSync(CONFIG_FILE);
+  const appInit = buildAppInitLine(clientScripts, hasConfig);
+  const updated = content.replace(/^const app = new Rain\(.*\);$/m, appInit);
+  if (updated !== content) {
+    fs.writeFileSync(ENTRY_FILE, updated);
+  }
+
+  const clientMsg =
+    clientFiles.length > 0 ? `${clientFiles.length} client` : "0 client";
+  console.log(`[gen:client] ${clientMsg} -> .rainjs/entry.ts`);
+}
+
 function generate() {
   if (!fs.existsSync(ROUTES_DIR)) {
     console.error(
@@ -618,6 +820,12 @@ function generate() {
     process.exit(1);
   }
 
+  const duplicateErrors = validateNoDuplicateUrls(files, pageFiles);
+  for (const err of duplicateErrors) {
+    console.error(err);
+    process.exit(1);
+  }
+
   const hasRootLayout = layoutFiles.some((f) => layoutPathToDir(f) === "");
 
   processMiddlewares(middlewareFiles, imports);
@@ -632,6 +840,10 @@ function generate() {
     registrations,
   );
 
+  const srcDir = path.join(PROJECT_ROOT, "src");
+  const clientFiles = getClientFiles(srcDir);
+  const clientScripts = bundleClientFilesSync(clientFiles, srcDir);
+
   const hasConfig = fs.existsSync(CONFIG_FILE);
   const fwPkg = BUILD_CONFIG.frameworkPackage;
   const frameworkImport =
@@ -647,9 +859,7 @@ function generate() {
     headerImports.push(`import config from "${configPath}";`);
   }
 
-  const appInit = hasConfig
-    ? "const app = new Rain(config);"
-    : "const app = new Rain();";
+  const appInit = buildAppInitLine(clientScripts, hasConfig);
 
   const content = [
     ...headerImports,
@@ -665,18 +875,22 @@ function generate() {
 
   fs.writeFileSync(ENTRY_FILE, content);
   const total = files.length + pageFiles.length;
+  const clientMsg =
+    clientFiles.length > 0 ? `, ${clientFiles.length} client` : "";
   console.log(
-    `[gen] ${total} route(s) (${files.length} api, ${pageFiles.length} page, ${layoutFiles.length} layout) -> .rainjs/entry.ts`,
+    `[gen] ${total} route(s) (${files.length} api, ${pageFiles.length} page, ${layoutFiles.length} layout${clientMsg}) -> .rainjs/entry.ts`,
   );
 }
 
 module.exports = {
   generate,
+  regenerateClient,
   loadBuildConfig,
   getRouteFiles,
   getMiddlewareFiles,
   getPageFiles,
   getLayoutFiles,
+  getClientFiles,
   getMiddlewaresForRoute,
   getLayoutsForPage,
   filePathToUrlPath,
@@ -691,7 +905,11 @@ module.exports = {
   detectMiddlewareExportFromContent,
   detectDefaultExport,
   detectDefaultExportFromContent,
+  detectUseClientDirective,
+  bundleClientFilesSync,
   validateNoPageRouteColocation,
+  validateNoDuplicateUrls,
+  stripRouteGroupSegments,
   ROUTES_DIR,
   ENTRY_FILE,
   HTTP_METHODS,