import { IUnleashConfig } from '../types/option'; import { IEventStore } from '../types/stores/event-store'; import { IUnleashStores } from '../types'; import { Logger } from '../logger'; import NameExistsError from '../error/name-exists-error'; import { ISegmentStore } from '../types/stores/segment-store'; import { IFeatureStrategy, ISegment } from '../types/model'; import { segmentSchema } from './segment-schema'; import { SEGMENT_CREATED, SEGMENT_DELETED, SEGMENT_UPDATED, } from '../types/events'; import User from '../types/user'; import { IFeatureStrategiesStore } from '../types/stores/feature-strategies-store'; import BadDataError from '../error/bad-data-error'; export class SegmentService { private logger: Logger; private segmentStore: ISegmentStore; private featureStrategiesStore: IFeatureStrategiesStore; private eventStore: IEventStore; private config: IUnleashConfig; constructor( { segmentStore, featureStrategiesStore, eventStore, }: Pick< IUnleashStores, 'segmentStore' | 'featureStrategiesStore' | 'eventStore' >, config: IUnleashConfig, ) { this.segmentStore = segmentStore; this.featureStrategiesStore = featureStrategiesStore; this.eventStore = eventStore; this.logger = config.getLogger('services/segment-service.ts'); this.config = config; } async get(id: number): Promise { return this.segmentStore.get(id); } async getAll(): Promise { return this.segmentStore.getAll(); } async getActive(): Promise { return this.segmentStore.getActive(); } // Used by unleash-enterprise. async getByStrategy(strategyId: string): Promise { return this.segmentStore.getByStrategy(strategyId); } // Used by unleash-enterprise. async getStrategies(id: number): Promise { return this.featureStrategiesStore.getStrategiesBySegment(id); } async create( data: unknown, user: Partial>, ): Promise { const input = await segmentSchema.validateAsync(data); this.validateSegmentValuesLimit(input); await this.validateName(input.name); const segment = await this.segmentStore.create(input, user); await this.eventStore.store({ type: SEGMENT_CREATED, createdBy: user.email || user.username, data: segment, }); return segment; } async update( id: number, data: unknown, user: Partial>, ): Promise { const input = await segmentSchema.validateAsync(data); this.validateSegmentValuesLimit(input); const preData = await this.segmentStore.get(id); if (preData.name !== input.name) { await this.validateName(input.name); } const segment = await this.segmentStore.update(id, input); await this.eventStore.store({ type: SEGMENT_UPDATED, createdBy: user.email || user.username, data: segment, preData, }); } async delete(id: number, user: User): Promise { const segment = await this.segmentStore.get(id); await this.segmentStore.delete(id); await this.eventStore.store({ type: SEGMENT_DELETED, createdBy: user.email || user.username, data: segment, }); } async cloneStrategySegments( sourceStrategyId: string, targetStrategyId: string, ): Promise { const sourceStrategySegments = await this.getByStrategy( sourceStrategyId, ); await Promise.all( sourceStrategySegments.map((sourceStrategySegment) => { return this.addToStrategy( sourceStrategySegment.id, targetStrategyId, ); }), ); } // Used by unleash-enterprise. async addToStrategy(id: number, strategyId: string): Promise { await this.validateStrategySegmentLimit(strategyId); await this.segmentStore.addToStrategy(id, strategyId); } // Used by unleash-enterprise. async removeFromStrategy(id: number, strategyId: string): Promise { await this.segmentStore.removeFromStrategy(id, strategyId); } async validateName(name: string): Promise { if (!name) { throw new BadDataError('Segment name cannot be empty'); } if (await this.segmentStore.existsByName(name)) { throw new NameExistsError('Segment name already exists'); } } private async validateStrategySegmentLimit( strategyId: string, ): Promise { const { strategySegmentsLimit } = this.config; if ( (await this.getByStrategy(strategyId)).length >= strategySegmentsLimit ) { throw new BadDataError( `Strategies may not have more than ${strategySegmentsLimit} segments`, ); } } private validateSegmentValuesLimit(segment: Omit): void { const { segmentValuesLimit } = this.config; const valuesCount = segment.constraints .flatMap((constraint) => constraint.values?.length ?? 0) .reduce((acc, length) => acc + length, 0); if (valuesCount > segmentValuesLimit) { throw new BadDataError( `Segments may not have more than ${segmentValuesLimit} values`, ); } } }