mirror of
https://github.com/Unleash/unleash.git
synced 2024-10-18 20:09:08 +02:00
90 lines
2.9 KiB
TypeScript
90 lines
2.9 KiB
TypeScript
|
import { IUnleashStores } from '../types/stores';
|
||
|
import { IUnleashConfig } from '../types/option';
|
||
|
import EnvironmentStore from '../db/environment-store';
|
||
|
import { Logger } from '../logger';
|
||
|
import { IEnvironment } from '../types/model';
|
||
|
import FeatureStrategiesStore from '../db/feature-strategy-store';
|
||
|
import { UNIQUE_CONSTRAINT_VIOLATION } from '../error/db-error';
|
||
|
import NameExistsError from '../error/name-exists-error';
|
||
|
import { environmentSchema } from './state-schema';
|
||
|
import NotFoundError from '../error/notfound-error';
|
||
|
|
||
|
export default class EnvironmentService {
|
||
|
private logger: Logger;
|
||
|
|
||
|
private environmentStore: EnvironmentStore;
|
||
|
|
||
|
private featureStrategiesStore: FeatureStrategiesStore;
|
||
|
|
||
|
constructor(
|
||
|
{
|
||
|
environmentStore,
|
||
|
featureStrategiesStore,
|
||
|
}: Pick<IUnleashStores, 'environmentStore' | 'featureStrategiesStore'>,
|
||
|
{ getLogger }: Pick<IUnleashConfig, 'getLogger'>,
|
||
|
) {
|
||
|
this.logger = getLogger('services/environment-service.ts');
|
||
|
this.environmentStore = environmentStore;
|
||
|
this.featureStrategiesStore = featureStrategiesStore;
|
||
|
}
|
||
|
|
||
|
async getAll(): Promise<IEnvironment[]> {
|
||
|
return this.environmentStore.getAll();
|
||
|
}
|
||
|
|
||
|
async get(name: string): Promise<IEnvironment> {
|
||
|
return this.environmentStore.getByName(name);
|
||
|
}
|
||
|
|
||
|
async delete(name: string): Promise<void> {
|
||
|
return this.environmentStore.delete(name);
|
||
|
}
|
||
|
|
||
|
async create(env: IEnvironment): Promise<IEnvironment> {
|
||
|
await environmentSchema.validateAsync(env);
|
||
|
return this.environmentStore.upsert(env);
|
||
|
}
|
||
|
|
||
|
async update(
|
||
|
name: string,
|
||
|
env: Pick<IEnvironment, 'displayName'>,
|
||
|
): Promise<IEnvironment> {
|
||
|
const exists = await this.environmentStore.exists(name);
|
||
|
if (exists) {
|
||
|
return this.environmentStore.upsert({ ...env, name });
|
||
|
}
|
||
|
throw new NotFoundError(`Could not find environment ${name}`);
|
||
|
}
|
||
|
|
||
|
async connectProjectToEnvironment(
|
||
|
environment: string,
|
||
|
projectId: string,
|
||
|
): Promise<void> {
|
||
|
try {
|
||
|
await this.environmentStore.connectProject(environment, projectId);
|
||
|
await this.environmentStore.connectFeatures(environment, projectId);
|
||
|
} catch (e) {
|
||
|
if (e.code === UNIQUE_CONSTRAINT_VIOLATION) {
|
||
|
throw new NameExistsError(
|
||
|
`${projectId} already has the environment ${environment} enabled`,
|
||
|
);
|
||
|
}
|
||
|
throw e;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
async removeEnvironmentFromProject(
|
||
|
environment: string,
|
||
|
projectId: string,
|
||
|
): Promise<void> {
|
||
|
await this.featureStrategiesStore.disconnectEnvironmentFromProject(
|
||
|
environment,
|
||
|
projectId,
|
||
|
);
|
||
|
await this.environmentStore.disconnectProjectFromEnv(
|
||
|
environment,
|
||
|
projectId,
|
||
|
);
|
||
|
}
|
||
|
}
|