forked from silsuer/bingo
/
make_controller.go
65 lines (54 loc) · 1.45 KB
/
make_controller.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package main
import (
"fmt"
"github.com/urfave/cli"
"io"
"io/ioutil"
"log"
"os"
"strings"
)
func makeController(c *cli.Context) error {
if c.Args().Get(0) == "" {
log.Fatal("Need a argument: controller's name. e.g. bingo sword make:controller example_controller")
}
//从templates文件夹中把控制器文件复制过来
//控制器模板文件夹路径
var controllerDir string
if c.Bool("resource") == true {
controllerDir = getTempDir() + "/resource_controller.gtpl"
} else {
controllerDir = getTempDir() + "/controller.gtpl"
}
if IfFileExist(controllerDir) == true {
// 读取文件内容,创建文件
f, err := os.Open(controllerDir)
if err != nil {
log.Fatal(err)
}
defer f.Close()
bytes, err := ioutil.ReadAll(f)
if err != nil {
log.Fatal(err)
}
str := strings.Replace(string(bytes), "${name}", c.Args().Get(0), -1)
// 得到要创建的env文件
p, err := os.Getwd()
if err != nil {
log.Fatal(err)
}
dstDir := p + "/" + new(Environment).GetWithDefault("CONTROLLER_DIR", "http/controllers") + "/" + c.Args().Get(0) + ".go"
// 创建文件并返回句柄
df, err := CreateFile(dstDir, 0755)
if err != nil {
log.Fatal(err)
}
defer df.Close()
io.WriteString(df, str)
fmt.Printf("\n%c[0;48;32m%s%c[0m\n", 0x1B, "Controller created successfully :"+dstDir, 0x1B)
} else {
//不存在,则弹出提示
log.Fatal("No such file named controller.gtpl in " + controllerDir)
}
return nil
}