Create handlers/handlers.go

handlers/handlers.go

@@ -0,0 +1,274 @@
+package handlers
+
+import (
+	"html/template"
+	"net/http"
+	"strings"
+	"tempmail-service/models"
+	"tempmail-service/services"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+var mailService = services.NewMailService()
+
+// Page Handlers
+
+func IndexPage(c *gin.Context) {
+	domains, err := mailService.GetDomains()
+	if err != nil {
+		c.HTML(http.StatusInternalServerError, "index.html", gin.H{
+			"Error": "Failed to load domains. Please try again.",
+		})
+		return
+	}
+
+	c.HTML(http.StatusOK, "index.html", gin.H{
+		"Domains":         domains,
+		"RandomUsername":  services.GenerateUsername(),
+	})
+}
+
+func InboxPage(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.Redirect(http.StatusFound, "/")
+		return
+	}
+
+	messages, err := mailService.GetMessages(session.Token, 1)
+	if err != nil {
+		// Try to refresh token
+		if refreshErr := mailService.RefreshToken(session); refreshErr == nil {
+			messages, err = mailService.GetMessages(session.Token, 1)
+		}
+	}
+
+	var messageList []models.MessageSummary
+	if messages != nil {
+		messageList = messages.HydraMember
+	}
+
+	c.HTML(http.StatusOK, "inbox.html", gin.H{
+		"SessionID":  sessionID,
+		"Email":      session.Email,
+		"Messages":   messageList,
+		"TotalCount": len(messageList),
+		"ExpiresAt":  session.ExpiresAt.Format("Jan 02, 2006 15:04"),
+	})
+}
+
+func MessagePage(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+	messageID := c.Param("messageId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.Redirect(http.StatusFound, "/")
+		return
+	}
+
+	message, err := mailService.GetMessage(session.Token, messageID)
+	if err != nil {
+		c.Redirect(http.StatusFound, "/inbox/"+sessionID)
+		return
+	}
+
+	// Mark as read
+	_ = mailService.MarkAsRead(session.Token, messageID)
+
+	// Process HTML content
+	var htmlContent template.HTML
+	if len(message.HTML) > 0 {
+		htmlContent = template.HTML(strings.Join(message.HTML, ""))
+	}
+
+	c.HTML(http.StatusOK, "message.html", gin.H{
+		"SessionID":   sessionID,
+		"Email":       session.Email,
+		"Message":     message,
+		"HTMLContent": htmlContent,
+	})
+}
+
+// API Handlers
+
+func GetDomains(c *gin.Context) {
+	domains, err := mailService.GetDomains()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"domains": domains})
+}
+
+func CreateAccount(c *gin.Context) {
+	var req models.CreateEmailRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		// If no JSON, generate random
+		domains, err := mailService.GetDomains()
+		if err != nil || len(domains) == 0 {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "No domains available"})
+			return
+		}
+		req.Username = services.GenerateUsername()
+		req.Domain = domains[0].Domain
+	}
+
+	// Validate
+	if req.Username == "" {
+		req.Username = services.GenerateUsername()
+	}
+	if req.Domain == "" {
+		domains, _ := mailService.GetDomains()
+		if len(domains) > 0 {
+			req.Domain = domains[0].Domain
+		}
+	}
+
+	// Validate username length (min 1 character)
+	if len(req.Username) < 1 {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Username must be at least 1 character"})
+		return
+	}
+
+	session, err := mailService.CreateSession(req.Username, req.Domain)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusCreated, gin.H{
+		"sessionId": session.SessionID,
+		"email":     session.Email,
+		"expiresAt": session.ExpiresAt.Format(time.RFC3339),
+	})
+}
+
+func DeleteAccount(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+
+	if err := mailService.DeleteSession(sessionID); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "Account deleted successfully"})
+}
+
+func GetAccount(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"sessionId": session.SessionID,
+		"email":     session.Email,
+		"expiresAt": session.ExpiresAt.Format(time.RFC3339),
+	})
+}
+
+func GetMessages(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
+		return
+	}
+
+	messages, err := mailService.GetMessages(session.Token, 1)
+	if err != nil {
+		// Try refresh token
+		if refreshErr := mailService.RefreshToken(session); refreshErr == nil {
+			messages, err = mailService.GetMessages(session.Token, 1)
+		}
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"messages": messages.HydraMember,
+		"total":    messages.TotalItems,
+	})
+}
+
+func GetMessage(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+	messageID := c.Param("messageId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": "Session not found"})
+		return
+	}
+
+	message, err := mailService.GetMessage(session.Token, messageID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, message)
+}
+
+func DeleteMessage(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+	messageID := c.Param("messageId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": "Session not found"})
+		return
+	}
+
+	if err := mailService.DeleteMessage(session.Token, messageID); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "Message deleted"})
+}
+
+func MarkAsRead(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+	messageID := c.Param("messageId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": "Session not found"})
+		return
+	}
+
+	if err := mailService.MarkAsRead(session.Token, messageID); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"seen": true})
+}
+
+func CheckSession(c *gin.Context) {
+	sessionID := c.Param("sessionId")
+
+	session, err := mailService.GetSession(sessionID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"valid": false, "error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"valid":     true,
+		"email":     session.Email,
+		"expiresAt": session.ExpiresAt.Format(time.RFC3339),
+	})
+}