114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package gorush
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gin-gonic/gin/binding"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"golang.org/x/crypto/acme/autocert"
|
|
api "gopkg.in/appleboy/gin-status-api.v1"
|
|
)
|
|
|
|
func init() {
|
|
// Support metrics
|
|
m := NewMetrics()
|
|
prometheus.MustRegister(m)
|
|
}
|
|
|
|
func abortWithError(c *gin.Context, code int, message string) {
|
|
c.AbortWithStatusJSON(code, gin.H{
|
|
"code": code,
|
|
"message": message,
|
|
})
|
|
}
|
|
|
|
func rootHandler(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"text": "Welcome to notification server.",
|
|
})
|
|
}
|
|
|
|
func pushHandler(c *gin.Context) {
|
|
var form RequestPush
|
|
var msg string
|
|
|
|
if err := c.ShouldBindWith(&form, binding.JSON); err != nil {
|
|
msg = "Missing notifications field."
|
|
LogAccess.Debug(msg)
|
|
abortWithError(c, http.StatusBadRequest, msg)
|
|
return
|
|
}
|
|
|
|
if len(form.Notifications) == 0 {
|
|
msg = "Notifications field is empty."
|
|
LogAccess.Debug(msg)
|
|
abortWithError(c, http.StatusBadRequest, msg)
|
|
return
|
|
}
|
|
|
|
if int64(len(form.Notifications)) > PushConf.Core.MaxNotification {
|
|
msg = fmt.Sprintf("Number of notifications(%d) over limit(%d)", len(form.Notifications), PushConf.Core.MaxNotification)
|
|
LogAccess.Debug(msg)
|
|
abortWithError(c, http.StatusBadRequest, msg)
|
|
return
|
|
}
|
|
|
|
counts, logs := queueNotification(form)
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": "ok",
|
|
"counts": counts,
|
|
"logs": logs,
|
|
})
|
|
}
|
|
|
|
func configHandler(c *gin.Context) {
|
|
c.YAML(http.StatusCreated, PushConf)
|
|
}
|
|
|
|
func metricsHandler(c *gin.Context) {
|
|
promhttp.Handler().ServeHTTP(c.Writer, c.Request)
|
|
}
|
|
|
|
func autoTLSServer() *http.Server {
|
|
m := autocert.Manager{
|
|
Prompt: autocert.AcceptTOS,
|
|
HostPolicy: autocert.HostWhitelist(PushConf.Core.AutoTLS.Host),
|
|
Cache: autocert.DirCache(PushConf.Core.AutoTLS.Folder),
|
|
}
|
|
|
|
return &http.Server{
|
|
Addr: ":https",
|
|
TLSConfig: &tls.Config{GetCertificate: m.GetCertificate},
|
|
Handler: routerEngine(),
|
|
}
|
|
}
|
|
|
|
func routerEngine() *gin.Engine {
|
|
// set server mode
|
|
gin.SetMode(PushConf.Core.Mode)
|
|
|
|
r := gin.New()
|
|
|
|
// Global middleware
|
|
r.Use(gin.Logger())
|
|
r.Use(gin.Recovery())
|
|
r.Use(VersionMiddleware())
|
|
r.Use(LogMiddleware())
|
|
r.Use(StatMiddleware())
|
|
|
|
r.GET(PushConf.API.StatGoURI, api.StatusHandler)
|
|
r.GET(PushConf.API.StatAppURI, appStatusHandler)
|
|
r.GET(PushConf.API.ConfigURI, configHandler)
|
|
r.GET(PushConf.API.SysStatURI, sysStatsHandler)
|
|
r.POST(PushConf.API.PushURI, pushHandler)
|
|
r.GET(PushConf.API.MetricURI, metricsHandler)
|
|
r.GET("/", rootHandler)
|
|
|
|
return r
|
|
}
|