feat: server entry (#43)

Author: do4ng

packages/dobs/src/builder/index.ts

@@ -47,7 +47,11 @@ export async function buildServer(config: ResolvedServerConfig) {
   const [, configFileTemp] = await compileModule('dobs.config', {
     extensions: ['.js', '.mjs', '.cjs', '.ts', '.mts', '.cts'],
   });
+  const [, serverEntryTemp] = await compileModule(config.serverEntry, {
+    extensions: ['', '.js', '.mjs', '.cjs', '.ts', '.mts', '.cts'],
+  });
   const formattedConfigFile = JSON.stringify(configFileTemp).replace(/\\/g, '/');
+  const formattedServerEntry = JSON.stringify(serverEntryTemp).replace(/\\/g, '/');
 
   writeFileSync(
     join(tempDirectory, '_temp.js'),
@@ -57,10 +61,13 @@ const _internal = require("dobs/_build");
 const _dobs_http = _internal.createHTTPServer;
 
 const _config = _dobs.resolveConfig(${configFileTemp ? `require(${formattedConfigFile})` : '{}'} ?? {});
+const _server_entry = (${serverEntryTemp ? `require(${formattedServerEntry})` : '(() => {})'} ?? (() => {}));
 const _middlewares = _config?.middlewares;
 
 const _app = _dobs_http();
 
+_server_entry(_app);
+
 _app.use(...[..._middlewares, _internal._buildInternalMiddleware({${rawRoutes.join(',\n')}}, _config)]);
 
 _app.listen(process.env.PORT ?? _config.port, () => {console.log("server is running on " + (process.env.PORT ?? _config.port))});
@@ -74,7 +81,15 @@ _app.listen(process.env.PORT ?? _config.port, () => {console.log("server is runn
       file: outputFile,
       minify: true,
     },
-    plugins: [nodeExternal({ allow: [formattedConfigFile.slice(1, -1), ...rawFiles] })],
+    plugins: [
+      nodeExternal({
+        allow: [
+          formattedConfigFile.slice(1, -1),
+          formattedServerEntry.slice(1, -1),
+          ...rawFiles,
+        ],
+      }),
+    ],
   });
 
   writeFileSync(outputFilePackageJSON, JSON.stringify({ type: 'commonjs' }));

packages/dobs/src/config.ts

@@ -1,8 +1,10 @@
 import { createServer } from 'node:http';
-import type { Middleware } from '@dobsjs/http';
+import type { BaseServer, Middleware } from '@dobsjs/http';
 import { deepmerge } from 'deepmerge-ts';
 import { createPluginRunner, Plugin } from './plugin';
 
+export type ServerEntry = (server: BaseServer) => void;
+
 export interface ResolvedServerConfig {
   /** port to serve (default: 8080) */
   port: number;
@@ -30,6 +32,8 @@ export interface ResolvedServerConfig {
   };
 
   plugins: Plugin[];
+
+  serverEntry: string;
 }
 
 export type ServerConfig = Partial<ResolvedServerConfig>;
@@ -45,6 +49,7 @@ export const DEFAULT_CONFIG: ResolvedServerConfig = {
     directory: 'dist',
   },
   plugins: [],
+  serverEntry: 'server.entry',
 };
 
 export function resolveConfig(config: ServerConfig): ResolvedServerConfig {

packages/dobs/src/server/index.ts

@@ -5,6 +5,7 @@ import { resolveConfig, ServerConfig } from '~/dobs/config';
 import { createPluginRunner } from '~/dobs/plugin';
 
 import { createRouterMiddleware } from './router';
+import { loadServerEntry } from './server-entry';
 
 type CreateServerReturn<T extends ServerConfig> = T['mode'] extends 'middleware'
   ? Middleware[]
@@ -21,10 +22,13 @@ export async function createDobsServer<T extends ServerConfig>(
 
   const resolvedConfig = resolveConfig(config);
   const server = httpServer(resolvedConfig.createServer);
+  const serverEntry = await loadServerEntry(resolvedConfig);
 
   // [plugin] execute plugin.server
   await runner.execute('server', server);
 
+  // user server entry
+  if (serverEntry) serverEntry(server);
   // user middleware
   server.middlewares.push(...resolvedConfig.middlewares);
   // router middleware

packages/dobs/src/server/server-entry.ts

@@ -0,0 +1,16 @@
+import { load } from 'module-loader-ts';
+import { isAbsolute, join } from 'node:path';
+
+import type { ResolvedServerConfig, ServerEntry } from '~/dobs/config';
+
+export async function loadServerEntry(
+  config: ResolvedServerConfig,
+): Promise<ServerEntry> {
+  let entry = config.serverEntry;
+
+  if (!isAbsolute(entry)) entry = join(config.cwd, entry);
+
+  return await load(entry, {
+    extensions: ['', '.ts', '.js', '.mts', '.cts', '.mjs', '.cjs'],
+  });
+}