bee/watch.go

188 lines
3.9 KiB
Go
Raw Normal View History

2013-09-03 17:23:58 +00:00
// 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.
2012-12-15 16:56:28 +00:00
package main
import (
"fmt"
"log"
"os"
"os/exec"
2013-07-24 06:58:13 +00:00
"strings"
2013-07-06 07:30:57 +00:00
"sync"
"time"
"github.com/Unknwon/com"
"github.com/howeyc/fsnotify"
2012-12-15 16:56:28 +00:00
)
var (
2013-07-06 08:13:30 +00:00
cmd *exec.Cmd
state sync.Mutex
eventTime = make(map[string]int64)
2012-12-15 16:56:28 +00:00
)
func NewWatcher(paths []string) {
watcher, err := fsnotify.NewWatcher()
if err != nil {
com.ColorLog("[ERRO] Fail to create new Watcher[ %s ]\n", err)
2013-08-15 07:24:23 +00:00
os.Exit(2)
2012-12-15 16:56:28 +00:00
}
go func() {
for {
select {
case e := <-watcher.Event:
2013-07-06 07:30:57 +00:00
isbuild := true
2013-07-24 06:58:13 +00:00
// Skip TMP files for Sublime Text.
if checkTMPFile(e.Name) {
continue
}
2013-07-30 07:23:14 +00:00
if !checkIsGoFile(e.Name) {
continue
}
2013-07-24 06:58:13 +00:00
mt := getFileModTime(e.Name)
if t := eventTime[e.Name]; mt == t {
com.ColorLog("[SKIP] # %s #\n", e.String())
isbuild = false
2013-07-06 07:30:57 +00:00
}
eventTime[e.Name] = mt
2013-07-06 07:30:57 +00:00
if isbuild {
com.ColorLog("[EVEN] %s\n", e)
2013-07-06 07:30:57 +00:00
go Autobuild()
}
2012-12-15 16:56:28 +00:00
case err := <-watcher.Error:
log.Fatal("error:", err)
}
}
2013-08-23 20:39:10 +00:00
time.Sleep(500 * time.Millisecond)
2012-12-15 16:56:28 +00:00
}()
com.ColorLog("[INFO] Initializing watcher...\n")
2012-12-15 16:56:28 +00:00
for _, path := range paths {
com.ColorLog("[TRAC] Directory( %s )\n", path)
2012-12-15 16:56:28 +00:00
err = watcher.Watch(path)
if err != nil {
com.ColorLog("[ERRO] Fail to watch directory[ %s ]\n", err)
2013-08-15 07:24:23 +00:00
os.Exit(2)
2012-12-15 16:56:28 +00:00
}
}
}
// getFileModTime retuens unix timestamp of `os.File.ModTime` by given path.
func getFileModTime(path string) int64 {
2013-08-23 20:39:10 +00:00
path = strings.Replace(path, "\\", "/", -1)
f, err := os.Open(path)
if err != nil {
com.ColorLog("[ERRO] Fail to open file[ %s ]\n", err)
return time.Now().Unix()
}
defer f.Close()
fi, err := f.Stat()
if err != nil {
com.ColorLog("[ERRO] Fail to get file information[ %s ]\n", err)
return time.Now().Unix()
}
return fi.ModTime().Unix()
}
2012-12-15 16:56:28 +00:00
func Autobuild() {
2013-07-06 08:13:30 +00:00
state.Lock()
defer state.Unlock()
2012-12-15 16:56:28 +00:00
com.ColorLog("[INFO] Start building...\n")
2012-12-15 16:56:28 +00:00
path, _ := os.Getwd()
os.Chdir(path)
2013-07-25 08:26:54 +00:00
var err error
// For applications use full import path like "github.com/.../.."
// are able to use "go install" to reduce build time.
if conf.GoInstall {
icmd := exec.Command("go", "install")
icmd.Stdout = os.Stdout
icmd.Stderr = os.Stderr
err = icmd.Run()
}
if err == nil {
bcmd := exec.Command("go", "build")
bcmd.Stdout = os.Stdout
bcmd.Stderr = os.Stderr
err = bcmd.Run()
}
2013-07-06 08:13:30 +00:00
2012-12-15 16:56:28 +00:00
if err != nil {
com.ColorLog("[ERRO] ============== Build failed ===================\n")
2013-07-06 07:30:57 +00:00
return
2012-12-15 16:56:28 +00:00
}
com.ColorLog("[SUCC] Build was successful\n")
2013-07-06 07:30:57 +00:00
Restart(appname)
2012-12-15 16:56:28 +00:00
}
func Kill() {
2013-07-06 08:13:30 +00:00
defer func() {
if e := recover(); e != nil {
fmt.Println("Kill -> ", e)
}
}()
if cmd != nil {
cmd.Process.Kill()
2012-12-15 16:56:28 +00:00
}
}
2013-07-06 07:30:57 +00:00
func Restart(appname string) {
2013-07-06 08:13:30 +00:00
Debugf("kill running process")
2013-07-06 07:30:57 +00:00
Kill()
2013-07-06 08:13:30 +00:00
go Start(appname)
2013-07-06 07:30:57 +00:00
}
2013-07-06 08:13:30 +00:00
2012-12-15 16:56:28 +00:00
func Start(appname string) {
com.ColorLog("[INFO] Restarting %s ...\n", appname)
if strings.Index(appname, "./") == -1 {
appname = "./" + appname
}
2013-07-06 07:30:57 +00:00
2012-12-15 16:56:28 +00:00
cmd = exec.Command(appname)
2013-07-06 07:30:57 +00:00
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
2013-07-06 08:13:30 +00:00
go cmd.Run()
com.ColorLog("[INFO] %s is running...\n", appname)
2013-08-24 11:00:08 +00:00
started <- true
2012-12-15 16:56:28 +00:00
}
2013-07-24 06:58:13 +00:00
// checkTMPFile returns true if the event was for TMP files.
func checkTMPFile(name string) bool {
if strings.HasSuffix(strings.ToLower(name), ".tmp") {
return true
}
return false
}
2013-07-30 07:23:14 +00:00
// checkIsGoFile returns true if the name HasSuffix ".go".
2013-07-30 07:23:14 +00:00
func checkIsGoFile(name string) bool {
if strings.HasSuffix(name, ".go") {
return true
}
return false
}