restapi: Initial commit

irc/rest_api.go

@@ -0,0 +1,70 @@
+// Copyright (c) 2016- Daniel Oaks <daniel@danieloaks.net>
+// released under the MIT license
+
+// viewing and modifying accounts, registered channels, dlines, rehashing, etc
+
+package irc
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"fmt"
+
+	"github.com/gorilla/mux"
+)
+
+const restErr = "{\"error\":\"An unknown error occurred\"}"
+
+// restAPIServer is used to keep a link to the current running server since this is the best
+// way to do it, given how HTTP handlers dispatch and work.
+var restAPIServer *Server
+
+type restStatusResp struct {
+	Clients  int `json:"clients"`
+	Opers    int `json:"opers"`
+	Channels int `json:"channels"`
+}
+
+type restDLinesResp struct {
+	DLines map[string]IPBanInfo `json:"dlines"`
+}
+
+func restStatus(w http.ResponseWriter, r *http.Request) {
+	rs := restStatusResp{
+		Clients:  restAPIServer.clients.Count(),
+		Opers:    len(restAPIServer.operators),
+		Channels: len(restAPIServer.channels),
+	}
+	b, err := json.Marshal(rs)
+	if err != nil {
+		fmt.Fprintln(w, restErr)
+	} else {
+		fmt.Fprintln(w, string(b))
+	}
+}
+
+func restDLines(w http.ResponseWriter, r *http.Request) {
+	rs := restDLinesResp{
+		DLines: restAPIServer.dlines.AllBans(),
+	}
+	b, err := json.Marshal(rs)
+	if err != nil {
+		fmt.Fprintln(w, restErr)
+	} else {
+		fmt.Fprintln(w, string(b))
+	}
+}
+
+func (s *Server) startRestAPI() {
+	// so handlers can ref it later
+	restAPIServer = s
+
+	// start router
+	r := mux.NewRouter()
+	r.HandleFunc("/status", restStatus)
+	r.HandleFunc("/dlines", restDLines)
+
+	// start api
+	go http.ListenAndServe(s.restAPI.Listen, r)
+}