feat: method for updating users metering directly (#2252)

* chore: cleanup ts extensions controller

* feat: method for updating users metering directly

src/backend/src/services/MeteringService/MeteringService.ts

@@ -405,6 +405,80 @@ export class MeteringService {
         });
     }
 
+    async setActorCurrentMonthUsageTotal (actor: Actor, totalCost: number) {
+        if ( ! actor.type?.user?.uuid ) {
+            throw new Error('Actor must be a user to set usage details');
+        }
+        if ( !Number.isFinite(totalCost) || totalCost < 0 ) {
+            throw new Error('Total cost must be a non-negative number');
+        }
+
+        const normalizedTotal = Math.round(totalCost);
+        const currentMonth = this.#getMonthYearString();
+        const userId = actor.type.user.uuid;
+        const appId = actor.type?.app?.uid || GLOBAL_APP_KEY;
+
+        return await this.#superUserService.sudo(async () => {
+            const actorUsageKey = `${METRICS_PREFIX}:actor:${userId}:${currentMonth}`;
+            const currentUsage = await this.#kvStore.get({ key: actorUsageKey }) as UsageByType | null;
+            const currentTotal = currentUsage?.total ?? 0;
+            const delta = normalizedTotal - currentTotal;
+
+            if ( delta === 0 ) {
+                return currentUsage || { total: 0 } as UsageByType;
+            }
+
+            const pathAndAmountMap = {
+                total: delta,
+                'manual_adjustment.cost': delta,
+                'manual_adjustment.units': delta,
+                'manual_adjustment.count': 1,
+            };
+
+            const updatedUsage = await this.#kvStore.incr({
+                key: actorUsageKey,
+                pathAndAmountMap,
+            }) as unknown as UsageByType;
+
+            const puterConsumptionKey = this.#generateGloabalUsageKey(userId, appId, currentMonth);
+            this.#kvStore.incr({
+                key: puterConsumptionKey,
+                pathAndAmountMap,
+            }).catch((e: Error) => {
+                console.warn('Failed to increment aux usage data \'puterConsumptionKey\' with error: ', e);
+            });
+
+            const actorAppUsageKey = `${METRICS_PREFIX}:actor:${userId}:app:${appId}:${currentMonth}`;
+            this.#kvStore.incr({
+                key: actorAppUsageKey,
+                pathAndAmountMap,
+            }).catch((e: Error) => {
+                console.warn('Failed to increment aux usage data \'actorAppUsageKey\' with error: ', e);
+            });
+
+            const actorAppTotalsKey = `${METRICS_PREFIX}:actor:${userId}:apps:${currentMonth}`;
+            this.#kvStore.incr({
+                key: actorAppTotalsKey,
+                pathAndAmountMap: {
+                    [`${appId}.total`]: delta,
+                    [`${appId}.count`]: 1,
+                },
+            }).catch((e: Error) => {
+                console.warn('Failed to increment aux usage data \'actorAppTotalsKey\' with error: ', e);
+            });
+
+            const lastUpdatedKey = `${METRICS_PREFIX}:actor:${userId}:lastUpdated`;
+            this.#kvStore.set({
+                key: lastUpdatedKey,
+                value: Date.now(),
+            }).catch((e: Error) => {
+                console.warn('Failed to set lastUpdatedKey with error: ', e);
+            });
+
+            return updatedUsage;
+        });
+    }
+
     async getActorCurrentMonthAppUsageDetails (actor: Actor, appId?: string) {
         if ( ! actor.type?.user?.uuid ) {
             throw new Error('Actor must be a user to get usage details');
@@ -557,4 +631,4 @@ export class MeteringService {
             });
         });
     }
-}
+}