feat: add koa router

Author: hubvue

src/core/index.ts

@@ -1,14 +1,13 @@
 import {instancePoll} from '../ioc'
 import {
   META_INJECT,
-  META_PROVIDE
-} from '../constants'
-import {
+  META_PROVIDE,
   META_CONTROLLER
 } from '../constants'
+import { RuntimeOptions } from '../interface'
 import KoaRuntime from '../runtime/koa'
 
-export const createApplication = () => {
+export const createApplication = (options: RuntimeOptions = {}) => {
   const controllers: Set<any> = new Set();
   const controllerPoll: Map<any, any> = instancePoll.getAll(META_INJECT)
   for(let [Controller, tag] of controllerPoll) {
@@ -21,12 +20,13 @@ export const createApplication = () => {
     }
 
   }
-  const app = KoaRuntime(controllers)
+  const app = KoaRuntime(controllers, options)
   return app
 }
 
 export const Controller = (path: string = '/') => {
   return function<T extends { new (...args: any[]): {} }>(constructor: T) {
+    console.log(constructor)
     Reflect.defineMetadata(META_CONTROLLER, path, constructor)
     return constructor
   }

src/decorator/httpRouter.ts

@@ -1,7 +1,7 @@
 import {
   META_ROUTER,
 } from '../constants'
-
+import { RouteAop } from '../interface'
 // create http request method decorator
 export const Get = createHTTPMethodDecorator('get')
 export const Post = createHTTPMethodDecorator('post')

src/interface/index.ts

@@ -0,0 +1,13 @@
+export interface RuntimeOptions {
+  plugins?: Array<Function>
+}
+
+export interface InstanceMeta {
+  key: symbol
+  value: (new () => {})
+}
+
+export interface RouteAop {
+  before?: Array<Function>;
+  after?: Array<Function>
+}

src/ioc/index.ts

@@ -3,18 +3,13 @@ import {
   META_PROVIDE,
   META_INJECT,
 } from '../constants'
-
-interface InstanceMeta {
-  key: symbol;
-  value: (new () => {})
-}
+import {InstanceMeta} from '../interface'
 
 class InstancePoll {
   private providePoll: Map<any, any> = new Map();
   private injectPoll: Map<any, any> = new Map();
   add(type: Symbol, instanceMeta: InstanceMeta) {
     if(type === META_PROVIDE) {
-      console.log(instanceMeta)
       this.providePoll.set(instanceMeta.key, instanceMeta.value)
     }
     if(type === META_INJECT) {
@@ -60,7 +55,7 @@ export const Inject = createIocDecorator(META_INJECT)
 
 
 function createIocDecorator(type) {
-  return (tag) => target => {
+  return tag => target => {
     instancePoll.add(type, {
       key:tag,
       value: target

src/runtime/koa/index.ts

@@ -1,8 +1,37 @@
 import * as Koa from 'koa'
+import * as Router from 'koa-router'
+import { RuntimeOptions } from '../../interface'
+import {
+  META_CONTROLLER,
+  META_ROUTER
+} from '../../constants'
 
-function KoaRuntime(controllers: Set<any>) {
+function KoaRuntime(controllers: Set<any>, options:RuntimeOptions) {
   const app = new Koa()
-  console.log(controllers)
+  const router = new Router()
+  const plugins: Array<Function> = options.plugins || []
+  if(plugins.length) {
+    for(let plugin of plugins){
+      app.use(plugin)
+    }
+  }
+  for(let controller of controllers) {
+    console.log(controller)
+    const rootPath = Reflect.getMetadata(META_CONTROLLER,controller.constructor)
+    Object.getOwnPropertyNames(Object.getPrototypeOf(controller))
+      .filter(name => name !== 'constructor')
+      .forEach(name => {
+        const metaRoute = Reflect.getMetadata(META_ROUTER,controller[name])
+        if(metaRoute){
+          router[metaRoute.method](`${rootPath}${metaRoute.path}`, async (ctx, next) => {
+            await controller[name](ctx, next)
+          },async(ctx,next) => {
+            console.log(123)
+          })
+        }
+      })
+  }
+  app.use(router.routes()).use(router.allowedMethods())
   return app
 }
 

test/app.ts

@@ -1,7 +1,16 @@
 import './loader'
 import {createApplication} from '../src/index'
 
-const app = createApplication()
+async function test(ctx, next: Function): Promise<any> {
+  console.log('golbal')
+  await next()
+  return;
+}
+
+
+const app = createApplication({
+  plugins: [test]
+})
 
 
 app.listen(8080, () => {

test/controller.ts

@@ -16,12 +16,12 @@ export class UserController {
     this._user = user;
   }
   @Get('/')
-  async getUser(ctx: any, next: Promise<any>): Promise<any> {
-    console.log(this)
+  async getUser(ctx: any, next: Function): Promise<any> {
     const result = this._user.getUser(1);
     ctx.body = {
       code: 200,
       data: result
     };
+    await next()
   }
 }