package handler import ( "blackforestbytes.com/simplecloudnotifier/api/apierr" "blackforestbytes.com/simplecloudnotifier/api/models" "blackforestbytes.com/simplecloudnotifier/common/ginresp" "blackforestbytes.com/simplecloudnotifier/logic" "database/sql" "github.com/gin-gonic/gin" "gogs.mikescher.com/BlackForestBytes/goext/langext" "net/http" "regexp" ) type APIHandler struct { app *logic.Application } // CreateUser swaggerdoc // // @Summary Create a new user // @ID api-user-create // // @Param post_body body handler.CreateUser.body false " " // // @Success 200 {object} models.UserJSON // @Failure 400 {object} ginresp.apiError // @Failure 500 {object} ginresp.apiError // // @Router /api-v2/user/ [POST] func (h APIHandler) CreateUser(g *gin.Context) ginresp.HTTPResponse { type body struct { FCMToken string `json:"fcm_token"` ProToken *string `json:"pro_token"` Username *string `json:"username"` AgentModel string `json:"agent_model"` AgentVersion string `json:"agent_version"` ClientType string `json:"client_type"` } var b body ctx, errResp := h.app.StartRequest(g, nil, nil, &b) if errResp != nil { return *errResp } defer ctx.Cancel() var clientType models.ClientType if b.ClientType == string(models.ClientTypeAndroid) { clientType = models.ClientTypeAndroid } else if b.ClientType == string(models.ClientTypeIOS) { clientType = models.ClientTypeIOS } else { return ginresp.InternAPIError(400, apierr.INVALID_CLIENTTYPE, "Invalid ClientType", nil) } if b.ProToken != nil { ptok, err := h.app.VerifyProToken(*b.ProToken) if err != nil { return ginresp.InternAPIError(500, apierr.FAILED_VERIFY_PRO_TOKEN, "Failed to query purchase status", err) } if !ptok { return ginresp.InternAPIError(400, apierr.INVALID_PRO_TOKEN, "Purchase token could not be verified", nil) } } readKey := h.app.GenerateRandomAuthKey() sendKey := h.app.GenerateRandomAuthKey() adminKey := h.app.GenerateRandomAuthKey() err := h.app.Database.ClearFCMTokens(ctx, b.FCMToken) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to clear existing fcm tokens", err) } if b.ProToken != nil { err := h.app.Database.ClearProTokens(ctx, *b.ProToken) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to clear existing fcm tokens", err) } } userobj, err := h.app.Database.CreateUser(ctx, readKey, sendKey, adminKey, b.ProToken, b.Username) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to create user in db", err) } _, err = h.app.Database.CreateClient(ctx, userobj.UserID, clientType, b.FCMToken, b.AgentModel, b.AgentVersion) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to create user in db", err) } return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, userobj.JSON())) } // GetUser swaggerdoc // // @Summary Get a user // @ID api-user-get // // @Param uid path int true "UserID" // // @Success 200 {object} models.UserJSON // @Failure 400 {object} ginresp.apiError // @Failure 401 {object} ginresp.apiError // @Failure 404 {object} ginresp.apiError // @Failure 500 {object} ginresp.apiError // // @Router /api-v2/user/{uid} [GET] func (h APIHandler) GetUser(g *gin.Context) ginresp.HTTPResponse { type uri struct { UserID int64 `uri:"uid"` } var u uri ctx, errResp := h.app.StartRequest(g, &u, nil, nil) if errResp != nil { return *errResp } defer ctx.Cancel() if permResp := ctx.CheckPermissionUserRead(u.UserID); permResp != nil { return *permResp } user, err := h.app.Database.GetUser(ctx, u.UserID) if err == sql.ErrNoRows { return ginresp.InternAPIError(404, apierr.USER_NOT_FOUND, "User not found", err) } if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to query user", err) } return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, user.JSON())) } // UpdateUser swaggerdoc // // @Summary (Partially) update a user // @Description The body-values are optional, only send the ones you want to update // @ID api-user-update // // @Param post_body body handler.UpdateUser.body false " " // // @Success 200 {object} models.UserJSON // @Failure 400 {object} ginresp.apiError // @Failure 401 {object} ginresp.apiError // @Failure 404 {object} ginresp.apiError // @Failure 500 {object} ginresp.apiError // // @Router /api-v2/user/{uid} [PATCH] func (h APIHandler) UpdateUser(g *gin.Context) ginresp.HTTPResponse { type uri struct { UserID int64 `uri:"uid"` } type body struct { Username *string `json:"username"` ProToken *string `json:"pro_token"` } var u uri var b body ctx, errResp := h.app.StartRequest(g, &u, nil, &b) if errResp != nil { return *errResp } defer ctx.Cancel() if permResp := ctx.CheckPermissionUserAdmin(u.UserID); permResp != nil { return *permResp } if b.Username != nil { username := langext.Ptr(regexp.MustCompile(`[[:alnum:]\-_]`).ReplaceAllString(*b.Username, "")) if *username == "" { username = nil } err := h.app.Database.UpdateUserUsername(ctx, u.UserID, b.Username) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to update user", err) } } if b.ProToken != nil { ptok, err := h.app.VerifyProToken(*b.ProToken) if err != nil { return ginresp.InternAPIError(500, apierr.FAILED_VERIFY_PRO_TOKEN, "Failed to query purchase status", err) } if !ptok { return ginresp.InternAPIError(400, apierr.INVALID_PRO_TOKEN, "Purchase token could not be verified", nil) } err = h.app.Database.ClearProTokens(ctx, *b.ProToken) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to clear existing fcm tokens", err) } err = h.app.Database.UpdateUserProToken(ctx, u.UserID, b.ProToken) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to update user", err) } } user, err := h.app.Database.GetUser(ctx, u.UserID) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to query (updated) user", err) } return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, user.JSON())) } // ListClients swaggerdoc // // @Summary List all clients // @ID api-clients-list // // @Param uid path int true "UserID" // // @Success 200 {object} handler.ListClients.result // @Failure 400 {object} ginresp.apiError // @Failure 401 {object} ginresp.apiError // @Failure 404 {object} ginresp.apiError // @Failure 500 {object} ginresp.apiError // // @Router /api-v2/user/{uid}/clients [GET] func (h APIHandler) ListClients(g *gin.Context) ginresp.HTTPResponse { type uri struct { UserID int64 `uri:"uid"` } type result struct { Clients []models.ClientJSON `json:"clients"` } var u uri ctx, errResp := h.app.StartRequest(g, &u, nil, nil) if errResp != nil { return *errResp } defer ctx.Cancel() if permResp := ctx.CheckPermissionUserRead(u.UserID); permResp != nil { return *permResp } clients, err := h.app.Database.ListClients(ctx, u.UserID) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to query clients", err) } res := langext.ArrMap(clients, func(v models.Client) models.ClientJSON { return v.JSON() }) return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, result{Clients: res})) } // GetClient swaggerdoc // // @Summary get a single clients // @ID api-clients-get // // @Param uid path int true "UserID" // @Param cid path int true "ClientID" // // @Success 200 {object} models.ClientJSON // @Failure 400 {object} ginresp.apiError // @Failure 401 {object} ginresp.apiError // @Failure 404 {object} ginresp.apiError // @Failure 500 {object} ginresp.apiError // // @Router /api-v2/user/{uid}/clients/{cid} [GET] func (h APIHandler) GetClient(g *gin.Context) ginresp.HTTPResponse { type uri struct { UserID int64 `uri:"uid"` ClientID int64 `uri:"cid"` } var u uri ctx, errResp := h.app.StartRequest(g, &u, nil, nil) if errResp != nil { return *errResp } defer ctx.Cancel() if permResp := ctx.CheckPermissionUserRead(u.UserID); permResp != nil { return *permResp } client, err := h.app.Database.GetClient(ctx, u.UserID, u.ClientID) if err == sql.ErrNoRows { return ginresp.InternAPIError(404, apierr.CLIENT_NOT_FOUND, "Client not found", err) } if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to query client", err) } return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, client.JSON())) } // AddClient swaggerdoc // // @Summary get a single clients // @ID api-clients-get // // @Param uid path int true "UserID" // // @Param post_body body handler.AddClient.body false " " // // @Success 200 {object} models.ClientJSON // @Failure 400 {object} ginresp.apiError // @Failure 401 {object} ginresp.apiError // @Failure 404 {object} ginresp.apiError // @Failure 500 {object} ginresp.apiError // // @Router /api-v2/user/{uid}/clients [POST] func (h APIHandler) AddClient(g *gin.Context) ginresp.HTTPResponse { type uri struct { UserID int64 `uri:"uid"` } type body struct { FCMToken string `json:"fcm_token"` AgentModel string `json:"agent_model"` AgentVersion string `json:"agent_version"` ClientType string `json:"client_type"` } var u uri var b body ctx, errResp := h.app.StartRequest(g, &u, nil, &b) if errResp != nil { return *errResp } defer ctx.Cancel() var clientType models.ClientType if b.ClientType == string(models.ClientTypeAndroid) { clientType = models.ClientTypeAndroid } else if b.ClientType == string(models.ClientTypeIOS) { clientType = models.ClientTypeIOS } else { return ginresp.InternAPIError(400, apierr.INVALID_CLIENTTYPE, "Invalid ClientType", nil) } if permResp := ctx.CheckPermissionUserAdmin(u.UserID); permResp != nil { return *permResp } client, err := h.app.Database.CreateClient(ctx, u.UserID, clientType, b.FCMToken, b.AgentModel, b.AgentVersion) if err != nil { return ginresp.InternAPIError(500, apierr.DATABASE_ERROR, "Failed to create user in db", err) } return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, client.JSON())) } func (h APIHandler) DeleteClient(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) ListChannels(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) GetChannel(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) GetChannelMessages(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) ListUserSubscriptions(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) ListChannelSubscriptions(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) GetSubscription(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) CancelSubscription(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) CreateSubscription(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) ListMessages(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) GetMessage(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func (h APIHandler) DeleteMessage(g *gin.Context) ginresp.HTTPResponse { return ginresp.NotImplemented() } func NewAPIHandler(app *logic.Application) APIHandler { return APIHandler{ app: app, } }