[server] When a user with no explicit 'usageAttributionId' joins a team with usage-based billing enabled, automatically re-attribute usage to that team

Author: jankeromnes

components/server/ee/src/workspace/gitpod-server-impl.ts

@@ -1428,13 +1428,27 @@ export class GitpodServerEEImpl extends GitpodServerImpl {
 
     protected async onTeamMemberAdded(userId: string, teamId: string): Promise<void> {
         const now = new Date();
-        const teamSubscription = await this.teamSubscription2DB.findForTeam(teamId, now.toISOString());
-        if (!teamSubscription) {
-            // No team subscription, nothing to do 🌴
-            return;
+        const ts2 = await this.teamSubscription2DB.findForTeam(teamId, now.toISOString());
+        if (ts2) {
+            await this.updateTeamSubscriptionQuantity(ts2);
+            await this.teamSubscription2Service.addTeamMemberSubscription(ts2, userId);
+        }
+        const [teamCustomer, user] = await Promise.all([
+            this.stripeService.findCustomerByTeamId(teamId),
+            this.userDB.findUserById(userId),
+        ]);
+        if (teamCustomer && user && !user.additionalData?.usageAttributionId) {
+            // If the user didn't explicitly choose yet where their usage should be attributed to, and
+            // they join a team which accepts usage attribution (i.e. with usage-based billing enabled),
+            // then we simplify the UX by automatically attributing the user's usage to that team.
+            // Note: This default choice can be changed at any time by the user in their billing settings.
+            const subscription = await this.stripeService.findUncancelledSubscriptionByCustomer(teamCustomer.id);
+            if (subscription) {
+                user.additionalData = user.additionalData || {};
+                user.additionalData.usageAttributionId = `team:${teamId}`;
+                await this.userDB.updateUserPartial(user);
+            }
         }
-        await this.updateTeamSubscriptionQuantity(teamSubscription);
-        await this.teamSubscription2Service.addTeamMemberSubscription(teamSubscription, userId);
     }
 
     protected async onTeamMemberRemoved(userId: string, teamId: string, teamMembershipId: string): Promise<void> {