context.go

/*
 * Copyright 2012-2019 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package SpringEcho

import (
	"net/http"

	"github.com/go-spring/spring-core/web"
	"github.com/labstack/echo/v4"
)

// EchoContext 将 web.Context 转换为 echo.Context
func EchoContext(c web.Context) echo.Context {
	v := c.NativeContext()
	if v == nil {
		return nil
	}
	ctx, _ := v.(echo.Context)
	return ctx
}

// WebContext 将 echo.Context 转换为 web.Context
func WebContext(c echo.Context) web.Context {
	v := c.Get(web.ContextKey)
	if v == nil {
		return nil
	}
	ctx, _ := v.(web.Context)
	return ctx
}

type Response struct {
	*echo.Response
}

func (resp *Response) Get() http.ResponseWriter {
	return resp.Response.Writer
}

func (resp *Response) Set(w http.ResponseWriter) {
	resp.Response.Writer = w
}

// context 适配 echo 的 Web 上下文
type context struct {
	*web.BaseContext

	echoCtx  echo.Context // echo 上下文
	wildcard string       // 通配符的名称
}

// newContext Context 的构造函数
func newContext(handler web.Handler, path, wildcard string, echoCtx echo.Context) *context {
	r := echoCtx.Request()
	w := &Response{Response: echoCtx.Response()}
	ctx := &context{
		echoCtx:     echoCtx,
		wildcard:    wildcard,
		BaseContext: web.NewBaseContext(path, handler, r, w),
	}
	echoCtx.Set(web.ContextKey, ctx)
	return ctx
}

// NativeContext 返回封装的底层上下文对象
func (c *context) NativeContext() interface{} {
	return c.echoCtx
}

// PathParam returns path parameter by name.
func (c *context) PathParam(name string) string {
	if name == c.wildcard {
		name = "*"
	}
	return c.echoCtx.Param(name)
}

// PathParamNames returns path parameter names.
func (c *context) PathParamNames() []string {
	return c.echoCtx.ParamNames()
}

// PathParamValues returns path parameter values.
func (c *context) PathParamValues() []string {
	return c.echoCtx.ParamValues()
}