.gitignore

@@ -1,2 +0,0 @@
-watchdog
-generated-version.go

README.md

@@ -16,9 +16,7 @@ Git:
 ```bash
 git clone https://git.coolaj86.com/coolaj86/watchdog.go.git
 pushd watchdog.go/
-go generate -mod=vendor ./...
+go build -o bin/watchdog
-pushd cmd/watchdog
-go build -mod=vendor
 ```
 
 Zip:

build.sh

@@ -1,45 +0,0 @@
-#!/usr/bin/env bash
-
-export CGO_ENABLED=0
-#GOOS=windows GOARCH=amd64 go install
-go tool dist list
-
-gocmd=watchdog.go
-golib=""
-echo ""
-
-echo ""
-echo "Windows amd64"
-GOOS=windows GOARCH=amd64 go build -o dist/windows-amd64/watchdog.exe $gocmd $golib
-echo "Windows 386"
-GOOS=windows GOARCH=386 go build -o dist/windows-386/watchdog.exe $gocmd $golib
-
-echo ""
-echo "Darwin (macOS) amd64"
-GOOS=darwin GOARCH=amd64 go build -o dist/darwin-amd64/watchdog $gocmd $golib
-
-echo ""
-echo "Linux amd64"
-GOOS=linux GOARCH=amd64 go build -o dist/linux-amd64/watchdog $gocmd $golib
-echo "Linux 386"
-
-echo ""
-GOOS=linux GOARCH=386 go build -o dist/linux-386/watchdog $gocmd $golib
-echo "RPi 3 B+ ARMv7"
-GOOS=linux GOARCH=arm GOARM=7 go build -o dist/linux-armv7/watchdog $gocmd $golib
-echo "RPi Zero ARMv5"
-GOOS=linux GOARCH=arm GOARM=5 go build -o dist/linux-armv5/watchdog $gocmd $golib
-
-my_ver=$(git describe --tags)
-pushd dist
-  ls -d *-* | while read my_dist
-  do
-    if [ -d "$my_dist" ]; then
-      #tar -czvf watchdog-$my_ver-$my_dist.tar.gz $my_dist
-      zip -r watchdog-$my_ver-$my_dist.zip $my_dist
-    fi
-  done
-popd
-
-echo ""
-echo ""

doc.go

@@ -4,6 +4,4 @@
 // The watchdog package is meant to be used as a binary only.
 // The git tag version describes the state of the binary,
 // not the state of the library. The API is not yet stable.
-//
-// See https://git.rootproject.org/root/watchdog.go for pre-built binaries.
 package watchdog

tools/version/version.go

@@ -1,123 +0,0 @@
-// +build tools
-
-package main
-
-import (
-	"bytes"
-	"fmt"
-	"go/format"
-	"os"
-	"os/exec"
-	"regexp"
-	"strconv"
-	"strings"
-	"text/template"
-	"time"
-)
-
-var exactVer *regexp.Regexp
-var gitVer *regexp.Regexp
-var verFile = "generated-version.go"
-
-func init() {
-	// exactly vX.Y.Z (go-compatible semver)
-	exactVer = regexp.MustCompile(`^v\d+\.\d+\.\d+$`)
-
-	// vX.Y.Z-n-g0000000 git post-release, semver prerelease
-	gitVer = regexp.MustCompile(`^(v\d+\.\d+)\.(\d+)-(\d+)-g`)
-}
-
-// Goal: Either use an exact version like v1.0.0
-//       or translate the git version like v1.0.0-4-g0000000
-//       to a semver like v1.0.1-pre4+g0000000
-//       Don't fail when git repo isn't available.
-func main() {
-	desc := gitDesc()
-	rev := gitRev()
-	ver := semVer(desc)
-	ts := time.Now().Format(time.RFC3339)
-
-	v := struct {
-		Timestamp string
-		Version   string
-		GitRev    string
-	}{
-		Timestamp: ts,
-		Version:   ver,
-		GitRev:    rev,
-	}
-
-	// Create or overwrite the go file from template
-	var buf bytes.Buffer
-	if err := versionTpl.Execute(&buf, v); nil != err {
-		panic(err)
-	}
-
-	// Format
-	src, err := format.Source(buf.Bytes())
-	if nil != err {
-		panic(err)
-	}
-
-	// Write to disk (in the Current Working Directory)
-	f, err := os.Create(verFile)
-	if nil != err {
-		panic(err)
-	}
-	if _, err := f.Write(src); nil != err {
-		panic(err)
-	}
-	if err := f.Close(); nil != err {
-		panic(err)
-	}
-}
-
-func gitDesc() string {
-	args := strings.Split("git describe --tags --dirty --always", " ")
-	cmd := exec.Command(args[0], args[1:]...)
-	out, err := cmd.CombinedOutput()
-	if nil != err {
-		// Don't panic, just carry on
-		out = []byte("v0.0.0-0-g0000000")
-	}
-	return strings.TrimSpace(string(out))
-}
-
-func gitRev() string {
-	args := strings.Split("git rev-parse HEAD", " ")
-	cmd := exec.Command(args[0], args[1:]...)
-	out, err := cmd.CombinedOutput()
-	if nil != err {
-		// Don't panic, just carry on
-		out = []byte("00000000000000000")
-	}
-	return strings.TrimSpace(string(out))
-}
-
-func semVer(desc string) string {
-	var ver string
-	if exactVer.MatchString(desc) {
-		// v1.0.0
-		ver = desc
-	} else if gitVer.MatchString(desc) {
-		// ((v1.0).(0)-(1))
-		vers := gitVer.FindStringSubmatch(desc)
-		patch, err := strconv.Atoi(vers[2])
-		if nil != err {
-			panic(err)
-		}
-		// v1.0.1-pre1
-		ver = fmt.Sprintf("%s.%d-pre%s", vers[1], patch+1, vers[3])
-	}
-	return ver
-}
-
-var versionTpl = template.Must(template.New("").Parse(`// Code generated by go generate; DO NOT EDIT.
-package main
-
-func init() {
-	Timestamp = "{{ .Timestamp }}"
-	Version = "{{ .Version }}"
-	GitRev = "{{ .GitRev }}"
-}
-`))

watchdog.go

@@ -1,4 +1,4 @@
-package watchdog
+package main
 
 import (
 	"bytes"
@@ -6,14 +6,90 @@ import (
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"log"
 	"net"
 	"net/http"
 	"net/url"
+	"os"
 	"os/exec"
 	"strings"
 	"time"
 )
 
+func usage() {
+	fmt.Println("Usage: watchdog -c config.json")
+}
+
+func main() {
+	if 3 != len(os.Args) {
+		usage()
+		os.Exit(1)
+		return
+	}
+	if "-c" != os.Args[1] {
+		usage()
+		os.Exit(1)
+		return
+	}
+
+	filename := os.Args[2]
+	f, err := os.Open(filename)
+	if nil != err {
+		log.Fatal(err)
+		return
+	}
+
+	configFile, err := ioutil.ReadAll(f)
+	if nil != err {
+		log.Fatal(err)
+		return
+	}
+
+	config := &Config{}
+	err = json.Unmarshal(configFile, config)
+	if nil != err {
+		log.Fatal(err)
+		return
+	}
+
+	//fmt.Printf("%#v\n", config)
+
+	done := make(chan struct{}, 1)
+
+	allWebhooks := make(map[string]ConfigWebhook)
+
+	for i := range config.Webhooks {
+		h := config.Webhooks[i]
+		allWebhooks[h.Name] = h
+	}
+
+	logQueue := make(chan string, 10)
+	go logger(logQueue)
+	for i := range config.Watches {
+		c := config.Watches[i]
+		logQueue <- fmt.Sprintf("Watching '%s'", c.Name)
+		go func(c ConfigWatch) {
+			d := New(&Dog{
+				Name:        c.Name,
+				CheckURL:    c.URL,
+				Keywords:    c.Keywords,
+				Recover:     c.RecoverScript,
+				Webhooks:    c.Webhooks,
+				AllWebhooks: allWebhooks,
+				logger:      logQueue,
+			})
+			d.Watch()
+		}(config.Watches[i])
+	}
+
+	if 0 == len(config.Watches) {
+		log.Fatal("Nothing to watch")
+		return
+	}
+
+	<-done
+}
+
 type Dog struct {
 	Name         string
 	CheckURL     string
@@ -21,7 +97,7 @@ type Dog struct {
 	Recover      string
 	Webhooks     []string
 	AllWebhooks  map[string]ConfigWebhook
-	Logger       chan string
+	logger       chan string
 	error        error
 	failures     int
 	passes       int
@@ -45,22 +121,13 @@ func (d *Dog) Watch() {
 }
 
 func (d *Dog) watch() {
-	d.Logger <- fmt.Sprintf("Check: '%s'", d.Name)
+	d.logger <- fmt.Sprintf("Check: '%s'", d.Name)
 
 	err := d.check()
 	if nil == err {
 		return
 	}
 
-	time.Sleep(time.Duration(2) * time.Second)
-	err2 := d.check()
-	if nil != err2 {
-		d.Logger <- fmt.Sprintf("Down: '%s': %s", d.Name, err2)
-	} else {
-		d.Logger <- fmt.Sprintf("Hiccup: '%s': %s", d.Name, err)
-		return
-	}
-
 	failure := false
 	t := 10
 	for {
@@ -70,7 +137,6 @@ func (d *Dog) watch() {
 		t *= 2
 		err := d.check()
 		if nil != err {
-			d.Logger <- fmt.Sprintf("Unrecoverable: '%s': %s", d.Name, err)
 			failure = true
 		} else {
 			failure = false
@@ -117,11 +183,11 @@ func (d *Dog) check() error {
 
 	if !bytes.Contains(b, []byte(d.Keywords)) {
 		err = fmt.Errorf("Down: '%s' Not Found for '%s'", d.Keywords, d.Name)
-		d.Logger <- fmt.Sprintf("%s", err)
+		d.logger <- fmt.Sprintf("%s", err)
 		d.error = err
 		return err
 	} else {
-		d.Logger <- fmt.Sprintf("Up: '%s'", d.Name)
+		d.logger <- fmt.Sprintf("Up: '%s'", d.Name)
 	}
 
 	return nil
@@ -137,25 +203,25 @@ func (d *Dog) recover() {
 	pipe, err := cmd.StdinPipe()
 	pipe.Write([]byte(d.Recover))
 	if nil != err {
-		d.Logger <- fmt.Sprintf("[Recover] Could not write to bash '%s': %s", d.Recover, err)
+		d.logger <- fmt.Sprintf("[Recover] Could not write to bash '%s': %s", d.Recover, err)
 	}
 	err = cmd.Start()
 	if nil != err {
-		d.Logger <- fmt.Sprintf("[Recover] Could not start '%s': %s", d.Recover, err)
+		d.logger <- fmt.Sprintf("[Recover] Could not start '%s': %s", d.Recover, err)
 	}
 	err = pipe.Close()
 	if nil != err {
-		d.Logger <- fmt.Sprintf("[Recover] Could not close '%s': %s", d.Recover, err)
+		d.logger <- fmt.Sprintf("[Recover] Could not close '%s': %s", d.Recover, err)
 	}
 	err = cmd.Wait()
 	cancel()
 	if nil != err {
-		d.Logger <- fmt.Sprintf("[Recover] '%s' failed for '%s': %s", d.Recover, d.Name, err)
+		d.logger <- fmt.Sprintf("[Recover] '%s' failed for '%s': %s", d.Recover, d.Name, err)
 	}
 }
 
 func (d *Dog) notify(hardFail bool) {
-	d.Logger <- fmt.Sprintf("Notifying the authorities of %s's failure", d.Name)
+	d.logger <- fmt.Sprintf("Notifying the authorities of %s's failure", d.Name)
 	d.lastNotified = time.Now()
 
 	for i := range d.Webhooks {
@@ -168,7 +234,7 @@ func (d *Dog) notify(hardFail bool) {
 		if !ok {
 			// TODO check in main when config is read
 			d.Webhooks[i] = ""
-			d.Logger <- fmt.Sprintf("[Warning] Could not find webhook '%s' for '%s'", name, h.Name)
+			d.logger <- fmt.Sprintf("[Warning] Could not find webhook '%s' for '%s'", name, h.Name)
 			continue
 		}
 
@@ -187,14 +253,14 @@ func (d *Dog) notify(hardFail bool) {
 				// because `{{` gets urlencoded
 				//k = strings.Replace(k, "{{ .Name }}", d.Name, -1)
 				v = strings.Replace(v, "{{ .Name }}", d.Name, -1)
-				d.Logger <- fmt.Sprintf("[HEADER] %s: %s", k, v)
+				d.logger <- fmt.Sprintf("[HEADER] %s: %s", k, v)
 				form.Set(k, v)
 			}
 			body = strings.NewReader(form.Encode())
 		} else if 0 != len(h.JSON) {
 			bodyBuf, err := json.Marshal(h.JSON)
 			if nil != err {
-				d.Logger <- fmt.Sprintf("[Notify] JSON Marshal Error for '%s': %s", h.Name, err)
+				d.logger <- fmt.Sprintf("[Notify] JSON Marshal Error for '%s': %s", h.Name, err)
 				continue
 			}
 			// `{{` should be left alone
@@ -204,7 +270,7 @@ func (d *Dog) notify(hardFail bool) {
 		client := NewHTTPClient()
 		req, err := http.NewRequest(h.Method, h.URL, body)
 		if nil != err {
-			d.Logger <- fmt.Sprintf("[Notify] HTTP Client Network Error for '%s': %s", h.Name, err)
+			d.logger <- fmt.Sprintf("[Notify] HTTP Client Network Error for '%s': %s", h.Name, err)
 			continue
 		}
 
@@ -233,12 +299,12 @@ func (d *Dog) notify(hardFail bool) {
 
 		resp, err := client.Do(req)
 		if nil != err {
-			d.Logger <- fmt.Sprintf("[Notify] HTTP Client Error for '%s': %s", h.Name, err)
+			d.logger <- fmt.Sprintf("[Notify] HTTP Client Error for '%s': %s", h.Name, err)
 			continue
 		}
 
 		if !(resp.StatusCode >= 200 && resp.StatusCode < 300) {
-			d.Logger <- fmt.Sprintf("[Notify] Response Error for '%s': %s", h.Name, resp.Status)
+			d.logger <- fmt.Sprintf("[Notify] Response Error for '%s': %s", h.Name, resp.Status)
 			continue
 		}
 
@@ -248,12 +314,12 @@ func (d *Dog) notify(hardFail bool) {
 		decoder := json.NewDecoder(resp.Body)
 		err = decoder.Decode(&data)
 		if err != nil {
-			d.Logger <- fmt.Sprintf("[Notify] Response Body Error for '%s': %s", h.Name, resp.Status)
+			d.logger <- fmt.Sprintf("[Notify] Response Body Error for '%s': %s", h.Name, resp.Status)
 			continue
 		}
 
 		// TODO some sort of way to determine if data is successful (keywords)
-		d.Logger <- fmt.Sprintf("[Notify] Success? %#v", data)
+		d.logger <- fmt.Sprintf("[Notify] Success? %#v", data)
 	}
 }
 
@@ -296,3 +362,12 @@ func NewHTTPClient() *http.Client {
 	}
 	return client
 }
+
+// This is so that the log messages don't trample
+// over each other when they happen simultaneously.
+func logger(msgs chan string) {
+	for {
+		msg := <-msgs
+		log.Println(msg)
+	}
+}