mirror of
https://github.com/beego/bee.git
synced 2024-11-22 15:10:54 +00:00
141 lines
3.2 KiB
Go
141 lines
3.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 (
|
|
"io/ioutil"
|
|
"os"
|
|
path "path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
)
|
|
|
|
var cmdRun = &Command{
|
|
UsageLine: "run [appname] [watchall] [-main=*.go] [-downdoc=true] [-gendoc=true]",
|
|
Short: "run the app and start a Web server for development",
|
|
Long: `
|
|
Run command will supervise the file system of the beego project using inotify,
|
|
it will recompile and restart the app after any modifications.
|
|
|
|
`,
|
|
}
|
|
|
|
var mainFiles ListOpts
|
|
|
|
var downdoc docValue
|
|
var gendoc docValue
|
|
|
|
func init() {
|
|
cmdRun.Run = runApp
|
|
cmdRun.Flag.Var(&mainFiles, "main", "specify main go files")
|
|
cmdRun.Flag.Var(&gendoc, "gendoc", "auto generate the docs")
|
|
cmdRun.Flag.Var(&downdoc, "downdoc", "auto download swagger file when not exist")
|
|
}
|
|
|
|
var appname string
|
|
|
|
func runApp(cmd *Command, args []string) int {
|
|
exit := make(chan bool)
|
|
crupath, _ := os.Getwd()
|
|
|
|
if len(args) == 0 || args[0] == "watchall" {
|
|
appname = path.Base(crupath)
|
|
ColorLog("[INFO] Uses '%s' as 'appname'\n", appname)
|
|
} else {
|
|
appname = args[0]
|
|
}
|
|
Debugf("current path:%s\n", crupath)
|
|
|
|
err := loadConfig()
|
|
if err != nil {
|
|
ColorLog("[ERRO] Fail to parse bee.json[ %s ]\n", err)
|
|
}
|
|
|
|
var paths []string
|
|
|
|
readAppDirectories(crupath, &paths)
|
|
|
|
// Because monitor files has some issues, we watch current directory
|
|
// and ignore non-go files.
|
|
gps := GetGOPATHs()
|
|
if len(gps) == 0 {
|
|
ColorLog("[ERRO] Fail to start[ %s ]\n", "$GOPATH is not set or empty")
|
|
os.Exit(2)
|
|
}
|
|
gopath := gps[0]
|
|
for _, p := range conf.DirStruct.Others {
|
|
paths = append(paths, strings.Replace(p, "$GOPATH", gopath, -1))
|
|
}
|
|
|
|
files := []string{}
|
|
for _, arg := range mainFiles {
|
|
if len(arg) > 0 {
|
|
files = append(files, arg)
|
|
}
|
|
}
|
|
|
|
if gendoc == "true" {
|
|
NewWatcher(paths, files, true)
|
|
Autobuild(files, true)
|
|
} else {
|
|
NewWatcher(paths, files, false)
|
|
Autobuild(files, false)
|
|
}
|
|
if downdoc == "true" {
|
|
if _, err := os.Stat(path.Join(crupath, "swagger")); err != nil {
|
|
if os.IsNotExist(err) {
|
|
downloadFromUrl(swaggerlink, "swagger.zip")
|
|
unzipAndDelete("swagger.zip", "swagger")
|
|
}
|
|
}
|
|
}
|
|
for {
|
|
select {
|
|
case <-exit:
|
|
runtime.Goexit()
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func readAppDirectories(directory string, paths *[]string) {
|
|
fileInfos, err := ioutil.ReadDir(directory)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
useDirectory := false
|
|
for _, fileInfo := range fileInfos {
|
|
if strings.HasSuffix(fileInfo.Name(), "docs") {
|
|
continue
|
|
}
|
|
if fileInfo.IsDir() == true && fileInfo.Name()[0] != '.' {
|
|
readAppDirectories(directory+"/"+fileInfo.Name(), paths)
|
|
continue
|
|
}
|
|
|
|
if useDirectory == true {
|
|
continue
|
|
}
|
|
|
|
if path.Ext(fileInfo.Name()) == ".go" {
|
|
*paths = append(*paths, directory)
|
|
useDirectory = true
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|