api: handler[handles service]

api/handler.go

@@ -0,0 +1,85 @@
+package api
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+
+	"github.com/krishpranav/remote-desktop/rdisplay"
+	"github.com/krishpranav/remote-desktop/rtc"
+)
+
+func handleError(w http.ResponseWriter, err error) {
+	fmt.Printf("Error: %v", err)
+	w.WriteHeader(http.StatusInternalServerError)
+}
+
+func MakeHandler(webrtc rtc.Service, display rdisplay.Service) http.Handler {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/session", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != http.MethodPost {
+			w.WriteHeader(http.StatusMethodNotAllowed)
+			return
+		}
+		dec := json.NewDecoder(r.Body)
+		req := newSessionRequest{}
+
+		if err := dec.Decode(&req); err != nil {
+			handleError(w, err)
+			return
+		}
+
+		peer, err := webrtc.CreateRemoteScreenConnection(req.Screen, 20)
+		if err != nil {
+			handleError(w, err)
+			return
+		}
+
+		answer, err := peer.ProcessOffer(req.Offer)
+
+		if err != nil {
+			handleError(w, err)
+			return
+		}
+
+		payload, err := json.Marshal(newSessionResponse{
+			Answer: answer,
+		})
+		if err != nil {
+			handleError(w, err)
+			return
+		}
+
+		w.Write(payload)
+	})
+
+	mux.HandleFunc("/screens", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != http.MethodGet {
+			w.WriteHeader(http.StatusMethodNotAllowed)
+			return
+		}
+
+		screens, err := display.Screens()
+
+		if err != nil {
+			handleError(w, err)
+			return
+		}
+
+		screensPayload := make([]screenPayload, len(screens))
+
+		for i, s := range screens {
+			screensPayload[i].Index = s.Index
+		}
+		payload, err := json.Marshal(screensResponse{
+			Screens: screensPayload,
+		})
+		if err != nil {
+			handleError(w, err)
+			return
+		}
+
+		w.Write(payload)
+	})
+	return mux
+}