feat: add all route

Author: hubvue

src/decorator/httpRouter.ts

@@ -7,9 +7,10 @@ export const Get = createHTTPMethodDecorator('get')
 export const Post = createHTTPMethodDecorator('post')
 export const Put = createHTTPMethodDecorator('put')
 export const Delete = createHTTPMethodDecorator('delete')
+export const All = createHTTPMethodDecorator('get', 'post', 'put', 'delete')
 
 
-function createHTTPMethodDecorator(method: string) {
+function createHTTPMethodDecorator(...methods: Array<string>) {
   return (path: string, aopPlugins: RouteAop = {}) => (
     target: any,
     propertyKey: string,
@@ -30,7 +31,7 @@ function createHTTPMethodDecorator(method: string) {
       }
     }
     Reflect.defineMetadata(META_ROUTER,{
-      method,
+      methods,
       path,
       beforePlugins,
       afterPlugins

src/index.ts

@@ -1,7 +1,7 @@
 import 'reflect-metadata'
 
 import {Inject, Provide} from './ioc'
-import {Get,Post,Put,Delete} from './decorator/httpRouter'
+import {Get,Post,Put,Delete, All} from './decorator/httpRouter'
 import {createApplication, Controller} from './core'
 
 export {
@@ -12,5 +12,6 @@ export {
   Post,
   Put,
   Delete,
+  All,
   Controller
 }

src/runtime/koa/index.ts

@@ -5,6 +5,7 @@ import {
   META_CONTROLLER,
   META_ROUTER
 } from '../../constants'
+import {resolvePath} from '../../utils'
 
 function KoaRuntime(controllers: Set<any>, options:RuntimeOptions) {
   const app = new Koa()
@@ -16,17 +17,23 @@ function KoaRuntime(controllers: Set<any>, options:RuntimeOptions) {
     }
   }
   for(let controller of controllers) {
-    console.log(controller)
     const rootPath = Reflect.getMetadata(META_CONTROLLER,controller.constructor)
+    if(!rootPath) {
+      throw new Error('this class is not controller')
+    }
     Object.getOwnPropertyNames(Object.getPrototypeOf(controller))
       .filter(name => name !== 'constructor')
       .forEach(name => {
         const metaRoute = Reflect.getMetadata(META_ROUTER,controller[name])
         if(metaRoute){
-          const {method, path, beforePlugins, afterPlugins} = metaRoute
-          router[method](`${rootPath}${path}`,...beforePlugins, async (ctx, next) => {
-            await controller[name](ctx, next)
-          }, ...afterPlugins)
+          const {methods, path, beforePlugins, afterPlugins} = metaRoute
+          const routePath = resolvePath(rootPath, path)
+          for(let method of methods) {
+            router[method](routePath,...beforePlugins, async (ctx, next) => {
+              // TODO
+              await controller[name](ctx, next)
+            }, ...afterPlugins)
+          }
         }
       })
   }

src/utils/index.ts

@@ -0,0 +1,6 @@
+export const resolvePath = (rootPath: string, path: string) => {
+  if(rootPath === '/') {
+    return path;
+  }
+  return `${rootPath}${path}`
+}

test/app.ts

@@ -2,7 +2,7 @@ import './loader'
 import {createApplication} from '../src/index'
 
 async function test(ctx, next: Function): Promise<any> {
-  console.log('golbal')
+  console.log(ctx)
   await next()
   return;
 }

test/controller.ts

@@ -4,6 +4,7 @@ import {
   Post,
   Put,
   Delete,
+  All,
   Controller
 } from '../src/index'
 import {UserInstance} from './constants'
@@ -15,12 +16,13 @@ export class UserController {
   constructor(user) {
     this._user = user;
   }
-  @Get('/', {
+  @Get('/getUser', {
     before: [before1, before2, before3],
     after: [after1, after2, after3]
   })
   async getUser(ctx: any, next: Function): Promise<any> {
-    console.log('middle')
+    // console.log(ctx)
+    // console.log('middle')
     const result = this._user.getUser(1);
     ctx.body = {
       code: 200,
@@ -30,6 +32,25 @@ export class UserController {
   }
 }
 
+
+@Inject(UserInstance)
+@Controller()
+export class TestController {
+  private _user
+  constructor(user) {
+    this._user = user
+  }
+  @Get('/getTestUser')
+  async getUser(ctx: any, next: Function): Promise<any> {
+    const result = this._user.getUser(2);
+    ctx.body = {
+      code: 200,
+      data: result
+    };
+  }
+
+}
+
 async function before1(ctx, next) {
   console.log(1)
   await next()