go-sjles-pta-vote/server/main.go

package main

import (
	"encoding/json"
	"log"
	"net/http"
	"os"
	"strconv"
	"path/filepath"

	"github.com/gorilla/mux"

	"go-sjles-pta-vote/server/common"
	"go-sjles-pta-vote/server/models"
	"go-sjles-pta-vote/server/services"
)

func voteHandler(resWriter http.ResponseWriter, request *http.Request) {
	var vote models.Vote
	if err := json.NewDecoder(request.Body).Decode(&vote); err != nil {
		common.SendError(resWriter, "Invalid JSON", http.StatusBadRequest)
		return
	}

	if err := services.SetVote(&vote); err != nil {
		common.SendError(resWriter, "Failed to set vote", http.StatusInternalServerError)
		return
	}

	resWriter.WriteHeader(http.StatusOK)
}

func voteIDHandler(resWriter http.ResponseWriter, request *http.Request) {
	vars := mux.Vars(request)
	idStr := vars["id"]
	id, err := strconv.ParseInt(idStr, 10, 64)
	if err != nil {
		common.SendError(resWriter, "Invalid poll ID", http.StatusBadRequest)
		return
	}

	vote := models.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 {
		common.SendError(resWriter, "Failed to set vote", http.StatusInternalServerError)
		return
	}

	resWriter.WriteHeader(http.StatusOK)
}

func statsHandler(resWriter http.ResponseWriter, request *http.Request) {
	if request.Method == http.MethodGet {
		filePath := "./server/templates/stats.html"
		http.ServeFile(resWriter, request, filePath)
	} else if request.Method == http.MethodPost {
		vars := mux.Vars(request)
		id := vars["id"]

		poll, err := services.GetPollByQuestion(id)
		if err != nil {
			common.SendError(resWriter, "Failed to get poll", http.StatusInternalServerError)
			return
		}

		json.NewEncoder(resWriter).Encode(poll)
	} else {
		resWriter.WriteHeader(http.StatusMethodNotAllowed)
	}
}

func statsIDHandler(resWriter http.ResponseWriter, request *http.Request) {
	vars := mux.Vars(request)
	id := vars["id"]

	poll, err := services.GetPollByQuestion(id)
	if err != nil {
		common.SendError(resWriter, "Failed to get poll", http.StatusInternalServerError)
		return
	}

	json.NewEncoder(resWriter).Encode(poll)
}

func adminLoginHandler(resWriter http.ResponseWriter, request *http.Request) {
	if request.Method != http.MethodPost {
		common.SendError(resWriter, "Method not allowed", http.StatusMethodNotAllowed)
		return
	}

	var loginReq services.LoginRequest
	if err := json.NewDecoder(request.Body).Decode(&loginReq); err != nil {
		common.SendError(resWriter, "Invalid JSON", http.StatusBadRequest)
		return
	}

	// Validate admin credentials
	isValid, err := services.ValidateAdminLogin(loginReq.Username, loginReq.Password)
	if err != nil {
		common.SendError(resWriter, "Invalid username or password", http.StatusBadRequest)
		return
	}

	if !isValid {
		common.SendError(resWriter, "Invalid username or password", http.StatusUnauthorized)
		return
	}

	// Generate JWT token
	token, err := services.GenerateAuthToken(loginReq.Username)
	if err != nil {
		common.SendError(resWriter, "Failed to generate auth token", http.StatusInternalServerError)
		return
	}

	resWriter.WriteHeader(http.StatusOK)
	json.NewEncoder(resWriter).Encode(services.LoginResponse{
		Success: true,
		Token:   token,
	})
}

func main() {
	log.SetOutput(os.Stdout)
	log.SetFlags(log.LstdFlags | log.Lshortfile)

	http.HandleFunc("/api/vote", voteHandler)
	http.HandleFunc("/api/vote/{id}", voteIDHandler)
	http.HandleFunc("/api/stats", statsHandler)
	http.HandleFunc("/api/stats/{id}", statsIDHandler)
	http.HandleFunc("/api/admin/new-vote", services.AdminNewVoteHandler)
	http.HandleFunc("/api/admin/view-votes", services.AdminViewVoteHandler)
	http.HandleFunc("/api/admin/login", adminLoginHandler)
	http.HandleFunc("/api/admin/members", services.AdminMembersHandler)
	http.HandleFunc("/api/admin/members/view", services.AdminMembersView)

	buildPath := filepath.Join(".", "client", "build")
	fs := http.FileServer(http.Dir(buildPath))
	
	http.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// If the file exists on disk, let the file server handle it.
		if _, err := os.Stat(filepath.Join(buildPath, r.URL.Path)); err == nil {
			fs.ServeHTTP(w, r)
			return
		}
		// Otherwise serve index.html (so React Router can handle the route)
		http.ServeFile(w, r, filepath.Join(buildPath, "index.html"))
	}))

	log.Printf("Starting server on :8080")
	log.Fatal(http.ListenAndServe(":8080", nil))
}