package middleware_ws

import (
	"gitee.com/os-lee/easy-paas/common"
	"gitee.com/os-lee/easy-paas/common/e"
	"gitee.com/os-lee/easy-paas/gateway/internal/dao"
	"github.com/gin-gonic/gin"
	"github.com/pkg/errors"
	"strings"
)

// StripUri uri去掉前缀
func StripUri() gin.HandlerFunc {
	return func(c *gin.Context) {
		serverInterface, ok := c.Get("service")
		if !ok {
			common.ResponseError(c, e.ErrGateway, errors.New("从上下文中获取服务失败"))
			c.Abort()
			return
		}
		serviceDetail := serverInterface.(*dao.ServiceDetail)
		if serviceDetail.HTTPRule.RuleType == common.RuleTypePrefixURL && serviceDetail.HTTPRule.NeedStripUri == 1 {
			c.Request.URL.Path = strings.Replace(c.Request.URL.Path, serviceDetail.HTTPRule.Rule, "", 1)
		}
		//例如
		//http://127.0.0.1:8080/test/abbb
		//http://127.0.0.1:2004/abbb
		c.Next()
	}
}