mirror of
https://github.com/h44z/wg-portal.git
synced 2025-04-19 08:55:12 +00:00
456 lines
14 KiB
Go
456 lines
14 KiB
Go
package handlers
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/h44z/wg-portal/internal/app"
|
|
"github.com/h44z/wg-portal/internal/app/api/v0/model"
|
|
"github.com/h44z/wg-portal/internal/domain"
|
|
)
|
|
|
|
type peerEndpoint struct {
|
|
app *app.App
|
|
authenticator *authenticationHandler
|
|
}
|
|
|
|
func (e peerEndpoint) GetName() string {
|
|
return "PeerEndpoint"
|
|
}
|
|
|
|
func (e peerEndpoint) RegisterRoutes(g *gin.RouterGroup, authenticator *authenticationHandler) {
|
|
apiGroup := g.Group("/peer", e.authenticator.LoggedIn())
|
|
|
|
apiGroup.GET("/iface/:iface/all", e.authenticator.LoggedIn(ScopeAdmin), e.handleAllGet())
|
|
apiGroup.GET("/iface/:iface/stats", e.authenticator.LoggedIn(ScopeAdmin), e.handleStatsGet())
|
|
apiGroup.GET("/iface/:iface/prepare", e.authenticator.LoggedIn(), e.handlePrepareGet())
|
|
apiGroup.POST("/iface/:iface/new", e.authenticator.LoggedIn(), e.handleCreatePost())
|
|
apiGroup.POST("/iface/:iface/multiplenew", e.authenticator.LoggedIn(ScopeAdmin), e.handleCreateMultiplePost())
|
|
apiGroup.GET("/config-qr/:id", e.handleQrCodeGet())
|
|
apiGroup.POST("/config-mail", e.handleEmailPost())
|
|
apiGroup.GET("/config/:id", e.handleConfigGet())
|
|
apiGroup.GET("/:id", e.handleSingleGet())
|
|
apiGroup.PUT("/:id", e.handleUpdatePut())
|
|
apiGroup.DELETE("/:id", e.handleDelete())
|
|
}
|
|
|
|
// handleAllGet returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleAllGet
|
|
// @Tags Peer
|
|
// @Summary Get peers for the given interface.
|
|
// @Produce json
|
|
// @Param iface path string true "The interface identifier"
|
|
// @Success 200 {object} []model.Peer
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/iface/{iface}/all [get]
|
|
func (e peerEndpoint) handleAllGet() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
interfaceId := Base64UrlDecode(c.Param("iface"))
|
|
if interfaceId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing iface parameter"})
|
|
return
|
|
}
|
|
|
|
_, peers, err := e.app.GetInterfaceAndPeers(ctx, domain.InterfaceIdentifier(interfaceId))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeers(peers))
|
|
}
|
|
}
|
|
|
|
// handleSingleGet returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleSingleGet
|
|
// @Tags Peer
|
|
// @Summary Get peer for the given identifier.
|
|
// @Produce json
|
|
// @Param id path string true "The peer identifier"
|
|
// @Success 200 {object} model.Peer
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/{id} [get]
|
|
func (e peerEndpoint) handleSingleGet() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
peerId := Base64UrlDecode(c.Param("id"))
|
|
if peerId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing id parameter"})
|
|
return
|
|
}
|
|
|
|
peer, err := e.app.GetPeer(ctx, domain.PeerIdentifier(peerId))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeer(peer))
|
|
}
|
|
}
|
|
|
|
// handlePrepareGet returns a gorm handler function.
|
|
//
|
|
// @ID peers_handlePrepareGet
|
|
// @Tags Peer
|
|
// @Summary Prepare a new peer for the given interface.
|
|
// @Produce json
|
|
// @Param iface path string true "The interface identifier"
|
|
// @Success 200 {object} model.Peer
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/iface/{iface}/prepare [get]
|
|
func (e peerEndpoint) handlePrepareGet() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
interfaceId := Base64UrlDecode(c.Param("iface"))
|
|
if interfaceId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing iface parameter"})
|
|
return
|
|
}
|
|
|
|
peer, err := e.app.PreparePeer(ctx, domain.InterfaceIdentifier(interfaceId))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeer(peer))
|
|
}
|
|
}
|
|
|
|
// handleCreatePost returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleCreatePost
|
|
// @Tags Peer
|
|
// @Summary Prepare a new peer for the given interface.
|
|
// @Produce json
|
|
// @Param iface path string true "The interface identifier"
|
|
// @Param request body model.Peer true "The peer data"
|
|
// @Success 200 {object} model.Peer
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/iface/{iface}/new [post]
|
|
func (e peerEndpoint) handleCreatePost() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
interfaceId := Base64UrlDecode(c.Param("iface"))
|
|
if interfaceId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing iface parameter"})
|
|
return
|
|
}
|
|
|
|
var p model.Peer
|
|
err := c.BindJSON(&p)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
if p.InterfaceIdentifier != interfaceId {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "interface id mismatch"})
|
|
return
|
|
}
|
|
|
|
newPeer, err := e.app.CreatePeer(ctx, model.NewDomainPeer(&p))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeer(newPeer))
|
|
}
|
|
}
|
|
|
|
// handleCreateMultiplePost returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleCreateMultiplePost
|
|
// @Tags Peer
|
|
// @Summary Create multiple new peers for the given interface.
|
|
// @Produce json
|
|
// @Param iface path string true "The interface identifier"
|
|
// @Param request body model.MultiPeerRequest true "The peer creation request data"
|
|
// @Success 200 {object} []model.Peer
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/iface/{iface}/multiplenew [post]
|
|
func (e peerEndpoint) handleCreateMultiplePost() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
interfaceId := Base64UrlDecode(c.Param("iface"))
|
|
if interfaceId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing iface parameter"})
|
|
return
|
|
}
|
|
|
|
var req model.MultiPeerRequest
|
|
err := c.BindJSON(&req)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
newPeers, err := e.app.CreateMultiplePeers(ctx, domain.InterfaceIdentifier(interfaceId),
|
|
model.NewDomainPeerCreationRequest(&req))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeers(newPeers))
|
|
}
|
|
}
|
|
|
|
// handleUpdatePut returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleUpdatePut
|
|
// @Tags Peer
|
|
// @Summary Update the given peer record.
|
|
// @Produce json
|
|
// @Param id path string true "The peer identifier"
|
|
// @Param request body model.Peer true "The peer data"
|
|
// @Success 200 {object} model.Peer
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/{id} [put]
|
|
func (e peerEndpoint) handleUpdatePut() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
peerId := Base64UrlDecode(c.Param("id"))
|
|
if peerId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing id parameter"})
|
|
return
|
|
}
|
|
|
|
var p model.Peer
|
|
err := c.BindJSON(&p)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
if p.Identifier != peerId {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "peer id mismatch"})
|
|
return
|
|
}
|
|
|
|
updatedPeer, err := e.app.UpdatePeer(ctx, model.NewDomainPeer(&p))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeer(updatedPeer))
|
|
}
|
|
}
|
|
|
|
// handleDelete returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleDelete
|
|
// @Tags Peer
|
|
// @Summary Delete the peer record.
|
|
// @Produce json
|
|
// @Param id path string true "The peer identifier"
|
|
// @Success 204 "No content if deletion was successful"
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/{id} [delete]
|
|
func (e peerEndpoint) handleDelete() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
id := Base64UrlDecode(c.Param("id"))
|
|
if id == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing peer id"})
|
|
return
|
|
}
|
|
|
|
err := e.app.DeletePeer(ctx, domain.PeerIdentifier(id))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.Status(http.StatusNoContent)
|
|
}
|
|
}
|
|
|
|
// handleConfigGet returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleConfigGet
|
|
// @Tags Peer
|
|
// @Summary Get peer configuration as string.
|
|
// @Produce json
|
|
// @Param id path string true "The peer identifier"
|
|
// @Success 200 {object} string
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/config/{id} [get]
|
|
func (e peerEndpoint) handleConfigGet() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
id := Base64UrlDecode(c.Param("id"))
|
|
if id == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{
|
|
Code: http.StatusInternalServerError, Message: "missing id parameter",
|
|
})
|
|
return
|
|
}
|
|
|
|
config, err := e.app.GetPeerConfig(ctx, domain.PeerIdentifier(id))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.Error{
|
|
Code: http.StatusInternalServerError, Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
configString, err := io.ReadAll(config)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.Error{
|
|
Code: http.StatusInternalServerError, Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, string(configString))
|
|
}
|
|
}
|
|
|
|
// handleQrCodeGet returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleQrCodeGet
|
|
// @Tags Peer
|
|
// @Summary Get peer configuration as qr code.
|
|
// @Produce png
|
|
// @Produce json
|
|
// @Param id path string true "The peer identifier"
|
|
// @Success 200 {file} binary
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/config-qr/{id} [get]
|
|
func (e peerEndpoint) handleQrCodeGet() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
id := Base64UrlDecode(c.Param("id"))
|
|
if id == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{
|
|
Code: http.StatusInternalServerError, Message: "missing id parameter",
|
|
})
|
|
return
|
|
}
|
|
|
|
config, err := e.app.GetPeerConfigQrCode(ctx, domain.PeerIdentifier(id))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.Error{
|
|
Code: http.StatusInternalServerError, Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
configData, err := io.ReadAll(config)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.Error{
|
|
Code: http.StatusInternalServerError, Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.Data(http.StatusOK, "image/png", configData)
|
|
}
|
|
}
|
|
|
|
// handleEmailPost returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleEmailPost
|
|
// @Tags Peer
|
|
// @Summary Send peer configuration via email.
|
|
// @Produce json
|
|
// @Param request body model.PeerMailRequest true "The peer mail request data"
|
|
// @Success 204 "No content if mail sending was successful"
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/config-mail [post]
|
|
func (e peerEndpoint) handleEmailPost() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
var req model.PeerMailRequest
|
|
err := c.BindJSON(&req)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
if len(req.Identifiers) == 0 {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing peer identifiers"})
|
|
return
|
|
}
|
|
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
peerIds := make([]domain.PeerIdentifier, len(req.Identifiers))
|
|
for i := range req.Identifiers {
|
|
peerIds[i] = domain.PeerIdentifier(req.Identifiers[i])
|
|
}
|
|
err = e.app.SendPeerEmail(ctx, req.LinkOnly, peerIds...)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.Status(http.StatusNoContent)
|
|
}
|
|
}
|
|
|
|
// handleStatsGet returns a gorm handler function.
|
|
//
|
|
// @ID peers_handleStatsGet
|
|
// @Tags Peer
|
|
// @Summary Get peer stats for the given interface.
|
|
// @Produce json
|
|
// @Param iface path string true "The interface identifier"
|
|
// @Success 200 {object} model.PeerStats
|
|
// @Failure 400 {object} model.Error
|
|
// @Failure 500 {object} model.Error
|
|
// @Router /peer/iface/{iface}/stats [get]
|
|
func (e peerEndpoint) handleStatsGet() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
ctx := domain.SetUserInfoFromGin(c)
|
|
|
|
interfaceId := Base64UrlDecode(c.Param("iface"))
|
|
if interfaceId == "" {
|
|
c.JSON(http.StatusBadRequest, model.Error{Code: http.StatusBadRequest, Message: "missing iface parameter"})
|
|
return
|
|
}
|
|
|
|
stats, err := e.app.GetPeerStats(ctx, domain.InterfaceIdentifier(interfaceId))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError,
|
|
model.Error{Code: http.StatusInternalServerError, Message: err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.NewPeerStats(e.app.Config.Statistics.CollectPeerData, stats))
|
|
}
|
|
}
|