update webhooks server thing to support multiple channels

1 files changed, 42 insertions, 18 deletions

diff --git a/webhooks.go b/webhooks.go
index 4e0d5b1..d544103 100644
--- a/webhooks.go
+++ b/webhooks.go
@@ -10,8 +10,12 @@ import (
 )
 
 const TOKEN_FNAME = "tokens.txt"
-const SOCKET_NAME = "./bees.sock"
+const SOCKET_DIR = "/srv/apiobot/ch/"
+// const SOCKET_DIR = "./socks/"
+var CHANNELS = [...]string{"a","b","g"}
+
 var tokens map[string]string
+var msg_chs map[string]chan<- beeMsg
 
 func loadTokens() map[string]string {
 	out := make(map[string]string)
@@ -45,19 +49,26 @@ func sendSock(msgs <-chan beeMsg, w io.Writer) {
 	}
 }
 
-func handleHttp(msgch chan<- beeMsg, w http.ResponseWriter, r *http.Request) {
-	tok := r.FormValue("token")
-	msg := r.FormValue("msg")
-	if tok == "" || msg == "" {
+func handleHttp(w http.ResponseWriter, r *http.Request) {
+	tok := r.FormValue("key")
+	msg := r.FormValue("message")
+	irc_ch := r.FormValue("channel")
+
+	if tok == "" {
+		w.WriteHeader(400)
+		fmt.Fprintln(w,"missing key")
+		return
+	}
+	if msg == "" {
 		w.WriteHeader(400)
-		fmt.Fprintln(w,"missing form values")
+		fmt.Fprintln(w,"missing message")
 		return
 	}
 
 	clientname, ok := tokens[tok]
 	if !ok {
 		w.WriteHeader(401)
-		fmt.Fprintln(w,"unknown token")
+		fmt.Fprintln(w,"bad key")
 		return
 	}
 
@@ -67,26 +78,39 @@ func handleHttp(msgch chan<- beeMsg, w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	msgch <- beeMsg{from: clientname, text:msg}
+	if irc_ch == "" {
+		irc_ch = "a"
+	}
+
+	msg_ch, ok := msg_chs[irc_ch]
+	if !ok {
+		w.WriteHeader(400)
+		fmt.Fprintf(w,"invalid channel name, valid channels are %v.",CHANNELS)
+		return		
+	}
+
+	msg_ch <- beeMsg{from: clientname, text:msg}
 }
 
 func init() {
 	tokens = loadTokens()
+	msg_chs = make(map[string]chan<- beeMsg)
+	for _,ch := range CHANNELS {
+		sock, err := net.Dial("unixgram", SOCKET_DIR+ch+".sock")
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		mc := make(chan beeMsg)
+		msg_chs[ch] = mc
+		go sendSock(mc, sock)
+	}
 }
 
 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)
-	})
+	http.HandleFunc("/", handleHttp)
 
 	log.Fatal(http.ListenAndServe(":3236", nil))