twenty/server/src/tenant/metadata/metadata.controller.ts

import { Controller, Get, UseGuards } from '@nestjs/common';

import { Workspace } from '@prisma/client';
import { EntitySchema } from 'typeorm';

import { AuthWorkspace } from 'src/decorators/auth-workspace.decorator';
import { JwtAuthGuard } from 'src/guards/jwt.auth.guard';

import { DataSourceMetadataService } from './data-source-metadata/data-source-metadata.service';
import { EntitySchemaGeneratorService } from './entity-schema-generator/entity-schema-generator.service';
import { DataSourceService } from './data-source/data-source.service';
import { MigrationGeneratorService } from './migration-generator/migration-generator.service';

@UseGuards(JwtAuthGuard)
@Controller('metadata')
export class MetadataController {
  constructor(
    private readonly entitySchemaGeneratorService: EntitySchemaGeneratorService,
    private readonly dataSourceMetadataService: DataSourceMetadataService,
    private readonly dataSourceService: DataSourceService,
    private readonly migrationGenerator: MigrationGeneratorService,
  ) {}

  @Get()
  async getMetadata(@AuthWorkspace() workspace: Workspace) {
    const dataSourcesMetadata =
      await this.dataSourceMetadataService.getDataSourcesMetadataFromWorkspaceId(
        workspace.id,
      );

    const entities: EntitySchema<{
      id: unknown;
    }>[] = [];

    for (const dataSource of dataSourcesMetadata) {
      const dataSourceEntities =
        await this.entitySchemaGeneratorService.getTypeORMEntitiesByDataSourceId(
          dataSource.id,
        );

      entities.push(...dataSourceEntities);
    }

    this.dataSourceService.createWorkspaceSchema(workspace.id);

    await this.migrationGenerator.executeMigrationFromPendingMigrations(
      workspace.id,
    );

    this.dataSourceService.connectToWorkspaceDataSource(workspace.id);

    return entities;
  }
}