From 375eb444793b93cfee0033212b4439d6180d9407 Mon Sep 17 00:00:00 2001 From: astaxie Date: Sat, 9 Aug 2014 01:37:55 +0800 Subject: [PATCH] bee support create controllers --- bee.go | 2 +- g.go | 14 +++++- g_controllers.go | 124 +++++++++++++++++++++++++++++++++++++++++++++++ g_migration.go | 2 +- 4 files changed, 138 insertions(+), 4 deletions(-) create mode 100644 g_controllers.go diff --git a/bee.go b/bee.go index 6055701..bf9029b 100644 --- a/bee.go +++ b/bee.go @@ -25,7 +25,7 @@ import ( "strings" ) -const version = "1.1.3" +const version = "1.2.0" type Command struct { // Run runs the command. diff --git a/g.go b/g.go index b91698f..c95a48c 100644 --- a/g.go +++ b/g.go @@ -30,8 +30,8 @@ bee generate model [-tables=""] [-driver=mysql] [-conn=root:@tcp(127.0.0.1:3306) bee generate migration [filename] generate migration file for making database schema update -bee generate controller [modelfile] - generate RESTFul controllers based on modelfile +bee generate controller [controllerfile] + generate RESTFul controllers bee generate router [controllerfile] generate router based on controllerfile @@ -102,6 +102,16 @@ func generateCode(cmd *Command, args []string) { ColorLog("[HINT] Usage: bee generate migration [filename]\n") os.Exit(2) } + case "controller": + if len(args) == 2 { + cname := args[1] + ColorLog("[INFO] Using '%s' as controller name\n", cname) + generateController(cname, curpath) + } else { + ColorLog("[ERRO] Wrong number of arguments\n") + ColorLog("[HINT] Usage: bee generate controller [filename]\n") + os.Exit(2) + } default: ColorLog("[ERRO] command is missing\n") } diff --git a/g_controllers.go b/g_controllers.go new file mode 100644 index 0000000..44a1a95 --- /dev/null +++ b/g_controllers.go @@ -0,0 +1,124 @@ +// 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 ( + "os" + "path" + "strings" +) + +// article +// cms/article +// +func generateController(cname, crupath string) { + 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] + } + fp := path.Join(crupath, "controllers", p) + if _, err := os.Stat(fp); os.IsNotExist(err) { + // create controller directory + if err := os.MkdirAll(fp, 0777); err != nil { + ColorLog("[ERRO] Could not create controllers directory: %s\n", err) + os.Exit(2) + } + } + 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 f.Close() + content := strings.Replace(controllerTpl, "{{packageName}}", packageName, -1) + content = strings.Replace(content, "{{controllerName}}", controllerName, -1) + f.WriteString(content) + // gofmt generated source code + formatSourceCode(fpath) + ColorLog("[INFO] controller file generated: %s\n", fpath) + } else { + // error creating file + ColorLog("[ERRO] Could not create controller file: %s\n", err) + os.Exit(2) + } +} + +var controllerTpl = `package {{packageName}} + +import ( + "github.com/astaxie/beego" +) + +// oprations for {{controllerName}} +type {{controllerName}}Controller struct { + beego.Controller +} + +// @Title Post +// @Description create {{controllerName}} +// @Param body body models.{{controllerName}} true "body for {{controllerName}} content" +// @Success 200 {int} models.{{controllerName}}.Id +// @Failure 403 body is empty +// @router / [post] +func (this *{{controllerName}}Controller) Post() { + +} + +// @Title Get +// @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 (this *{{controllerName}}Controller) GetOne() { + +} + +// @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 (this *{{controllerName}}Controller) GetAll() { + +} + +// @Title Update +// @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 (this *{{controllerName}}Controller) Put() { + +} + +// @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 (this *{{controllerName}}Controller) Delete() { + +} +` diff --git a/g_migration.go b/g_migration.go index aff7527..0c59eaa 100644 --- a/g_migration.go +++ b/g_migration.go @@ -35,7 +35,7 @@ func generateMigration(mname string, curpath string) { migrationFilePath := path.Join(curpath, M_PATH) if _, err := os.Stat(migrationFilePath); os.IsNotExist(err) { // create migrations directory - if err := os.Mkdir(migrationFilePath, 0777); err != nil { + if err := os.MkdirAll(migrationFilePath, 0777); err != nil { ColorLog("[ERRO] Could not create migration directory: %s\n", err) os.Exit(2) }