Add backend framework selection between hono, elysiajs

apps/cli/src/helpers/runtime-setup.ts

@@ -1,11 +1,12 @@
 import path from "node:path";
 import fs from "fs-extra";
-import type { Runtime } from "../types";
+import type { BackendFramework, Runtime } from "../types";
 import { addPackageDependency } from "../utils/add-package-deps";
 
 export async function setupRuntime(
 	projectDir: string,
 	runtime: Runtime,
+	backendFramework: BackendFramework,
 ): Promise<void> {
 	const serverDir = path.join(projectDir, "apps/server");
 	const serverIndexPath = path.join(serverDir, "src/index.ts");
@@ -13,9 +14,19 @@ export async function setupRuntime(
 	const indexContent = await fs.readFile(serverIndexPath, "utf-8");
 
 	if (runtime === "bun") {
-		await setupBunRuntime(serverDir, serverIndexPath, indexContent);
+		await setupBunRuntime(
+			serverDir,
+			serverIndexPath,
+			indexContent,
+			backendFramework,
+		);
 	} else if (runtime === "node") {
-		await setupNodeRuntime(serverDir, serverIndexPath, indexContent);
+		await setupNodeRuntime(
+			serverDir,
+			serverIndexPath,
+			indexContent,
+			backendFramework,
+		);
 	}
 }
 
@@ -23,6 +34,7 @@ async function setupBunRuntime(
 	serverDir: string,
 	serverIndexPath: string,
 	indexContent: string,
+	backendFramework: BackendFramework,
 ): Promise<void> {
 	const packageJsonPath = path.join(serverDir, "package.json");
 	const packageJson = await fs.readJson(packageJsonPath);
@@ -40,7 +52,7 @@ async function setupBunRuntime(
 		projectDir: serverDir,
 	});
 
-	if (!indexContent.includes("export default app")) {
+	if (backendFramework === "hono") {
 		const updatedContent = `${indexContent}\n\nexport default app;\n`;
 		await fs.writeFile(serverIndexPath, updatedContent);
 	}
@@ -50,13 +62,8 @@ async function setupNodeRuntime(
 	serverDir: string,
 	serverIndexPath: string,
 	indexContent: string,
+	backendFramework: BackendFramework,
 ): Promise<void> {
-	addPackageDependency({
-		dependencies: ["@hono/node-server"],
-		devDependencies: ["tsx", "@types/node"],
-		projectDir: serverDir,
-	});
-
 	const packageJsonPath = path.join(serverDir, "package.json");
 	const packageJson = await fs.readJson(packageJsonPath);
 
@@ -68,27 +75,62 @@ async function setupNodeRuntime(
 
 	await fs.writeJson(packageJsonPath, packageJson, { spaces: 2 });
 
-	const importLine = 'import { serve } from "@hono/node-server";\n';
-	const serverCode = `
+	addPackageDependency({
+		devDependencies: ["tsx", "@types/node"],
+		projectDir: serverDir,
+	});
+
+	if (backendFramework === "hono") {
+		addPackageDependency({
+			dependencies: ["@hono/node-server"],
+			projectDir: serverDir,
+		});
+
+		const importLine = 'import { serve } from "@hono/node-server";\n';
+		const serverCode = `
 serve(
-  {
-    fetch: app.fetch,
-    port: 3000,
-  },
-  (info) => {
-    console.log(\`Server is running on http://localhost:\${info.port}\`);
-  },
+		{
+				fetch: app.fetch,
+				port: 3000,
+		},
+		(info) => {
+				console.log(\`Server is running on http://localhost:\${info.port}\`);
+		},
 );\n`;
 
-	if (!indexContent.includes("@hono/node-server")) {
-		const importEndIndex = indexContent.lastIndexOf("import");
-		const importSection = indexContent.substring(0, importEndIndex);
-		const restOfFile = indexContent.substring(importEndIndex);
+		if (!indexContent.includes("@hono/node-server")) {
+			const importEndIndex = indexContent.lastIndexOf("import");
+			const importSection = indexContent.substring(0, importEndIndex);
+			const restOfFile = indexContent.substring(importEndIndex);
 
-		const updatedContent = importSection + importLine + restOfFile + serverCode;
-		await fs.writeFile(serverIndexPath, updatedContent);
-	} else if (!indexContent.includes("serve(")) {
-		const updatedContent = indexContent + serverCode;
-		await fs.writeFile(serverIndexPath, updatedContent);
+			const updatedContent =
+				importSection + importLine + restOfFile + serverCode;
+			await fs.writeFile(serverIndexPath, updatedContent);
+		}
+	} else if (backendFramework === "elysia") {
+		addPackageDependency({
+			dependencies: ["@elysiajs/node"],
+			projectDir: serverDir,
+		});
+
+		if (!indexContent.includes("@elysiajs/node")) {
+			const nodeImport = 'import { node } from "@elysiajs/node";\n';
+
+			const firstImportEnd = indexContent.indexOf(
+				"\n",
+				indexContent.indexOf("import"),
+			);
+			const before = indexContent.substring(0, firstImportEnd + 1);
+			const after = indexContent.substring(firstImportEnd + 1);
+
+			let updatedContent = before + nodeImport + after;
+
+			updatedContent = updatedContent.replace(
+				/const app = new Elysia\([^)]*\)/,
+				"const app = new Elysia({ adapter: node() })",
+			);
+
+			await fs.writeFile(serverIndexPath, updatedContent);
+		}
 	}
 }