[server] Tie increased resource access to payment status

Author: csweichel

components/server/ee/src/user/eligibility-service.ts

@@ -247,6 +247,27 @@ export class EligibilityService {
         return subscriptions.filter(s => eligblePlans.includes(s.planId!)).length > 0;
     }
 
+    /**
+     * Returns true if the user ought to land on a workspace cluster that provides more resources
+     * compared to the default case.
+     */
+    async userGetsMoreResources(user: User): Promise<boolean> {
+        if (!this.config.enablePayment) {
+            // when payment is disabled users can do everything
+            return true;
+        }
+
+        const subscriptions = await this.subscriptionService.getNotYetCancelledSubscriptions(user, new Date().toISOString());
+        const eligblePlans = [
+            Plans.PROFESSIONAL_EUR,
+            Plans.PROFESSIONAL_USD,
+            Plans.TEAM_PROFESSIONAL_EUR,
+            Plans.TEAM_PROFESSIONAL_USD,
+        ].map(p => p.chargebeeId);
+
+        return subscriptions.filter(s => eligblePlans.includes(s.planId!)).length > 0;
+    }
+
     protected async getUser(user: User | string): Promise<User> {
         if (typeof user === 'string') {
             const realUser = await this.userDb.findUserById(user);

components/server/ee/src/user/user-service.ts

@@ -36,6 +36,14 @@ export class UserServiceEE extends UserService {
         return "60m";
     }
 
+    async userGetsMoreResources(user: User): Promise<boolean> {
+        if (this.config.enablePayment) {
+            return this.eligibilityService.userGetsMoreResources(user);
+        }
+
+        return false;
+    }
+
     async checkSignUp(params: CheckSignUpParams) {
 
         // todo@at: check if we need an optimization for SaaS here. used to be a no-op there.

components/server/src/user/user-service.ts

@@ -158,6 +158,17 @@ export class UserService {
         return "30m";
     }
 
+    /**
+     * Returns true if the user ought land in a cluster which offers more resources than
+     * the default.
+     *
+     * @param user user to check for
+     * @returns
+     */
+    async userGetsMoreResources(user: User): Promise<boolean> {
+        return false;
+    }
+
     /**
      * This might throw `AuthException`s.
      *

components/server/src/workspace/workspace-starter.ts

@@ -35,6 +35,7 @@ import { EnvVarWithValue } from "@gitpod/gitpod-protocol/src/protocol";
 import { WithReferrerContext } from "@gitpod/gitpod-protocol/lib/protocol";
 import { IDEOption } from "@gitpod/gitpod-protocol/lib/ide-protocol";
 import { Deferred } from "@gitpod/gitpod-protocol/lib/util/deferred";
+import { ExtendedUser } from "@gitpod/ws-manager/lib/constraints";
 
 export interface StartWorkspaceOptions {
     rethrow?: boolean;
@@ -174,10 +175,16 @@ export class WorkspaceStarter {
             startRequest.setSpec(spec);
             startRequest.setServicePrefix(workspace.id);
 
+            // we add additional information to the user to help with cluster selection
+            const euser: ExtendedUser = {
+                ...user,
+                getsMoreResources: await this.userService.userGetsMoreResources(user),
+            }
+
             // tell the world we're starting this instance
             let resp: StartWorkspaceResponse.AsObject | undefined;
             let lastInstallation = "";
-            const clusters = await this.clientProvider.getStartClusterSets(user, workspace, instance);
+            const clusters = await this.clientProvider.getStartClusterSets(euser, workspace, instance);
             for await (let cluster of clusters) {
                 try {
                     // getStartManager will throw an exception if there's no cluster available and hence exit the loop