Poll-system/app/internal/handlers/volunteer_poll.go

package handlers

import (
	"database/sql"
	"fmt"
	"log"

	"net/http"
	"strconv"

	"github.com/patel-mann/poll-system/app/internal/models"
	"github.com/patel-mann/poll-system/app/internal/utils"
)

func PollHandler(w http.ResponseWriter, r *http.Request) {
	username, _ := models.GetCurrentUserName(r)

	if r.Method == http.MethodGet {
		addressID := r.URL.Query().Get("address_id")
		if addressID == "" {
			http.Error(w, "Address ID required", http.StatusBadRequest)
			return
		}

		// Get address details
		var address string
		var userID int
		fmt.Print(addressID, userID)
		err := models.DB.QueryRow(`
			SELECT a.address, ap.user_id
			FROM appointment AS ap
			JOIN address_database a ON a.address_id = ap.address_id
			WHERE ap.address_id = $1
		`, addressID).Scan(&address, &userID)

		if err != nil {
			http.Error(w, "Address not found", http.StatusNotFound)
			return
		}

		// Check if poll already exists for this address
		var pollID int
		err = models.DB.QueryRow(`
			SELECT poll_id
			FROM poll
			WHERE address_id = $1 AND user_id = $2
		`, addressID, userID).Scan(&pollID)

		// If no poll exists, create one
		if err == sql.ErrNoRows {
			err = models.DB.QueryRow(`
				INSERT INTO poll (user_id, address_id, poll_title, poll_description, is_active)
				VALUES ($1, $2, 'Door-to-Door Poll', 'Campaign polling questions', true)
				RETURNING poll_id
			`, userID, addressID).Scan(&pollID)

			if err != nil {
				log.Printf("Failed to create poll: %v", err)
				http.Error(w, "Failed to create poll", http.StatusInternalServerError)
				return
			}
		} else if err != nil {
			log.Printf("Database error: %v", err)
			http.Error(w, "Database error", http.StatusInternalServerError)
			return
		}

		utils.Render(w, "poll_form.html", map[string]interface{}{
			"Title":            "Poll Questions",
			"IsAuthenticated":  true,
			"ShowVolunteerNav": true,
			"ActiveSection":    "schedule",
			"UserName":         username,
			"PollID":           pollID,
			"AddressID":        addressID,
			"Address":          address,
			"PageIcon":         "fas fa-poll",
		})
		return
	}

	if r.Method == http.MethodPost {
		err := r.ParseForm()
		if err != nil {
			http.Error(w, "Invalid form", http.StatusBadRequest)
			return
		}

		pollID := r.FormValue("poll_id")
		postalCode := r.FormValue("postal_code")

		// Parse integer values
		question3LawnSigns, _ := strconv.Atoi(r.FormValue("question3_lawn_signs"))
		question4BannerSigns, _ := strconv.Atoi(r.FormValue("question4_banner_signs"))
		question5Thoughts := r.FormValue("question5_thoughts")
		question6donation := r.FormValue("question6_amount")

		// Parse boolean values
		var question1VotedBefore *bool
		if val := r.FormValue("question1_voted_before"); val != "" {
			switch val {
			case "true":
				b := true
				question1VotedBefore = &b
			case "false":
				b := false
				question1VotedBefore = &b
			}
		}

		var question2VoteAgain *bool
		if val := r.FormValue("question2_vote_again"); val != "" {
			switch val {
			case "true":
				b := true
				question2VoteAgain = &b
			case "false":
				b := false
				question2VoteAgain = &b
			}
		}

		// Insert poll response
		_, err = models.DB.Exec(`
			INSERT INTO poll_response (
				poll_id, respondent_postal_code, question1_voted_before,
				question2_vote_again, question3_lawn_signs, question4_banner_signs,
				question5_thoughts, question6_donation_amount
			) VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
		`, pollID, postalCode, question1VotedBefore, question2VoteAgain,
			question3LawnSigns, question4BannerSigns, question5Thoughts, question6donation)

		if err != nil {
			fmt.Print(err)
			http.Error(w, "Failed to save poll response", http.StatusInternalServerError)
			return
		} else {
			_, err := models.DB.Exec(`
				UPDATE address_database
				SET visited_validated = true
				WHERE address_id IN (
					SELECT address_id
					FROM poll
					WHERE poll_id = $1
				)
			`, pollID)
			if err != nil {
				fmt.Print(err)
				http.Error(w, "Failed to save poll response", http.StatusInternalServerError)
				return
			}

		}

		http.Redirect(w, r, "/volunteer/Addresses", http.StatusSeeOther)
	}
}