All files / src/platform/feature-flags feature-flag.service.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { Injectable, Logger } from '@nestjs/common';
import { InjectModel } from '@nestjs/mongoose';
import { Model } from 'mongoose';
import { FeatureFlag } from './schemas/feature-flag.schema';
 
@Injectable()
export class FeatureFlagService {
  private readonly logger = new Logger(FeatureFlagService.name);
 
  constructor(
    @InjectModel(FeatureFlag.name)
    private readonly featureFlagModel: Model<FeatureFlag>,
  ) {}
 
  async evaluate(
    flagKey: string,
    context: { tenantId: string; userId: string; userEmail: string },
  ): Promise<boolean> {
    const flag = await this.featureFlagModel
      .findOne({ key: flagKey })
      .lean()
      .exec();
 
    if (!flag || !flag.isEnabled) return false;
    if (!flag.rules) return true;
 
    try {
      const rules = flag.rules;
 
      if (rules.emails && Array.isArray(rules.emails)) {
        if (rules.emails.includes(context.userEmail)) return true;
      }
      if (rules.tenants && Array.isArray(rules.tenants)) {
        if (rules.tenants.includes(context.tenantId)) return true;
      }
      if (rules.defaultRule !== undefined) {
        return !!rules.defaultRule;
      }
    } catch (e) {
      this.logger.error(`Error parsing rules for feature flag: ${flagKey}`, e);
    }
 
    return false;
  }
 
  async configureFlag(key: string, isEnabled: boolean, rules?: any) {
    return this.featureFlagModel
      .findOneAndUpdate(
        { key },
        { key, isEnabled, rules },
        { upsert: true, new: true },
      )
      .lean()
      .exec();
  }
}