feat(server): user feature model (#9843)

close CLOUD-108
2026-07-21 20:16:26 +08:00 · 2025-01-22 10:38:04 +00:00
parent 994d758c07
commit f8a515e89a
7 changed files with 580 additions and 39 deletions
@@ -0,0 +1,120 @@
+import { Injectable } from '@nestjs/common';
+import { Transactional } from '@nestjs-cls/transactional';
+
+import { BaseModel } from './base';
+import type { UserFeatureName } from './common';
+
+@Injectable()
+export class UserFeatureModel extends BaseModel {
+  async get<T extends UserFeatureName>(userId: string, name: T) {
+    // TODO(@forehalo):
+    // all feature query-and-use queries should be simplified like the below when `feature(name)` becomes a unique index
+    //
+    // this.db.userFeature.findFirst({
+    //   include: {
+    //     feature: true
+    //   },
+    //   where: {
+    //     userId,
+    //     activated: true,
+    //     feature: {
+    //       feature: name,
+    //     }
+    //   }
+    // })
+    const feature = await this.models.feature.get(name);
+
+    const count = await this.db.userFeature.count({
+      where: {
+        userId,
+        featureId: feature.id,
+        activated: true,
+      },
+    });
+
+    return count > 0 ? feature : null;
+  }
+
+  async has(userId: string, name: UserFeatureName) {
+    const feature = await this.models.feature.get_unchecked(name);
+
+    const count = await this.db.userFeature.count({
+      where: {
+        userId,
+        featureId: feature.id,
+        activated: true,
+      },
+    });
+
+    return count > 0;
+  }
+
+  async list(userId: string) {
+    const userFeatures = await this.db.userFeature.findMany({
+      include: {
+        feature: true,
+      },
+      where: {
+        userId,
+        activated: true,
+      },
+    });
+
+    return userFeatures.map(
+      userFeature => userFeature.feature.feature
+    ) as UserFeatureName[];
+  }
+
+  async add(userId: string, featureName: UserFeatureName, reason: string) {
+    const feature = await this.models.feature.get_unchecked(featureName);
+
+    const existing = await this.tx.userFeature.findFirst({
+      where: {
+        userId,
+        featureId: feature.id,
+        activated: true,
+      },
+    });
+
+    if (existing) {
+      return existing;
+    }
+
+    const userFeature = await this.tx.userFeature.create({
+      data: {
+        userId,
+        featureId: feature.id,
+        activated: true,
+        reason,
+      },
+    });
+
+    this.logger.verbose(`Feature ${featureName} added to user ${userId}`);
+
+    return userFeature;
+  }
+
+  async remove(userId: string, featureName: UserFeatureName) {
+    const feature = await this.models.feature.get_unchecked(featureName);
+
+    await this.tx.userFeature.deleteMany({
+      where: {
+        userId,
+        featureId: feature.id,
+      },
+    });
+
+    this.logger.verbose(`Feature ${featureName} removed from user ${userId}`);
+  }
+
+  @Transactional()
+  async switch(
+    userId: string,
+    from: UserFeatureName,
+    to: UserFeatureName,
+    reason: string
+  ) {
+    await this.remove(userId, from);
+    await this.add(userId, to, reason);
+  }
+}