response.go

package http

import (
	"errors"
	"fmt"
	"github.com/goal-web/contracts"
	"github.com/goal-web/supports/logs"
	"go/types"
	"net/http"
	"os"
)

var (
	FileTypeError = errors.New("文件参数类型错误")
)

// HandleResponse 处理控制器函数的响应
func HandleResponse(response interface{}, ctx contracts.HttpRequest) {
	if response == nil {
		return
	}
	switch res := response.(type) {
	case error:
		logs.WithError(ctx.String(http.StatusInternalServerError, res.Error())).Debug("response error")
	case string:
		logs.WithError(ctx.String(http.StatusOK, res)).Debug("response error")
	case fmt.Stringer:
		logs.WithError(ctx.String(http.StatusOK, res.String())).Debug("response error")
	case contracts.Json:
		logs.WithError(ctx.String(http.StatusOK, res.ToJson())).Debug("response error")
	case contracts.HttpResponse:
		logs.WithError(res.Response(ctx)).Debug("response error")
	case types.Nil:
		return
	default:
		logs.WithError(ctx.JSON(200, res)).Debug("response json error")
	}

}

type Response struct {
	status   int
	Json     interface{}
	String   string
	FilePath string
	File     *os.File
}

func StringResponse(str string, code ...int) contracts.HttpResponse {
	status := 200
	if len(code) > 0 {
		status = code[0]
	}
	return Response{
		status: status,
		String: str,
	}
}

func JsonResponse(json interface{}, code ...int) contracts.HttpResponse {
	status := 200
	if len(code) > 0 {
		status = code[0]
	}
	return Response{
		status: status,
		Json:   json,
	}
}

// FileResponse 响应文件
func FileResponse(file interface{}) contracts.HttpResponse {
	switch f := file.(type) {
	case *os.File:
		return Response{File: f}
	case string:
		return Response{FilePath: f}
	default:
		panic(FileTypeError)
	}
}

func (res Response) Status() int {
	return res.status
}

func (res Response) Response(ctx contracts.HttpContext) error {
	if res.Json != nil {
		return ctx.JSON(res.Status(), res.Json)
	}
	if res.FilePath != "" {
		return ctx.File(res.FilePath)
	}
	if res.File != nil {
		return ctx.File(res.File.Name())
	}

	return ctx.String(res.Status(), res.String)
}