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/modules/auth/services/google-apis.service.ts
+++ b/packages/twenty-server/src/engine/modules/auth/services/google-apis.service.ts
@ -0,0 +1,140 @@
+import { ConflictException, Inject, Injectable } from '@nestjs/common';
+import { InjectRepository } from '@nestjs/typeorm';
+
+import { v4 } from 'uuid';
+import { Repository } from 'typeorm';
+
+import { DataSourceService } from 'src/engine-metadata/data-source/data-source.service';
+import { TypeORMService } from 'src/database/typeorm/typeorm.service';
+import { SaveConnectedAccountInput } from 'src/engine/modules/auth/dto/save-connected-account';
+import { MessageQueue } from 'src/integrations/message-queue/message-queue.constants';
+import { MessageQueueService } from 'src/integrations/message-queue/services/message-queue.service';
+import {
+  GmailFullSyncJob,
+  GmailFullSyncJobData,
+} from 'src/business/modules/message/jobs/gmail-full-sync.job';
+import {
+  GoogleCalendarFullSyncJob,
+  GoogleCalendarFullSyncJobData,
+} from 'src/business/modules/calendar/jobs/google-calendar-full-sync.job';
+import { EnvironmentService } from 'src/integrations/environment/environment.service';
+import {
+  FeatureFlagEntity,
+  FeatureFlagKeys,
+} from 'src/engine/modules/feature-flag/feature-flag.entity';
+
+@Injectable()
+export class GoogleAPIsService {
+  constructor(
+    private readonly dataSourceService: DataSourceService,
+    private readonly typeORMService: TypeORMService,
+    @Inject(MessageQueue.messagingQueue)
+    private readonly messageQueueService: MessageQueueService,
+    @Inject(MessageQueue.calendarQueue)
+    private readonly calendarQueueService: MessageQueueService,
+    private readonly environmentService: EnvironmentService,
+    @InjectRepository(FeatureFlagEntity, 'core')
+    private readonly featureFlagRepository: Repository<FeatureFlagEntity>,
+  ) {}
+
+  providerName = 'google';
+
+  async saveConnectedAccount(
+    saveConnectedAccountInput: SaveConnectedAccountInput,
+  ) {
+    const {
+      handle,
+      workspaceId,
+      accessToken,
+      refreshToken,
+      workspaceMemberId,
+    } = saveConnectedAccountInput;
+
+    const dataSourceMetadata =
+      await this.dataSourceService.getLastDataSourceMetadataFromWorkspaceIdOrFail(
+        workspaceId,
+      );
+
+    const workspaceDataSource =
+      await this.typeORMService.connectToDataSource(dataSourceMetadata);
+
+    const connectedAccount = await workspaceDataSource?.query(
+      `SELECT * FROM ${dataSourceMetadata.schema}."connectedAccount" WHERE "handle" = $1 AND "provider" = $2 AND "accountOwnerId" = $3`,
+      [handle, this.providerName, workspaceMemberId],
+    );
+
+    if (connectedAccount.length > 0) {
+      throw new ConflictException('Connected account already exists');
+    }
+
+    const connectedAccountId = v4();
+
+    const IsCalendarEnabled = await this.featureFlagRepository.findOneBy({
+      workspaceId,
+      key: FeatureFlagKeys.IsCalendarEnabled,
+      value: true,
+    });
+
+    await workspaceDataSource?.transaction(async (manager) => {
+      await manager.query(
+        `INSERT INTO ${dataSourceMetadata.schema}."connectedAccount" ("id", "handle", "provider", "accessToken", "refreshToken", "accountOwnerId") VALUES ($1, $2, $3, $4, $5, $6)`,
+        [
+          connectedAccountId,
+          handle,
+          this.providerName,
+          accessToken,
+          refreshToken,
+          workspaceMemberId,
+        ],
+      );
+
+      if (this.environmentService.get('MESSAGING_PROVIDER_GMAIL_ENABLED')) {
+        await manager.query(
+          `INSERT INTO ${dataSourceMetadata.schema}."messageChannel" ("visibility", "handle", "connectedAccountId", "type") VALUES ($1, $2, $3, $4)`,
+          ['share_everything', handle, connectedAccountId, 'email'],
+        );
+      }
+
+      if (
+        this.environmentService.get('CALENDAR_PROVIDER_GOOGLE_ENABLED') &&
+        IsCalendarEnabled
+      ) {
+        await manager.query(
+          `INSERT INTO ${dataSourceMetadata.schema}."calendarChannel" ("visibility", "handle", "connectedAccountId") VALUES ($1, $2, $3)`,
+          ['SHARE_EVERYTHING', handle, connectedAccountId],
+        );
+      }
+    });
+
+    if (this.environmentService.get('MESSAGING_PROVIDER_GMAIL_ENABLED')) {
+      await this.messageQueueService.add<GmailFullSyncJobData>(
+        GmailFullSyncJob.name,
+        {
+          workspaceId,
+          connectedAccountId,
+        },
+        {
+          retryLimit: 2,
+        },
+      );
+    }
+
+    if (
+      this.environmentService.get('CALENDAR_PROVIDER_GOOGLE_ENABLED') &&
+      IsCalendarEnabled
+    ) {
+      await this.calendarQueueService.add<GoogleCalendarFullSyncJobData>(
+        GoogleCalendarFullSyncJob.name,
+        {
+          workspaceId,
+          connectedAccountId,
+        },
+        {
+          retryLimit: 2,
+        },
+      );
+    }
+
+    return;
+  }
+}