package main import ( "encoding/json" "io/ioutil" "net/http" "git.andreafazzi.eu/andrea/testhub/logger" "git.andreafazzi.eu/andrea/testhub/models" "git.andreafazzi.eu/andrea/testhub/store" ) const jsonContentType = "application/json" type QuizHubCollectorServer struct { store store.QuizHubCollectorStore http.Handler } type Response struct { Status string `json:"success"` Content interface{} `json:"content"` } func NewQuizHubCollectorServer(store store.QuizHubCollectorStore) *QuizHubCollectorServer { ps := new(QuizHubCollectorServer) ps.store = store router := http.NewServeMux() router.Handle("/tests", logger.WithLogging(http.HandlerFunc(ps.testHandler))) ps.Handler = router return ps } func (ps *QuizHubCollectorServer) testHandler(w http.ResponseWriter, r *http.Request) { switch r.Method { case http.MethodGet: w.Header().Set("content-type", jsonContentType) json.NewEncoder(w).Encode(ps.readAllQuizs(w, r)) case http.MethodPost: w.WriteHeader(http.StatusAccepted) json.NewEncoder(w).Encode(ps.createQuiz(w, r)) } } func (ps *QuizHubCollectorServer) readAllQuizs(w http.ResponseWriter, r *http.Request) *Response { tests, err := ps.store.ReadAllQuizzes() if err != nil { return &Response{"error", err.Error()} } return &Response{"success", tests} } func (ps *QuizHubCollectorServer) createQuiz(w http.ResponseWriter, r *http.Request) *models.Quiz { body, err := ioutil.ReadAll(r.Body) if err != nil { panic(err) } createQuizReq := new(store.CreateQuizRequest) err = json.Unmarshal(body, &createQuizReq) if err != nil { panic(err) } createdQuiz := ps.store.CreateQuiz(createQuizReq) return createdQuiz }