diff options
Diffstat (limited to 'template')
| -rw-r--r-- | template/Dockerfile | 2 | ||||
| -rw-r--r-- | template/args/args.go | 12 | ||||
| -rw-r--r-- | template/main.go | 15 | ||||
| -rw-r--r-- | template/ntfy/publisher.go | 16 | ||||
| -rw-r--r-- | template/ntfy/watcher.go | 96 |
5 files changed, 139 insertions, 2 deletions
diff --git a/template/Dockerfile b/template/Dockerfile index 87a2422..ff39f7c 100644 --- a/template/Dockerfile +++ b/template/Dockerfile @@ -10,4 +10,4 @@ RUN go build -o /app/{{ service }} EXPOSE 8080 -CMD ["/app/{{ service }}", "--server", "--migrate", "--port", "8080", "--template-path", "/app/templates", "--database-path", "/app/db/{{ service }}.db", "--static-path", "/app/static", "--scheduler"] +CMD ["/app/{{ service }}", "--server", "--migrate", "--port", "8080", "--template-path", "/app/templates", "--database-path", "/app/db/{{ service }}.db", "--static-path", "/app/static", "--scheduler", "--ntfy-topics", "whois", "--ntfy-endpoint", "https://ntfy.simponic.hatecomputers.club"] diff --git a/template/args/args.go b/template/args/args.go index 6e4aff1..dcba9f7 100644 --- a/template/args/args.go +++ b/template/args/args.go @@ -4,6 +4,7 @@ import ( "flag" "fmt" "os" + "strings" "sync" ) @@ -15,6 +16,10 @@ type Arguments struct { Migrate bool Scheduler bool + NtfyEndpoint string + NtfyTopics []string + NtfyListener bool + Port int Server bool } @@ -56,6 +61,10 @@ func GetArgs() (*Arguments, error) { templatePath := flag.String("template-path", "./templates", "Path to the template directory") staticPath := flag.String("static-path", "./static", "Path to the static directory") + ntfyEndpoint := flag.String("ntfy-endpoint", "https://ntfy.simponic.hatecomputers.club", "NTFY Endpoint") + ntfyTopics := flag.String("ntfy-topics", "testtopic", "Comma-separated NTFY Topics") + ntfyListener := flag.Bool("ntfy-listener", false, "Listen to NTFY Topic and propagate messages") + scheduler := flag.Bool("scheduler", false, "Run scheduled jobs via cron") migrate := flag.Bool("migrate", false, "Run the migrations") @@ -72,6 +81,9 @@ func GetArgs() (*Arguments, error) { Server: *server, Migrate: *migrate, Scheduler: *scheduler, + NtfyEndpoint: *ntfyEndpoint, + NtfyTopics: strings.Split(*ntfyTopics, ","), + NtfyListener: *ntfyListener, } err := validateArgs(args) if err != nil { diff --git a/template/main.go b/template/main.go index 6d2b657..3ddb39c 100644 --- a/template/main.go +++ b/template/main.go @@ -9,6 +9,7 @@ import ( "{{ service_repo }}/args" "{{ service_repo }}/database" "{{ service_repo }}/scheduler" + "{{ service_repo }}/ntfy" "github.com/joho/godotenv" ) @@ -36,6 +37,18 @@ func main() { log.Println("database migrated successfully") } + if argv.NtfyListener { + ntfy := ntfy.MakeNtfyWatcher(argv.NtfyEndpoint, argv.NtfyTopics) + notifications := ntfy.Watch() + + go func() { + for notification := range notifications { + message := notification.Message + log.Println("got message", message) + } + }() + } + if argv.Scheduler { go func() { scheduler.StartScheduler(dbConn, argv) @@ -57,7 +70,7 @@ func main() { }() } - if argv.Server || argv.Scheduler { + if argv.Server || argv.Scheduler || argv.NtfyListener { select {} // block forever } } diff --git a/template/ntfy/publisher.go b/template/ntfy/publisher.go new file mode 100644 index 0000000..68f8e49 --- /dev/null +++ b/template/ntfy/publisher.go @@ -0,0 +1,16 @@ +package ntfy + +import ( + "net/http" + "strings" +) + +func SendMessage(message string, endpoint string, topics []string) error { + for _, topic := range topics { + _, err := http.Post(endpoint+"/"+topic, "text/plain", strings.NewReader(message)) + if err != nil { + return err + } + } + return nil +} diff --git a/template/ntfy/watcher.go b/template/ntfy/watcher.go new file mode 100644 index 0000000..d8959cd --- /dev/null +++ b/template/ntfy/watcher.go @@ -0,0 +1,96 @@ +package ntfy + +import ( + "bufio" + "encoding/json" + "log" + "net/http" + "net/url" + "path" + "time" +) + +type Message struct { + Id string `json:"id"` + Time int `json:"time"` + Message string `json:"message"` + Event string `json:"event"` +} + +type NtfyWatcher struct { + Endpoint string + Topics []string +} + +func (w *NtfyWatcher) Watch() chan Message { + notifications := make(chan Message) + + for _, topic := range w.Topics { + log.Println("subscribing to topic:", topic) + + go func() { + retryCount := 5 + retryTime := 5 * time.Second + retries := retryCount + + sleepAndDecrementRetry := func() { + log.Println("waiting 5 seconds before reconnecting. retries left:", retries, "topic:", topic, "endpoint:", w.Endpoint) + time.Sleep(retryTime) + retries-- + } + + for true { + if retries == 0 { + log.Fatal("too many retries, exiting") + } + + endpoint, _ := url.JoinPath(w.Endpoint, path.Join(topic, "json")) + resp, err := http.Get(endpoint) + if err != nil { + log.Println("error connecting to endpoint:", err) + sleepAndDecrementRetry() + continue + } + + defer resp.Body.Close() + scanner := bufio.NewScanner(resp.Body) + for scanner.Scan() { + bytes := scanner.Bytes() + var msg Message + err := json.Unmarshal(bytes, &msg) + if err != nil { + log.Println("could not unmarshal message:", err) + continue + } + + if msg.Event == "keepalive" { + log.Println("received keepalive message") + continue + } + if msg.Event != "message" { + log.Println("received unknown event:", msg.Event) + continue + } + + log.Println("received notification:", msg) + notifications <- msg + retries = retryCount // reset retries + } + + if err := scanner.Err(); err != nil { + log.Println("error reading response body:", err) + sleepAndDecrementRetry() + } + } + }() + } + + return notifications +} + +func MakeNtfyWatcher(endpoint string, topics []string) *NtfyWatcher { + return &NtfyWatcher{ + Endpoint: endpoint, + Topics: topics, + } +} |