unleash.unleash/src/lib/routes/admin-api/addon.ts

import { Request, Response } from 'express';
import Controller from '../controller';
import { IUnleashConfig } from '../../types/option';
import { IUnleashServices } from '../../types/services';
import { Logger } from '../../logger';
import AddonService from '../../services/addon-service';

import { extractUsername } from '../../util/extract-user';
import {
    CREATE_ADDON,
    UPDATE_ADDON,
    DELETE_ADDON,
} from '../../types/permissions';
import { IAuthRequest } from '../unleash-types';

class AddonController extends Controller {
    private logger: Logger;

    private addonService: AddonService;

    constructor(
        config: IUnleashConfig,
        { addonService }: Pick<IUnleashServices, 'addonService'>,
    ) {
        super(config);
        this.logger = config.getLogger('/admin-api/addon.ts');
        this.addonService = addonService;

        this.get('/', this.getAddons);
        this.post('/', this.createAddon, CREATE_ADDON);
        this.get('/:id', this.getAddon);
        this.put('/:id', this.updateAddon, UPDATE_ADDON);
        this.delete('/:id', this.deleteAddon, DELETE_ADDON);
    }

    async getAddons(req: Request, res: Response): Promise<void> {
        const addons = await this.addonService.getAddons();
        const providers = this.addonService.getProviderDefinitions();
        res.json({ addons, providers });
    }

    async getAddon(
        req: Request<{ id: number }, any, any, any>,
        res: Response,
    ): Promise<void> {
        const { id } = req.params;
        const addon = await this.addonService.getAddon(id);
        res.json(addon);
    }

    async updateAddon(
        req: IAuthRequest<{ id: number }, any, any, any>,
        res: Response,
    ): Promise<void> {
        const { id } = req.params;
        const createdBy = extractUsername(req);
        const data = req.body;

        const addon = await this.addonService.updateAddon(id, data, createdBy);
        res.status(200).json(addon);
    }

    async createAddon(req: IAuthRequest, res: Response): Promise<void> {
        const createdBy = extractUsername(req);
        const data = req.body;
        const addon = await this.addonService.createAddon(data, createdBy);
        res.status(201).json(addon);
    }

    async deleteAddon(
        req: IAuthRequest<{ id: number }, any, any, any>,
        res: Response,
    ): Promise<void> {
        const { id } = req.params;
        const username = extractUsername(req);
        await this.addonService.removeAddon(id, username);
        res.status(200).end();
    }
}
export default AddonController;
module.exports = AddonController;