mirror of
https://github.com/toeverything/AFFiNE.git
synced 2025-01-03 22:52:43 +03:00
638fc62601
--- <details open="true"><summary>Generated summary (powered by <a href="https://app.graphite.dev">Graphite</a>)</summary> > ## TL;DR > This pull request adds a new migration file, a new model, and new modules related to runtime settings. It also introduces a new `Runtime` service that allows getting, setting, and updating runtime configurations. > > ## What changed > - Added a new migration file `migration.sql` that creates a table called `application_settings` with columns `key` and `value`. > - Added a new model `ApplicationSetting` with properties `key` and `value`. > - Added a new module `RuntimeSettingModule` that exports the `Runtime` service. > - Added a new service `Runtime` that provides methods for getting, setting, and updating runtime configurations. > - Modified the `app.module.ts` file to import the `RuntimeSettingModule`. > - Modified the `index.ts` file in the `fundamentals` directory to export the `Runtime` service. > - Added a new file `def.ts` in the `runtime` directory that defines the runtime configurations and provides a default implementation. > - Added a new file `service.ts` in the `runtime` directory that implements the `Runtime` service. > > ## How to test > 1. Run the migration script to create the `application_settings` table. > 2. Use the `Runtime` service to get, set, and update runtime configurations. > 3. Verify that the runtime configurations are stored correctly in the database and can be retrieved and modified using the `Runtime` service. > > ## Why make this change > This change introduces a new feature related to runtime settings. The `Runtime` service allows the application to dynamically manage and modify runtime configurations without requiring a restart. This provides flexibility and allows for easier customization and configuration of the application. </details>
148 lines
3.5 KiB
TypeScript
148 lines
3.5 KiB
TypeScript
import { INestApplication, ModuleMetadata } from '@nestjs/common';
|
|
import { APP_GUARD } from '@nestjs/core';
|
|
import { Query, Resolver } from '@nestjs/graphql';
|
|
import { Test, TestingModuleBuilder } from '@nestjs/testing';
|
|
import { PrismaClient } from '@prisma/client';
|
|
import cookieParser from 'cookie-parser';
|
|
import graphqlUploadExpress from 'graphql-upload/graphqlUploadExpress.mjs';
|
|
import type { Response } from 'supertest';
|
|
|
|
import { AppModule, FunctionalityModules } from '../../src/app.module';
|
|
import { AuthGuard, AuthModule } from '../../src/core/auth';
|
|
import { UserFeaturesInit1698652531198 } from '../../src/data/migrations/1698652531198-user-features-init';
|
|
import { GqlModule } from '../../src/fundamentals/graphql';
|
|
|
|
async function flushDB(client: PrismaClient) {
|
|
const result: { tablename: string }[] =
|
|
await client.$queryRaw`SELECT tablename
|
|
FROM pg_catalog.pg_tables
|
|
WHERE schemaname != 'pg_catalog'
|
|
AND schemaname != 'information_schema'`;
|
|
|
|
// remove all table data
|
|
await client.$executeRawUnsafe(
|
|
`TRUNCATE TABLE ${result
|
|
.map(({ tablename }) => tablename)
|
|
.filter(name => !name.includes('migrations'))
|
|
.join(', ')}`
|
|
);
|
|
}
|
|
|
|
async function initFeatureConfigs(db: PrismaClient) {
|
|
await UserFeaturesInit1698652531198.up(db);
|
|
}
|
|
|
|
export async function initTestingDB(db: PrismaClient) {
|
|
await flushDB(db);
|
|
await initFeatureConfigs(db);
|
|
}
|
|
|
|
interface TestingModuleMeatdata extends ModuleMetadata {
|
|
tapModule?(m: TestingModuleBuilder): void;
|
|
tapApp?(app: INestApplication): void;
|
|
}
|
|
|
|
function dedupeModules(modules: NonNullable<ModuleMetadata['imports']>) {
|
|
const map = new Map();
|
|
|
|
modules.forEach(m => {
|
|
if ('module' in m) {
|
|
map.set(m.module, m);
|
|
} else {
|
|
map.set(m, m);
|
|
}
|
|
});
|
|
|
|
return Array.from(map.values());
|
|
}
|
|
|
|
@Resolver(() => String)
|
|
class MockResolver {
|
|
@Query(() => String)
|
|
hello() {
|
|
return 'hello world';
|
|
}
|
|
}
|
|
|
|
export async function createTestingModule(
|
|
moduleDef: TestingModuleMeatdata = {}
|
|
) {
|
|
// setting up
|
|
let imports = moduleDef.imports ?? [];
|
|
imports =
|
|
imports[0] === AppModule
|
|
? [AppModule]
|
|
: dedupeModules([
|
|
...FunctionalityModules,
|
|
AuthModule,
|
|
GqlModule,
|
|
...imports,
|
|
]);
|
|
|
|
const builder = Test.createTestingModule({
|
|
imports,
|
|
providers: [
|
|
{
|
|
provide: APP_GUARD,
|
|
useClass: AuthGuard,
|
|
},
|
|
MockResolver,
|
|
...(moduleDef.providers ?? []),
|
|
],
|
|
controllers: moduleDef.controllers,
|
|
});
|
|
|
|
if (moduleDef.tapModule) {
|
|
moduleDef.tapModule(builder);
|
|
}
|
|
|
|
const m = await builder.compile();
|
|
|
|
const prisma = m.get(PrismaClient);
|
|
if (prisma instanceof PrismaClient) {
|
|
await initTestingDB(prisma);
|
|
}
|
|
|
|
return m;
|
|
}
|
|
|
|
export async function createTestingApp(moduleDef: TestingModuleMeatdata = {}) {
|
|
const m = await createTestingModule(moduleDef);
|
|
|
|
const app = m.createNestApplication({
|
|
cors: true,
|
|
bodyParser: true,
|
|
rawBody: true,
|
|
logger: ['warn'],
|
|
});
|
|
|
|
app.use(
|
|
graphqlUploadExpress({
|
|
maxFileSize: 10 * 1024 * 1024,
|
|
maxFiles: 5,
|
|
})
|
|
);
|
|
|
|
app.use(cookieParser());
|
|
|
|
if (moduleDef.tapApp) {
|
|
moduleDef.tapApp(app);
|
|
}
|
|
|
|
await app.init();
|
|
|
|
return {
|
|
module: m,
|
|
app,
|
|
};
|
|
}
|
|
|
|
export function handleGraphQLError(resp: Response) {
|
|
const { errors } = resp.body;
|
|
if (errors) {
|
|
const cause = errors[0];
|
|
const stacktrace = cause.extensions?.stacktrace;
|
|
throw new Error(stacktrace ? stacktrace.join('\n') : cause.message, cause);
|
|
}
|
|
}
|