mirror of https://github.com/beego/bee.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
140 lines
3.2 KiB
140 lines
3.2 KiB
// 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 |
|
}
|
|
|