@rainfw/core 0.1.1 → 0.2.0

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) {
@@ -27,7 +34,11 @@ function extractStringProps(obj, sourceFile, keys) {
 
 function loadBuildConfig() {
   const configPath = path.join(PROJECT_ROOT, "rain.config.ts");
-  const defaults = { routesDir: "src/routes", outDir: ".rainjs" };
+  const defaults = {
+    routesDir: "src/routes",
+    outDir: ".rainjs",
+    frameworkPackage: "@rainfw/core",
+  };
   if (!fs.existsSync(configPath)) return defaults;
 
   const content = fs.readFileSync(configPath, "utf-8");
@@ -46,7 +57,11 @@ function loadBuildConfig() {
     if (!ts.isObjectLiteralExpression(obj)) return;
     Object.assign(
       config,
-      extractStringProps(obj, sourceFile, ["routesDir", "outDir"]),
+      extractStringProps(obj, sourceFile, [
+        "routesDir",
+        "outDir",
+        "frameworkPackage",
+      ]),
     );
   });
 
@@ -158,6 +173,104 @@ 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 routePathToDir(filePath) {
   return filePath
     .replace(/\\/g, "/")
@@ -533,6 +646,63 @@ function processPages(
   }
 }
 
+function validateCompatibilityFlags() {
+  const wranglerPath = path.join(PROJECT_ROOT, "wrangler.toml");
+  if (!fs.existsSync(wranglerPath)) return;
+
+  const content = fs.readFileSync(wranglerPath, "utf-8");
+  const flagsMatch = content.match(/compatibility_flags\s*=\s*\[([^\]]*)\]/);
+  const flags = flagsMatch
+    ? (flagsMatch[1].match(/"([^"]+)"/g) || []).map((s) => s.replace(/"/g, ""))
+    : [];
+
+  if (!(flags.includes("nodejs_compat") || flags.includes("nodejs_als"))) {
+    console.error(
+      "[Rain] Error: Missing required compatibility flag.\n" +
+        '  → Rain.js requires "nodejs_compat" in your wrangler.toml.\n' +
+        "  → Add the following to your wrangler.toml:\n\n" +
+        '    compatibility_flags = ["nodejs_compat"]\n\n' +
+        "  → Without this flag, the Workers runtime will fail with:\n" +
+        '    "No such module node:async_hooks"\n' +
+        "  → See: https://developers.cloudflare.com/workers/" +
+        "configuration/compatibility-flags/" +
+        "#nodejs-compatibility-flag",
+    );
+    process.exit(1);
+  }
+}
+
+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(
@@ -544,6 +714,8 @@ function generate() {
     process.exit(1);
   }
 
+  validateCompatibilityFlags();
+
   try {
     execSync("npx wrangler types", {
       cwd: PROJECT_ROOT,
@@ -592,12 +764,18 @@ 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 frameworkPath = relativeImportPath(
-    path.join(PROJECT_ROOT, "src", "framework"),
-  );
+  const fwPkg = BUILD_CONFIG.frameworkPackage;
+  const frameworkImport =
+    fwPkg.startsWith(".") || fwPkg.startsWith("/")
+      ? relativeImportPath(path.join(PROJECT_ROOT, fwPkg))
+      : fwPkg;
 
-  const headerImports = [`import { Rain } from "${frameworkPath}";`];
+  const headerImports = [`import { Rain } from "${frameworkImport}";`];
   if (hasConfig) {
     const configPath = relativeImportPath(
       path.join(PROJECT_ROOT, "rain.config"),
@@ -605,9 +783,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,
@@ -623,18 +799,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,
@@ -649,6 +829,8 @@ module.exports = {
   detectMiddlewareExportFromContent,
   detectDefaultExport,
   detectDefaultExportFromContent,
+  detectUseClientDirective,
+  bundleClientFilesSync,
   validateNoPageRouteColocation,
   ROUTES_DIR,
   ENTRY_FILE,