From f7e7fab6f22180b4ca6f924f7807f81cb2c173ae Mon Sep 17 00:00:00 2001
From: astaxie <xiemengjun@gmail.com>
Date: Sat, 27 Jul 2013 10:25:14 +0800
Subject: [PATCH] support autorouter

---
 beego.go  | 10 ++++++++
 router.go | 72 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 81 insertions(+), 1 deletion(-)

diff --git a/beego.go b/beego.go
index c1354828..66ebf6de 100644
--- a/beego.go
+++ b/beego.go
@@ -138,6 +138,11 @@ func (app *App) Router(path string, c ControllerInterface, mappingMethods ...str
 	return app
 }
 
+func (app *App) AutoRouter(c ControllerInterface) *App {
+	app.Handlers.AddAuto(c)
+	return app
+}
+
 func (app *App) Filter(filter http.HandlerFunc) *App {
 	app.Handlers.Filter(filter)
 	return app
@@ -192,6 +197,11 @@ func RESTRouter(rootpath string, c ControllerInterface) *App {
 	return BeeApp
 }
 
+func AutoRouter(c ControllerInterface) *App {
+	BeeApp.AutoRouter(c)
+	return BeeApp
+}
+
 func RouterHandler(path string, c http.Handler) *App {
 	BeeApp.Handlers.AddHandler(path, c)
 	return BeeApp
diff --git a/router.go b/router.go
index e7409ecc..1892ce13 100644
--- a/router.go
+++ b/router.go
@@ -10,6 +10,7 @@ import (
 	"reflect"
 	"regexp"
 	"runtime"
+	"strconv"
 	"strings"
 )
 
@@ -35,10 +36,15 @@ type ControllerRegistor struct {
 	fixrouters   []*controllerInfo
 	filters      []http.HandlerFunc
 	userHandlers map[string]*userHandler
+	autoRouter   map[string]map[string]reflect.Type //key:controller key:method value:reflect.type
 }
 
 func NewControllerRegistor() *ControllerRegistor {
-	return &ControllerRegistor{routers: make([]*controllerInfo, 0), userHandlers: make(map[string]*userHandler)}
+	return &ControllerRegistor{
+		routers:      make([]*controllerInfo, 0),
+		userHandlers: make(map[string]*userHandler),
+		autoRouter:   make(map[string]map[string]reflect.Type),
+	}
 }
 
 //methods support like this:
@@ -148,6 +154,21 @@ func (p *ControllerRegistor) Add(pattern string, c ControllerInterface, mappingM
 	}
 }
 
+func (p *ControllerRegistor) AddAuto(c ControllerInterface) {
+	reflectVal := reflect.ValueOf(c)
+	rt := reflectVal.Type()
+	ct := reflect.Indirect(reflectVal).Type()
+	firstParam := strings.ToLower(strings.TrimSuffix(ct.Name(), "Controller"))
+	if _, ok := p.autoRouter[firstParam]; ok {
+		return
+	} else {
+		p.autoRouter[firstParam] = make(map[string]reflect.Type)
+	}
+	for i := 0; i < rt.NumMethod(); i++ {
+		p.autoRouter[firstParam][rt.Method(i).Name] = ct
+	}
+}
+
 func (p *ControllerRegistor) AddHandler(pattern string, c http.Handler) {
 	parts := strings.Split(pattern, "/")
 
@@ -518,6 +539,55 @@ func (p *ControllerRegistor) ServeHTTP(rw http.ResponseWriter, r *http.Request)
 		method.Call(in)
 	}
 
+	//start autorouter
+
+	if !findrouter {
+		for cName, methodmap := range p.autoRouter {
+			if strings.HasPrefix(strings.ToLower(requestPath), "/"+cName) {
+				for mName, controllerType := range methodmap {
+					if strings.HasPrefix(strings.ToLower(requestPath), "/"+cName+"/"+strings.ToLower(mName)) {
+						//parse params
+						otherurl := requestPath[len("/"+cName+"/"+strings.ToLower(mName)):]
+						if len(otherurl) > 1 && otherurl[0] != '/' {
+							plist := strings.Split(otherurl, "/")
+							for k, v := range plist {
+								params[strconv.Itoa(k)] = v
+							}
+						}
+						//Invoke the request handler
+						vc := reflect.New(controllerType)
+
+						//call the controller init function
+						init := vc.MethodByName("Init")
+						in := make([]reflect.Value, 2)
+						ct := &Context{ResponseWriter: w, Request: r, Params: params, RequestBody: requestbody}
+
+						in[0] = reflect.ValueOf(ct)
+						in[1] = reflect.ValueOf(controllerType.Name())
+						init.Call(in)
+						//call prepare function
+						in = make([]reflect.Value, 0)
+						method := vc.MethodByName("Prepare")
+						method.Call(in)
+						method = vc.MethodByName(mName)
+						method.Call(in)
+						if !w.started {
+							if AutoRender {
+								method = vc.MethodByName("Render")
+								method.Call(in)
+							}
+							method = vc.MethodByName("Finish")
+							method.Call(in)
+						}
+						method = vc.MethodByName("Destructor")
+						method.Call(in)
+						// set find
+						findrouter = true
+					}
+				}
+			}
+		}
+	}
 	//if no matches to url, throw a not found exception
 	if !findrouter {
 		if h, ok := ErrorMaps["404"]; ok {