117 lines
2 KiB
Go
117 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/fsnotify/fsnotify"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
once := false
|
|
if len(os.Args) > 1 {
|
|
once = os.Args[1] == "--once"
|
|
}
|
|
|
|
command := ""
|
|
for i, arg := range os.Args {
|
|
if arg == "--command" {
|
|
command = os.Args[i+1]
|
|
}
|
|
}
|
|
|
|
if command == "" {
|
|
panic("command is required")
|
|
}
|
|
|
|
if once {
|
|
runCommand(command)
|
|
return
|
|
}
|
|
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
fmt.Println("Recovered from fatal error:", r)
|
|
// You can log the error here or take other corrective actions
|
|
}
|
|
}()
|
|
|
|
runCommand(command)
|
|
// Create new watcher.
|
|
watcher, err := fsnotify.NewWatcher()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer watcher.Close()
|
|
// Start listening for events.
|
|
go func() {
|
|
for {
|
|
select {
|
|
case event, ok := <-watcher.Events:
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
if strings.HasSuffix(event.Name, "generated.go") {
|
|
continue
|
|
}
|
|
|
|
if event.Has(fsnotify.Write) {
|
|
success := runCommand(command)
|
|
if success {
|
|
log.Println(fmt.Sprintf("file changed. code generation successful"))
|
|
} else {
|
|
log.Println(fmt.Sprintf("file changed. code generation failed"))
|
|
}
|
|
}
|
|
case err, ok := <-watcher.Errors:
|
|
if !ok {
|
|
return
|
|
}
|
|
log.Println("error:", err)
|
|
}
|
|
}
|
|
}()
|
|
|
|
cwd, _ := os.Getwd()
|
|
pagesDir := filepath.Join(cwd, "pages")
|
|
partialsDir := filepath.Join(cwd, "partials")
|
|
|
|
toWatch := []string{pagesDir, partialsDir}
|
|
|
|
for _, watch := range toWatch {
|
|
err = watcher.Add(watch)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// Block main goroutine forever.
|
|
<-make(chan struct{})
|
|
}
|
|
|
|
func runCommand(command string) bool {
|
|
// Create a new command
|
|
cmd := exec.Command("bash", "-c", command)
|
|
|
|
// Capture stdout and stderr in buffers
|
|
var out bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
cmd.Stdout = &out
|
|
cmd.Stderr = &stderr
|
|
|
|
// Run the command
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
log.Println(fmt.Sprintf("error: %v", err))
|
|
println(stderr.String())
|
|
return false
|
|
} else {
|
|
println(out.String())
|
|
return true
|
|
}
|
|
}
|