frp-panel/biz/server/auth_handler.go

package server

import (
	"net/http"

	"github.com/VaalaCat/frp-panel/pb"
	"github.com/VaalaCat/frp-panel/rpc"
	plugin "github.com/fatedier/frp/pkg/plugin/server"
	"github.com/gin-gonic/gin"
	"github.com/sirupsen/logrus"
)

func NewRouter() *gin.Engine {
	router := gin.Default()
	router.POST("/auth", MakeGinHandlerFunc(HandleLogin))
	return router
}

type Response struct {
	Msg string `json:"msg"`
}

type HTTPError struct {
	Code int
	Err  error
}

func (e *HTTPError) Error() string {
	return e.Err.Error()
}

type HandlerFunc func(ctx *gin.Context) (interface{}, error)

func MakeGinHandlerFunc(handler HandlerFunc) gin.HandlerFunc {
	return func(ctx *gin.Context) {
		res, err := handler(ctx)
		if err != nil {
			logrus.Infof("handle %s error: %v", ctx.Request.URL.Path, err)
			switch e := err.(type) {
			case *HTTPError:
				ctx.JSON(e.Code, &Response{Msg: e.Err.Error()})
			default:
				ctx.JSON(500, &Response{Msg: err.Error()})
			}
			return
		}
		ctx.JSON(http.StatusOK, res)
	}
}

func HandleLogin(ctx *gin.Context) (interface{}, error) {
	var r plugin.Request
	var content plugin.LoginContent
	r.Content = &content
	if err := ctx.BindJSON(&r); err != nil {
		return nil, &HTTPError{
			Code: http.StatusBadRequest,
			Err:  err,
		}
	}

	var res plugin.Response
	token := content.Metas["token"]
	if len(content.User) == 0 || len(token) == 0 {
		res.Reject = true
		res.RejectReason = "user or meta token can not be empty"
		return res, nil
	}
	cli, err := rpc.MasterCli(ctx)
	if err != nil {
		res.Reject = true
		res.RejectReason = "cannot connect to master, please check master is running"
		return res, nil
	}

	authResponse, err := cli.FRPCAuth(ctx, &pb.FRPAuthRequest{User: content.User, Token: token})
	if err != nil {
		res.Reject = true
		res.RejectReason = "invalid meta token"
		return res, nil
	}

	if authResponse.GetOk() {
		res.Unchange = true
		return res, nil
	}

	res.Reject = true
	res.RejectReason = "invalid meta token"
	return res, nil
}