probo/server.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"net/http"

	"git.andreafazzi.eu/andrea/probo/client"
	"git.andreafazzi.eu/andrea/probo/logger"
	"git.andreafazzi.eu/andrea/probo/models"
	"git.andreafazzi.eu/andrea/probo/store"
)

const jsonContentType = "application/json"

type QuizHubCollectorServer struct {
	store store.QuizHubCollectorStore
	http.Handler
}

func NewQuizHubCollectorServer(store store.QuizHubCollectorStore) *QuizHubCollectorServer {
	ps := new(QuizHubCollectorServer)
	ps.store = store

	router := http.NewServeMux()

	router.Handle("/quizzes", 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.readAllQuizzes(w, r))

	case http.MethodPost:
		response := new(client.Response)

		quiz, err := ps.createQuiz(w, r)
		if err != nil {
			response = &client.Response{Status: "error", Content: err.Error()}
		}

		response = &client.Response{Status: "success", Content: quiz}

		w.WriteHeader(http.StatusAccepted)
		json.NewEncoder(w).Encode(response)
	}
}

func (ps *QuizHubCollectorServer) readAllQuizzes(w http.ResponseWriter, r *http.Request) *client.Response {
	tests, err := ps.store.ReadAllQuizzes()
	if err != nil {
		return &client.Response{Status: "error", Content: err.Error()}
	}
	return &client.Response{Status: "success", Content: tests}
}

func (ps *QuizHubCollectorServer) createQuiz(w http.ResponseWriter, r *http.Request) (*models.Quiz, error) {
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		return nil, err
	}

	createQuizReq := new(client.CreateQuizRequest)

	err = json.Unmarshal(body, &createQuizReq)
	if err != nil {
		return nil, err
	}

	createdQuiz, err := ps.store.CreateQuiz(createQuizReq)
	if err != nil {
		return nil, err
	}

	return createdQuiz, nil
}