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

package handlers

import (
	"database/sql"
	"errors"
	"log"
	"net/http"
	"strconv"
	"time"

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

// View model for listing/assigning schedules
type AssignmentVM struct {
	ID               int
	VolunteerID      int
	VolunteerName    string
	AddressID        int
	Address          string
	Date             string // YYYY-MM-DD (for input[type=date])
	AppointmentTime  string // HH:MM
	VisitedValidated bool
}

// GET + POST in one handler:
// - GET: show assignments + form to assign
// - POST: create a new assignment
func AdminAssignmentsHandler(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case http.MethodPost:
		if err := createAssignmentFromForm(r); err != nil {
			log.Println("create assignment error:", err)
			volunteers, _ := fetchVolunteers()
			addresses, _ := fetchAddresses()
			assignments, _ := fetchAssignments()

			utils.Render(w, "schedual/assignments.html", map[string]interface{}{
				"Title":           "Admin — Assign Addresses",
				"IsAuthenticated": true,
				"ActiveSection":   "admin_assignments",
				"Volunteers":      volunteers,
				"Addresses":       addresses,
				"Assignments":     assignments,
				"Error":           err.Error(),
			})
			return
		}
		http.Redirect(w, r, "/admin/assignments", http.StatusSeeOther)
		return
	}

	// GET: fetch volunteers, addresses, and existing assignments
	volunteers, err := fetchVolunteers()
	if err != nil {
		log.Println("fetch volunteers error:", err)
		http.Error(w, "Failed to load volunteers", http.StatusInternalServerError)
		return
	}
	addresses, err := fetchAddresses()
	if err != nil {
		log.Println("fetch addresses error:", err)
		http.Error(w, "Failed to load addresses", http.StatusInternalServerError)
		return
	}
	assignments, err := fetchAssignments()
	if err != nil {
		log.Println("fetch assignments error:", err)
		http.Error(w, "Failed to load assignments", http.StatusInternalServerError)
		return
	}

	utils.Render(w, "assignments.html", map[string]interface{}{
		"Title":           "Admin — Assign Addresses",
		"IsAuthenticated": true,
		"ActiveSection":   "admin_assignments",
		"Volunteers":      volunteers,
		"Addresses":       addresses,
		"Assignments":     assignments,
	})
}

// GET (edit form) + POST (update/delete)
func AdminAssignmentEditHandler(w http.ResponseWriter, r *http.Request) {
	idStr := r.URL.Query().Get("id")
	id, _ := strconv.Atoi(idStr)
	if id <= 0 {
		http.NotFound(w, r)
		return
	}

	if r.Method == http.MethodPost {
		action := r.FormValue("action")
		switch action {
		case "delete":
			if err := deleteAssignment(id); err != nil {
				log.Println("delete assignment error:", err)
				http.Error(w, "Failed to delete assignment", http.StatusInternalServerError)
				return
			}
			http.Redirect(w, r, "/admin/assignments", http.StatusSeeOther)
			return
		case "update":
			if err := updateAssignmentFromForm(id, r); err != nil {
				log.Println("update assignment error:", err)
				vm, _ := fetchAssignmentByID(id)
				volunteers, _ := fetchVolunteers()
				addresses, _ := fetchAddresses()

				utils.Render(w, "assignment_edit.html", map[string]interface{}{
					"Title":      "Edit Assignment",
					"Assignment": vm,
					"Volunteers": volunteers,
					"Addresses":  addresses,
					"Error":      err.Error(),
				})
				return
			}
			http.Redirect(w, r, "/admin/assignments", http.StatusSeeOther)
			return
		default:
			http.Error(w, "Unknown action", http.StatusBadRequest)
			return
		}
	}

	// GET edit
	vm, err := fetchAssignmentByID(id)
	if err != nil {
		if err == sql.ErrNoRows {
			http.NotFound(w, r)
			return
		}
		log.Println("fetch assignment by ID error:", err)
		http.Error(w, "Failed to load assignment", http.StatusInternalServerError)
		return
	}

	volunteers, err := fetchVolunteers()
	if err != nil {
		log.Println("fetch volunteers error:", err)
		http.Error(w, "Failed to load volunteers", http.StatusInternalServerError)
		return
	}
	addresses, err := fetchAddresses()
	if err != nil {
		log.Println("fetch addresses error:", err)
		http.Error(w, "Failed to load addresses", http.StatusInternalServerError)
		return
	}

	utils.Render(w, "assignment_edit.html", map[string]interface{}{
		"Title":      "Edit Assignment",
		"Assignment": vm,
		"Volunteers": volunteers,
		"Addresses":  addresses,
	})
}

// ----- Helpers -----

func createAssignmentFromForm(r *http.Request) error {
	volID, _ := strconv.Atoi(r.FormValue("volunteer_id"))
	addrID, _ := strconv.Atoi(r.FormValue("address_id"))
	dateStr := r.FormValue("date")
	timeStr := r.FormValue("appointment_time")

	if volID <= 0 || addrID <= 0 || dateStr == "" || timeStr == "" {
		return errors.New("please fill all required fields")
	}

	if _, err := time.Parse("2006-01-02", dateStr); err != nil {
		return errors.New("invalid date format")
	}
	if _, err := time.Parse("15:04", timeStr); err != nil {
		return errors.New("invalid time format")
	}

	_, err := models.DB.Exec(`
		INSERT INTO schedual (user_id, address_id, appointment_date, appointment_time, created_at, updated_at)
		VALUES ($1,$2,$3,$4,NOW(),NOW())
	`, volID, addrID, dateStr, timeStr)

	if err != nil {
		log.Println("database insert error:", err)
		return errors.New("failed to create assignment")
	}

	return nil
}

func updateAssignmentFromForm(id int, r *http.Request) error {
	volID, _ := strconv.Atoi(r.FormValue("volunteer_id"))
	addrID, _ := strconv.Atoi(r.FormValue("address_id"))
	dateStr := r.FormValue("date")
	timeStr := r.FormValue("appointment_time")

	if volID <= 0 || addrID <= 0 || dateStr == "" || timeStr == "" {
		return errors.New("please fill all required fields")
	}

	if _, err := time.Parse("2006-01-02", dateStr); err != nil {
		return errors.New("invalid date format")
	}
	if _, err := time.Parse("15:04", timeStr); err != nil {
		return errors.New("invalid time format")
	}

	result, err := models.DB.Exec(`
		UPDATE schedual
		SET user_id=$1, address_id=$2, appointment_date=$3, appointment_time=$4, updated_at=NOW()
		WHERE schedual_id=$5
	`, volID, addrID, dateStr, timeStr, id)

	if err != nil {
		log.Println("database update error:", err)
		return errors.New("failed to update assignment")
	}

	rowsAffected, _ := result.RowsAffected()
	if rowsAffected == 0 {
		return errors.New("assignment not found")
	}
	return nil
}

func deleteAssignment(id int) error {
	result, err := models.DB.Exec(`DELETE FROM schedual WHERE schedual_id=$1`, id)
	if err != nil {
		log.Println("database delete error:", err)
		return errors.New("failed to delete assignment")
	}
	rowsAffected, _ := result.RowsAffected()
	if rowsAffected == 0 {
		return errors.New("assignment not found")
	}
	return nil
}

// Fetch volunteers
type VolunteerPick struct {
	ID        int
	FirstName string
	LastName  string
	Email     string
}

func fetchVolunteers() ([]VolunteerPick, error) {
	rows, err := models.DB.Query(`
		SELECT users_id, first_name, last_name, email
		FROM "user"
		WHERE role='volunteer'
		ORDER BY first_name, last_name
	`)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var out []VolunteerPick
	for rows.Next() {
		var v VolunteerPick
		if err := rows.Scan(&v.ID, &v.FirstName, &v.LastName, &v.Email); err != nil {
			log.Println("fetchVolunteers scan:", err)
			continue
		}
		out = append(out, v)
	}
	return out, rows.Err()
}

// Fetch addresses
type AddressPick struct {
	ID               int
	Label            string
	VisitedValidated bool
}

func fetchAddresses() ([]AddressPick, error) {
	rows, err := models.DB.Query(`
		SELECT
			address_id,
			address,
			street_name,
			street_type,
			street_quadrant,
			house_number,
			house_alpha,
			longitude,
			latitude,
			visited_validated
		FROM address_database
		ORDER BY address_id DESC
	`)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var out []AddressPick
	for rows.Next() {
		var addr models.AddressDatabase
		if err := rows.Scan(
			&addr.AddressID,
			&addr.Address,
			&addr.StreetName,
			&addr.StreetType,
			&addr.StreetQuadrant,
			&addr.HouseNumber,
			&addr.HouseAlpha,
			&addr.Longitude,
			&addr.Latitude,
			&addr.VisitedValidated,
		); err != nil {
			log.Println("fetchAddresses scan:", err)
			continue
		}

		label := addr.Address
		if label == "" {
			label = addr.HouseNumber
			if addr.StreetName != "" {
				if label != "" {
					label += " "
				}
				label += addr.StreetName
			}
			if addr.StreetType != "" {
				label += " " + addr.StreetType
			}
			if addr.StreetQuadrant != "" {
				label += " " + addr.StreetQuadrant
			}
			if addr.HouseAlpha != nil {
				label += " " + *addr.HouseAlpha
			}
		}

		out = append(out, AddressPick{
			ID:               addr.AddressID,
			Label:            label,
			VisitedValidated: addr.VisitedValidated,
		})
	}
	return out, rows.Err()
}

// Add this missing function
func fetchAssignments() ([]AssignmentVM, error) {
	rows, err := models.DB.Query(`
		SELECT
			s.schedual_id,
			u.users_id,
			COALESCE(u.first_name,'') || ' ' || COALESCE(u.last_name,'') AS volunteer_name,
			a.address_id,
			COALESCE(a.address,'') AS address,
			s.appointment_date,
			s.appointment_time
		FROM schedual s
		JOIN "user" u ON u.users_id = s.user_id
		JOIN address_database a ON a.address_id = s.address_id
		ORDER BY s.appointment_date DESC, s.appointment_time DESC
	`)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var assignments []AssignmentVM
	for rows.Next() {
		var vm AssignmentVM
		if err := rows.Scan(&vm.ID, &vm.VolunteerID, &vm.VolunteerName, &vm.AddressID, &vm.Address,
			&vm.Date, &vm.AppointmentTime); err != nil {
			log.Println("fetchAssignments scan:", err)
			continue
		}
		assignments = append(assignments, vm)
	}
	return assignments, rows.Err()
}

func fetchAssignmentByID(id int) (AssignmentVM, error) {
	var vm AssignmentVM
	err := models.DB.QueryRow(`
		SELECT
			s.schedual_id,
			u.users_id,
			COALESCE(u.first_name,'') || ' ' || COALESCE(u.last_name,'') AS volunteer_name,
			a.address_id,
			COALESCE(a.address,'') AS address,
			s.appointment_date,
			s.appointment_time
		FROM schedual s
		JOIN "user" u ON u.users_id = s.user_id
		JOIN address_database a ON a.address_id = s.address_id
		WHERE s.schedual_id = $1
	`, id).Scan(&vm.ID, &vm.VolunteerID, &vm.VolunteerName, &vm.AddressID, &vm.Address,
		&vm.Date, &vm.AppointmentTime)

	return vm, err
}