unleash.unleash/src/lib/features/project-environments/environments.ts

import type { Response } from 'express';
import Controller from '../../routes/controller';
import {
    type IUnleashConfig,
    type IUnleashServices,
    serializeDates,
    UPDATE_PROJECT,
    PROJECT_DEFAULT_STRATEGY_WRITE,
} from '../../types';
import type { Logger } from '../../logger';
import type EnvironmentService from './environment-service';
import {
    createFeatureStrategySchema,
    type CreateFeatureStrategySchema,
    createRequestSchema,
    createResponseSchema,
    emptyResponse,
    getStandardResponses,
    type ProjectEnvironmentSchema,
} from '../../openapi';
import type { OpenApiService, ProjectService } from '../../services';
import type { IAuthRequest } from '../../routes/unleash-types';
import type { WithTransactional } from '../../db/transaction';

const PREFIX = '/:projectId/environments';

interface IProjectEnvironmentParams {
    projectId: string;
    environment: string;
}

export default class EnvironmentsController extends Controller {
    private logger: Logger;

    private environmentService: WithTransactional<EnvironmentService>;

    private openApiService: OpenApiService;

    private projectService: ProjectService;

    constructor(
        config: IUnleashConfig,
        {
            transactionalEnvironmentService,
            openApiService,
            projectService,
        }: Pick<
            IUnleashServices,
            | 'transactionalEnvironmentService'
            | 'openApiService'
            | 'projectService'
        >,
    ) {
        super(config);

        this.logger = config.getLogger('admin-api/project/environments.ts');
        this.environmentService = transactionalEnvironmentService;
        this.openApiService = openApiService;
        this.projectService = projectService;

        this.route({
            method: 'post',
            path: PREFIX,
            handler: this.addEnvironmentToProject,
            permission: UPDATE_PROJECT,
            middleware: [
                openApiService.validPath({
                    tags: ['Projects'],
                    operationId: 'addEnvironmentToProject',
                    summary: 'Add an environment to a project.',
                    description:
                        'This endpoint adds the provided environment to the specified project, with optional support for enabling and disabling change requests for the environment and project.',
                    requestBody: createRequestSchema(
                        'projectEnvironmentSchema',
                    ),
                    responses: {
                        200: emptyResponse,
                        ...getStandardResponses(401, 403, 409),
                    },
                }),
            ],
        });

        this.route({
            method: 'delete',
            path: `${PREFIX}/:environment`,
            acceptAnyContentType: true,
            handler: this.removeEnvironmentFromProject,
            permission: UPDATE_PROJECT,
            middleware: [
                openApiService.validPath({
                    tags: ['Projects'],
                    operationId: 'removeEnvironmentFromProject',
                    summary: 'Remove an environment from a project.',
                    description:
                        'This endpoint removes the specified environment from the project.',
                    responses: {
                        200: emptyResponse,
                        ...getStandardResponses(400, 401, 403),
                    },
                }),
            ],
        });

        this.route({
            method: 'post',
            path: `${PREFIX}/:environment/default-strategy`,
            handler: this.updateDefaultStrategyForProjectEnvironment,
            permission: [UPDATE_PROJECT, PROJECT_DEFAULT_STRATEGY_WRITE],
            middleware: [
                openApiService.validPath({
                    tags: ['Projects'],
                    operationId: 'addDefaultStrategyToProjectEnvironment',
                    summary: 'Set environment-default strategy',
                    description:
                        'Sets a default strategy for this environment. Unleash will use this strategy by default when enabling a toggle. Use the wild card "*" for `:environment` to add to all environments. ',
                    requestBody: createRequestSchema(
                        'createFeatureStrategySchema',
                    ),
                    responses: {
                        200: createResponseSchema(
                            'createFeatureStrategySchema',
                        ),
                        ...getStandardResponses(400),
                    },
                }),
            ],
        });
    }

    async addEnvironmentToProject(
        req: IAuthRequest<
            Omit<IProjectEnvironmentParams, 'environment'>,
            void,
            ProjectEnvironmentSchema
        >,
        res: Response,
    ): Promise<void> {
        const { projectId } = req.params;
        const { environment } = req.body;
        await this.projectService.getProject(projectId); // Validates that the project exists

        await this.environmentService.transactional((service) =>
            service.addEnvironmentToProject(environment, projectId, req.audit),
        );

        res.status(200).end();
    }

    async removeEnvironmentFromProject(
        req: IAuthRequest<IProjectEnvironmentParams>,
        res: Response<void>,
    ): Promise<void> {
        const { projectId, environment } = req.params;

        await this.environmentService.transactional((service) =>
            service.removeEnvironmentFromProject(
                environment,
                projectId,
                req.audit,
            ),
        );

        res.status(200).end();
    }

    async updateDefaultStrategyForProjectEnvironment(
        req: IAuthRequest<
            IProjectEnvironmentParams,
            CreateFeatureStrategySchema
        >,
        res: Response<CreateFeatureStrategySchema>,
    ): Promise<void> {
        const { projectId, environment } = req.params;
        const strategy = req.body;

        const saved = await this.environmentService.transactional((service) =>
            service.updateDefaultStrategy(
                environment,
                projectId,
                strategy,
                req.audit,
            ),
        );

        this.openApiService.respondWithValidation(
            200,
            res,
            createFeatureStrategySchema.$id,
            serializeDates(saved),
        );
    }
}