3 Commits

Author SHA1 Message Date
Zoe
f7ac354bcf fix config typos and misleading errors 2025-12-06 23:27:54 -06:00
Zoe
dcd8d810f0 More robust script handling and code cleanup
In this commit, I have made the script handling dramatically more
robust, and the code has been cleaned up a fair amount. We no longer
have to manually set is restarting or anything like that, and we dont
rely on a fragile exit code system.
2025-10-07 16:15:05 +00:00
Zoe
4a70260ea5 Better debugging, and several bug fixes
This commit brings a greater experience for developers working on ZQDGR,
and also fixes a number of bugs. The following bugs have been fixed:
- Scripts that died in watch mode did not restart once changes were made
- Created and deleted files did not cause a reload
- script.Stop is now used in every place where we kill the process, this
  ensures that the process is *eventually* actually killed
2025-10-06 15:54:31 -05:00
5 changed files with 216 additions and 173 deletions

View File

@@ -5,6 +5,13 @@ how you would use npm. ZQDGR lets you watch files and rebuild your project as yo
optional websocket server that will notify listeners that a rebuild has occurred, this is very useful for live reloading optional websocket server that will notify listeners that a rebuild has occurred, this is very useful for live reloading
when doing web development with Go. when doing web development with Go.
**Status: Alpha**
This project is not recommend for production use, you are allowed to, and I try
my best to keep it as stable as possible, but features and APIs will change
in nearly every release without backwards compatibility, upgrade carefilly, and
make sure to depend on a specific version of ZQDGR.
## Install ## Install
```bash ```bash

View File

@@ -9,5 +9,5 @@
"dev": "go run main.go" "dev": "go run main.go"
}, },
"pattern": "**/*.go", "pattern": "**/*.go",
"excluded_dirs": [] "excluded_globs": []
} }

330
main.go
View File

@@ -8,6 +8,7 @@ import (
"flag" "flag"
"fmt" "fmt"
"log" "log"
"log/slog"
"net/http" "net/http"
"os" "os"
"os/exec" "os/exec"
@@ -15,6 +16,7 @@ import (
"path" "path"
"path/filepath" "path/filepath"
"runtime" "runtime"
"strconv"
"strings" "strings"
"sync" "sync"
"syscall" "syscall"
@@ -38,24 +40,20 @@ type Config struct {
Type string `json:"type"` Type string `json:"type"`
URL string `json:"url"` URL string `json:"url"`
} `json:"repository"` } `json:"repository"`
Scripts map[string]string `json:"scripts"` Scripts map[string]string `json:"scripts"`
Pattern string `json:"pattern"` Pattern string `json:"pattern"`
// Deprecated: use excludedGlobs instead ExcludedGlobs []string `json:"excluded_globs"`
ExcludedDirs []string `json:"excluded_dirs"` ShutdownSignal string `json:"shutdown_signal"`
ExcludedGlobs []string `json:"excluded_files"` ShutdownTimeout int `json:"shutdown_timeout"`
ShutdownSignal string `json:"shutdown_signal"`
ShutdownTimeout int `json:"shutdown_timeout"`
} }
type Script struct { type Script struct {
zqdgr *ZQDGR zqdgr *ZQDGR
command *exec.Cmd command *exec.Cmd
mutex sync.Mutex mutex sync.Mutex
scriptName string scriptName string
isRestarting bool // notified with the exit code of the script when it exits
wg sync.WaitGroup exitCode chan int
// the exit code of the script, only set after the script has exited
exitCode int
} }
type ZQDGR struct { type ZQDGR struct {
@@ -132,10 +130,10 @@ func (zqdgr *ZQDGR) NewScript(scriptName string, args ...string) *Script {
} }
return &Script{ return &Script{
zqdgr: zqdgr, zqdgr: zqdgr,
command: command, command: command,
scriptName: scriptName, scriptName: scriptName,
isRestarting: false, exitCode: make(chan int),
} }
} }
@@ -143,74 +141,66 @@ func (s *Script) Start() error {
s.mutex.Lock() s.mutex.Lock()
defer s.mutex.Unlock() defer s.mutex.Unlock()
s.wg.Add(1)
err := s.command.Start() err := s.command.Start()
if err != nil { if err != nil {
s.wg.Done()
return err return err
} }
go func() { go func() {
err := s.command.Wait() processState, err := s.command.Process.Wait()
if err != nil { slog.Debug("Script exited", "script", s.scriptName, "error", err)
if exitError, ok := err.(*exec.ExitError); ok {
s.exitCode = exitError.ExitCode()
} else {
// Other errors (e.g., process not found, permission denied)
if !s.isRestarting {
log.Printf("Error waiting for script %s: %v", s.scriptName, err)
s.exitCode = 1
}
}
}
if !s.isRestarting { s.exitCode <- processState.ExitCode()
s.wg.Done()
}
}() }()
return err return err
} }
func (s *Script) Stop(lock bool) error { // it is the caller's responsibility to lock the mutex before calling this function
if lock { func (s *Script) Stop() error {
s.mutex.Lock() slog.Debug("Making sure process is still alive")
defer s.mutex.Unlock() if runtime.GOOS == "windows" {
} if _, err := os.FindProcess(s.command.Process.Pid); err != nil {
// process is already dead
s.isRestarting = true
if s.command.Process != nil {
var signal syscall.Signal
switch s.zqdgr.Config.ShutdownSignal {
case "SIGINT":
signal = syscall.SIGINT
case "SIGTERM":
signal = syscall.SIGTERM
case "SIGQUIT":
signal = syscall.SIGQUIT
default:
signal = syscall.SIGKILL
}
// make sure the process is not dead
if s.command.ProcessState != nil && s.command.ProcessState.Exited() {
return nil return nil
} }
} else {
if err := syscall.Kill(-s.command.Process.Pid, signal); err != nil { process, err := os.FindProcess(s.command.Process.Pid)
log.Printf("error killing previous process: %v", err) if err != nil {
return err return err
} }
// Sending signal 0 checks for existence and permissions
err = process.Signal(syscall.Signal(0))
if err != nil {
return nil
}
} }
slog.Debug("Process is still alive, sending signal")
dead := make(chan bool) dead := make(chan bool)
go func() { go func() {
s.command.Wait() s.command.Wait()
dead <- true dead <- true
}() }()
var signal syscall.Signal
switch s.zqdgr.Config.ShutdownSignal {
case "SIGINT":
signal = syscall.SIGINT
case "SIGTERM":
signal = syscall.SIGTERM
case "SIGQUIT":
signal = syscall.SIGQUIT
default:
signal = syscall.SIGKILL
}
if err := syscall.Kill(-s.command.Process.Pid, signal); err != nil {
log.Printf("error killing previous process: %v", err)
return err
}
shutdownTimeout := s.zqdgr.Config.ShutdownTimeout shutdownTimeout := s.zqdgr.Config.ShutdownTimeout
if shutdownTimeout == 0 { if shutdownTimeout == 0 {
shutdownTimeout = 1 shutdownTimeout = 1
@@ -231,9 +221,11 @@ func (s *Script) Stop(lock bool) error {
} }
func (s *Script) Restart() error { func (s *Script) Restart() error {
slog.Debug("Restarting script", "script", s.scriptName)
s.mutex.Lock() s.mutex.Lock()
err := s.Stop(false) err := s.Stop()
if err != nil { if err != nil {
s.mutex.Unlock() s.mutex.Unlock()
return err return err
@@ -247,8 +239,6 @@ func (s *Script) Restart() error {
return nil return nil
} }
s.isRestarting = false
s.mutex.Unlock() s.mutex.Unlock()
err = s.Start() err = s.Start()
@@ -270,10 +260,6 @@ func (s *Script) Restart() error {
return err return err
} }
func (s *Script) Wait() {
s.wg.Wait()
}
func (wsServer *WSServer) handleWs(w http.ResponseWriter, r *http.Request) { func (wsServer *WSServer) handleWs(w http.ResponseWriter, r *http.Request) {
conn, err := wsServer.upgrader.Upgrade(w, r, nil) conn, err := wsServer.upgrader.Upgrade(w, r, nil)
if err != nil { if err != nil {
@@ -315,27 +301,79 @@ func (zqdgr *ZQDGR) loadConfig() error {
} }
} }
if zqdgr.Config.ExcludedDirs != nil {
fmt.Printf("WARNING: the 'excluded_dirs' key is deprecated, please use 'excluded_globs' instead\n")
zqdgr.Config.ExcludedGlobs = append(zqdgr.Config.ExcludedGlobs, zqdgr.Config.ExcludedDirs...)
}
return nil return nil
} }
func validatePattern(pattern string) ([]string, error) {
var paternArray []string
var currentPattern string
inMatch := false
// iterate over every letter in the pattern
for _, p := range pattern {
if string(p) == "{" {
if inMatch {
return nil, fmt.Errorf("unexpected { in pattern")
}
inMatch = true
}
if string(p) == "}" {
if !inMatch {
return nil, fmt.Errorf("enexpected } in pattern")
}
inMatch = false
}
if string(p) == "," && !inMatch {
paternArray = append(paternArray, currentPattern)
currentPattern = ""
continue
}
currentPattern += string(p)
}
if inMatch {
return nil, fmt.Errorf("unmatched } in pattern")
}
if currentPattern != "" {
paternArray = append(paternArray, currentPattern)
}
return paternArray, nil
}
func main() { func main() {
// var err error
var debugMode bool
noWs := flag.Bool("no-ws", false, "Disable WebSocket server") noWs := flag.Bool("no-ws", false, "Disable WebSocket server")
configDir := flag.String("config", ".", "Path to the config directory") configDir := flag.String("config", ".", "Path to the config directory")
disableReloadConfig := flag.Bool("no-reload-config", false, "Do not restart ZQDGR on config file change")
flag.StringVar(configDir, "C", *configDir, "Path to the config directory") flag.StringVar(configDir, "C", *configDir, "Path to the config directory")
disableReloadConfig := flag.Bool("no-reload-config", false, "Do not restart ZQDGR on config file change")
flag.Parse() flag.Parse()
// if ParseBool returns an error, ZQDGR_DEBUG is not a bool
debugMode, _ = strconv.ParseBool(os.Getenv("ZQDGR_DEBUG"))
if debugMode {
slog.SetLogLoggerLevel(slog.LevelDebug)
}
originalArgs := os.Args originalArgs := os.Args
os.Args = flag.Args() os.Args = flag.Args()
zqdgr := NewZQDGR(*noWs, *configDir) expandedConfigDir, err := filepath.Abs(*configDir)
if err != nil {
log.Fatal(err)
}
slog.Debug("noWS", "noWs", *noWs, "configDir", *configDir, "expandedConfigDir", expandedConfigDir, "disableReloadConfig", *disableReloadConfig)
zqdgr := NewZQDGR(*noWs, expandedConfigDir)
if zqdgr == nil { if zqdgr == nil {
return return
} }
@@ -343,10 +381,13 @@ func main() {
var command string var command string
var commandArgs []string var commandArgs []string
// command name trimmed by flags.Args()
// os.Args ~= [script, --, arguments]
for i, arg := range os.Args { for i, arg := range os.Args {
if arg == "--" { if arg == "--" {
if i+2 < len(os.Args) { slog.Debug("Found double-dash", "i", i, "len(os.Args)", len(os.Args), "os.Args", os.Args)
commandArgs = os.Args[i+2:] if len(os.Args)-i > 1 {
commandArgs = os.Args[i+1:]
} }
break break
} }
@@ -359,6 +400,8 @@ func main() {
commandArgs = append(commandArgs, arg) commandArgs = append(commandArgs, arg)
} }
slog.Debug("Collected", "command", command, "commandArgs", commandArgs)
watchMode := false watchMode := false
var scriptName string var scriptName string
switch command { switch command {
@@ -501,13 +544,7 @@ func main() {
log.Fatal("please specify a script to run") log.Fatal("please specify a script to run")
} }
watchMode = true watchMode = true
for i := range commandArgs { scriptName = commandArgs[0]
if strings.HasPrefix(commandArgs[i], "-") {
continue
}
scriptName = commandArgs[i]
}
default: default:
scriptName = command scriptName = command
} }
@@ -525,19 +562,9 @@ func main() {
log.Println("Received signal, exiting...") log.Println("Received signal, exiting...")
if script.command != nil { if script.command != nil {
var signal syscall.Signal script.mutex.Lock()
switch zqdgr.Config.ShutdownSignal { script.Stop()
case "SIGINT": script.mutex.Unlock()
signal = syscall.SIGINT
case "SIGTERM":
signal = syscall.SIGTERM
case "SIGQUIT":
signal = syscall.SIGQUIT
default:
signal = syscall.SIGKILL
}
syscall.Kill(-script.command.Process.Pid, signal)
} }
os.Exit(0) os.Exit(0)
@@ -561,48 +588,21 @@ func main() {
} }
// make sure the pattern is valid // make sure the pattern is valid
var paternArray []string patternArray, err := validatePattern(zqdgr.Config.Pattern)
var currentPattern string if err != nil {
inMatch := false log.Fatal(err)
// iterate over every letter in the pattern
for _, p := range zqdgr.Config.Pattern {
if string(p) == "{" {
if inMatch {
log.Fatal("unmatched { in pattern")
}
inMatch = true
}
if string(p) == "}" {
if !inMatch {
log.Fatal("unmatched } in pattern")
}
inMatch = false
}
if string(p) == "," && !inMatch {
paternArray = append(paternArray, currentPattern)
currentPattern = ""
inMatch = false
continue
}
currentPattern += string(p)
} }
if inMatch { for _, pattern := range zqdgr.Config.ExcludedGlobs {
log.Fatal("unmatched } in pattern") _, err := validatePattern(pattern)
} if err != nil {
log.Fatal(err)
if currentPattern != "" { }
paternArray = append(paternArray, currentPattern)
} }
watcherConfig := WatcherConfig{ watcherConfig := WatcherConfig{
excludedGlobs: globList(zqdgr.Config.ExcludedGlobs), excludedGlobs: globList(zqdgr.Config.ExcludedGlobs),
pattern: paternArray, pattern: patternArray,
} }
watcher, err := NewWatcher(&watcherConfig) watcher, err := NewWatcher(&watcherConfig)
@@ -622,13 +622,12 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
// We use this timer to deduplicate events. // tailing edge debounce of file system events
var ( var (
// Wait 100ms for new events; each new event resets the timer.
waitFor = 100 * time.Millisecond waitFor = 100 * time.Millisecond
// Keep track of the timers, as path → timer. mu sync.Mutex
mu sync.Mutex // watched filepath -> timer
timers = make(map[string]*time.Timer) timers = make(map[string]*time.Timer)
) )
go func() { go func() {
@@ -645,25 +644,37 @@ func main() {
if !ok { if !ok {
timer = time.AfterFunc(waitFor, func() { timer = time.AfterFunc(waitFor, func() {
if event.Op&fsnotify.Remove == fsnotify.Remove || event.Op&fsnotify.Write == fsnotify.Write || event.Op&fsnotify.Create == fsnotify.Create { slog.Debug("FSnotify event received", "event", event)
if os.Getenv("ZQDGR_DEBUG") != "" {
fmt.Println("File changed:", event.Name)
}
if strings.HasSuffix(event.Name, "zqdgr.config.json") { if event.Op&fsnotify.Remove == fsnotify.Remove || event.Op&fsnotify.Write == fsnotify.Write || event.Op&fsnotify.Create == fsnotify.Create {
slog.Debug("File changed", "file", event.Name)
fullEventPath, err := filepath.Abs(event.Name)
if err != nil {
log.Fatal(err)
}
slog.Debug("expanded event path", "path", fullEventPath)
// check against the fullpath to make sure that the config file that was changed is
// actually the one we are using
if fullEventPath == filepath.Join(zqdgr.WorkingDirectory, "zqdgr.config.json") {
// re-exec the exact same command // re-exec the exact same command
if !*disableReloadConfig { if !*disableReloadConfig {
log.Println("zqdgr.config.json has changed, restarting...") fmt.Println("zqdgr.config.json has changed, restarting...")
executable, err := os.Executable() executable, err := os.Executable()
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
err = script.Stop(true) script.mutex.Lock()
err = script.Stop()
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
script.mutex.Unlock()
err = syscall.Exec(executable, originalArgs, os.Environ()) err = syscall.Exec(executable, originalArgs, os.Environ())
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
@@ -673,8 +684,16 @@ func main() {
} }
} }
if directoryShouldBeTracked(&watcherConfig, event.Name) { if pathShouldBeTracked(&watcherConfig, event.Name) {
watcher.(NotifyWatcher).watcher.Add(event.Name) if event.Op&fsnotify.Create == fsnotify.Create {
slog.Debug("Adding new file to watcher", "file", event.Name)
watcher.(NotifyWatcher).watcher.Add(event.Name)
}
if event.Op&fsnotify.Remove == fsnotify.Remove {
slog.Debug("Removing file from watcher", "file", event.Name)
watcher.(NotifyWatcher).watcher.Remove(event.Name)
}
} }
if pathMatches(&watcherConfig, event.Name) { if pathMatches(&watcherConfig, event.Name) {
@@ -707,8 +726,17 @@ func main() {
} }
} }
}() }()
// block until the script exits with a zero (aka, it normally exited on its own)
for {
exitCode := <-script.exitCode
slog.Debug("Script exited", "script", scriptName, "exitCode", exitCode)
if exitCode == 0 {
os.Exit(0)
}
}
} }
script.Wait() // block until the script exits
os.Exit(script.exitCode) os.Exit(<-script.exitCode)
} }

View File

@@ -4,7 +4,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"log" "log"
"os" "log/slog"
"path/filepath" "path/filepath"
"strings" "strings"
@@ -34,7 +34,9 @@ func (g *globList) Matches(value string) bool {
func matchesPattern(pattern []string, path string) bool { func matchesPattern(pattern []string, path string) bool {
for _, p := range pattern { for _, p := range pattern {
slog.Debug("checking path against pattern", "pattern", p, "path", path)
if matched, _ := doublestar.Match(p, path); matched { if matched, _ := doublestar.Match(p, path); matched {
slog.Debug("path matches pattern", "pattern", p, "path", path)
return true return true
} }
} }
@@ -42,17 +44,13 @@ func matchesPattern(pattern []string, path string) bool {
return false return false
} }
func directoryShouldBeTracked(cfg *WatcherConfig, path string) bool { func pathShouldBeTracked(cfg *WatcherConfig, path string) bool {
base := filepath.Dir(path) base := filepath.Dir(path)
if os.Getenv("ZQDGR_DEBUG") != "" { slog.Debug("checking file against path", "path", path, "base", base)
log.Printf("checking %s against %s %v\n", path, base, *cfg)
}
if cfg.excludedGlobs.Matches(base) { if cfg.excludedGlobs.Matches(base) {
if os.Getenv("ZQDGR_DEBUG") != "" { slog.Debug("file is excluded", "path", path)
log.Printf("%s is excluded\n", base)
}
return false return false
} }
@@ -86,9 +84,7 @@ func (n NotifyWatcher) Close() error {
} }
func (n NotifyWatcher) AddFile(path string) error { func (n NotifyWatcher) AddFile(path string) error {
if os.Getenv("ZQDGR_DEBUG") != "" { slog.Debug("manually adding file", "file", path)
log.Printf("manually adding file\n")
}
return n.add(path) return n.add(path)
} }
@@ -125,24 +121,36 @@ func NewWatcher(cfg *WatcherConfig) (FileWatcher, error) {
func addFiles(fw FileWatcher) error { func addFiles(fw FileWatcher) error {
cfg := fw.getConfig() cfg := fw.getConfig()
for _, pattern := range cfg.pattern { for _, pattern := range cfg.pattern {
if os.Getenv("ZQDGR_DEBUG") != "" { slog.Debug("adding pattern", "pattern", pattern)
fmt.Printf("processing glob %s\n", pattern)
}
matches, err := doublestar.Glob(pattern) matches, err := doublestar.Glob(pattern)
if err != nil { if err != nil {
log.Fatalf("Bad pattern \"%s\": %s", pattern, err.Error()) log.Fatalf("Bad pattern \"%s\": %s", pattern, err.Error())
} }
trackedDirs := make(map[string]bool)
for _, match := range matches { for _, match := range matches {
if os.Getenv("ZQDGR_DEBUG") != "" { base := filepath.Dir(match)
log.Printf("checking %s\n", match) // this allows us to track file creations and deletions
if !trackedDirs[base] {
if cfg.excludedGlobs.Matches(base) {
slog.Debug("directory is excluded", "file", match)
continue
}
trackedDirs[base] = true
slog.Debug("adding directory", "dir", base)
if err := fw.add(base); err != nil {
return fmt.Errorf("FileWatcher.Add(): %v", err)
}
} }
if directoryShouldBeTracked(cfg, match) { slog.Debug("adding file", "file", match)
if os.Getenv("ZQDGR_DEBUG") != "" {
log.Printf("%s is not excluded\n", match) if pathShouldBeTracked(cfg, match) {
} slog.Debug("path should be tracked", "file", match)
if err := fw.add(match); err != nil { if err := fw.add(match); err != nil {
return fmt.Errorf("FileWatcher.Add(): %v", err) return fmt.Errorf("FileWatcher.Add(): %v", err)

View File

@@ -1,6 +1,6 @@
{ {
"name": "zqdgr", "name": "zqdgr",
"version": "0.0.4a", "version": "0.0.6.1",
"description": "zqdgr is a quick and dirty Golang runner", "description": "zqdgr is a quick and dirty Golang runner",
"author": "juls0730", "author": "juls0730",
"license": "BSL-1.0", "license": "BSL-1.0",