mirror of
https://github.com/VaalaCat/frp-panel.git
synced 2025-10-04 15:02:54 +08:00
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/VaalaCat/frp-panel/pb"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gin-gonic/gin/binding"
|
|
"google.golang.org/protobuf/encoding/protojson"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
)
|
|
|
|
type CommonReq interface {
|
|
pb.UpdateFRPCRequest | pb.RemoveFRPCRequest |
|
|
pb.UpdateFRPSRequest | pb.RemoveFRPSRequest |
|
|
pb.CommonRequest | pb.RegisterRequest | pb.LoginRequest |
|
|
pb.InitClientRequest | pb.ListClientsRequest | pb.GetClientRequest |
|
|
pb.DeleteClientRequest |
|
|
pb.InitServerRequest | pb.ListServersRequest | pb.GetServerRequest |
|
|
pb.DeleteServerRequest |
|
|
pb.GetUserInfoRequest | pb.UpdateUserInfoRequest |
|
|
pb.GetPlatformInfoRequest | pb.GetClientsStatusRequest |
|
|
pb.GetClientCertRequest
|
|
}
|
|
|
|
func GetProtoRequest[T CommonReq](c *gin.Context) (r *T, err error) {
|
|
r = new(T)
|
|
if c.ContentType() == "application/x-protobuf" {
|
|
err = c.Copy().ShouldBindWith(r, binding.ProtoBuf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
b, err := c.Copy().GetRawData()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = GetServerMessageRequest(b, r, protojson.Unmarshal)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
func GetServerMessageRequest[T CommonReq](b []byte, r *T, trans func(b []byte, m protoreflect.ProtoMessage) error) (err error) {
|
|
switch ptr := any(r).(type) {
|
|
case *pb.CommonRequest:
|
|
return trans(b, ptr)
|
|
case *pb.UpdateFRPCRequest:
|
|
return trans(b, ptr)
|
|
case *pb.RemoveFRPCRequest:
|
|
return trans(b, ptr)
|
|
case *pb.UpdateFRPSRequest:
|
|
return trans(b, ptr)
|
|
case *pb.RemoveFRPSRequest:
|
|
return trans(b, ptr)
|
|
case *pb.RegisterRequest:
|
|
return trans(b, ptr)
|
|
case *pb.LoginRequest:
|
|
return trans(b, ptr)
|
|
case *pb.InitClientRequest:
|
|
return trans(b, ptr)
|
|
case *pb.ListClientsRequest:
|
|
return trans(b, ptr)
|
|
case *pb.GetClientRequest:
|
|
return trans(b, ptr)
|
|
case *pb.DeleteClientRequest:
|
|
return trans(b, ptr)
|
|
case *pb.InitServerRequest:
|
|
return trans(b, ptr)
|
|
case *pb.ListServersRequest:
|
|
return trans(b, ptr)
|
|
case *pb.GetServerRequest:
|
|
return trans(b, ptr)
|
|
case *pb.DeleteServerRequest:
|
|
return trans(b, ptr)
|
|
case *pb.GetUserInfoRequest:
|
|
return trans(b, ptr)
|
|
case *pb.UpdateUserInfoRequest:
|
|
return trans(b, ptr)
|
|
case *pb.GetPlatformInfoRequest:
|
|
return trans(b, ptr)
|
|
case *pb.GetClientsStatusRequest:
|
|
return trans(b, ptr)
|
|
case *pb.GetClientCertRequest:
|
|
return trans(b, ptr)
|
|
default:
|
|
}
|
|
return fmt.Errorf("cannot unmarshal unknown type: %T", r)
|
|
}
|