add sync customer command and drop subscription customer constraint

packages/twenty-server/src/database/typeorm/core/migrations/billing/1734532875877-dropSubscriptionCustomerConstraint.ts

@@ -0,0 +1,19 @@
+import { MigrationInterface, QueryRunner } from 'typeorm';
+
+export class DropSubscriptionCustomerConstraint1734532875877
+  implements MigrationInterface
+{
+  name = 'DropSubscriptionCustomerConstraint1734532875877';
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "core"."billingSubscription" DROP CONSTRAINT "FK_9120b7586c3471463480b58d20a"`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "core"."billingSubscription" ADD CONSTRAINT "FK_9120b7586c3471463480b58d20a" FOREIGN KEY ("stripeCustomerId") REFERENCES "core"."billingCustomer"("stripeCustomerId") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+  }
+}

packages/twenty-server/src/engine/core-modules/billing/billing.module.ts

@@ -3,6 +3,7 @@ import { TypeOrmModule } from '@nestjs/typeorm';
 
 import { BillingController } from 'src/engine/core-modules/billing/billing.controller';
 import { BillingResolver } from 'src/engine/core-modules/billing/billing.resolver';
+import { BillingSyncCustomerDataCommand } from 'src/engine/core-modules/billing/commands/billing-sync-customer-data.command';
 import { BillingCustomer } from 'src/engine/core-modules/billing/entities/billing-customer.entity';
 import { BillingEntitlement } from 'src/engine/core-modules/billing/entities/billing-entitlement.entity';
 import { BillingMeter } from 'src/engine/core-modules/billing/entities/billing-meter.entity';
@@ -59,6 +60,7 @@ import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
     BillingWebhookProductService,
     BillingWebhookPriceService,
     BillingRestApiExceptionFilter,
+    BillingSyncCustomerDataCommand,
   ],
   exports: [
     BillingSubscriptionService,

packages/twenty-server/src/engine/core-modules/billing/commands/billing-sync-customer-data.command.ts

@@ -0,0 +1,97 @@
+import { InjectRepository } from '@nestjs/typeorm';
+
+import chalk from 'chalk';
+import { Command } from 'nest-commander';
+import { Repository } from 'typeorm';
+
+import {
+  ActiveWorkspacesCommandOptions,
+  ActiveWorkspacesCommandRunner,
+} from 'src/database/commands/active-workspaces.command';
+import { BillingCustomer } from 'src/engine/core-modules/billing/entities/billing-customer.entity';
+import { StripeService } from 'src/engine/core-modules/billing/stripe/stripe.service';
+import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
+
+interface SyncCustomerDataCommandOptions
+  extends ActiveWorkspacesCommandOptions {}
+
+@Command({
+  name: 'billing:sync-customer-data',
+  description: 'Sync customer data from Stripe for all active workspaces',
+})
+export class BillingSyncCustomerDataCommand extends ActiveWorkspacesCommandRunner {
+  constructor(
+    @InjectRepository(Workspace, 'core')
+    protected readonly workspaceRepository: Repository<Workspace>,
+    private readonly stripeService: StripeService,
+    @InjectRepository(BillingCustomer, 'core')
+    protected readonly billingCustomerRepository: Repository<BillingCustomer>,
+  ) {
+    super(workspaceRepository);
+  }
+
+  async executeActiveWorkspacesCommand(
+    _passedParam: string[],
+    options: SyncCustomerDataCommandOptions,
+    workspaceIds: string[],
+  ): Promise<void> {
+    this.logger.log('Running command to sync customer data');
+
+    for (const workspaceId of workspaceIds) {
+      this.logger.log(`Running command for workspace ${workspaceId}`);
+
+      try {
+        await this.syncCustomerDataForWorkspace(workspaceId, options);
+      } catch (error) {
+        this.logger.log(
+          chalk.red(
+            `Running command on workspace ${workspaceId} failed with error: ${error}, ${error.stack}`,
+          ),
+        );
+        continue;
+      } finally {
+        this.logger.log(
+          chalk.green(`Finished running command for workspace ${workspaceId}.`),
+        );
+      }
+    }
+
+    this.logger.log(chalk.green(`Command completed!`));
+  }
+
+  private async syncCustomerDataForWorkspace(
+    workspaceId: string,
+    options: SyncCustomerDataCommandOptions,
+  ): Promise<void> {
+    const billingCustomer = await this.billingCustomerRepository.findOne({
+      where: {
+        workspaceId,
+      },
+    });
+
+    if (!options.dryRun && !billingCustomer) {
+      const stripeCustomerId =
+        await this.stripeService.getStripeCustomerIdFromWorkspaceId(
+          workspaceId,
+        );
+
+      if (stripeCustomerId) {
+        await this.billingCustomerRepository.upsert(
+          {
+            stripeCustomerId,
+            workspaceId,
+          },
+          {
+            conflictPaths: ['workspaceId'],
+          },
+        );
+      }
+    }
+
+    if (options.verbose) {
+      this.logger.log(
+        chalk.yellow(`Added ${workspaceId} to billingCustomer table`),
+      );
+    }
+  }
+}

packages/twenty-server/src/engine/core-modules/billing/entities/billing-subscription.entity.ts

@@ -82,6 +82,7 @@ export class BillingSubscription {
     {
       nullable: false,
       onDelete: 'CASCADE',
+      createForeignKeyConstraints: false,
     },
   )
   @JoinColumn({

packages/twenty-server/src/engine/core-modules/billing/stripe/stripe.service.ts

@@ -194,4 +194,16 @@ export class StripeService {
 
     return productPrices.sort((a, b) => a.unitAmount - b.unitAmount);
   }
+
+  async getStripeCustomerIdFromWorkspaceId(workspaceId: string) {
+    const subscription = await this.stripe.subscriptions.search({
+      query: `metadata['workspaceId']:'${workspaceId}'`,
+      limit: 1,
+    });
+    const stripeCustomerId = subscription.data[0].customer
+      ? String(subscription.data[0].customer)
+      : undefined;
+
+    return stripeCustomerId;
+  }
 }