index.ts

import {Context, Middleware} from 'koa';
import * as Router from 'koa-router';
import * as fs from 'fs';

export enum HttpMethod {
  HEAD,
  OPTIONS,
  GET,
  PUT,
  PATCH,
  POST,
  DELETE
}

function getFiles(dir: string, files_?: string[]): string[] {
  files_ = files_ || [];
  const files = fs.readdirSync(dir);
  for (let i in files) {
    const name = dir + '/' + files[i];
    if (fs.statSync(name).isDirectory()) {
      getFiles(name, files_);
    } else {
      files_.push(name);
    }
  }
  return files_;
}

/**
 * 格式化返回数据的格式
 */
async function formatResponse(descriptor: any, ctx: Context) {
  const ret = descriptor.value(ctx);
  if (ret != null) {
    const data = await Promise.resolve(ret);
    if (data != null) {
      // 正常格式
      ctx.body = {
        data: data,
      };
    }
  }
}

const router = new Router();

// @router 装饰器
export function route(path: string, method?: HttpMethod, ...middleware: Array<Middleware>) {
  return (target: any, key?: string | symbol, descriptor?: any): void => {
    // Decorator applied to Class (for Constructor injection).
    if (typeof target === 'function' && key === undefined && descriptor === undefined) {
      if (!target.prototype.router) {
        target.prototype.router = new Router();
      }
      if (middleware.length > 0) {
        target.prototype.router.use(...middleware);
      }
      router.use(path, target.prototype.router.routes(), target.prototype.router.allowedMethods());
      return;
    } else if (typeof target === 'object') {
      if (!target.router) {
        target.router = new Router();
      }
      const handleReturnMiddleware = async function (ctx: Context) {
        await formatResponse(descriptor, ctx);
      };
      // Decorator applied to member (method or property).
      switch (method) {
        case HttpMethod.HEAD:
          target.router.head(path, ...middleware, handleReturnMiddleware);
          break;
        case HttpMethod.OPTIONS:
          target.router.options(path, ...middleware, handleReturnMiddleware);
          break;
        case HttpMethod.GET:
          target.router.get(path, ...middleware, handleReturnMiddleware);
          break;
        case HttpMethod.PUT:
          target.router.put(path, ...middleware, handleReturnMiddleware);
          break;
        case HttpMethod.PATCH:
          target.router.patch(path, ...middleware, handleReturnMiddleware);
          break;
        case HttpMethod.POST:
          target.router.post(path, ...middleware, handleReturnMiddleware);
          break;
        case HttpMethod.DELETE:
          target.router.del(path, ...middleware, handleReturnMiddleware);
          break;
        default:
          target.router.all(path, ...middleware, handleReturnMiddleware);
          break;
      }
    }
  };
}

// 加载所有controller文件
export function load(controllersDir: string, extension?: string): Router {
  getFiles(controllersDir).forEach((file) => {
    if (file.endsWith(extension || '.js')) {
      require(file);
    }
  });
  return router;
}