From 042cd3b473d0bfe32f2ebcac907437f456e3e112 Mon Sep 17 00:00:00 2001
From: Mo Tarbin <mhed.t91@gmail.com>
Date: Sat, 10 Aug 2024 00:26:33 -0400
Subject: Add endpoint to update user notification target

---
 internal/user/handler.go     | 27 +++++++++++++++++++++++++++
 internal/user/model/model.go | 24 +++++++++++++++++++++---
 2 files changed, 48 insertions(+), 3 deletions(-)

(limited to 'internal')

diff --git a/internal/user/handler.go b/internal/user/handler.go
index 15e881f..ff885f3 100644
--- a/internal/user/handler.go
+++ b/internal/user/handler.go
@@ -486,6 +486,32 @@ func (h *Handler) DeleteUserToken(c *gin.Context) {
 	c.JSON(http.StatusOK, gin.H{})
 }
 
+func (h *Handler) UpdateNotificationTarget(c *gin.Context) {
+	currentUser, ok := auth.CurrentUser(c)
+	if !ok {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get current user"})
+		return
+	}
+
+	type Request struct {
+		Type  uModel.UserNotificationType `json:"type" binding:"required"`
+		Token string                      `json:"token" binding:"required"`
+	}
+
+	var req Request
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request"})
+		return
+	}
+
+	err := h.userRepo.UpdateNotificationTarget(c, currentUser.ID, req.Token, req.Type)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update notification target"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{})
+}
 func Routes(router *gin.Engine, h *Handler, auth *jwt.GinJWTMiddleware, limiter *limiter.Limiter) {
 
 	userRoutes := router.Group("users")
@@ -497,6 +523,7 @@ func Routes(router *gin.Engine, h *Handler, auth *jwt.GinJWTMiddleware, limiter
 		userRoutes.POST("/tokens", h.CreateLongLivedToken)
 		userRoutes.GET("/tokens", h.GetAllUserToken)
 		userRoutes.DELETE("/tokens/:id", h.DeleteUserToken)
+		userRoutes.PUT("/targets", h.UpdateNotificationTarget)
 	}
 
 	authRoutes := router.Group("auth")
diff --git a/internal/user/model/model.go b/internal/user/model/model.go
index 4874ac1..4cfb38b 100644
--- a/internal/user/model/model.go
+++ b/internal/user/model/model.go
@@ -16,9 +16,10 @@ type User struct {
 	UpdatedAt   time.Time `json:"updated_at" gorm:"column:updated_at"`    // Updated at
 	Disabled    bool      `json:"disabled" gorm:"column:disabled"`        // Disabled
 	// Email    string `json:"email" gorm:"column:email"`       // Email
-	CustomerID   *string `gorm:"column:customer_id;<-:false"`                      // read one column
-	Subscription *string `json:"subscription" gorm:"column:subscription;<-:false"` // read one column
-	Expiration   *string `json:"expiration" gorm:"column:expiration;<-:false"`     // read one column
+	CustomerID              *string                  `gorm:"column:customer_id;<-:false"`                      // read only column
+	Subscription            *string                  `json:"subscription" gorm:"column:subscription;<-:false"` // read only column
+	Expiration              *string                  `json:"expiration" gorm:"column:expiration;<-:false"`     // read only column
+	UserNotificationTargets []UserNotificationTarget `json:"-" gorm:"foreignKey:UserID;references:ID"`
 }
 
 type UserPasswordReset struct {
@@ -36,3 +37,20 @@ type APIToken struct {
 	Token     string    `json:"token" gorm:"column:token;index"`    // Index on token
 	CreatedAt time.Time `json:"createdAt" gorm:"column:created_at"`
 }
+
+type UserNotificationTarget struct {
+	ID        int                  `json:"id" gorm:"primary_key"`              // Unique identifier
+	UserID    int                  `json:"userId" gorm:"column:user_id;index"` // Index on userID
+	Type      UserNotificationType `json:"type" gorm:"column:type"`            // Type
+	TargetID  string               `json:"targetId" gorm:"column:target_id"`   // Target ID
+	CreatedAt time.Time            `json:"createdAt" gorm:"column:created_at"`
+}
+
+type UserNotificationType int8
+
+const (
+	_ UserNotificationType = iota
+	Android
+	IOS
+	Telegram
+)
-- 
cgit