Merge pull request #1 from idbentley/feat/accept_header_multiplex

feat: introduce endpoint that multiplexes the response based on accept header.

Author: alexadrake

cmd/server/main.go

@@ -12,6 +12,7 @@ import (
 	"github.com/gorilla/mux"
 	"github.com/speakeasy-api/speakeasy-api-test-service/internal/auth"
 	"github.com/speakeasy-api/speakeasy-api-test-service/internal/requestbody"
+	"github.com/speakeasy-api/speakeasy-api-test-service/internal/acceptHeaders"
 )
 
 func main() {
@@ -27,6 +28,7 @@ func main() {
 	r.HandleFunc("/pagination/cursor", pagination.HandleCursor).Methods(http.MethodGet, http.MethodPut)
 	r.HandleFunc("/retries", retries.HandleRetries).Methods(http.MethodGet)
 	r.HandleFunc("/errors/{status_code}", errors.HandleErrors).Methods(http.MethodGet)
+	r.HandleFunc("/optional", acceptHeaders.HandleAcceptHeaderMultiplexing).Methods(http.MethodGet)
 
 	log.Println("Listening on :8080")
 	if err := http.ListenAndServe(":8080", r); err != nil {

internal/acceptHeaders/service.go

@@ -0,0 +1,40 @@
+package acceptHeaders
+
+import (
+	"encoding/json"
+	"net/http"
+	"strings"
+
+	"github.com/speakeasy-api/speakeasy-api-test-service/internal/utils"
+)
+
+func headersContains(headers []string, toCheck string) bool {
+	for _, a := range headers {
+		if strings.Contains(a, toCheck) {
+			return true
+		}
+	}
+	return false
+}
+
+func HandleAcceptHeaderMultiplexing(w http.ResponseWriter, r *http.Request) {
+	var obj interface{}
+	if headersContains(r.Header["Accept"], "application/json") {
+		err := json.Unmarshal([]byte("{\"type\":\"obj1\", \"value\": \"JSON\"}"), &obj)
+		if err != nil {
+			utils.HandleError(w, err)
+			return
+		}
+	
+		w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	
+		if err := json.NewEncoder(w).Encode(obj); err != nil {
+			utils.HandleError(w, err)
+		}
+	} else if headersContains(r.Header["Accept"], "text/plain") {
+
+		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+		w.Write([]byte("Success"))
+	}
+
+}