index.ts

import {Request, Response} from 'express';

import faker from 'faker';
import path from 'path';

const sleep = (time: number) => {
    return new Promise(resolve => setTimeout(resolve, time));
};

export type Options = {
    path?: string;
    delay?: number | ((method: string) => number);
};

export default (options: Options) => {
    return async (req: Request, res: Response) => {
        let method = req.path;
        if (!method) {
            method = Array.isArray(req.query.method) ? req.query.method.join('/') : req.query.method;
        } else {
            method = method.replace(/^\//, '');
        }

        if (!method) {
            res.status(404).send({});
            return;
        }

        try {
            let {default: mock} = await import(path.resolve(options.path || path.join(__dirname, 'data'), method));

            if ('function' === typeof mock) {
                mock = await mock(req, res);
            }

            let delay = 0;
            if ('function' === typeof options.delay) {
                delay = options.delay(method);
            } else if (options.delay) {
                delay = options.delay;
            }

            if (delay) {
                await sleep(delay);
            }

            if (mock instanceof ArrayBuffer) {
                res.send(Buffer.from(mock));
            } else {
                const result = JSON.parse(faker.fake(JSON.stringify(mock, null, 4)));
                if (result && 'status' in result && 'data' in result) {
                    res.json(result);
                } else {
                    res.json({status: 0, msg: '', data: result});
                }
            }
        } catch (e) {
            res.status(500).send(e.message);
            // eslint-disable-next-line no-console
            console.error(e);
        }
    };
};