[server] Remove dead code in Entitlement- and UserService

gitpod-io · Sep 21, 2023 · eba6a0f · eba6a0f
1 parent 267805a
commit eba6a0f
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 78 deletions.
diff --git a/components/gitpod-protocol/src/billing-mode.ts b/components/gitpod-protocol/src/billing-mode.ts
@@ -36,5 +36,5 @@ interface UsageBased {
     mode: "usage-based";
 
     /** True if the org has a paid plan. */
-    paid?: boolean;
+    paid: boolean;
 }
diff --git a/components/server/src/billing/billing-mode.ts b/components/server/src/billing/billing-mode.ts
@@ -31,20 +31,4 @@ export class BillingModes {
         const paid = billingStrategy === CostCenter_BillingStrategy.BILLING_STRATEGY_STRIPE;
         return { mode: "usage-based", paid };
     }
-
-    /**
-     * @deprecated use getBillingMode(userId, organizationId) instead
-     * @returns
-     */
-    async getBillingModeForUser(): Promise<BillingMode> {
-        if (!this.config.enablePayment) {
-            // Payment is not enabled. E.g. Self-Hosted.
-            return { mode: "none" };
-        }
-
-        // "paid" is not set here, just as before. Also, it's we should remove this whole method once the org-migration is done, and center all capabilities around Organizations
-        return {
-            mode: "usage-based",
-        };
-    }
 }
diff --git a/components/server/src/billing/entitlement-service-ubp.ts b/components/server/src/billing/entitlement-service-ubp.ts
@@ -4,7 +4,6 @@
  * See License.AGPL.txt in the project root for license information.
  */
 
-import { TeamDB } from "@gitpod/gitpod-db/lib";
 import {
     WorkspaceInstance,
     WorkspaceTimeoutDuration,
@@ -14,7 +13,6 @@ import {
     WORKSPACE_LIFETIME_SHORT,
     User,
     BillingTier,
-    Team,
 } from "@gitpod/gitpod-protocol";
 import { AttributionId } from "@gitpod/gitpod-protocol/lib/attribution";
 import { inject, injectable } from "inversify";
@@ -31,10 +29,7 @@ const MAX_PARALLEL_WORKSPACES_PAID = 16;
  */
 @injectable()
 export class EntitlementServiceUBP implements EntitlementService {
-    constructor(
-        @inject(UsageService) private readonly usageService: UsageService,
-        @inject(TeamDB) private readonly teamDB: TeamDB,
-    ) {}
+    constructor(@inject(UsageService) private readonly usageService: UsageService) {}
 
     async mayStartWorkspace(
         user: User,
@@ -79,7 +74,7 @@ export class EntitlementServiceUBP implements EntitlementService {
         }
     }
 
-    async maySetTimeout(userId: string, organizationId?: string): Promise<boolean> {
+    async maySetTimeout(userId: string, organizationId: string): Promise<boolean> {
         return this.hasPaidSubscription(userId, organizationId);
     }
 
@@ -109,48 +104,15 @@ export class EntitlementServiceUBP implements EntitlementService {
         return true;
     }
 
-    private async hasPaidSubscription(userId: string, organizationId?: string): Promise<boolean> {
-        if (organizationId) {
-            try {
-                // This is the "stricter", more correct version: We only allow privileges on the Organization that is paying for it
-                const { billingStrategy } = await this.usageService.getCostCenter(userId, organizationId);
-                return billingStrategy === CostCenter_BillingStrategy.BILLING_STRATEGY_STRIPE;
-            } catch (err) {
-                log.warn({ userId, organizationId }, "Error checking if user is subscribed to organization", err);
-                return false;
-            }
-        }
-
-        // TODO(gpl) Remove everything below once organizations are fully rolled out
-        // This is the old behavior, stemming from our transition to PAYF, where our API did-/doesn't pass organizationId, yet
-        // Member of paid team?
-        const teams = await this.teamDB.findTeamsByUser(userId);
-        const isTeamSubscribedPromises = teams.map(async (team: Team) => {
-            const { billingStrategy } = await this.usageService.getCostCenter(userId, team.id);
+    private async hasPaidSubscription(userId: string, organizationId: string): Promise<boolean> {
+        try {
+            // This is the "stricter", more correct version: We only allow privileges on the Organization that is paying for it
+            const { billingStrategy } = await this.usageService.getCostCenter(userId, organizationId);
             return billingStrategy === CostCenter_BillingStrategy.BILLING_STRATEGY_STRIPE;
-        });
-        // Return the first truthy promise, or false if all the promises were falsy.
-        // Source: https://gist.github.com/jbreckmckye/66364021ebaa0785e426deec0410a235
-        return new Promise((resolve, reject) => {
-            // If any promise returns true, immediately resolve with true
-            isTeamSubscribedPromises.forEach(async (isTeamSubscribedPromise: Promise<boolean>) => {
-                try {
-                    const isTeamSubscribed = await isTeamSubscribedPromise;
-                    if (isTeamSubscribed) resolve(true);
-                } catch (err) {
-                    log.warn({ userId, organizationId }, "Error checking if user is subscribed to organization", err);
-                    resolve(false);
-                }
-            });
-
-            // If neither of the above fires, resolve with false
-            // Check truthiness just in case callbacks fire out-of-band
-            Promise.all(isTeamSubscribedPromises)
-                .then((areTeamsSubscribed) => {
-                    resolve(!!areTeamsSubscribed.find((isTeamSubscribed: boolean) => !!isTeamSubscribed));
-                })
-                .catch(reject);
-        });
+        } catch (err) {
+            log.warn({ userId, organizationId }, "Error checking if user is subscribed to organization", err);
+            return false;
+        }
     }
 
     async getBillingTier(userId: string, organizationId: string): Promise<BillingTier> {

diff --git a/components/server/src/billing/entitlement-service.ts b/components/server/src/billing/entitlement-service.ts
@@ -54,7 +54,7 @@ export interface EntitlementService {
      * @param userId
      * @param organizationId
      */
-    maySetTimeout(userId: string, organizationId?: string): Promise<boolean>;
+    maySetTimeout(userId: string, organizationId: string): Promise<boolean>;
 
     /**
      * Returns the default workspace timeout for the given user at a given point in time
@@ -127,10 +127,9 @@ export class EntitlementServiceImpl implements EntitlementService {
         }
     }
 
-    async maySetTimeout(userId: string, organizationId?: string): Promise<boolean> {
+    async maySetTimeout(userId: string, organizationId: string): Promise<boolean> {
         try {
-            // TODO(gpl): We need to replace this with ".getBillingMode(user.id, organizationId);" once all callers forward organizationId
-            const billingMode = await this.billingModes.getBillingModeForUser();
+            const billingMode = await this.billingModes.getBillingMode(userId, organizationId);
             switch (billingMode.mode) {
                 case "none":
                     // when payment is disabled users can do everything

diff --git a/components/server/src/user/user-service.ts b/components/server/src/user/user-service.ts
@@ -23,7 +23,6 @@ import { CreateUserParams } from "./user-authentication";
 import { IAnalyticsWriter } from "@gitpod/gitpod-protocol/lib/analytics";
 import { TransactionalContext } from "@gitpod/gitpod-db/lib/typeorm/transactional-db-impl";
 import { RelationshipUpdater } from "../authorization/relationship-updater";
-import { EntitlementService } from "../billing/entitlement-service";
 
 @injectable()
 export class UserService {
@@ -33,7 +32,6 @@ export class UserService {
         @inject(Authorizer) private readonly authorizer: Authorizer,
         @inject(IAnalyticsWriter) private readonly analytics: IAnalyticsWriter,
         @inject(RelationshipUpdater) private readonly relationshipUpdater: RelationshipUpdater,
-        @inject(EntitlementService) private readonly entitlementService: EntitlementService,
     ) {}
 
     public async createUser(
@@ -144,13 +142,6 @@ export class UserService {
             }
         }
 
-        if (!(await this.entitlementService.maySetTimeout(targetUserId))) {
-            throw new ApplicationError(
-                ErrorCodes.PERMISSION_DENIED,
-                "Configure workspace timeout only available for paid user.",
-            );
-        }
-
         const user = await this.findUserById(userId, targetUserId);
         AdditionalUserData.set(user, setting);
         await this.userDb.updateUserPartial(user);