testhub/server.go

72 lines
1.6 KiB
Go

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
}