Refactor connected account module (#6225)

- Refactor connected account module
- Move blocklist into it's own module
- Move contact-creation-manager into it's own module

---------

Co-authored-by: Charles Bochet <charles@twenty.com>

packages/twenty-server/src/modules/connected-account/refresh-access-token-manager/drivers/google/google-api-refresh-access-token.module.ts

@@ -0,0 +1,18 @@
+import { Module } from '@nestjs/common';
+
+import { ObjectMetadataRepositoryModule } from 'src/engine/object-metadata-repository/object-metadata-repository.module';
+import { GoogleAPIRefreshAccessTokenService } from 'src/modules/connected-account/refresh-access-token-manager/drivers/google/services/google-api-refresh-access-token.service';
+import { ConnectedAccountWorkspaceEntity } from 'src/modules/connected-account/standard-objects/connected-account.workspace-entity';
+import { MessagingCommonModule } from 'src/modules/messaging/common/messaging-common.module';
+
+@Module({
+  imports: [
+    ObjectMetadataRepositoryModule.forFeature([
+      ConnectedAccountWorkspaceEntity,
+    ]),
+    MessagingCommonModule,
+  ],
+  providers: [GoogleAPIRefreshAccessTokenService],
+  exports: [GoogleAPIRefreshAccessTokenService],
+})
+export class GoogleAPIRefreshAccessTokenModule {}

packages/twenty-server/src/modules/connected-account/refresh-access-token-manager/drivers/google/services/google-api-refresh-access-token.service.ts

@@ -0,0 +1,29 @@
+import { Injectable } from '@nestjs/common';
+
+import axios from 'axios';
+
+import { EnvironmentService } from 'src/engine/integrations/environment/environment.service';
+
+@Injectable()
+export class GoogleAPIRefreshAccessTokenService {
+  constructor(private readonly environmentService: EnvironmentService) {}
+
+  async refreshAccessToken(refreshToken: string): Promise<string> {
+    const response = await axios.post(
+      'https://oauth2.googleapis.com/token',
+      {
+        client_id: this.environmentService.get('AUTH_GOOGLE_CLIENT_ID'),
+        client_secret: this.environmentService.get('AUTH_GOOGLE_CLIENT_SECRET'),
+        refresh_token: refreshToken,
+        grant_type: 'refresh_token',
+      },
+      {
+        headers: {
+          'Content-Type': 'application/json',
+        },
+      },
+    );
+
+    return response.data.access_token;
+  }
+}

packages/twenty-server/src/modules/connected-account/refresh-access-token-manager/refresh-access-token-manager.module.ts

@@ -0,0 +1,11 @@
+import { Module } from '@nestjs/common';
+
+import { GoogleAPIRefreshAccessTokenModule } from 'src/modules/connected-account/refresh-access-token-manager/drivers/google/google-api-refresh-access-token.module';
+import { RefreshAccessTokenService } from 'src/modules/connected-account/refresh-access-token-manager/services/refresh-access-token.service';
+
+@Module({
+  imports: [GoogleAPIRefreshAccessTokenModule],
+  providers: [RefreshAccessTokenService],
+  exports: [RefreshAccessTokenService],
+})
+export class RefreshAccessTokenManagerModule {}

packages/twenty-server/src/modules/connected-account/refresh-access-token-manager/services/refresh-access-token.service.ts

@@ -0,0 +1,62 @@
+import { Injectable } from '@nestjs/common';
+
+import { InjectObjectMetadataRepository } from 'src/engine/object-metadata-repository/object-metadata-repository.decorator';
+import { GoogleAPIRefreshAccessTokenService } from 'src/modules/connected-account/refresh-access-token-manager/drivers/google/services/google-api-refresh-access-token.service';
+import { ConnectedAccountRepository } from 'src/modules/connected-account/repositories/connected-account.repository';
+import { ConnectedAccountWorkspaceEntity } from 'src/modules/connected-account/standard-objects/connected-account.workspace-entity';
+
+@Injectable()
+export class RefreshAccessTokenService {
+  constructor(
+    private readonly googleAPIRefreshAccessTokenService: GoogleAPIRefreshAccessTokenService,
+    @InjectObjectMetadataRepository(ConnectedAccountWorkspaceEntity)
+    private readonly connectedAccountRepository: ConnectedAccountRepository,
+  ) {}
+
+  async refreshAndSaveAccessToken(
+    connectedAccount: ConnectedAccountWorkspaceEntity,
+    workspaceId: string,
+  ): Promise<string> {
+    const refreshToken = connectedAccount.refreshToken;
+
+    if (!refreshToken) {
+      throw new Error(
+        `No refresh token found for connected account ${connectedAccount.id} in workspace ${workspaceId}`,
+      );
+    }
+    const accessToken = await this.refreshAccessToken(
+      connectedAccount,
+      refreshToken,
+    );
+
+    await this.connectedAccountRepository.updateAccessToken(
+      accessToken,
+      connectedAccount.id,
+      workspaceId,
+    );
+
+    await this.connectedAccountRepository.updateAccessToken(
+      accessToken,
+      connectedAccount.id,
+      workspaceId,
+    );
+
+    return accessToken;
+  }
+
+  async refreshAccessToken(
+    connectedAccount: ConnectedAccountWorkspaceEntity,
+    refreshToken: string,
+  ): Promise<string> {
+    switch (connectedAccount.provider) {
+      case 'google':
+        return this.googleAPIRefreshAccessTokenService.refreshAccessToken(
+          refreshToken,
+        );
+      default:
+        throw new Error(
+          `Provider ${connectedAccount.provider} is not supported.`,
+        );
+    }
+  }
+}