feat: refactor folder structure (#4498)

* feat: wip refactor folder structure * Fix * fix position --------- Co-authored-by: Charles Bochet <charles@twenty.com>
2024-03-15 14:40:58 +01:00
parent 52f1b3ac98
commit 94487f6737
760 changed files with 3215 additions and 3155 deletions
--- a/packages/twenty-server/src/engine/workspace-manager/workspace-manager.service.ts
+++ b/packages/twenty-server/src/engine/workspace-manager/workspace-manager.service.ts
@ -0,0 +1,187 @@
+import { Injectable } from '@nestjs/common';
+
+import { DataSourceService } from 'src/engine-metadata/data-source/data-source.service';
+import { ObjectMetadataService } from 'src/engine-metadata/object-metadata/object-metadata.service';
+import { WorkspaceMigrationService } from 'src/engine-metadata/workspace-migration/workspace-migration.service';
+import { standardObjectsPrefillData } from 'src/engine/workspace-manager/standard-objects-prefill-data/standard-objects-prefill-data';
+import { demoObjectsPrefillData } from 'src/engine/workspace-manager/demo-objects-prefill-data/demo-objects-prefill-data';
+import { WorkspaceDataSourceService } from 'src/engine/workspace-datasource/workspace-datasource.service';
+import { DataSourceEntity } from 'src/engine-metadata/data-source/data-source.entity';
+import { WorkspaceSyncMetadataService } from 'src/engine/workspace-manager/workspace-sync-metadata/workspace-sync-metadata.service';
+
+@Injectable()
+export class WorkspaceManagerService {
+  constructor(
+    private readonly workspaceDataSourceService: WorkspaceDataSourceService,
+    private readonly workspaceMigrationService: WorkspaceMigrationService,
+    private readonly objectMetadataService: ObjectMetadataService,
+    private readonly dataSourceService: DataSourceService,
+    private readonly workspaceSyncMetadataService: WorkspaceSyncMetadataService,
+  ) {}
+
+  /**
+   * Init a workspace by creating a new data source and running all migrations
+   * @param workspaceId
+   * @returns Promise<void>
+   */
+  public async init(workspaceId: string): Promise<void> {
+    const schemaName =
+      await this.workspaceDataSourceService.createWorkspaceDBSchema(
+        workspaceId,
+      );
+
+    const dataSourceMetadata =
+      await this.dataSourceService.createDataSourceMetadata(
+        workspaceId,
+        schemaName,
+      );
+
+    await this.setWorkspaceMaxRow(workspaceId, schemaName);
+
+    await this.workspaceSyncMetadataService.synchronize({
+      workspaceId,
+      dataSourceId: dataSourceMetadata.id,
+    });
+
+    await this.prefillWorkspaceWithStandardObjects(
+      dataSourceMetadata,
+      workspaceId,
+    );
+  }
+
+  /**
+   * InitDemo a workspace by creating a new data source and running all migrations
+   * @param workspaceId
+   * @returns Promise<void>
+   */
+  public async initDemo(workspaceId: string): Promise<void> {
+    const schemaName =
+      await this.workspaceDataSourceService.createWorkspaceDBSchema(
+        workspaceId,
+      );
+
+    const dataSourceMetadata =
+      await this.dataSourceService.createDataSourceMetadata(
+        workspaceId,
+        schemaName,
+      );
+
+    await this.setWorkspaceMaxRow(workspaceId, schemaName);
+
+    await this.workspaceSyncMetadataService.synchronize({
+      workspaceId,
+      dataSourceId: dataSourceMetadata.id,
+    });
+
+    await this.prefillWorkspaceWithDemoObjects(dataSourceMetadata, workspaceId);
+  }
+
+  /**
+   *
+   * Check if the workspace schema has already been created or not
+   *
+   * @param workspaceId
+   * @Returns Promise<boolean>
+   */
+  public async doesDataSourceExist(workspaceId: string): Promise<boolean> {
+    const dataSource =
+      await this.dataSourceService.getDataSourcesMetadataFromWorkspaceId(
+        workspaceId,
+      );
+
+    return dataSource.length > 0;
+  }
+
+  /**
+   *
+   * We are updating the pg_graphql max_rows from 30 (default value) to 60
+   *
+   * @params workspaceId, schemaName
+   * @param workspaceId
+   */
+  private async setWorkspaceMaxRow(workspaceId, schemaName) {
+    const workspaceDataSource =
+      await this.workspaceDataSourceService.connectToWorkspaceDataSource(
+        workspaceId,
+      );
+
+    await workspaceDataSource.query(
+      `comment on schema ${schemaName} is e'@graphql({"max_rows": 60})'`,
+    );
+  }
+
+  /**
+   *
+   * We are prefilling a few standard objects with data to make it easier for the user to get started.
+   *
+   * @param dataSourceMetadata
+   * @param workspaceId
+   */
+  private async prefillWorkspaceWithStandardObjects(
+    dataSourceMetadata: DataSourceEntity,
+    workspaceId: string,
+  ) {
+    const workspaceDataSource =
+      await this.workspaceDataSourceService.connectToWorkspaceDataSource(
+        workspaceId,
+      );
+
+    if (!workspaceDataSource) {
+      throw new Error('Could not connect to workspace data source');
+    }
+
+    const createdObjectMetadata =
+      await this.objectMetadataService.findManyWithinWorkspace(workspaceId);
+
+    await standardObjectsPrefillData(
+      workspaceDataSource,
+      dataSourceMetadata.schema,
+      createdObjectMetadata,
+    );
+  }
+
+  /**
+   *
+   * We are prefilling a few demo objects with data to make it easier for the user to get started.
+   *
+   * @param dataSourceMetadata
+   * @param workspaceId
+   */
+  private async prefillWorkspaceWithDemoObjects(
+    dataSourceMetadata: DataSourceEntity,
+    workspaceId: string,
+  ) {
+    const workspaceDataSource =
+      await this.workspaceDataSourceService.connectToWorkspaceDataSource(
+        workspaceId,
+      );
+
+    if (!workspaceDataSource) {
+      throw new Error('Could not connect to workspace data source');
+    }
+
+    const createdObjectMetadata =
+      await this.objectMetadataService.findManyWithinWorkspace(workspaceId);
+
+    await demoObjectsPrefillData(
+      workspaceDataSource,
+      dataSourceMetadata.schema,
+      createdObjectMetadata,
+    );
+  }
+
+  /**
+   *
+   * Delete a workspace by deleting all metadata and the schema
+   *
+   * @param workspaceId
+   */
+  public async delete(workspaceId: string): Promise<void> {
+    // Delete data from metadata tables
+    await this.objectMetadataService.deleteObjectsMetadata(workspaceId);
+    await this.workspaceMigrationService.delete(workspaceId);
+    await this.dataSourceService.delete(workspaceId);
+    // Delete schema
+    await this.workspaceDataSourceService.deleteWorkspaceDBSchema(workspaceId);
+  }
+}