216 lines
4.1 KiB
Go
216 lines
4.1 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"path"
|
||
"path/filepath"
|
||
"strings"
|
||
|
||
"github.com/AlecAivazis/survey"
|
||
"github.com/urfave/cli"
|
||
)
|
||
|
||
var (
|
||
// 允许建立项目的部门
|
||
depts = map[string]bool{
|
||
"main": true,
|
||
"live": true,
|
||
"openplatform": true,
|
||
"search": true,
|
||
"ep": true,
|
||
"bbq": true,
|
||
"video": true,
|
||
"bplus": true,
|
||
"ops": true,
|
||
}
|
||
// 允许建立的项目类型
|
||
types = map[string]bool{
|
||
"interface": true,
|
||
"admin": true,
|
||
"job": true,
|
||
"service": true,
|
||
}
|
||
)
|
||
|
||
const (
|
||
_textModeFastInit = "一键初始化项目"
|
||
_textModeInteraction = "自定义项目参数"
|
||
_textYes = "是"
|
||
_textNo = "否"
|
||
)
|
||
|
||
func runInit(ctx *cli.Context) (err error) {
|
||
if ctx.NumFlags() == 0 {
|
||
if err = interact(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
if ok := check(); !ok {
|
||
return nil
|
||
}
|
||
if err = create(); err != nil {
|
||
println("项目初始化失败: ", err.Error())
|
||
return nil
|
||
}
|
||
fmt.Printf(`项目初始化成功！
|
||
注意：请先创建rider、服务树节点、在配置中心创建uat环境配置文件，否则提交mr后无法运行单元测试！
|
||
相关帮助信息见 http://info.bilibili.co/pages/viewpage.action?pageId=7567510
|
||
`)
|
||
return nil
|
||
}
|
||
|
||
func initPwd() (ok bool) {
|
||
pwd, err := os.Getwd()
|
||
if err != nil {
|
||
return
|
||
}
|
||
ps := strings.Split(pwd, string(os.PathSeparator))
|
||
plen := len(ps)
|
||
if plen < 3 {
|
||
// 至少要有三个目录层级：部门、项目类型、项目名
|
||
return
|
||
}
|
||
name := ps[plen-1]
|
||
dept := ps[plen-2]
|
||
typ := ps[plen-3]
|
||
if !depts[dept] {
|
||
return
|
||
}
|
||
if !types[typ] {
|
||
return
|
||
}
|
||
if name == "" {
|
||
return
|
||
}
|
||
p.Name = name
|
||
p.Department = dept
|
||
p.Type = typ
|
||
p.Path = pwd
|
||
return true
|
||
}
|
||
|
||
func check() (ok bool) {
|
||
root, err := goPath()
|
||
if err != nil || root == "" {
|
||
log.Printf("can not read GOPATH, use ~/go as default GOPATH")
|
||
root = path.Join(os.Getenv("HOME"), "go")
|
||
}
|
||
if !validate() {
|
||
return
|
||
}
|
||
p.Path = fmt.Sprintf("%s/src/go-common/app/%s/%s/%s", strings.TrimRight(root, "/"), p.Type, p.Department, p.Name)
|
||
return true
|
||
}
|
||
|
||
func goPath() (string, error) {
|
||
gopaths := strings.Split(os.Getenv("GOPATH"), ":")
|
||
if len(gopaths) == 1 {
|
||
return gopaths[0], nil
|
||
}
|
||
pwd, err := os.Getwd()
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
abspwd, err := filepath.Abs(pwd)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
for _, gp := range gopaths {
|
||
absgp, err := filepath.Abs(gp)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
if strings.HasPrefix(abspwd, absgp) {
|
||
return absgp, nil
|
||
}
|
||
}
|
||
return "", fmt.Errorf("can't found current gopath")
|
||
}
|
||
|
||
func interact() (err error) {
|
||
qs1 := &survey.Select{
|
||
Message: "你想怎么玩？",
|
||
Options: []string{_textModeFastInit, _textModeInteraction},
|
||
}
|
||
var ans1 string
|
||
if err = survey.AskOne(qs1, &ans1, nil); err != nil {
|
||
return
|
||
}
|
||
switch ans1 {
|
||
case _textModeFastInit:
|
||
if ok := initPwd(); !ok {
|
||
println("Notice: Not in project directory. Skipped fast init.")
|
||
}
|
||
return
|
||
case _textModeInteraction:
|
||
// go on
|
||
default:
|
||
return
|
||
}
|
||
var ds, ts []string
|
||
for d := range depts {
|
||
ds = append(ds, d)
|
||
}
|
||
for t := range types {
|
||
ts = append(ts, t)
|
||
}
|
||
qs := []*survey.Question{
|
||
{
|
||
Name: "department",
|
||
Prompt: &survey.Select{
|
||
Message: "请选择选择部门：",
|
||
Options: ds,
|
||
Default: "main",
|
||
},
|
||
},
|
||
{
|
||
Name: "type",
|
||
Prompt: &survey.Select{
|
||
Message: "请选择项目类型：",
|
||
Options: ts,
|
||
},
|
||
},
|
||
{
|
||
Name: "name",
|
||
Prompt: &survey.Input{
|
||
Message: "请输入项目名称：",
|
||
},
|
||
Validate: survey.Required,
|
||
},
|
||
{
|
||
Name: "owner",
|
||
Prompt: &survey.Input{
|
||
Message: "请输入项目负责人：",
|
||
},
|
||
},
|
||
{
|
||
Name: "useGRPC",
|
||
Prompt: &survey.Select{
|
||
Message: "是否使用 gRPC ？",
|
||
Options: []string{_textYes, _textNo},
|
||
Default: _textNo,
|
||
},
|
||
},
|
||
}
|
||
ans := struct {
|
||
Department string
|
||
Type string
|
||
Name string
|
||
Owner string
|
||
UseGRPC string
|
||
}{}
|
||
if err = survey.Ask(qs, &ans); err != nil {
|
||
return
|
||
}
|
||
p.Name = ans.Name
|
||
p.Department = ans.Department
|
||
p.Type = ans.Type
|
||
p.Owner = ans.Owner
|
||
if ans.UseGRPC == _textYes {
|
||
p.WithGRPC = true
|
||
}
|
||
return
|
||
}
|