127 lines
2.9 KiB
Go
127 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/mux"
|
|
"go-sjles-pta-vote/server/services"
|
|
)
|
|
|
|
func voteHandler(w http.ResponseWriter, r *http.Request) {
|
|
var vote services.Vote
|
|
if err := json.NewDecoder(r.Body).Decode(&vote); err != nil {
|
|
http.Error(w, "Invalid request payload", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
err := services.SetVote(&vote)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func voteIDHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
id := vars["id"]
|
|
|
|
vote := services.Vote{
|
|
PollId: id,
|
|
Email: "example@example.com", // Replace with actual email retrieval logic
|
|
Vote: true, // Replace with actual vote retrieval logic
|
|
}
|
|
|
|
err := services.SetVote(&vote)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func statsHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
id := vars["id"]
|
|
|
|
poll, err := services.GetPollByQuestion(id)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
json.NewEncoder(w).Encode(poll)
|
|
}
|
|
|
|
func statsIDHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
id := vars["id"]
|
|
|
|
poll, err := services.GetPollByQuestion(id)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
json.NewEncoder(w).Encode(poll)
|
|
}
|
|
|
|
func adminHandler(w http.ResponseWriter, r *http.Request) {
|
|
// Add admin functionality here
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func adminIDHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
id := vars["id"]
|
|
|
|
// Add admin functionality here
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func membersHandler(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "POST" {
|
|
file, handler, err := r.FormFile("members.csv")
|
|
if err != nil {
|
|
http.Error(w, "Failed to upload file", http.StatusBadRequest)
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
fileBytes, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
http.Error(w, "Failed to read file", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
members, err := services.ParseMembersFromBytes(2023, fileBytes) // Assuming year 2023 for demonstration purposes
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
json.NewEncoder(w).Encode(members)
|
|
} else {
|
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
r := mux.NewRouter()
|
|
|
|
r.HandleFunc("/vote", voteHandler).Methods("POST")
|
|
r.HandleFunc("/vote/{id}", voteIDHandler).Methods("POST")
|
|
r.HandleFunc("/stats", statsHandler).Methods("POST")
|
|
r.HandleFunc("/stats/{id}", statsIDHandler).Methods("POST")
|
|
r.HandleFunc("/admin", adminHandler).Methods("GET")
|
|
r.HandleFunc("/admin/{id}", adminIDHandler).Methods("GET")
|
|
r.HandleFunc("/admin/members", membersHandler).Methods("POST")
|
|
|
|
log.Fatal(http.ListenAndServe(":8080", r))
|
|
}
|