feat: create token schema; apns development mode for dev

apis/token/apis.go

@@ -1,11 +1,13 @@
 package token
 
 import (
+	"time"
+
 	"github.com/gofiber/fiber/v2"
 	"github.com/opentreehole/go-common"
 	"gorm.io/gorm"
+
 	. "notification/models"
-	"time"
 )
 
 // ListTokens
@@ -29,16 +31,17 @@ func ListTokens(c *fiber.Ctx) (err error) {
 	return c.JSON(tokens)
 }
 
-// AddToken
+// CreateToken
 // @Summary Add Token of a User
 // @Tags Token
 // @Produce application/json
-// @Param json body models.PushToken true "json"
+// @Param json body CreateTokenRequest true "json"
 // @Router /users/push-tokens [post]
+// @Router /users/push-tokens [put]
 // @Success 200 {object} PushToken
-func AddToken(c *fiber.Ctx) (err error) {
-	var token PushToken
-	err = common.ValidateBody(c, &token)
+func CreateToken(c *fiber.Ctx) (err error) {
+	var body CreateTokenRequest
+	err = common.ValidateBody(c, &body)
 	if err != nil {
 		return err
 	}
@@ -47,17 +50,24 @@ func AddToken(c *fiber.Ctx) (err error) {
 	if !ok {
 		return common.Unauthorized()
 	}
-	token.UserID = userID
+
+	token := PushToken{
+		UserID:      userID,
+		Service:     NewPushService(body.Service),
+		DeviceID:    body.DeviceID,
+		Token:       body.Token,
+		PackageName: body.PackageName,
+		CreatedAt:   time.Now(),
+		UpdatedAt:   time.Now(),
+	}
+
 	err = DB.Transaction(func(tx *gorm.DB) (err error) {
 		// remove all device_id duplicates
 		err = tx.Where("device_id = ?", token.DeviceID).Delete(&PushToken{}).Error
 		if err != nil {
 			return err
 		}
 
-		token.CreatedAt = time.Now()
-		token.UpdatedAt = time.Now()
-
 		// create or update new token
 		return tx.Save(&token).Error
 	})

apis/token/routes.go

@@ -2,13 +2,14 @@ package token
 
 import (
 	"context"
+
 	"github.com/gofiber/fiber/v2"
 )
 
 func RegisterRoutes(app fiber.Router) {
 	app.Get("/users/push-tokens", ListTokens)
-	app.Post("/users/push-tokens", AddToken)
-	app.Put("/users/push-tokens", AddToken)
+	app.Post("/users/push-tokens", CreateToken)
+	app.Put("/users/push-tokens", CreateToken)
 	app.Delete("/users/push-tokens", DeleteToken)
 	app.Delete("/users/push-tokens/_all", DeleteAllTokens)
 }

apis/token/schema.go

@@ -0,0 +1,8 @@
+package token
+
+type CreateTokenRequest struct {
+	DeviceID    string `json:"device_id" validate:"max=64"`
+	Service     string `json:"service" validate:"required,oneof=apns fcm mipush"`
+	Token       string `json:"token" validate:"max=64"`
+	PackageName string `json:"package_name"`
+}

docs/swagger.json

@@ -98,7 +98,7 @@
                     }
                 }
             },
-            "post": {
+            "put": {
                 "produces": [
                     "application/json"
                 ],
@@ -112,10 +112,38 @@
                         "name": "json",
                         "in": "body",
                         "required": true,
+                        "schema": {
+                            "$ref": "#/definitions/token.CreateTokenRequest"
+                        }
+                    }
+                ],
+                "responses": {
+                    "200": {
+                        "description": "OK",
                         "schema": {
                             "$ref": "#/definitions/models.PushToken"
                         }
                     }
+                }
+            },
+            "post": {
+                "produces": [
+                    "application/json"
+                ],
+                "tags": [
+                    "Token"
+                ],
+                "summary": "Add Token of a User",
+                "parameters": [
+                    {
+                        "description": "json",
+                        "name": "json",
+                        "in": "body",
+                        "required": true,
+                        "schema": {
+                            "$ref": "#/definitions/token.CreateTokenRequest"
+                        }
+                    }
                 ],
                 "responses": {
                     "200": {
@@ -216,23 +244,11 @@
                 }
             }
         },
-        "models.PushService": {
-            "type": "integer",
-            "enum": [
-                0,
-                1,
-                2
-            ],
-            "x-enum-varnames": [
-                "ServiceAPNS",
-                "ServiceFCM",
-                "ServiceMipush"
-            ]
-        },
         "models.PushToken": {
             "type": "object",
             "required": [
                 "device_id",
+                "service",
                 "token"
             ],
             "properties": {
@@ -247,7 +263,12 @@
                     "type": "string"
                 },
                 "service": {
-                    "$ref": "#/definitions/models.PushService"
+                    "type": "string",
+                    "enum": [
+                        "apns",
+                        "fcm",
+                        "mipush"
+                    ]
                 },
                 "token": {
                     "type": "string",
@@ -262,6 +283,33 @@
                 }
             }
         },
+        "token.CreateTokenRequest": {
+            "type": "object",
+            "required": [
+                "service"
+            ],
+            "properties": {
+                "device_id": {
+                    "type": "string",
+                    "maxLength": 64
+                },
+                "package_name": {
+                    "type": "string"
+                },
+                "service": {
+                    "type": "string",
+                    "enum": [
+                        "apns",
+                        "fcm",
+                        "mipush"
+                    ]
+                },
+                "token": {
+                    "type": "string",
+                    "maxLength": 64
+                }
+            }
+        },
         "token.DeleteModel": {
             "type": "object",
             "properties": {