The spending limit is reached

Author: AlexTugarev

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

@@ -2096,6 +2096,26 @@ export class GitpodServerEEImpl extends GitpodServerImpl {
         });
     }
 
+    async getNotifications(ctx: TraceContext): Promise<string[]> {
+        const result = await super.getNotifications(ctx);
+        const user = this.checkAndBlockUser("getNotifications");
+        if (user.usageAttributionId) {
+            const allSessions = await this.listBilledUsage(ctx, user.usageAttributionId);
+            const totalUsage = allSessions.map((s) => s.credits).reduce((a, b) => a + b, 0);
+            const costCenter = await this.costCenterDB.findById(user.usageAttributionId);
+            if (costCenter) {
+                if (totalUsage > costCenter.spendingLimit) {
+                    result.unshift("The spending limit is reached.");
+                } else if (totalUsage > 0.8 * costCenter.spendingLimit * 0.8) {
+                    result.unshift("The spending limit is almost reached.");
+                }
+            } else {
+                log.warn("No costcenter found.", { userId: user.id, attributionId: user.usageAttributionId });
+            }
+        }
+        return result;
+    }
+
     async listBilledUsage(ctx: TraceContext, attributionId: string): Promise<BillableSession[]> {
         traceAPIParams(ctx, { attributionId });
         const user = this.checkAndBlockUser("listBilledUsage");

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

@@ -108,11 +108,7 @@ import {
 } from "@gitpod/gitpod-protocol/lib/team-subscription-protocol";
 import { Cancelable } from "@gitpod/gitpod-protocol/lib/util/cancelable";
 import { log, LogContext } from "@gitpod/gitpod-protocol/lib/util/logging";
-import {
-    InterfaceWithTraceContext,
-    TraceContext,
-    TraceContextWithSpan,
-} from "@gitpod/gitpod-protocol/lib/util/tracing";
+import { InterfaceWithTraceContext, TraceContext } from "@gitpod/gitpod-protocol/lib/util/tracing";
 import {
     IdentifyMessage,
     RemoteIdentifyMessage,
@@ -3248,8 +3244,8 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
         }
     }
 
-    async getNotifications(ctx: TraceContextWithSpan): Promise<string[]> {
-        const user = this.checkAndBlockUser("getNotifications");
-        return [`Hi ${user.name}`];
+    async getNotifications(ctx: TraceContext): Promise<string[]> {
+        this.checkAndBlockUser("getNotifications");
+        return [];
     }
 }