Implement RPC

rpc/request.go

@@ -0,0 +1,61 @@
+package rpc
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"log/slog"
+
+	"nhooyr.io/websocket"
+)
+
+func (s *server) handleRequest(ctx context.Context, c *websocket.Conn, data []byte) {
+	var request Request
+	err := json.Unmarshal(data, &request)
+	if err != nil {
+		respondError(ctx, c, "", ERROR_JRPC2_PARSE_ERROR, fmt.Errorf("Error Parsing Request: %w", err), nil)
+		return
+	}
+
+	// Get the Requested function
+	fun, ok := s.methods[request.Method]
+	if !ok {
+		respondError(ctx, c, request.ID, ERROR_JRPC2_METHOD_NOT_FOUND, fmt.Errorf("Method Not Found"), nil)
+		return
+	}
+
+	reqCtx, cancel := context.WithCancel(ctx)
+	defer cancel()
+
+	// Run the Requested function
+	result, err := fun(reqCtx, request)
+	if err != nil {
+		respondError(ctx, c, request.ID, 1000, fmt.Errorf("Method Error: %w", err), result)
+		return
+	}
+
+	rData := json.RawMessage{}
+	if data != nil {
+		rData, err = json.Marshal(result)
+		if err != nil {
+			respondError(ctx, c, request.ID, ERROR_JRPC2_INTERNAL, fmt.Errorf("Error Marshalling Response Data: %w", err), nil)
+			return
+		}
+	}
+	slog.InfoContext(ctx, "response data", "rdata", rData)
+
+	resp, err := json.Marshal(Response{
+		ID:     request.ID,
+		Result: &rData,
+	})
+	if err != nil {
+		respondError(ctx, c, request.ID, ERROR_JRPC2_INTERNAL, fmt.Errorf("Error Marshalling Response: %w", err), nil)
+		return
+	}
+
+	err = c.Write(ctx, websocket.MessageText, resp)
+	if err != nil {
+		respondError(ctx, c, request.ID, ERROR_JRPC2_INTERNAL, fmt.Errorf("Error Sending Response: %w", err), nil)
+		return
+	}
+}