tachybase_todo/packages/core/server/src/app-manager.ts
ChengLei Shao 43e52843fa
fix: add schema (#1464)
* fix: add schema

* chore: format
2023-02-18 09:27:54 +08:00

97 lines
2.6 KiB
TypeScript

import { applyMixins, AsyncEmitter } from '@nocobase/utils';
import EventEmitter from 'events';
import http, { IncomingMessage, ServerResponse } from 'http';
import Application, { ApplicationOptions } from './application';
type AppSelector = (req: IncomingMessage) => Application | string | undefined | null;
export class AppManager extends EventEmitter {
public applications: Map<string, Application> = new Map<string, Application>();
constructor(public app: Application) {
super();
app.on('beforeStop', async (mainApp, options) => {
return await Promise.all(
[...this.applications.values()].map((application: Application) => application.stop(options)),
);
});
app.on('afterDestroy', async (mainApp, options) => {
return await Promise.all(
[...this.applications.values()].map((application: Application) => application.destroy(options)),
);
});
}
appSelector: AppSelector = (req: IncomingMessage) => this.app;
createApplication(name: string, options: ApplicationOptions): Application {
const application = new Application(options);
this.applications.set(name, application);
return application;
}
async removeApplication(name: string) {
const application = this.applications.get(name);
if (!application) {
return;
}
await application.destroy();
this.applications.delete(name);
}
setAppSelector(selector: AppSelector) {
this.appSelector = selector;
}
listen(...args) {
const server = http.createServer(this.callback());
return server.listen(...args);
}
async getApplication(appName: string, options = {}): Promise<null | Application> {
await this.emitAsync('beforeGetApplication', {
appManager: this,
name: appName,
options,
});
return this.applications.get(appName);
}
callback() {
return async (req: IncomingMessage, res: ServerResponse) => {
const appManager = this.app.appManager;
let handleApp: any = appManager.appSelector(req) || appManager.app;
if (typeof handleApp === 'string') {
handleApp = await appManager.getApplication(handleApp);
if (!handleApp) {
res.statusCode = 404;
return res.end(
JSON.stringify({
redirectTo: process.env.APP_NOT_FOUND_REDIRECT_TO,
errors: [
{
message: 'Application Not Found',
},
],
}),
);
}
}
handleApp.callback()(req, res);
};
}
declare emitAsync: (event: string | symbol, ...args: any[]) => Promise<boolean>;
}
applyMixins(AppManager, [AsyncEmitter]);