From abdb9c23fd5db2752369c8c807c66080d86ebef9 Mon Sep 17 00:00:00 2001 From: ubq323 Date: Wed, 9 Mar 2022 03:29:23 +0000 Subject: web hooks --- webhooks.go | 93 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 93 insertions(+) create mode 100644 webhooks.go diff --git a/webhooks.go b/webhooks.go new file mode 100644 index 0000000..4e0d5b1 --- /dev/null +++ b/webhooks.go @@ -0,0 +1,93 @@ +package main + +import ( + "fmt" + "io" + "log" + "net" + "net/http" + "os" +) + +const TOKEN_FNAME = "tokens.txt" +const SOCKET_NAME = "./bees.sock" +var tokens map[string]string + +func loadTokens() map[string]string { + out := make(map[string]string) + file, err := os.Open(TOKEN_FNAME) + if err != nil { + log.Fatal(err) + } + defer file.Close() + + for { + var token, name string + _, err := fmt.Fscanf(file, "%s %s\n", &token, &name) + if err == io.EOF { + break + } else if err != nil { + log.Fatalf("%#v", err) + } + out[token] = name + } + return out +} + +type beeMsg struct { + from string + text string +} + +func sendSock(msgs <-chan beeMsg, w io.Writer) { + for msg := range msgs { + fmt.Fprintf(w, "%s: %s", msg.from, msg.text) + } +} + +func handleHttp(msgch chan<- beeMsg, w http.ResponseWriter, r *http.Request) { + tok := r.FormValue("token") + msg := r.FormValue("msg") + if tok == "" || msg == "" { + w.WriteHeader(400) + fmt.Fprintln(w,"missing form values") + return + } + + clientname, ok := tokens[tok] + if !ok { + w.WriteHeader(401) + fmt.Fprintln(w,"unknown token") + return + } + + if len(msg) > 300 { + w.WriteHeader(400) + fmt.Fprintln(w,"msg too long") + return + } + + msgch <- beeMsg{from: clientname, text:msg} +} + +func init() { + tokens = loadTokens() +} + +func main() { + msgs := make(chan beeMsg) + fmt.Println("ok") + + sock, err := net.Dial("unixgram", SOCKET_NAME) + if err != nil { + fmt.Println(err) + } + go sendSock(msgs, sock) + + http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { + handleHttp(msgs,w,r) + }) + + log.Fatal(http.ListenAndServe(":3236", nil)) + +} -- cgit v1.2.3