diff --git a/app.go b/app.go index dc8fb1f5..c8d91d78 100644 --- a/app.go +++ b/app.go @@ -10,19 +10,22 @@ import ( "github.com/astaxie/beego/context" ) +// FilterFunc defines filter function type. type FilterFunc func(*context.Context) +// App defines beego application with a new PatternServeMux. type App struct { Handlers *ControllerRegistor } -// New returns a new PatternServeMux. +// NewApp returns a new beego application. func NewApp() *App { cr := NewControllerRegistor() app := &App{Handlers: cr} return app } +// Run beego application. func (app *App) Run() { addr := HttpAddr @@ -84,39 +87,77 @@ func (app *App) Run() { } } +// Router adds a url-patterned controller handler. +// The path argument supports regex rules and specific placeholders. +// The c argument needs a controller handler implemented beego.ControllerInterface. +// The mapping methods argument only need one string to define custom router rules. +// usage: +// simple router +// beego.Router("/admin", &admin.UserController{}) +// beego.Router("/admin/index", &admin.ArticleController{}) +// +// regex router +// +// beego.Router(“/api/:id([0-9]+)“, &controllers.RController{}) +// +// custom rules +// beego.Router("/api/list",&RestController{},"*:ListFood") +// beego.Router("/api/create",&RestController{},"post:CreateFood") +// beego.Router("/api/update",&RestController{},"put:UpdateFood") +// beego.Router("/api/delete",&RestController{},"delete:DeleteFood") func (app *App) Router(path string, c ControllerInterface, mappingMethods ...string) *App { app.Handlers.Add(path, c, mappingMethods...) return app } +// AutoRouter adds beego-defined controller handler. +// if beego.AddAuto(&MainContorlller{}) and MainController has methods List and Page, +// visit the url /main/list to exec List function or /main/page to exec Page function. func (app *App) AutoRouter(c ControllerInterface) *App { app.Handlers.AddAuto(c) return app } +// UrlFor does another controller handler with params in current context. +// The endpoint is formed as path.controller.name to defined the controller method which will run. +// The values need key-pair data to assign into controller method. func (app *App) UrlFor(endpoint string, values ...string) string { return app.Handlers.UrlFor(endpoint, values...) } + +// [Deprecated] use InsertFilter. +// Filter adds a FilterFunc under pattern condition and named action. +// The actions contains BeforeRouter,AfterStatic,BeforeExec,AfterExec and FinishRouter. func (app *App) Filter(pattern, action string, filter FilterFunc) *App { app.Handlers.AddFilter(pattern, action, filter) return app } +// InsertFilter adds a FilterFunc with pattern condition and action constant. +// The pos means action constant including +// beego.BeforeRouter, beego.AfterStatic, beego.BeforeExec, beego.AfterExec and beego.FinishRouter. func (app *App) InsertFilter(pattern string, pos int, filter FilterFunc) *App { app.Handlers.InsertFilter(pattern, pos, filter) return app } +// SetViewsPath sets view directory path in beego application. +// it returns beego application self. func (app *App) SetViewsPath(path string) *App { ViewsPath = path return app } +// SetStaticPath sets static directory path and proper url pattern in beego application. +// if beego.SetStaticPath("static","public"), visit /static/* to load static file in folder "public". +// it returns beego application self. func (app *App) SetStaticPath(url string, path string) *App { StaticDir[url] = path return app } +// DelStaticPath removes the static folder setting in this url pattern in beego application. +// it returns beego application self. func (app *App) DelStaticPath(url string) *App { delete(StaticDir, url) return app diff --git a/beego.go b/beego.go index 1b16a422..b2d22cc2 100644 --- a/beego.go +++ b/beego.go @@ -10,34 +10,50 @@ import ( "github.com/astaxie/beego/session" ) +// beego web framework version. const VERSION = "1.0.0" +// Router adds a patterned controller handler to BeeApp. +// it's an alias method of App.Router. func Router(rootpath string, c ControllerInterface, mappingMethods ...string) *App { BeeApp.Router(rootpath, c, mappingMethods...) return BeeApp } +// RESTRouter adds a restful controller handler to BeeApp. +// its' controller implements beego.ControllerInterface and +// defines a param "pattern/:objectId" to visit each resource. func RESTRouter(rootpath string, c ControllerInterface) *App { Router(rootpath, c) Router(path.Join(rootpath, ":objectId"), c) return BeeApp } +// AutoRouter adds defined controller handler to BeeApp. +// it's same to App.AutoRouter. func AutoRouter(c ControllerInterface) *App { BeeApp.AutoRouter(c) return BeeApp } +// ErrorHandler registers http.HandlerFunc to each http err code string. +// usage: +// beego.ErrorHandler("404",NotFound) +// beego.ErrorHandler("500",InternalServerError) func Errorhandler(err string, h http.HandlerFunc) *App { middleware.Errorhandler(err, h) return BeeApp } +// SetViewsPath sets view directory to BeeApp. +// it's alias of App.SetViewsPath. func SetViewsPath(path string) *App { BeeApp.SetViewsPath(path) return BeeApp } +// SetStaticPath sets static directory and url prefix to BeeApp. +// it's alias of App.SetStaticPath. func SetStaticPath(url string, path string) *App { if !strings.HasPrefix(url, "/") { url = "/" + url @@ -46,27 +62,33 @@ func SetStaticPath(url string, path string) *App { return BeeApp } +// DelStaticPath removes the static folder setting in this url pattern in beego application. +// it's alias of App.DelStaticPath. func DelStaticPath(url string) *App { delete(StaticDir, url) return BeeApp } -//!!DEPRECATED!! use InsertFilter -//action has four values: -//BeforRouter -//AfterStatic -//BeforExec -//AfterExec +// [Deprecated] use InsertFilter. +// Filter adds a FilterFunc under pattern condition and named action. +// The actions contains BeforeRouter,AfterStatic,BeforeExec,AfterExec and FinishRouter. +// it's alias of App.Filter. func AddFilter(pattern, action string, filter FilterFunc) *App { BeeApp.Filter(pattern, action, filter) return BeeApp } +// InsertFilter adds a FilterFunc with pattern condition and action constant. +// The pos means action constant including +// beego.BeforeRouter, beego.AfterStatic, beego.BeforeExec, beego.AfterExec and beego.FinishRouter. +// it's alias of App.InsertFilter. func InsertFilter(pattern string, pos int, filter FilterFunc) *App { BeeApp.InsertFilter(pattern, pos, filter) return BeeApp } +// Run beego application. +// it's alias of App.Run. func Run() { // if AppConfigPath not In the conf/app.conf reParse config if AppConfigPath != filepath.Join(AppPath, "conf", "app.conf") { diff --git a/config.go b/config.go index 3a03a45c..9baf9aa5 100644 --- a/config.go +++ b/config.go @@ -14,54 +14,53 @@ import ( ) var ( - BeeApp *App + BeeApp *App // beego application AppName string AppPath string AppConfigPath string StaticDir map[string]string - TemplateCache map[string]*template.Template - StaticExtensionsToGzip []string //Files which should also be compressed with gzip (.js, .css, etc) + TemplateCache map[string]*template.Template // template caching map + StaticExtensionsToGzip []string // files with should be compressed with gzip (.js,.css,etc) HttpAddr string HttpPort int HttpTLS bool HttpCertFile string HttpKeyFile string - RecoverPanic bool - AutoRender bool + RecoverPanic bool // flag of auto recover panic + AutoRender bool // flag of render template automatically ViewsPath string - RunMode string //"dev" or "prod" + RunMode string // run mode, "dev" or "prod" AppConfig config.ConfigContainer - //related to session - GlobalSessions *session.Manager //GlobalSessions - SessionOn bool // whether auto start session,default is false - SessionProvider string // default session provider memory mysql redis - SessionName string // sessionName cookie's name - SessionGCMaxLifetime int64 // session's gc maxlifetime - SessionSavePath string // session savepath if use mysql/redis/file this set to the connectinfo - SessionHashFunc string - SessionHashKey string - SessionCookieLifeTime int - UseFcgi bool - MaxMemory int64 - EnableGzip bool // enable gzip - DirectoryIndex bool //enable DirectoryIndex default is false - EnableHotUpdate bool //enable HotUpdate default is false - HttpServerTimeOut int64 //set httpserver timeout - ErrorsShow bool //set weather show errors - XSRFKEY string //set XSRF - EnableXSRF bool - XSRFExpire int - CopyRequestBody bool //When in raw application, You want to the reqeustbody - TemplateLeft string - TemplateRight string - BeegoServerName string - EnableAdmin bool //enable admin module to log api time - AdminHttpAddr string //admin module http addr - AdminHttpPort int + GlobalSessions *session.Manager // global session mananger + SessionOn bool // flag of starting session auto. default is false. + SessionProvider string // default session provider, memory, mysql , redis ,etc. + SessionName string // the cookie name when saving session id into cookie. + SessionGCMaxLifetime int64 // session gc time for auto cleaning expired session. + SessionSavePath string // if use mysql/redis/file provider, define save path to connection info. + SessionHashFunc string // session hash generation func. + SessionHashKey string // session hash salt string. + SessionCookieLifeTime int // the life time of session id in cookie. + UseFcgi bool + MaxMemory int64 + EnableGzip bool // flag of enable gzip + DirectoryIndex bool // flag of display directory index. default is false. + EnableHotUpdate bool // flag of hot update checking by app self. default is false. + HttpServerTimeOut int64 + ErrorsShow bool // flag of show errors in page. if true, show error and trace info in page rendered with error template. + XSRFKEY string // xsrf hash salt string. + EnableXSRF bool // flag of enable xsrf. + XSRFExpire int // the expiry of xsrf value. + CopyRequestBody bool // flag of copy raw request body in context. + TemplateLeft string + TemplateRight string + BeegoServerName string // beego server name exported in response header. + EnableAdmin bool // flag of enable admin module to log every request info. + AdminHttpAddr string // http server configurations for admin module. + AdminHttpPort int ) func init() { - // create beeapp + // create beego application BeeApp = NewApp() // initialize default configurations @@ -135,7 +134,8 @@ func init() { } } -//parse config now only support ini, next will support json +// ParseConfig parsed default config file. +// now only support ini, next will support json. func ParseConfig() (err error) { AppConfig, err = config.NewConfig("ini", AppConfigPath) if err != nil { diff --git a/controller.go b/controller.go index f33d38dc..f14593a5 100644 --- a/controller.go +++ b/controller.go @@ -25,9 +25,12 @@ import ( ) var ( + // custom error when user stop request handler manually. USERSTOPRUN = errors.New("User stop run") ) +// Controller defines some basic http request handler operations, such as +// http context, template and view, session and xsrf. type Controller struct { Ctx *context.Context Data map[interface{}]interface{} @@ -43,6 +46,7 @@ type Controller struct { AppController interface{} } +// ControllerInterface is an interface to uniform all controller handler. type ControllerInterface interface { Init(ct *context.Context, controllerName, actionName string, app interface{}) Prepare() @@ -59,6 +63,7 @@ type ControllerInterface interface { CheckXsrfCookie() bool } +// Init generates default values of controller operations. func (c *Controller) Init(ctx *context.Context, controllerName, actionName string, app interface{}) { c.Data = make(map[interface{}]interface{}) c.Layout = "" @@ -70,42 +75,52 @@ func (c *Controller) Init(ctx *context.Context, controllerName, actionName strin c.AppController = app } +// Prepare runs after Init before request function execution. func (c *Controller) Prepare() { } +// Finish runs after request function execution. func (c *Controller) Finish() { } +// Get adds a request function to handle GET request. func (c *Controller) Get() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Post adds a request function to handle POST request. func (c *Controller) Post() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Delete adds a request function to handle DELETE request. func (c *Controller) Delete() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Put adds a request function to handle PUT request. func (c *Controller) Put() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Head adds a request function to handle HEAD request. func (c *Controller) Head() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Patch adds a request function to handle PATCH request. func (c *Controller) Patch() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Options adds a request function to handle OPTIONS request. func (c *Controller) Options() { http.Error(c.Ctx.ResponseWriter, "Method Not Allowed", 405) } +// Render sends the response with rendered template bytes as text/html type. func (c *Controller) Render() error { rb, err := c.RenderBytes() @@ -118,11 +133,13 @@ func (c *Controller) Render() error { return nil } +// RenderString returns the rendered template string. Do not send out response. func (c *Controller) RenderString() (string, error) { b, e := c.RenderBytes() return string(b), e } +// RenderBytes returns the bytes of renderd tempate string. Do not send out response. func (c *Controller) RenderBytes() ([]byte, error) { //if the controller has set layout, then first get the tplname's content set the content to the layout if c.Layout != "" { @@ -175,10 +192,12 @@ func (c *Controller) RenderBytes() ([]byte, error) { return []byte{}, nil } +// Redirect sends the redirection response to url with status code. func (c *Controller) Redirect(url string, code int) { c.Ctx.Redirect(code, url) } +// Aborts stops controller handler and show the error data if code is defined in ErrorMap or code string. func (c *Controller) Abort(code string) { status, err := strconv.Atoi(code) if err == nil { @@ -188,10 +207,13 @@ func (c *Controller) Abort(code string) { } } +// StopRun makes panic of USERSTOPRUN error and go to recover function if defined. func (c *Controller) StopRun() { panic(USERSTOPRUN) } +// UrlFor does another controller handler in this request function. +// it goes to this controller method if endpoint is not clear. func (c *Controller) UrlFor(endpoint string, values ...string) string { if len(endpoint) <= 0 { return "" @@ -201,8 +223,10 @@ func (c *Controller) UrlFor(endpoint string, values ...string) string { } else { return UrlFor(endpoint, values...) } + return "" } +// ServeJson sends a json response with encoding charset. func (c *Controller) ServeJson(encoding ...bool) { var hasIndent bool var hasencoding bool @@ -217,6 +241,7 @@ func (c *Controller) ServeJson(encoding ...bool) { c.Ctx.Output.Json(c.Data["json"], hasIndent, hasencoding) } +// ServeJson sends a jsonp response. func (c *Controller) ServeJsonp() { var hasIndent bool if RunMode == "prod" { @@ -227,6 +252,7 @@ func (c *Controller) ServeJsonp() { c.Ctx.Output.Jsonp(c.Data["jsonp"], hasIndent) } +// ServeJson sends xml response. func (c *Controller) ServeXml() { var hasIndent bool if RunMode == "prod" { @@ -237,6 +263,7 @@ func (c *Controller) ServeXml() { c.Ctx.Output.Xml(c.Data["xml"], hasIndent) } +// Input returns the input data map from POST or PUT request body and query string. func (c *Controller) Input() url.Values { ct := c.Ctx.Request.Header.Get("Content-Type") if strings.Contains(ct, "multipart/form-data") { @@ -247,14 +274,18 @@ func (c *Controller) Input() url.Values { return c.Ctx.Request.Form } +// ParseForm maps input data map to obj struct. func (c *Controller) ParseForm(obj interface{}) error { return ParseForm(c.Input(), obj) } +// GetString returns the input value by key string. func (c *Controller) GetString(key string) string { return c.Input().Get(key) } +// GetStrings returns the input string slice by key string. +// it's designed for multi-value input field such as checkbox(input[type=checkbox]), multi-selection. func (c *Controller) GetStrings(key string) []string { r := c.Ctx.Request if r.Form == nil { @@ -267,22 +298,29 @@ func (c *Controller) GetStrings(key string) []string { return []string{} } +// GetInt returns input value as int64. func (c *Controller) GetInt(key string) (int64, error) { return strconv.ParseInt(c.Input().Get(key), 10, 64) } +// GetBool returns input value as bool. func (c *Controller) GetBool(key string) (bool, error) { return strconv.ParseBool(c.Input().Get(key)) } +// GetFloat returns input value as float64. func (c *Controller) GetFloat(key string) (float64, error) { return strconv.ParseFloat(c.Input().Get(key), 64) } +// GetFile returns the file data in file upload field named as key. +// it returns the first one of multi-uploaded files. func (c *Controller) GetFile(key string) (multipart.File, *multipart.FileHeader, error) { return c.Ctx.Request.FormFile(key) } +// SaveToFile saves uploaded file to new path. +// it only operates the first one of mutil-upload form file field. func (c *Controller) SaveToFile(fromfile, tofile string) error { file, _, err := c.Ctx.Request.FormFile(fromfile) if err != nil { @@ -298,6 +336,7 @@ func (c *Controller) SaveToFile(fromfile, tofile string) error { return nil } +// StartSession starts session and load old session data info this controller. func (c *Controller) StartSession() session.SessionStore { if c.CruSession == nil { c.CruSession = c.Ctx.Input.CruSession @@ -305,6 +344,7 @@ func (c *Controller) StartSession() session.SessionStore { return c.CruSession } +// SetSession puts value into session. func (c *Controller) SetSession(name interface{}, value interface{}) { if c.CruSession == nil { c.StartSession() @@ -312,6 +352,7 @@ func (c *Controller) SetSession(name interface{}, value interface{}) { c.CruSession.Set(name, value) } +// GetSession gets value from session. func (c *Controller) GetSession(name interface{}) interface{} { if c.CruSession == nil { c.StartSession() @@ -319,6 +360,7 @@ func (c *Controller) GetSession(name interface{}) interface{} { return c.CruSession.Get(name) } +// SetSession removes value from session. func (c *Controller) DelSession(name interface{}) { if c.CruSession == nil { c.StartSession() @@ -326,19 +368,24 @@ func (c *Controller) DelSession(name interface{}) { c.CruSession.Delete(name) } +// SessionRegenerateID regenerates session id for this session. +// the session data have no changes. func (c *Controller) SessionRegenerateID() { c.CruSession = GlobalSessions.SessionRegenerateId(c.Ctx.ResponseWriter, c.Ctx.Request) c.Ctx.Input.CruSession = c.CruSession } +// DestroySession cleans session data and session cookie. func (c *Controller) DestroySession() { GlobalSessions.SessionDestroy(c.Ctx.ResponseWriter, c.Ctx.Request) } +// IsAjax returns this request is ajax or not. func (c *Controller) IsAjax() bool { return c.Ctx.Input.IsAjax() } +// GetSecureCookie returns decoded cookie value from encoded browser cookie values. func (c *Controller) GetSecureCookie(Secret, key string) (string, bool) { val := c.Ctx.GetCookie(key) if val == "" { @@ -365,6 +412,7 @@ func (c *Controller) GetSecureCookie(Secret, key string) (string, bool) { return string(res), true } +// SetSecureCookie puts value into cookie after encoded the value. func (c *Controller) SetSecureCookie(Secret, name, val string, age int64) { vs := base64.URLEncoding.EncodeToString([]byte(val)) timestamp := strconv.FormatInt(time.Now().UnixNano(), 10) @@ -375,6 +423,7 @@ func (c *Controller) SetSecureCookie(Secret, name, val string, age int64) { c.Ctx.SetCookie(name, cookie, age, "/") } +// XsrfToken creates a xsrf token string and returns. func (c *Controller) XsrfToken() string { if c._xsrf_token == "" { token, ok := c.GetSecureCookie(XSRFKEY, "_xsrf") @@ -393,6 +442,9 @@ func (c *Controller) XsrfToken() string { return c._xsrf_token } +// CheckXsrfCookie checks xsrf token in this request is valid or not. +// the token can provided in request header "X-Xsrftoken" and "X-CsrfToken" +// or in form field value named as "_xsrf". func (c *Controller) CheckXsrfCookie() bool { token := c.GetString("_xsrf") if token == "" { @@ -409,16 +461,18 @@ func (c *Controller) CheckXsrfCookie() bool { return true } +// XsrfFormHtml writes an input field contains xsrf token value. func (c *Controller) XsrfFormHtml() string { return "" } +// GetControllerAndAction gets the executing controller name and action name. func (c *Controller) GetControllerAndAction() (controllerName, actionName string) { return c.controllerName, c.actionName } -//utils func for controller internal +// getRandomString returns random string. func getRandomString(n int) string { const alphanum = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz" var bytes = make([]byte, n) diff --git a/filter.go b/filter.go index 05e5fd49..7e0245a6 100644 --- a/filter.go +++ b/filter.go @@ -5,6 +5,8 @@ import ( "strings" ) +// FilterRouter defines filter operation before controller handler execution. +// it can match patterned url and do filter function when action arrives. type FilterRouter struct { pattern string regex *regexp.Regexp @@ -14,6 +16,8 @@ type FilterRouter struct { parseParams map[string]string } +// ValidRouter check current request is valid for this filter. +// if matched, returns parsed params in this request by defined filter router pattern. func (mr *FilterRouter) ValidRouter(router string) (bool, map[string]string) { if mr.pattern == "" { return true, nil diff --git a/flash.go b/flash.go index 5a670176..f435c155 100644 --- a/flash.go +++ b/flash.go @@ -6,18 +6,22 @@ import ( "strings" ) +// the separation string when encoding flash data. const BEEGO_FLASH_SEP = "#BEEGOFLASH#" +// FlashData is a tools to maintain data when using across request. type FlashData struct { Data map[string]string } +// NewFlash return a new empty FlashData struct. func NewFlash() *FlashData { return &FlashData{ Data: make(map[string]string), } } +// Notice writes notice message to flash. func (fd *FlashData) Notice(msg string, args ...interface{}) { if len(args) == 0 { fd.Data["notice"] = msg @@ -26,6 +30,7 @@ func (fd *FlashData) Notice(msg string, args ...interface{}) { } } +// Warning writes warning message to flash. func (fd *FlashData) Warning(msg string, args ...interface{}) { if len(args) == 0 { fd.Data["warning"] = msg @@ -34,6 +39,7 @@ func (fd *FlashData) Warning(msg string, args ...interface{}) { } } +// Error writes error message to flash. func (fd *FlashData) Error(msg string, args ...interface{}) { if len(args) == 0 { fd.Data["error"] = msg @@ -42,6 +48,8 @@ func (fd *FlashData) Error(msg string, args ...interface{}) { } } +// Store does the saving operation of flash data. +// the data are encoded and saved in cookie. func (fd *FlashData) Store(c *Controller) { c.Data["flash"] = fd.Data var flashValue string @@ -51,6 +59,7 @@ func (fd *FlashData) Store(c *Controller) { c.Ctx.SetCookie("BEEGO_FLASH", url.QueryEscape(flashValue), 0, "/") } +// ReadFromRequest parsed flash data from encoded values in cookie. func ReadFromRequest(c *Controller) *FlashData { flash := &FlashData{ Data: make(map[string]string),