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

package main

import (
	"encoding/json"
	"fmt"
	"log"
	"math/rand"
	"net/http"
	"os"
	"path/filepath"
	"strconv"
	"time"

	"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 pollsIDHandler(resWriter http.ResponseWriter, request *http.Request) {
	vars := mux.Vars(request)
	id, err := strconv.ParseInt(vars["id"], 10, 64)
	if err != nil {
		common.SendError(resWriter, "Invalid poll ID", http.StatusBadRequest)
		return
	}

	poll, err := services.GetPollById(id)
	if err == services.ErrPollNotFound {
		common.SendError(resWriter, "Poll not found", http.StatusNotFound)
		return
	} else 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 initDatabase() error {
	// Seed random generator for reproducible results in tests
	rand.Seed(42)

	polls := []models.Poll{
		{
			ID:           1,
			Question:     "Should we increase the budget?",
			MemberYes:    rand.Int63n(50),
			MemberNo:     rand.Int63n(50),
			NonMemberYes: rand.Int63n(20),
			NonMemberNo:  rand.Int63n(20),
			TotalVotes:   int(rand.Int63n(100)),
			WhoVoted:     []string{"email1@example.com", "email2@example.com", "email3@example.com", "email4@example.com"},
			CreatedAt:    time.Now().Format(time.RFC3339),
			UpdatedAt:    time.Now().Format(time.RFC3339),
			ExpiresAt:    time.Now().Add(24 * time.Hour).Format(time.RFC3339),
		},
		{
			ID:           2,
			Question:     "Should we hire more staff?",
			MemberYes:    rand.Int63n(50),
			MemberNo:     rand.Int63n(50),
			NonMemberYes: rand.Int63n(20),
			NonMemberNo:  rand.Int63n(20),
			TotalVotes:   int(rand.Int63n(100)),
			WhoVoted:     []string{"email1@example.com", "email2@example.com", "email3@example.com", "email4@example.com"},
			CreatedAt:    time.Now().Format(time.RFC3339),
			UpdatedAt:    time.Now().Format(time.RFC3339),
			ExpiresAt:    time.Now().Add(24 * time.Hour).Format(time.RFC3339),
		},
		{
			ID:           3,
			Question:     "Should we renovate the building?",
			MemberYes:    rand.Int63n(50),
			MemberNo:     rand.Int63n(50),
			NonMemberYes: rand.Int63n(20),
			NonMemberNo:  rand.Int63n(20),
			TotalVotes:   int(rand.Int63n(100)),
			WhoVoted:     []string{"email1@example.com", "email2@example.com", "email3@example.com", "email4@example.com"},
			CreatedAt:    time.Now().Format(time.RFC3339),
			UpdatedAt:    time.Now().Format(time.RFC3339),
			ExpiresAt:    time.Now().Add(24 * time.Hour).Format(time.RFC3339),
		},
	}

	for _, poll := range polls {
		if err := services.CreatePollIgnore(&poll); err != nil {
			return fmt.Errorf("failed to create poll %d: %v", poll.ID, err)
		}
	}
	return nil
}

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

	// Initialize database with sample data
	if err := initDatabase(); err != nil {
		log.Fatalf("Failed to initialize database: %v", err)
	}

	http.HandleFunc("/api/vote", voteHandler)
	http.HandleFunc("/api/vote/{id}", voteIDHandler)
	http.HandleFunc("/api/stats", statsHandler)
	http.HandleFunc("/api/polls/{id}", pollsIDHandler)
	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))
}