mirror of
https://github.com/beego/bee.git
synced 2024-11-14 22:10:55 +00:00
cf7aef47f0
Moved logging to the new logging infrastructure by removing the use of ColorLog() function. Added more documentation. Also fixed some typos in comments and function names.
322 lines
9.2 KiB
Go
322 lines
9.2 KiB
Go
// Copyright 2013 bee authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
// License for the specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
func generateController(cname, currpath string) {
|
|
w := NewColorWriter(os.Stdout)
|
|
|
|
p, f := path.Split(cname)
|
|
controllerName := strings.Title(f)
|
|
packageName := "controllers"
|
|
|
|
if p != "" {
|
|
i := strings.LastIndex(p[:len(p)-1], "/")
|
|
packageName = p[i+1 : len(p)-1]
|
|
}
|
|
|
|
logger.Infof("Using '%s' as controller name", controllerName)
|
|
logger.Infof("Using '%s' as package name", packageName)
|
|
|
|
fp := path.Join(currpath, "controllers", p)
|
|
if _, err := os.Stat(fp); os.IsNotExist(err) {
|
|
// Create the controller's directory
|
|
if err := os.MkdirAll(fp, 0777); err != nil {
|
|
logger.Fatalf("Could not create controllers directory: %s", err)
|
|
}
|
|
}
|
|
|
|
fpath := path.Join(fp, strings.ToLower(controllerName)+".go")
|
|
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
|
defer CloseFile(f)
|
|
|
|
modelPath := path.Join(currpath, "models", strings.ToLower(controllerName)+".go")
|
|
|
|
var content string
|
|
if _, err := os.Stat(modelPath); err == nil {
|
|
logger.Infof("Using matching model '%s'", controllerName)
|
|
content = strings.Replace(controllerModelTpl, "{{packageName}}", packageName, -1)
|
|
pkgPath := getPackagePath(currpath)
|
|
content = strings.Replace(content, "{{pkgPath}}", pkgPath, -1)
|
|
} else {
|
|
content = strings.Replace(controllerTpl, "{{packageName}}", packageName, -1)
|
|
}
|
|
|
|
content = strings.Replace(content, "{{controllerName}}", controllerName, -1)
|
|
f.WriteString(content)
|
|
|
|
// Run 'gofmt' on the generated source code
|
|
formatSourceCode(fpath)
|
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
|
} else {
|
|
logger.Fatalf("Could not create controller file: %s", err)
|
|
}
|
|
}
|
|
|
|
var controllerTpl = `package {{packageName}}
|
|
|
|
import (
|
|
"github.com/astaxie/beego"
|
|
)
|
|
|
|
// {{controllerName}}Controller operations for {{controllerName}}
|
|
type {{controllerName}}Controller struct {
|
|
beego.Controller
|
|
}
|
|
|
|
// URLMapping ...
|
|
func (c *{{controllerName}}Controller) URLMapping() {
|
|
c.Mapping("Post", c.Post)
|
|
c.Mapping("GetOne", c.GetOne)
|
|
c.Mapping("GetAll", c.GetAll)
|
|
c.Mapping("Put", c.Put)
|
|
c.Mapping("Delete", c.Delete)
|
|
}
|
|
|
|
// Post ...
|
|
// @Title Create
|
|
// @Description create {{controllerName}}
|
|
// @Param body body models.{{controllerName}} true "body for {{controllerName}} content"
|
|
// @Success 201 {object} models.{{controllerName}}
|
|
// @Failure 403 body is empty
|
|
// @router / [post]
|
|
func (c *{{controllerName}}Controller) Post() {
|
|
|
|
}
|
|
|
|
// GetOne ...
|
|
// @Title GetOne
|
|
// @Description get {{controllerName}} by id
|
|
// @Param id path string true "The key for staticblock"
|
|
// @Success 200 {object} models.{{controllerName}}
|
|
// @Failure 403 :id is empty
|
|
// @router /:id [get]
|
|
func (c *{{controllerName}}Controller) GetOne() {
|
|
|
|
}
|
|
|
|
// GetAll ...
|
|
// @Title GetAll
|
|
// @Description get {{controllerName}}
|
|
// @Param query query string false "Filter. e.g. col1:v1,col2:v2 ..."
|
|
// @Param fields query string false "Fields returned. e.g. col1,col2 ..."
|
|
// @Param sortby query string false "Sorted-by fields. e.g. col1,col2 ..."
|
|
// @Param order query string false "Order corresponding to each sortby field, if single value, apply to all sortby fields. e.g. desc,asc ..."
|
|
// @Param limit query string false "Limit the size of result set. Must be an integer"
|
|
// @Param offset query string false "Start position of result set. Must be an integer"
|
|
// @Success 200 {object} models.{{controllerName}}
|
|
// @Failure 403
|
|
// @router / [get]
|
|
func (c *{{controllerName}}Controller) GetAll() {
|
|
|
|
}
|
|
|
|
// Put ...
|
|
// @Title Put
|
|
// @Description update the {{controllerName}}
|
|
// @Param id path string true "The id you want to update"
|
|
// @Param body body models.{{controllerName}} true "body for {{controllerName}} content"
|
|
// @Success 200 {object} models.{{controllerName}}
|
|
// @Failure 403 :id is not int
|
|
// @router /:id [put]
|
|
func (c *{{controllerName}}Controller) Put() {
|
|
|
|
}
|
|
|
|
// Delete ...
|
|
// @Title Delete
|
|
// @Description delete the {{controllerName}}
|
|
// @Param id path string true "The id you want to delete"
|
|
// @Success 200 {string} delete success!
|
|
// @Failure 403 id is empty
|
|
// @router /:id [delete]
|
|
func (c *{{controllerName}}Controller) Delete() {
|
|
|
|
}
|
|
`
|
|
|
|
var controllerModelTpl = `package {{packageName}}
|
|
|
|
import (
|
|
"{{pkgPath}}/models"
|
|
"encoding/json"
|
|
"errors"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/astaxie/beego"
|
|
)
|
|
|
|
// {{controllerName}}Controller oprations for {{controllerName}}
|
|
type {{controllerName}}Controller struct {
|
|
beego.Controller
|
|
}
|
|
|
|
// URLMapping ...
|
|
func (c *{{controllerName}}Controller) URLMapping() {
|
|
c.Mapping("Post", c.Post)
|
|
c.Mapping("GetOne", c.GetOne)
|
|
c.Mapping("GetAll", c.GetAll)
|
|
c.Mapping("Put", c.Put)
|
|
c.Mapping("Delete", c.Delete)
|
|
}
|
|
|
|
// Post ...
|
|
// @Title Post
|
|
// @Description create {{controllerName}}
|
|
// @Param body body models.{{controllerName}} true "body for {{controllerName}} content"
|
|
// @Success 201 {int} models.{{controllerName}}
|
|
// @Failure 403 body is empty
|
|
// @router / [post]
|
|
func (c *{{controllerName}}Controller) Post() {
|
|
var v models.{{controllerName}}
|
|
json.Unmarshal(c.Ctx.Input.RequestBody, &v)
|
|
if _, err := models.Add{{controllerName}}(&v); err == nil {
|
|
c.Ctx.Output.SetStatus(201)
|
|
c.Data["json"] = v
|
|
} else {
|
|
c.Data["json"] = err.Error()
|
|
}
|
|
c.ServeJSON()
|
|
}
|
|
|
|
// GetOne ...
|
|
// @Title Get One
|
|
// @Description get {{controllerName}} by id
|
|
// @Param id path string true "The key for staticblock"
|
|
// @Success 200 {object} models.{{controllerName}}
|
|
// @Failure 403 :id is empty
|
|
// @router /:id [get]
|
|
func (c *{{controllerName}}Controller) GetOne() {
|
|
idStr := c.Ctx.Input.Param(":id")
|
|
id, _ := strconv.ParseInt(idStr, 0, 64)
|
|
v, err := models.Get{{controllerName}}ById(id)
|
|
if err != nil {
|
|
c.Data["json"] = err.Error()
|
|
} else {
|
|
c.Data["json"] = v
|
|
}
|
|
c.ServeJSON()
|
|
}
|
|
|
|
// GetAll ...
|
|
// @Title Get All
|
|
// @Description get {{controllerName}}
|
|
// @Param query query string false "Filter. e.g. col1:v1,col2:v2 ..."
|
|
// @Param fields query string false "Fields returned. e.g. col1,col2 ..."
|
|
// @Param sortby query string false "Sorted-by fields. e.g. col1,col2 ..."
|
|
// @Param order query string false "Order corresponding to each sortby field, if single value, apply to all sortby fields. e.g. desc,asc ..."
|
|
// @Param limit query string false "Limit the size of result set. Must be an integer"
|
|
// @Param offset query string false "Start position of result set. Must be an integer"
|
|
// @Success 200 {object} models.{{controllerName}}
|
|
// @Failure 403
|
|
// @router / [get]
|
|
func (c *{{controllerName}}Controller) GetAll() {
|
|
var fields []string
|
|
var sortby []string
|
|
var order []string
|
|
var query = make(map[string]string)
|
|
var limit int64 = 10
|
|
var offset int64
|
|
|
|
// fields: col1,col2,entity.col3
|
|
if v := c.GetString("fields"); v != "" {
|
|
fields = strings.Split(v, ",")
|
|
}
|
|
// limit: 10 (default is 10)
|
|
if v, err := c.GetInt64("limit"); err == nil {
|
|
limit = v
|
|
}
|
|
// offset: 0 (default is 0)
|
|
if v, err := c.GetInt64("offset"); err == nil {
|
|
offset = v
|
|
}
|
|
// sortby: col1,col2
|
|
if v := c.GetString("sortby"); v != "" {
|
|
sortby = strings.Split(v, ",")
|
|
}
|
|
// order: desc,asc
|
|
if v := c.GetString("order"); v != "" {
|
|
order = strings.Split(v, ",")
|
|
}
|
|
// query: k:v,k:v
|
|
if v := c.GetString("query"); v != "" {
|
|
for _, cond := range strings.Split(v, ",") {
|
|
kv := strings.SplitN(cond, ":", 2)
|
|
if len(kv) != 2 {
|
|
c.Data["json"] = errors.New("Error: invalid query key/value pair")
|
|
c.ServeJSON()
|
|
return
|
|
}
|
|
k, v := kv[0], kv[1]
|
|
query[k] = v
|
|
}
|
|
}
|
|
|
|
l, err := models.GetAll{{controllerName}}(query, fields, sortby, order, offset, limit)
|
|
if err != nil {
|
|
c.Data["json"] = err.Error()
|
|
} else {
|
|
c.Data["json"] = l
|
|
}
|
|
c.ServeJSON()
|
|
}
|
|
|
|
// Put ...
|
|
// @Title Put
|
|
// @Description update the {{controllerName}}
|
|
// @Param id path string true "The id you want to update"
|
|
// @Param body body models.{{controllerName}} true "body for {{controllerName}} content"
|
|
// @Success 200 {object} models.{{controllerName}}
|
|
// @Failure 403 :id is not int
|
|
// @router /:id [put]
|
|
func (c *{{controllerName}}Controller) Put() {
|
|
idStr := c.Ctx.Input.Param(":id")
|
|
id, _ := strconv.ParseInt(idStr, 0, 64)
|
|
v := models.{{controllerName}}{Id: id}
|
|
json.Unmarshal(c.Ctx.Input.RequestBody, &v)
|
|
if err := models.Update{{controllerName}}ById(&v); err == nil {
|
|
c.Data["json"] = "OK"
|
|
} else {
|
|
c.Data["json"] = err.Error()
|
|
}
|
|
c.ServeJSON()
|
|
}
|
|
|
|
// Delete ...
|
|
// @Title Delete
|
|
// @Description delete the {{controllerName}}
|
|
// @Param id path string true "The id you want to delete"
|
|
// @Success 200 {string} delete success!
|
|
// @Failure 403 id is empty
|
|
// @router /:id [delete]
|
|
func (c *{{controllerName}}Controller) Delete() {
|
|
idStr := c.Ctx.Input.Param(":id")
|
|
id, _ := strconv.ParseInt(idStr, 0, 64)
|
|
if err := models.Delete{{controllerName}}(id); err == nil {
|
|
c.Data["json"] = "OK"
|
|
} else {
|
|
c.Data["json"] = err.Error()
|
|
}
|
|
c.ServeJSON()
|
|
}
|
|
`
|