Refactor message sending into logic package (+ more tests for uptime-kuma)
This commit is contained in:
parent
a4a651229c
commit
9f656bdefe
@ -49,167 +49,6 @@
|
|||||||
|
|
||||||
- route to re-check all pro-token (for me)
|
- route to re-check all pro-token (for me)
|
||||||
|
|
||||||
- /send endpoint should be compatible with the [ webhook ] notifier of uptime-kuma
|
|
||||||
(or add another /kuma endpoint)
|
|
||||||
-> https://webhook.site/
|
|
||||||
|
|
||||||
|
|
||||||
````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
{"heartbeat":{"monitorID":89,"status":0,"time":"2023-07-31 18:56:15.374","msg":"timeout of 16000ms exceeded","important":true,"duration":36,"timezone":"Europe/Berlin","timezoneOffset":"+02:00","localDateTime":"2023-07-31 20:56:15"},"monitor":{"id":89,"name":"test","description":null,"pathName":"test","parent":null,"childrenIDs":[],"url":"https://exampleXYZ.com","method":"GET","hostname":null,"port":null,"maxretries":1,"weight":2000,"active":true,"forceInactive":false,"type":"http","interval":20,"retryInterval":20,"resendInterval":0,"keyword":null,"expiryNotification":false,"ignoreTls":false,"upsideDown":false,"packetSize":56,"maxredirects":10,"accepted_statuscodes":["200-299"],"dns_resolve_type":"A","dns_resolve_server":"1.1.1.1","dns_last_result":null,"docker_container":"","docker_host":null,"proxyId":null,"notificationIDList":{"2":true},"tags":[],"maintenance":false,"mqttTopic":"","mqttSuccessMessage":"","databaseQuery":null,"authMethod":null,"grpcUrl":null,"grpcProtobuf":null,"grpcMethod":null,"grpcServiceName":null,"grpcEnableTls":false,"radiusCalledStationId":null,"radiusCallingStationId":null,"game":null,"httpBodyEncoding":"json","includeSensitiveData":false},"msg":"[test] [🔴 Down] timeout of 16000ms exceeded"}
|
|
||||||
|
|
||||||
|
|
||||||
=====================================================================================================================================================================================================
|
|
||||||
|
|
||||||
|
|
||||||
{
|
|
||||||
"heartbeat": {
|
|
||||||
"monitorID": 89,
|
|
||||||
"status": 1,
|
|
||||||
"time": "2023-07-31 18:56:57.151",
|
|
||||||
"msg": "200 - OK",
|
|
||||||
"ping": 55,
|
|
||||||
"important": true,
|
|
||||||
"duration": 41,
|
|
||||||
"timezone": "Europe/Berlin",
|
|
||||||
"timezoneOffset": "+02:00",
|
|
||||||
"localDateTime": "2023-07-31 20:56:57"
|
|
||||||
},
|
|
||||||
"monitor": {
|
|
||||||
"id": 89,
|
|
||||||
"name": "test",
|
|
||||||
"description": null,
|
|
||||||
"pathName": "test",
|
|
||||||
"parent": null,
|
|
||||||
"childrenIDs": [],
|
|
||||||
"url": "https://example.com",
|
|
||||||
"method": "GET",
|
|
||||||
"hostname": null,
|
|
||||||
"port": null,
|
|
||||||
"maxretries": 1,
|
|
||||||
"weight": 2000,
|
|
||||||
"active": true,
|
|
||||||
"forceInactive": false,
|
|
||||||
"type": "http",
|
|
||||||
"interval": 20,
|
|
||||||
"retryInterval": 20,
|
|
||||||
"resendInterval": 0,
|
|
||||||
"keyword": null,
|
|
||||||
"expiryNotification": false,
|
|
||||||
"ignoreTls": false,
|
|
||||||
"upsideDown": false,
|
|
||||||
"packetSize": 56,
|
|
||||||
"maxredirects": 10,
|
|
||||||
"accepted_statuscodes": [
|
|
||||||
"200-299"
|
|
||||||
],
|
|
||||||
"dns_resolve_type": "A",
|
|
||||||
"dns_resolve_server": "1.1.1.1",
|
|
||||||
"dns_last_result": null,
|
|
||||||
"docker_container": "",
|
|
||||||
"docker_host": null,
|
|
||||||
"proxyId": null,
|
|
||||||
"notificationIDList": {
|
|
||||||
"2": true
|
|
||||||
},
|
|
||||||
"tags": [],
|
|
||||||
"maintenance": false,
|
|
||||||
"mqttTopic": "",
|
|
||||||
"mqttSuccessMessage": "",
|
|
||||||
"databaseQuery": null,
|
|
||||||
"authMethod": null,
|
|
||||||
"grpcUrl": null,
|
|
||||||
"grpcProtobuf": null,
|
|
||||||
"grpcMethod": null,
|
|
||||||
"grpcServiceName": null,
|
|
||||||
"grpcEnableTls": false,
|
|
||||||
"radiusCalledStationId": null,
|
|
||||||
"radiusCallingStationId": null,
|
|
||||||
"game": null,
|
|
||||||
"httpBodyEncoding": "json",
|
|
||||||
"includeSensitiveData": false
|
|
||||||
},
|
|
||||||
"msg": "[test] [✅ Up] 200 - OK"
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
=====================================================================================================================================================================================================
|
|
||||||
|
|
||||||
|
|
||||||
{
|
|
||||||
"heartbeat": {
|
|
||||||
"monitorID": 89,
|
|
||||||
"status": 0,
|
|
||||||
"time": "2023-07-31 18:57:44.037",
|
|
||||||
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
|
||||||
"important": true,
|
|
||||||
"duration": 20,
|
|
||||||
"timezone": "Europe/Berlin",
|
|
||||||
"timezoneOffset": "+02:00",
|
|
||||||
"localDateTime": "2023-07-31 20:57:44"
|
|
||||||
},
|
|
||||||
"monitor": {
|
|
||||||
"id": 89,
|
|
||||||
"name": "test",
|
|
||||||
"description": null,
|
|
||||||
"pathName": "test",
|
|
||||||
"parent": null,
|
|
||||||
"childrenIDs": [],
|
|
||||||
"url": "https://exampleasdsda.com",
|
|
||||||
"method": "GET",
|
|
||||||
"hostname": null,
|
|
||||||
"port": null,
|
|
||||||
"maxretries": 1,
|
|
||||||
"weight": 2000,
|
|
||||||
"active": true,
|
|
||||||
"forceInactive": false,
|
|
||||||
"type": "http",
|
|
||||||
"interval": 20,
|
|
||||||
"retryInterval": 20,
|
|
||||||
"resendInterval": 0,
|
|
||||||
"keyword": null,
|
|
||||||
"expiryNotification": false,
|
|
||||||
"ignoreTls": false,
|
|
||||||
"upsideDown": false,
|
|
||||||
"packetSize": 56,
|
|
||||||
"maxredirects": 10,
|
|
||||||
"accepted_statuscodes": [
|
|
||||||
"200-299"
|
|
||||||
],
|
|
||||||
"dns_resolve_type": "A",
|
|
||||||
"dns_resolve_server": "1.1.1.1",
|
|
||||||
"dns_last_result": null,
|
|
||||||
"docker_container": "",
|
|
||||||
"docker_host": null,
|
|
||||||
"proxyId": null,
|
|
||||||
"notificationIDList": {
|
|
||||||
"2": true
|
|
||||||
},
|
|
||||||
"tags": [],
|
|
||||||
"maintenance": false,
|
|
||||||
"mqttTopic": "",
|
|
||||||
"mqttSuccessMessage": "",
|
|
||||||
"databaseQuery": null,
|
|
||||||
"authMethod": null,
|
|
||||||
"grpcUrl": null,
|
|
||||||
"grpcProtobuf": null,
|
|
||||||
"grpcMethod": null,
|
|
||||||
"grpcServiceName": null,
|
|
||||||
"grpcEnableTls": false,
|
|
||||||
"radiusCalledStationId": null,
|
|
||||||
"radiusCallingStationId": null,
|
|
||||||
"game": null,
|
|
||||||
"httpBodyEncoding": "json",
|
|
||||||
"includeSensitiveData": false
|
|
||||||
},
|
|
||||||
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com"
|
|
||||||
}
|
|
||||||
|
|
||||||
````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````````
|
|
||||||
|
|
||||||
|
|
||||||
- endpoint to list all servernames of user (distinct select)
|
- endpoint to list all servernames of user (distinct select)
|
||||||
|
|
||||||
- weblogin, webapp, ...
|
- weblogin, webapp, ...
|
||||||
|
@ -29,7 +29,7 @@ func NewCompatHandler(app *logic.Application) CompatHandler {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendMessageCompat swaggerdoc
|
// SendMessage swaggerdoc
|
||||||
//
|
//
|
||||||
// @Deprecated
|
// @Deprecated
|
||||||
//
|
//
|
||||||
@ -37,17 +37,17 @@ func NewCompatHandler(app *logic.Application) CompatHandler {
|
|||||||
// @Description All parameter can be set via query-parameter or form-data body. Only UserID, UserKey and Title are required
|
// @Description All parameter can be set via query-parameter or form-data body. Only UserID, UserKey and Title are required
|
||||||
// @Tags External
|
// @Tags External
|
||||||
//
|
//
|
||||||
// @Param query_data query handler.SendMessageCompat.combined false " "
|
// @Param query_data query handler.SendMessage.combined false " "
|
||||||
// @Param form_data formData handler.SendMessageCompat.combined false " "
|
// @Param form_data formData handler.SendMessage.combined false " "
|
||||||
//
|
//
|
||||||
// @Success 200 {object} handler.SendMessageCompat.response
|
// @Success 200 {object} handler.SendMessage.response
|
||||||
// @Failure 400 {object} ginresp.apiError
|
// @Failure 400 {object} ginresp.apiError
|
||||||
// @Failure 401 {object} ginresp.apiError
|
// @Failure 401 {object} ginresp.apiError
|
||||||
// @Failure 403 {object} ginresp.apiError
|
// @Failure 403 {object} ginresp.apiError
|
||||||
// @Failure 500 {object} ginresp.apiError
|
// @Failure 500 {object} ginresp.apiError
|
||||||
//
|
//
|
||||||
// @Router /send.php [POST]
|
// @Router /send.php [POST]
|
||||||
func (h MessageHandler) SendMessageCompat(g *gin.Context) ginresp.HTTPResponse {
|
func (h CompatHandler) SendMessage(g *gin.Context) ginresp.HTTPResponse {
|
||||||
type combined struct {
|
type combined struct {
|
||||||
UserID *int64 `json:"user_id" form:"user_id"`
|
UserID *int64 `json:"user_id" form:"user_id"`
|
||||||
UserKey *string `json:"user_key" form:"user_key"`
|
UserKey *string `json:"user_key" form:"user_key"`
|
||||||
@ -88,7 +88,7 @@ func (h MessageHandler) SendMessageCompat(g *gin.Context) ginresp.HTTPResponse {
|
|||||||
return ginresp.SendAPIError(g, 400, apierr.USER_NOT_FOUND, hl.USER_ID, "User not found (compat)", nil)
|
return ginresp.SendAPIError(g, 400, apierr.USER_NOT_FOUND, hl.USER_ID, "User not found (compat)", nil)
|
||||||
}
|
}
|
||||||
|
|
||||||
okResp, errResp := h.sendMessageInternal(g, ctx, langext.Ptr(models.UserID(*newid)), data.UserKey, nil, data.Title, data.Content, data.Priority, data.UserMessageID, data.SendTimestamp, nil)
|
okResp, errResp := h.app.SendMessage(g, ctx, langext.Ptr(models.UserID(*newid)), data.UserKey, nil, data.Title, data.Content, data.Priority, data.UserMessageID, data.SendTimestamp, nil)
|
||||||
if errResp != nil {
|
if errResp != nil {
|
||||||
return *errResp
|
return *errResp
|
||||||
} else {
|
} else {
|
||||||
|
134
scnserver/api/handler/external.go
Normal file
134
scnserver/api/handler/external.go
Normal file
@ -0,0 +1,134 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/api/apierr"
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/api/ginresp"
|
||||||
|
primarydb "blackforestbytes.com/simplecloudnotifier/db/impl/primary"
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/logic"
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/models"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ExternalHandler struct {
|
||||||
|
app *logic.Application
|
||||||
|
database *primarydb.Database
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewExternalHandler(app *logic.Application) ExternalHandler {
|
||||||
|
return ExternalHandler{
|
||||||
|
app: app,
|
||||||
|
database: app.Database.Primary,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// UptimeKuma swaggerdoc
|
||||||
|
//
|
||||||
|
// @Summary Send a new message
|
||||||
|
// @Description All parameter can be set via query-parameter or the json body. Only UserID, UserKey and Title are required
|
||||||
|
// @Tags External
|
||||||
|
//
|
||||||
|
// @Param query_data query handler.UptimeKuma.query false " "
|
||||||
|
// @Param post_body body handler.UptimeKuma.body false " "
|
||||||
|
//
|
||||||
|
// @Success 200 {object} handler.UptimeKuma.response
|
||||||
|
// @Failure 400 {object} ginresp.apiError
|
||||||
|
// @Failure 401 {object} ginresp.apiError "The user_id was not found or the user_key is wrong"
|
||||||
|
// @Failure 403 {object} ginresp.apiError "The user has exceeded its daily quota - wait 24 hours or upgrade your account"
|
||||||
|
// @Failure 500 {object} ginresp.apiError "An internal server error occurred - try again later"
|
||||||
|
//
|
||||||
|
// @Router /external/v1/uptime-kuma [POST]
|
||||||
|
func (h ExternalHandler) UptimeKuma(g *gin.Context) ginresp.HTTPResponse {
|
||||||
|
type query struct {
|
||||||
|
UserID *models.UserID `form:"user_id" example:"7725"`
|
||||||
|
KeyToken *string `form:"key" example:"P3TNH8mvv14fm"`
|
||||||
|
Channel *string `form:"channel"`
|
||||||
|
ChannelUp *string `form:"channel_up"`
|
||||||
|
ChannelDown *string `form:"channel_down"`
|
||||||
|
Priority *int `form:"priority"`
|
||||||
|
PriorityUp *int `form:"priority_up"`
|
||||||
|
PriorityDown *int `form:"priority_down"`
|
||||||
|
SenderName *string `form:"senderName"`
|
||||||
|
}
|
||||||
|
type body struct {
|
||||||
|
Heartbeat *struct {
|
||||||
|
Time string `json:"time"`
|
||||||
|
Status int `json:"status"`
|
||||||
|
Msg string `json:"msg"`
|
||||||
|
Timezone string `json:"timezone"`
|
||||||
|
TimezoneOffset string `json:"timezoneOffset"`
|
||||||
|
LocalDateTime string `json:"localDateTime"`
|
||||||
|
} `json:"heartbeat"`
|
||||||
|
Monitor *struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Url *string `json:"url"`
|
||||||
|
} `json:"monitor"`
|
||||||
|
Msg *string `json:"msg"`
|
||||||
|
}
|
||||||
|
type response struct {
|
||||||
|
MessageID models.MessageID `json:"message_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var b body
|
||||||
|
var q query
|
||||||
|
ctx, httpErr := h.app.StartRequest(g, nil, &q, &b, nil)
|
||||||
|
if httpErr != nil {
|
||||||
|
return *httpErr
|
||||||
|
}
|
||||||
|
defer ctx.Cancel()
|
||||||
|
|
||||||
|
if b.Heartbeat == nil {
|
||||||
|
return ginresp.APIError(g, 400, apierr.BINDFAIL_BODY_PARAM, "missing field 'heartbeat' in request body", nil)
|
||||||
|
}
|
||||||
|
if b.Monitor == nil {
|
||||||
|
return ginresp.APIError(g, 400, apierr.BINDFAIL_BODY_PARAM, "missing field 'monitor' in request body", nil)
|
||||||
|
}
|
||||||
|
if b.Msg == nil {
|
||||||
|
return ginresp.APIError(g, 400, apierr.BINDFAIL_BODY_PARAM, "missing field 'msg' in request body", nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
title := langext.Conditional(b.Heartbeat.Status == 1, fmt.Sprintf("Monitor %v is back online", b.Monitor.Name), fmt.Sprintf("Monitor %v went down!", b.Monitor.Name))
|
||||||
|
|
||||||
|
content := b.Heartbeat.Msg
|
||||||
|
|
||||||
|
var timestamp *float64 = nil
|
||||||
|
if tz, err := time.LoadLocation(b.Heartbeat.Timezone); err == nil {
|
||||||
|
if ts, err := time.ParseInLocation("2006-01-02 15:04:05", b.Heartbeat.LocalDateTime, tz); err == nil {
|
||||||
|
timestamp = langext.Ptr(float64(ts.Unix()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var channel *string = nil
|
||||||
|
if q.Channel != nil {
|
||||||
|
channel = q.Channel
|
||||||
|
}
|
||||||
|
if q.ChannelUp != nil && b.Heartbeat.Status == 1 {
|
||||||
|
channel = q.ChannelUp
|
||||||
|
}
|
||||||
|
if q.ChannelDown != nil && b.Heartbeat.Status != 1 {
|
||||||
|
channel = q.ChannelDown
|
||||||
|
}
|
||||||
|
|
||||||
|
var priority *int = nil
|
||||||
|
if q.Priority != nil {
|
||||||
|
priority = q.Priority
|
||||||
|
}
|
||||||
|
if q.PriorityUp != nil && b.Heartbeat.Status == 1 {
|
||||||
|
priority = q.PriorityUp
|
||||||
|
}
|
||||||
|
if q.PriorityDown != nil && b.Heartbeat.Status != 1 {
|
||||||
|
priority = q.PriorityDown
|
||||||
|
}
|
||||||
|
|
||||||
|
okResp, errResp := h.app.SendMessage(g, ctx, q.UserID, q.KeyToken, channel, &title, &content, priority, nil, timestamp, q.SenderName)
|
||||||
|
if errResp != nil {
|
||||||
|
return *errResp
|
||||||
|
}
|
||||||
|
|
||||||
|
return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, response{
|
||||||
|
MessageID: okResp.Message.MessageID,
|
||||||
|
}))
|
||||||
|
}
|
@ -2,23 +2,14 @@ package handler
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"blackforestbytes.com/simplecloudnotifier/api/apierr"
|
"blackforestbytes.com/simplecloudnotifier/api/apierr"
|
||||||
hl "blackforestbytes.com/simplecloudnotifier/api/apihighlight"
|
|
||||||
"blackforestbytes.com/simplecloudnotifier/api/ginresp"
|
"blackforestbytes.com/simplecloudnotifier/api/ginresp"
|
||||||
primarydb "blackforestbytes.com/simplecloudnotifier/db/impl/primary"
|
primarydb "blackforestbytes.com/simplecloudnotifier/db/impl/primary"
|
||||||
"blackforestbytes.com/simplecloudnotifier/logic"
|
"blackforestbytes.com/simplecloudnotifier/logic"
|
||||||
"blackforestbytes.com/simplecloudnotifier/models"
|
"blackforestbytes.com/simplecloudnotifier/models"
|
||||||
"database/sql"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/rs/zerolog/log"
|
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/dataext"
|
"gogs.mikescher.com/BlackForestBytes/goext/dataext"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/mathext"
|
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/timeext"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type SendMessageResponse struct {
|
type SendMessageResponse struct {
|
||||||
@ -96,7 +87,7 @@ func (h MessageHandler) SendMessage(g *gin.Context) ginresp.HTTPResponse {
|
|||||||
// query has highest prio, then form, then json
|
// query has highest prio, then form, then json
|
||||||
data := dataext.ObjectMerge(dataext.ObjectMerge(b, f), q)
|
data := dataext.ObjectMerge(dataext.ObjectMerge(b, f), q)
|
||||||
|
|
||||||
okResp, errResp := h.sendMessageInternal(g, ctx, data.UserID, data.KeyToken, data.Channel, data.Title, data.Content, data.Priority, data.UserMessageID, data.SendTimestamp, data.SenderName)
|
okResp, errResp := h.app.SendMessage(g, ctx, data.UserID, data.KeyToken, data.Channel, data.Title, data.Content, data.Priority, data.UserMessageID, data.SendTimestamp, data.SenderName)
|
||||||
if errResp != nil {
|
if errResp != nil {
|
||||||
return *errResp
|
return *errResp
|
||||||
} else {
|
} else {
|
||||||
@ -114,284 +105,3 @@ func (h MessageHandler) SendMessage(g *gin.Context) ginresp.HTTPResponse {
|
|||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h MessageHandler) sendMessageInternal(g *gin.Context, ctx *logic.AppContext, UserID *models.UserID, Key *string, Channel *string, Title *string, Content *string, Priority *int, UserMessageID *string, SendTimestamp *float64, SenderName *string) (*SendMessageResponse, *ginresp.HTTPResponse) {
|
|
||||||
if Title != nil {
|
|
||||||
Title = langext.Ptr(strings.TrimSpace(*Title))
|
|
||||||
}
|
|
||||||
if UserMessageID != nil {
|
|
||||||
UserMessageID = langext.Ptr(strings.TrimSpace(*UserMessageID))
|
|
||||||
}
|
|
||||||
|
|
||||||
if UserID == nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.MISSING_UID, hl.USER_ID, "Missing parameter [[user_id]]", nil))
|
|
||||||
}
|
|
||||||
if Key == nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.MISSING_TOK, hl.USER_KEY, "Missing parameter [[key]]", nil))
|
|
||||||
}
|
|
||||||
if Title == nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.MISSING_TITLE, hl.TITLE, "Missing parameter [[title]]", nil))
|
|
||||||
}
|
|
||||||
if Priority != nil && (*Priority != 0 && *Priority != 1 && *Priority != 2) {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.INVALID_PRIO, hl.PRIORITY, "Invalid priority", nil))
|
|
||||||
}
|
|
||||||
if len(*Title) == 0 {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.NO_TITLE, hl.TITLE, "No title specified", nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
user, err := h.database.GetUser(ctx, *UserID)
|
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.USER_NOT_FOUND, hl.USER_ID, "User not found", err))
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query user", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
channelDisplayName := user.DefaultChannel()
|
|
||||||
channelInternalName := user.DefaultChannel()
|
|
||||||
if Channel != nil {
|
|
||||||
channelDisplayName = h.app.NormalizeChannelDisplayName(*Channel)
|
|
||||||
channelInternalName = h.app.NormalizeChannelInternalName(*Channel)
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(*Title) > user.MaxTitleLength() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.TITLE_TOO_LONG, hl.TITLE, fmt.Sprintf("Title too long (max %d characters)", user.MaxTitleLength()), nil))
|
|
||||||
}
|
|
||||||
if Content != nil && len(*Content) > user.MaxContentLength() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CONTENT_TOO_LONG, hl.CONTENT, fmt.Sprintf("Content too long (%d characters; max := %d characters)", len(*Content), user.MaxContentLength()), nil))
|
|
||||||
}
|
|
||||||
if len(channelDisplayName) > user.MaxChannelNameLength() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_TOO_LONG, hl.CHANNEL, fmt.Sprintf("Channel too long (max %d characters)", user.MaxChannelNameLength()), nil))
|
|
||||||
}
|
|
||||||
if len(strings.TrimSpace(channelDisplayName)) == 0 {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_NAME_EMPTY, hl.CHANNEL, fmt.Sprintf("Channel displayname cannot be empty"), nil))
|
|
||||||
}
|
|
||||||
if len(channelInternalName) > user.MaxChannelNameLength() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_TOO_LONG, hl.CHANNEL, fmt.Sprintf("Channel too long (max %d characters)", user.MaxChannelNameLength()), nil))
|
|
||||||
}
|
|
||||||
if len(strings.TrimSpace(channelInternalName)) == 0 {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_NAME_EMPTY, hl.CHANNEL, fmt.Sprintf("Channel internalname cannot be empty"), nil))
|
|
||||||
}
|
|
||||||
if SenderName != nil && len(*SenderName) > user.MaxSenderNameLength() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.SENDERNAME_TOO_LONG, hl.SENDER_NAME, fmt.Sprintf("SenderName too long (max %d characters)", user.MaxSenderNameLength()), nil))
|
|
||||||
}
|
|
||||||
if UserMessageID != nil && len(*UserMessageID) > user.MaxUserMessageIDLength() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.USR_MSG_ID_TOO_LONG, hl.USER_MESSAGE_ID, fmt.Sprintf("MessageID too long (max %d characters)", user.MaxUserMessageIDLength()), nil))
|
|
||||||
}
|
|
||||||
if SendTimestamp != nil && mathext.Abs(*SendTimestamp-float64(time.Now().Unix())) > timeext.FromHours(user.MaxTimestampDiffHours()).Seconds() {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.TIMESTAMP_OUT_OF_RANGE, hl.NONE, fmt.Sprintf("The timestamp mus be within %d hours of now()", user.MaxTimestampDiffHours()), nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
if UserMessageID != nil {
|
|
||||||
msg, err := h.database.GetMessageByUserMessageID(ctx, *UserMessageID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query existing message", err))
|
|
||||||
}
|
|
||||||
if msg != nil {
|
|
||||||
|
|
||||||
existingCompID, _, err := h.database.ConvertToCompatID(ctx, msg.MessageID.String())
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query compat-id", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
if existingCompID == nil {
|
|
||||||
v, err := h.database.CreateCompatID(ctx, "messageid", msg.MessageID.String())
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create compat-id", err))
|
|
||||||
}
|
|
||||||
existingCompID = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
//the found message can be deleted (!), but we still return NO_ERROR here...
|
|
||||||
return &SendMessageResponse{
|
|
||||||
User: user,
|
|
||||||
Message: *msg,
|
|
||||||
MessageIsOld: true,
|
|
||||||
CompatMessageID: *existingCompID,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if user.QuotaRemainingToday() <= 0 {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 403, apierr.QUOTA_REACHED, hl.NONE, fmt.Sprintf("Daily quota reached (%d)", user.QuotaPerDay()), nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
channel, err := h.app.GetOrCreateChannel(ctx, *UserID, channelDisplayName, channelInternalName)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query/create (owned) channel", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
keytok, permResp := ctx.CheckPermissionSend(channel, *Key)
|
|
||||||
if permResp != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 401, apierr.USER_AUTH_FAILED, hl.USER_KEY, "You are not authorized for this action", nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
var sendTimestamp *time.Time = nil
|
|
||||||
if SendTimestamp != nil {
|
|
||||||
sendTimestamp = langext.Ptr(timeext.UnixFloatSeconds(*SendTimestamp))
|
|
||||||
}
|
|
||||||
|
|
||||||
priority := langext.Coalesce(Priority, user.DefaultPriority())
|
|
||||||
|
|
||||||
clientIP := g.ClientIP()
|
|
||||||
|
|
||||||
msg, err := h.database.CreateMessage(ctx, *UserID, channel, sendTimestamp, *Title, Content, priority, UserMessageID, clientIP, SenderName, keytok.KeyTokenID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create message in db", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
compatMsgID, err := h.database.CreateCompatID(ctx, "messageid", msg.MessageID.String())
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create compat-id", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
subFilter := models.SubscriptionFilter{ChannelID: langext.Ptr([]models.ChannelID{channel.ChannelID}), Confirmed: langext.PTrue}
|
|
||||||
activeSubscriptions, err := h.database.ListSubscriptions(ctx, subFilter)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query subscriptions", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
err = h.database.IncUserMessageCounter(ctx, &user)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to inc user msg-counter", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
err = h.database.IncChannelMessageCounter(ctx, &channel)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to inc channel msg-counter", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
err = h.database.IncKeyTokenMessageCounter(ctx, keytok)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to inc token msg-counter", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Info().Msg(fmt.Sprintf("Sending new notification %s for user %s", msg.MessageID, UserID))
|
|
||||||
|
|
||||||
for _, sub := range activeSubscriptions {
|
|
||||||
clients, err := h.database.ListClients(ctx, sub.SubscriberUserID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query clients", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, client := range clients {
|
|
||||||
|
|
||||||
isCompatClient, err := h.database.IsCompatClient(ctx, client.ClientID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query compat_clients", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
var titleOverride *string = nil
|
|
||||||
var msgidOverride *string = nil
|
|
||||||
if isCompatClient {
|
|
||||||
titleOverride = langext.Ptr(h.app.CompatizeMessageTitle(ctx, msg))
|
|
||||||
msgidOverride = langext.Ptr(fmt.Sprintf("%d", compatMsgID))
|
|
||||||
}
|
|
||||||
|
|
||||||
fcmDelivID, err := h.app.DeliverMessage(ctx, client, msg, titleOverride, msgidOverride)
|
|
||||||
if err != nil {
|
|
||||||
_, err = h.database.CreateRetryDelivery(ctx, client, msg)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create delivery", err))
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
_, err = h.database.CreateSuccessDelivery(ctx, client, msg, fcmDelivID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create delivery", err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return &SendMessageResponse{
|
|
||||||
User: user,
|
|
||||||
Message: msg,
|
|
||||||
MessageIsOld: false,
|
|
||||||
CompatMessageID: compatMsgID,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// UptimeKumaWebHook swaggerdoc
|
|
||||||
//
|
|
||||||
// @Summary Send a new message
|
|
||||||
// @Description All parameter can be set via query-parameter or the json body. Only UserID, UserKey and Title are required
|
|
||||||
// @Tags External
|
|
||||||
//
|
|
||||||
// @Param query_data query handler.UptimeKumaWebHook.query false " "
|
|
||||||
// @Param post_body body handler.UptimeKumaWebHook.uptimeKumaWebhookBody false " "
|
|
||||||
//
|
|
||||||
// @Success 200 {object} any
|
|
||||||
// @Failure 400 {object} ginresp.apiError
|
|
||||||
// @Failure 401 {object} ginresp.apiError "The user_id was not found or the user_key is wrong"
|
|
||||||
// @Failure 403 {object} ginresp.apiError "The user has exceeded its daily quota - wait 24 hours or upgrade your account"
|
|
||||||
// @Failure 500 {object} ginresp.apiError "An internal server error occurred - try again later"
|
|
||||||
//
|
|
||||||
// @Router /webhook/uptime-kuma [POST]
|
|
||||||
func (h MessageHandler) UptimeKumaWebHook(g *gin.Context) ginresp.HTTPResponse {
|
|
||||||
type query struct {
|
|
||||||
UserID *models.UserID `form:"user_id" example:"7725"`
|
|
||||||
KeyToken *string `form:"key" example:"P3TNH8mvv14fm"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type uptimeKumaWebhookBody struct {
|
|
||||||
Heartbeat *struct {
|
|
||||||
Time string `json:"time"`
|
|
||||||
Status int `json:"status"`
|
|
||||||
Msg string `json:"msg"`
|
|
||||||
Timezone string `json:"timezone"`
|
|
||||||
TimezoneOffset string `json:"timezoneOffset"`
|
|
||||||
LocalDateTime string `json:"localDateTime"`
|
|
||||||
} `json:"heartbeat"`
|
|
||||||
Monitor *struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Url *string `json:"url"`
|
|
||||||
} `json:"monitor"`
|
|
||||||
Msg string `json:"msg"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var b uptimeKumaWebhookBody
|
|
||||||
var q query
|
|
||||||
|
|
||||||
ctx, httpErr := h.app.StartRequest(g, nil, &q, &b, nil)
|
|
||||||
if httpErr != nil {
|
|
||||||
return *httpErr
|
|
||||||
}
|
|
||||||
defer ctx.Cancel()
|
|
||||||
|
|
||||||
var title = ""
|
|
||||||
|
|
||||||
var content = ""
|
|
||||||
content += fmt.Sprintf("%v\n", b.Msg)
|
|
||||||
if b.Monitor != nil {
|
|
||||||
content += fmt.Sprintf("%v\n", b.Monitor.Name)
|
|
||||||
if b.Monitor.Url != nil {
|
|
||||||
content += fmt.Sprintf("url: %v\n", *b.Monitor.Url)
|
|
||||||
}
|
|
||||||
|
|
||||||
if b.Heartbeat != nil {
|
|
||||||
statusString := "down"
|
|
||||||
|
|
||||||
if b.Heartbeat.Status == 1 {
|
|
||||||
statusString = "up"
|
|
||||||
}
|
|
||||||
title = fmt.Sprintf("%v %v!", b.Monitor.Name, statusString)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
if b.Heartbeat != nil {
|
|
||||||
content += "\n===== Heartbeat ======\n"
|
|
||||||
content += fmt.Sprintf("msg: %v\n", b.Heartbeat.Msg)
|
|
||||||
content += fmt.Sprintf("timestamp: %v\n", b.Heartbeat.Time)
|
|
||||||
content += fmt.Sprintf("timezone: %v\n", b.Heartbeat.Timezone)
|
|
||||||
content += fmt.Sprintf("timezone offset: %v\n", b.Heartbeat.TimezoneOffset)
|
|
||||||
content += fmt.Sprintf("local date time: %v\n", b.Heartbeat.TimezoneOffset)
|
|
||||||
}
|
|
||||||
okResp, errResp := h.sendMessageInternal(g, ctx, q.UserID, q.KeyToken, nil, &title, &content, langext.Ptr(1), nil, nil, nil)
|
|
||||||
|
|
||||||
if errResp != nil {
|
|
||||||
return *errResp
|
|
||||||
}
|
|
||||||
return ctx.FinishSuccess(ginresp.JSON(http.StatusOK, okResp))
|
|
||||||
}
|
|
||||||
|
@ -21,6 +21,7 @@ type Router struct {
|
|||||||
websiteHandler handler.WebsiteHandler
|
websiteHandler handler.WebsiteHandler
|
||||||
apiHandler handler.APIHandler
|
apiHandler handler.APIHandler
|
||||||
messageHandler handler.MessageHandler
|
messageHandler handler.MessageHandler
|
||||||
|
externalHandler handler.ExternalHandler
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewRouter(app *logic.Application) *Router {
|
func NewRouter(app *logic.Application) *Router {
|
||||||
@ -32,6 +33,7 @@ func NewRouter(app *logic.Application) *Router {
|
|||||||
websiteHandler: handler.NewWebsiteHandler(app),
|
websiteHandler: handler.NewWebsiteHandler(app),
|
||||||
apiHandler: handler.NewAPIHandler(app),
|
apiHandler: handler.NewAPIHandler(app),
|
||||||
messageHandler: handler.NewMessageHandler(app),
|
messageHandler: handler.NewMessageHandler(app),
|
||||||
|
externalHandler: handler.NewExternalHandler(app),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -162,9 +164,9 @@ func (r *Router) Init(e *gin.Engine) error {
|
|||||||
{
|
{
|
||||||
sendAPI.POST("/", r.Wrap(r.messageHandler.SendMessage))
|
sendAPI.POST("/", r.Wrap(r.messageHandler.SendMessage))
|
||||||
sendAPI.POST("/send", r.Wrap(r.messageHandler.SendMessage))
|
sendAPI.POST("/send", r.Wrap(r.messageHandler.SendMessage))
|
||||||
sendAPI.POST("/send.php", r.Wrap(r.messageHandler.SendMessageCompat))
|
sendAPI.POST("/send.php", r.Wrap(r.compatHandler.SendMessage))
|
||||||
|
|
||||||
sendAPI.POST("/webhook/uptime-kuma", r.Wrap(r.messageHandler.UptimeKumaWebHook))
|
sendAPI.POST("/external/v1/uptime-kuma", r.Wrap(r.externalHandler.UptimeKuma))
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
222
scnserver/logic/message.go
Normal file
222
scnserver/logic/message.go
Normal file
@ -0,0 +1,222 @@
|
|||||||
|
package logic
|
||||||
|
|
||||||
|
import (
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/api/apierr"
|
||||||
|
hl "blackforestbytes.com/simplecloudnotifier/api/apihighlight"
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/api/ginresp"
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/models"
|
||||||
|
"database/sql"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/rs/zerolog/log"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/mathext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/timeext"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SendMessageResponse struct {
|
||||||
|
User models.User
|
||||||
|
Message models.Message
|
||||||
|
MessageIsOld bool
|
||||||
|
CompatMessageID int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func (app *Application) SendMessage(g *gin.Context, ctx *AppContext, UserID *models.UserID, Key *string, Channel *string, Title *string, Content *string, Priority *int, UserMessageID *string, SendTimestamp *float64, SenderName *string) (*SendMessageResponse, *ginresp.HTTPResponse) {
|
||||||
|
if Title != nil {
|
||||||
|
Title = langext.Ptr(strings.TrimSpace(*Title))
|
||||||
|
}
|
||||||
|
if UserMessageID != nil {
|
||||||
|
UserMessageID = langext.Ptr(strings.TrimSpace(*UserMessageID))
|
||||||
|
}
|
||||||
|
|
||||||
|
if UserID == nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.MISSING_UID, hl.USER_ID, "Missing parameter [[user_id]]", nil))
|
||||||
|
}
|
||||||
|
if Key == nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.MISSING_TOK, hl.USER_KEY, "Missing parameter [[key]]", nil))
|
||||||
|
}
|
||||||
|
if Title == nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.MISSING_TITLE, hl.TITLE, "Missing parameter [[title]]", nil))
|
||||||
|
}
|
||||||
|
if Priority != nil && (*Priority != 0 && *Priority != 1 && *Priority != 2) {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.INVALID_PRIO, hl.PRIORITY, "Invalid priority", nil))
|
||||||
|
}
|
||||||
|
if len(*Title) == 0 {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.NO_TITLE, hl.TITLE, "No title specified", nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
user, err := app.Database.Primary.GetUser(ctx, *UserID)
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.USER_NOT_FOUND, hl.USER_ID, "User not found", err))
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query user", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
channelDisplayName := user.DefaultChannel()
|
||||||
|
channelInternalName := user.DefaultChannel()
|
||||||
|
if Channel != nil {
|
||||||
|
channelDisplayName = app.NormalizeChannelDisplayName(*Channel)
|
||||||
|
channelInternalName = app.NormalizeChannelInternalName(*Channel)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(*Title) > user.MaxTitleLength() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.TITLE_TOO_LONG, hl.TITLE, fmt.Sprintf("Title too long (max %d characters)", user.MaxTitleLength()), nil))
|
||||||
|
}
|
||||||
|
if Content != nil && len(*Content) > user.MaxContentLength() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CONTENT_TOO_LONG, hl.CONTENT, fmt.Sprintf("Content too long (%d characters; max := %d characters)", len(*Content), user.MaxContentLength()), nil))
|
||||||
|
}
|
||||||
|
if len(channelDisplayName) > user.MaxChannelNameLength() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_TOO_LONG, hl.CHANNEL, fmt.Sprintf("Channel too long (max %d characters)", user.MaxChannelNameLength()), nil))
|
||||||
|
}
|
||||||
|
if len(strings.TrimSpace(channelDisplayName)) == 0 {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_NAME_EMPTY, hl.CHANNEL, fmt.Sprintf("Channel displayname cannot be empty"), nil))
|
||||||
|
}
|
||||||
|
if len(channelInternalName) > user.MaxChannelNameLength() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_TOO_LONG, hl.CHANNEL, fmt.Sprintf("Channel too long (max %d characters)", user.MaxChannelNameLength()), nil))
|
||||||
|
}
|
||||||
|
if len(strings.TrimSpace(channelInternalName)) == 0 {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.CHANNEL_NAME_EMPTY, hl.CHANNEL, fmt.Sprintf("Channel internalname cannot be empty"), nil))
|
||||||
|
}
|
||||||
|
if SenderName != nil && len(*SenderName) > user.MaxSenderNameLength() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.SENDERNAME_TOO_LONG, hl.SENDER_NAME, fmt.Sprintf("SenderName too long (max %d characters)", user.MaxSenderNameLength()), nil))
|
||||||
|
}
|
||||||
|
if UserMessageID != nil && len(*UserMessageID) > user.MaxUserMessageIDLength() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.USR_MSG_ID_TOO_LONG, hl.USER_MESSAGE_ID, fmt.Sprintf("MessageID too long (max %d characters)", user.MaxUserMessageIDLength()), nil))
|
||||||
|
}
|
||||||
|
if SendTimestamp != nil && mathext.Abs(*SendTimestamp-float64(time.Now().Unix())) > timeext.FromHours(user.MaxTimestampDiffHours()).Seconds() {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 400, apierr.TIMESTAMP_OUT_OF_RANGE, hl.NONE, fmt.Sprintf("The timestamp mus be within %d hours of now()", user.MaxTimestampDiffHours()), nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
if UserMessageID != nil {
|
||||||
|
msg, err := app.Database.Primary.GetMessageByUserMessageID(ctx, *UserMessageID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query existing message", err))
|
||||||
|
}
|
||||||
|
if msg != nil {
|
||||||
|
|
||||||
|
existingCompID, _, err := app.Database.Primary.ConvertToCompatID(ctx, msg.MessageID.String())
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query compat-id", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
if existingCompID == nil {
|
||||||
|
v, err := app.Database.Primary.CreateCompatID(ctx, "messageid", msg.MessageID.String())
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create compat-id", err))
|
||||||
|
}
|
||||||
|
existingCompID = &v
|
||||||
|
}
|
||||||
|
|
||||||
|
//the found message can be deleted (!), but we still return NO_ERROR here...
|
||||||
|
return &SendMessageResponse{
|
||||||
|
User: user,
|
||||||
|
Message: *msg,
|
||||||
|
MessageIsOld: true,
|
||||||
|
CompatMessageID: *existingCompID,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if user.QuotaRemainingToday() <= 0 {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 403, apierr.QUOTA_REACHED, hl.NONE, fmt.Sprintf("Daily quota reached (%d)", user.QuotaPerDay()), nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
channel, err := app.GetOrCreateChannel(ctx, *UserID, channelDisplayName, channelInternalName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query/create (owned) channel", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
keytok, permResp := ctx.CheckPermissionSend(channel, *Key)
|
||||||
|
if permResp != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 401, apierr.USER_AUTH_FAILED, hl.USER_KEY, "You are not authorized for this action", nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
var sendTimestamp *time.Time = nil
|
||||||
|
if SendTimestamp != nil {
|
||||||
|
sendTimestamp = langext.Ptr(timeext.UnixFloatSeconds(*SendTimestamp))
|
||||||
|
}
|
||||||
|
|
||||||
|
priority := langext.Coalesce(Priority, user.DefaultPriority())
|
||||||
|
|
||||||
|
clientIP := g.ClientIP()
|
||||||
|
|
||||||
|
msg, err := app.Database.Primary.CreateMessage(ctx, *UserID, channel, sendTimestamp, *Title, Content, priority, UserMessageID, clientIP, SenderName, keytok.KeyTokenID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create message in db", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
compatMsgID, err := app.Database.Primary.CreateCompatID(ctx, "messageid", msg.MessageID.String())
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create compat-id", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
subFilter := models.SubscriptionFilter{ChannelID: langext.Ptr([]models.ChannelID{channel.ChannelID}), Confirmed: langext.PTrue}
|
||||||
|
activeSubscriptions, err := app.Database.Primary.ListSubscriptions(ctx, subFilter)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query subscriptions", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
err = app.Database.Primary.IncUserMessageCounter(ctx, &user)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to inc user msg-counter", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
err = app.Database.Primary.IncChannelMessageCounter(ctx, &channel)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to inc channel msg-counter", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
err = app.Database.Primary.IncKeyTokenMessageCounter(ctx, keytok)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to inc token msg-counter", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Info().Msg(fmt.Sprintf("Sending new notification %s for user %s", msg.MessageID, UserID))
|
||||||
|
|
||||||
|
for _, sub := range activeSubscriptions {
|
||||||
|
clients, err := app.Database.Primary.ListClients(ctx, sub.SubscriberUserID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query clients", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, client := range clients {
|
||||||
|
|
||||||
|
isCompatClient, err := app.Database.Primary.IsCompatClient(ctx, client.ClientID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to query compat_clients", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
var titleOverride *string = nil
|
||||||
|
var msgidOverride *string = nil
|
||||||
|
if isCompatClient {
|
||||||
|
titleOverride = langext.Ptr(app.CompatizeMessageTitle(ctx, msg))
|
||||||
|
msgidOverride = langext.Ptr(fmt.Sprintf("%d", compatMsgID))
|
||||||
|
}
|
||||||
|
|
||||||
|
fcmDelivID, err := app.DeliverMessage(ctx, client, msg, titleOverride, msgidOverride)
|
||||||
|
if err != nil {
|
||||||
|
_, err = app.Database.Primary.CreateRetryDelivery(ctx, client, msg)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create delivery", err))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
_, err = app.Database.Primary.CreateSuccessDelivery(ctx, client, msg, fcmDelivID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, langext.Ptr(ginresp.SendAPIError(g, 500, apierr.DATABASE_ERROR, hl.NONE, "Failed to create delivery", err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return &SendMessageResponse{
|
||||||
|
User: user,
|
||||||
|
Message: msg,
|
||||||
|
MessageIsOld: false,
|
||||||
|
CompatMessageID: compatMsgID,
|
||||||
|
}, nil
|
||||||
|
}
|
@ -1780,43 +1780,3 @@ func TestSendWithPermissionSendKey(t *testing.T) {
|
|||||||
func TestSendDeliveryRetry(t *testing.T) {
|
func TestSendDeliveryRetry(t *testing.T) {
|
||||||
t.SkipNow() //TODO
|
t.SkipNow() //TODO
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestUptimeKuma(t *testing.T) {
|
|
||||||
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
|
||||||
defer stop()
|
|
||||||
|
|
||||||
pusher := ws.Pusher.(*push.TestSink)
|
|
||||||
|
|
||||||
r0 := tt.RequestPost[gin.H](t, baseUrl, "/api/v2/users", gin.H{
|
|
||||||
"agent_model": "DUMMY_PHONE",
|
|
||||||
"agent_version": "4X",
|
|
||||||
"client_type": "ANDROID",
|
|
||||||
"fcm_token": "DUMMY_FCM",
|
|
||||||
})
|
|
||||||
|
|
||||||
uid := r0["user_id"].(string)
|
|
||||||
admintok := r0["admin_key"].(string)
|
|
||||||
sendtok := r0["send_key"].(string)
|
|
||||||
|
|
||||||
suffix := fmt.Sprintf("/webhook/uptime-kuma?user_id=%v&key=%v", uid, sendtok)
|
|
||||||
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
|
||||||
"msg": "Uptime Kuma failed with 503!",
|
|
||||||
"heartbeat": gin.H{
|
|
||||||
"status": 0,
|
|
||||||
},
|
|
||||||
"monitor": gin.H{
|
|
||||||
"name": "Test-Kuma",
|
|
||||||
},
|
|
||||||
})
|
|
||||||
|
|
||||||
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
|
||||||
tt.AssertStrRepEqual(t, "msg.title", "Test-Kuma down!", pusher.Last().Message.Title)
|
|
||||||
|
|
||||||
type mglist struct {
|
|
||||||
Messages []gin.H `json:"messages"`
|
|
||||||
}
|
|
||||||
|
|
||||||
msgList1 := tt.RequestAuthGet[mglist](t, admintok, baseUrl, "/api/v2/messages")
|
|
||||||
tt.AssertEqual(t, "len(messages)", 1, len(msgList1.Messages))
|
|
||||||
tt.AssertStrRepEqual(t, "msg.title", "Test-Kuma down!", msgList1.Messages[0]["title"])
|
|
||||||
}
|
|
||||||
|
447
scnserver/test/uptimekuma_test.go
Normal file
447
scnserver/test/uptimekuma_test.go
Normal file
@ -0,0 +1,447 @@
|
|||||||
|
package test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"blackforestbytes.com/simplecloudnotifier/push"
|
||||||
|
tt "blackforestbytes.com/simplecloudnotifier/test/util"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestUptimeKumaDown(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test went down!", pusher.Last().Message.Title)
|
||||||
|
|
||||||
|
type mglist struct {
|
||||||
|
Messages []gin.H `json:"messages"`
|
||||||
|
}
|
||||||
|
|
||||||
|
msgList1 := tt.RequestAuthGet[mglist](t, data.AdminKey, baseUrl, "/api/v2/messages")
|
||||||
|
tt.AssertEqual(t, "len(messages)", 1, len(msgList1.Messages))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test went down!", msgList1.Messages[0]["title"])
|
||||||
|
tt.AssertStrRepEqual(t, "msg.content", "getaddrinfo ENOTFOUND exampleasdsda.com", msgList1.Messages[0]["content"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaUp(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [✅ Up] 200 - OK",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 1,
|
||||||
|
"msg": "200 - OK",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test is back online", pusher.Last().Message.Title)
|
||||||
|
|
||||||
|
type mglist struct {
|
||||||
|
Messages []gin.H `json:"messages"`
|
||||||
|
}
|
||||||
|
|
||||||
|
msgList1 := tt.RequestAuthGet[mglist](t, data.AdminKey, baseUrl, "/api/v2/messages")
|
||||||
|
tt.AssertEqual(t, "len(messages)", 1, len(msgList1.Messages))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test is back online", msgList1.Messages[0]["title"])
|
||||||
|
tt.AssertStrRepEqual(t, "msg.content", "200 - OK", msgList1.Messages[0]["content"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaFullDown(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
ts := time.Now().Add(-time.Hour).Format("2006-01-02 15:04:05")
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, tt.RawJSON{ContentType: "application/json", Body: `{"heartbeat":{"monitorID":89,"status":0,"time":"` + ts + `","msg":"timeout of 16000ms exceeded","important":true,"duration":36,"timezone":"Europe/Berlin","timezoneOffset":"+02:00","localDateTime":"` + ts + `"},"monitor":{"id":89,"name":"test","description":null,"pathName":"test","parent":null,"childrenIDs":[],"url":"https://exampleXYZ.com","method":"GET","hostname":null,"port":null,"maxretries":1,"weight":2000,"active":true,"forceInactive":false,"type":"http","interval":20,"retryInterval":20,"resendInterval":0,"keyword":null,"expiryNotification":false,"ignoreTls":false,"upsideDown":false,"packetSize":56,"maxredirects":10,"accepted_statuscodes":["200-299"],"dns_resolve_type":"A","dns_resolve_server":"1.1.1.1","dns_last_result":null,"docker_container":"","docker_host":null,"proxyId":null,"notificationIDList":{"2":true},"tags":[],"maintenance":false,"mqttTopic":"","mqttSuccessMessage":"","databaseQuery":null,"authMethod":null,"grpcUrl":null,"grpcProtobuf":null,"grpcMethod":null,"grpcServiceName":null,"grpcEnableTls":false,"radiusCalledStationId":null,"radiusCallingStationId":null,"game":null,"httpBodyEncoding":"json","includeSensitiveData":false},"msg":"[test] [🔴 Down] timeout of 16000ms exceeded"}`})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test went down!", pusher.Last().Message.Title)
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "timeout of 16000ms exceeded", pusher.Last().Message.Content)
|
||||||
|
|
||||||
|
type mglist struct {
|
||||||
|
Messages []gin.H `json:"messages"`
|
||||||
|
}
|
||||||
|
|
||||||
|
msgList1 := tt.RequestAuthGet[mglist](t, data.AdminKey, baseUrl, "/api/v2/messages")
|
||||||
|
tt.AssertEqual(t, "len(messages)", 1, len(msgList1.Messages))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test went down!", msgList1.Messages[0]["title"])
|
||||||
|
tt.AssertStrRepEqual(t, "msg.content", "timeout of 16000ms exceeded", msgList1.Messages[0]["content"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaFullUp(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
ts := time.Now().Add(-time.Hour).Format("2006-01-02 15:04:05")
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, tt.RawJSON{ContentType: "application/json", Body: `{"heartbeat":{"monitorID":89,"status":1,"time":"` + ts + `","msg":"200 - OK","ping":55,"important":true,"duration":41,"timezone":"Europe/Berlin","timezoneOffset":"+02:00","localDateTime":"` + ts + `"},"monitor":{"id":89,"name":"test","description":null,"pathName":"test","parent":null,"childrenIDs":[],"url":"https://example.com","method":"GET","hostname":null,"port":null,"maxretries":1,"weight":2000,"active":true,"forceInactive":false,"type":"http","interval":20,"retryInterval":20,"resendInterval":0,"keyword":null,"expiryNotification":false,"ignoreTls":false,"upsideDown":false,"packetSize":56,"maxredirects":10,"accepted_statuscodes":["200-299"],"dns_resolve_type":"A","dns_resolve_server":"1.1.1.1","dns_last_result":null,"docker_container":"","docker_host":null,"proxyId":null,"notificationIDList":{"2":true},"tags":[],"maintenance":false,"mqttTopic":"","mqttSuccessMessage":"","databaseQuery":null,"authMethod":null,"grpcUrl":null,"grpcProtobuf":null,"grpcMethod":null,"grpcServiceName":null,"grpcEnableTls":false,"radiusCalledStationId":null,"radiusCallingStationId":null,"game":null,"httpBodyEncoding":"json","includeSensitiveData":false},"msg":"[test] [✅ Up] 200 - OK"}`})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test is back online", pusher.Last().Message.Title)
|
||||||
|
|
||||||
|
type mglist struct {
|
||||||
|
Messages []gin.H `json:"messages"`
|
||||||
|
}
|
||||||
|
|
||||||
|
msgList1 := tt.RequestAuthGet[mglist](t, data.AdminKey, baseUrl, "/api/v2/messages")
|
||||||
|
tt.AssertEqual(t, "len(messages)", 1, len(msgList1.Messages))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.title", "Monitor test is back online", msgList1.Messages[0]["title"])
|
||||||
|
tt.AssertStrRepEqual(t, "msg.content", "200 - OK", msgList1.Messages[0]["content"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaChannelNone(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", "main", pusher.Last().Message.ChannelInternalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaChannelSingle(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&channel=CTEST", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", "CTEST", pusher.Last().Message.ChannelInternalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaChannelAllDown(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&channel=CTEST&channel_up=CTEST_UP&channel_down=CTEST_DOWN", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", "CTEST_DOWN", pusher.Last().Message.ChannelInternalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaChannelSpecDown(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&channel_up=CTEST_UP&channel_down=CTEST_DOWN", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", "CTEST_DOWN", pusher.Last().Message.ChannelInternalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaChannelAllUp(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&channel=CTEST&channel_up=CTEST_UP&channel_down=CTEST_DOWN", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [✅ Up] 200 - OK",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 1,
|
||||||
|
"msg": "200 - OK",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", "CTEST_UP", pusher.Last().Message.ChannelInternalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaChannelSpecUp(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&channel_up=CTEST_UP&channel_down=CTEST_DOWN", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [✅ Up] 200 - OK",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 1,
|
||||||
|
"msg": "200 - OK",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", "CTEST_UP", pusher.Last().Message.ChannelInternalName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaPriorityNone(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.channel", 1, pusher.Last().Message.Priority)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaPrioritySingle(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix0 := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority=0", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix0, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 0, pusher.Last().Message.Priority)
|
||||||
|
|
||||||
|
suffix1 := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority=1", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix1, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 2, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 1, pusher.Last().Message.Priority)
|
||||||
|
|
||||||
|
suffix2 := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority=2", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix2, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 3, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 2, pusher.Last().Message.Priority)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaPriorityAllDown(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority=1&priority_up=2&priority_down=0", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 0, pusher.Last().Message.Priority)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaPrioritySpecDown(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority_up=2&priority_down=0", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [🔴 Down] getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 0,
|
||||||
|
"msg": "getaddrinfo ENOTFOUND exampleasdsda.com",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 0, pusher.Last().Message.Priority)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaPriorityAllUp(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority=1&priority_up=2&priority_down=0", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [✅ Up] 200 - OK",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 1,
|
||||||
|
"msg": "200 - OK",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 2, pusher.Last().Message.Priority)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUptimeKumaPrioritySpecUp(t *testing.T) {
|
||||||
|
ws, baseUrl, stop := tt.StartSimpleWebserver(t)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
data := tt.InitSingleData(t, ws)
|
||||||
|
|
||||||
|
pusher := ws.Pusher.(*push.TestSink)
|
||||||
|
|
||||||
|
suffix := fmt.Sprintf("/external/v1/uptime-kuma?user_id=%v&key=%v&priority_up=2&priority_down=0", data.UID, data.SendKey)
|
||||||
|
_ = tt.RequestPost[gin.H](t, baseUrl, suffix, gin.H{
|
||||||
|
"msg": "[test] [✅ Up] 200 - OK",
|
||||||
|
"heartbeat": gin.H{
|
||||||
|
"status": 1,
|
||||||
|
"msg": "200 - OK",
|
||||||
|
},
|
||||||
|
"monitor": gin.H{
|
||||||
|
"name": "test",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
tt.AssertEqual(t, "messageCount", 1, len(pusher.Data))
|
||||||
|
tt.AssertStrRepEqual(t, "msg.prio", 2, pusher.Last().Message.Priority)
|
||||||
|
}
|
@ -1,3 +1,8 @@
|
|||||||
package util
|
package util
|
||||||
|
|
||||||
type FormData map[string]string
|
type FormData map[string]string
|
||||||
|
|
||||||
|
type RawJSON struct {
|
||||||
|
ContentType string
|
||||||
|
Body string
|
||||||
|
}
|
||||||
|
@ -114,6 +114,9 @@ func RequestAny[TResult any](t *testing.T, akey string, method string, baseURL s
|
|||||||
}
|
}
|
||||||
bytesbody = bodybuffer.Bytes()
|
bytesbody = bodybuffer.Bytes()
|
||||||
contentType = writer.FormDataContentType()
|
contentType = writer.FormDataContentType()
|
||||||
|
case RawJSON:
|
||||||
|
bytesbody = []byte(body.(RawJSON).Body)
|
||||||
|
contentType = "application/json"
|
||||||
default:
|
default:
|
||||||
bjson, err := json.Marshal(body)
|
bjson, err := json.Marshal(body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
Loading…
Reference in New Issue
Block a user