Refactor and remove sqlite and replace with MySQL

internal/handlers/messages.go

@@ -0,0 +1,186 @@
+package handlers
+
+import (
+	"database/sql"
+	"log"
+	"net/http"
+
+	templateHandlers "synlotto-website/handlers/template"
+	"synlotto-website/helpers"
+	httpHelpers "synlotto-website/helpers/http"
+	securityHelpers "synlotto-website/helpers/security"
+	templateHelpers "synlotto-website/helpers/template"
+
+	"synlotto-website/storage"
+)
+
+func MessagesInboxHandler(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		userID, ok := securityHelpers.GetCurrentUserID(r)
+		if !ok {
+			templateHelpers.RenderError(w, r, 403)
+			return
+		}
+
+		page := helpers.Atoi(r.URL.Query().Get("page"))
+		if page < 1 {
+			page = 1
+		}
+		perPage := 10
+
+		totalCount := storage.GetInboxMessageCount(db, userID)
+		totalPages := (totalCount + perPage - 1) / perPage
+		if totalPages == 0 {
+			totalPages = 1
+		}
+
+		messages := storage.GetInboxMessages(db, userID, page, perPage)
+
+		data := templateHandlers.BuildTemplateData(db, w, r)
+		context := templateHelpers.TemplateContext(w, r, data)
+
+		context["Messages"] = messages
+		context["CurrentPage"] = page
+		context["TotalPages"] = totalPages
+		context["PageRange"] = templateHelpers.PageRange(page, totalPages)
+
+		tmpl := templateHelpers.LoadTemplateFiles("messages.html", "templates/account/messages/index.html")
+
+		if err := tmpl.ExecuteTemplate(w, "layout", context); err != nil {
+			// ToDo: Make this load all error pages without defining explictly.
+			templateHelpers.RenderError(w, r, 500)
+		}
+	}
+}
+
+func ReadMessageHandler(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		idStr := r.URL.Query().Get("id")
+		messageID := helpers.Atoi(idStr)
+
+		session, _ := httpHelpers.GetSession(w, r)
+		userID, ok := session.Values["user_id"].(int)
+		if !ok {
+			templateHelpers.RenderError(w, r, 403)
+			return
+		}
+
+		message, err := storage.GetMessageByID(db, userID, messageID)
+		if err != nil {
+			log.Printf("❌ Message not found: %v", err)
+			message = nil
+		} else if !message.IsRead {
+			_ = storage.MarkMessageAsRead(db, messageID, userID)
+		}
+
+		data := templateHandlers.BuildTemplateData(db, w, r)
+		context := templateHelpers.TemplateContext(w, r, data)
+		context["Message"] = message
+
+		tmpl := templateHelpers.LoadTemplateFiles("read-message.html", "templates/account/messages/read.html")
+
+		tmpl.ExecuteTemplate(w, "layout", context)
+	}
+}
+
+func ArchiveMessageHandler(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		id := helpers.Atoi(r.URL.Query().Get("id"))
+		userID, ok := securityHelpers.GetCurrentUserID(r)
+		if !ok {
+			templateHelpers.RenderError(w, r, 403)
+			return
+		}
+
+		err := storage.ArchiveMessage(db, userID, id)
+		if err != nil {
+			templateHelpers.SetFlash(w, r, "Failed to archive message.")
+		} else {
+			templateHelpers.SetFlash(w, r, "Message archived.")
+		}
+
+		http.Redirect(w, r, "/account/messages", http.StatusSeeOther)
+	}
+}
+
+func ArchivedMessagesHandler(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		userID, ok := securityHelpers.GetCurrentUserID(r)
+		if !ok {
+			templateHelpers.RenderError(w, r, 403)
+			return
+		}
+
+		page := helpers.Atoi(r.URL.Query().Get("page"))
+		if page < 1 {
+			page = 1
+		}
+		perPage := 10
+
+		messages := storage.GetArchivedMessages(db, userID, page, perPage)
+		hasMore := len(messages) == perPage
+
+		data := templateHandlers.BuildTemplateData(db, w, r)
+		context := templateHelpers.TemplateContext(w, r, data)
+		context["Messages"] = messages
+		context["Page"] = page
+		context["HasMore"] = hasMore
+
+		tmpl := templateHelpers.LoadTemplateFiles("archived.html", "templates/account/messages/archived.html")
+		tmpl.ExecuteTemplate(w, "layout", context)
+	}
+}
+
+func SendMessageHandler(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		switch r.Method {
+		case http.MethodGet:
+			data := templateHandlers.BuildTemplateData(db, w, r)
+			context := templateHelpers.TemplateContext(w, r, data)
+			tmpl := templateHelpers.LoadTemplateFiles("send-message.html", "templates/account/messages/send.html")
+
+			if err := tmpl.ExecuteTemplate(w, "layout", context); err != nil {
+				templateHelpers.RenderError(w, r, 500)
+			}
+		case http.MethodPost:
+			senderID, ok := securityHelpers.GetCurrentUserID(r)
+			if !ok {
+				templateHelpers.RenderError(w, r, 403)
+				return
+			}
+
+			recipientID := helpers.Atoi(r.FormValue("recipient_id"))
+			subject := r.FormValue("subject")
+			body := r.FormValue("message")
+
+			if err := storage.SendMessage(db, senderID, recipientID, subject, body); err != nil {
+				templateHelpers.SetFlash(w, r, "Failed to send message.")
+			} else {
+				templateHelpers.SetFlash(w, r, "Message sent.")
+			}
+			http.Redirect(w, r, "/account/messages", http.StatusSeeOther)
+		default:
+			templateHelpers.RenderError(w, r, 405)
+		}
+	}
+}
+
+func RestoreMessageHandler(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		id := helpers.Atoi(r.URL.Query().Get("id"))
+		userID, ok := securityHelpers.GetCurrentUserID(r)
+		if !ok {
+			templateHelpers.RenderError(w, r, 403)
+			return
+		}
+
+		err := storage.RestoreMessage(db, userID, id)
+		if err != nil {
+			templateHelpers.SetFlash(w, r, "Failed to restore message.")
+		} else {
+			templateHelpers.SetFlash(w, r, "Message restored.")
+		}
+
+		http.Redirect(w, r, "/account/messages/archived", http.StatusSeeOther)
+	}
+}