package webs

import (
	"io/ioutil"
	"strings"

	"github.com/kataras/iris/v12"
	"github.com/lithammer/shortuuid"
	"gopkg.in/yaml.v3"
)

// Handler 句柄
type Handler struct {
	Method string // GET POST PUT DELETE
	Path   string // 路径
	File   string // 文件
	Type   string // Binary,Text,HTML,JSON,JSONP,XML
	data   []byte //
}

// Handler 句柄
func (h *Handler) Handler() iris.Handler {
	if h.data == nil {
		h.data, _ = ioutil.ReadFile(h.File)
	}
	return func(ctx iris.Context) {
		if h.data == nil {
			return
		}
		if strings.EqualFold(h.Type, "Json") {
			json := string(h.data)
			ctx.Text(strings.ReplaceAll(json, "UUID", shortuuid.New()))
			ctx.Header("Content-Type", "application/json")
			return
		}
		if strings.EqualFold(h.Type, "Text") {
			ctx.Text(string(h.data))
			return
		}
		if strings.EqualFold(h.Type, "Binary") {
			ctx.Binary(h.data)
			return
		}
	}
}

// Load 加载
func Load(file string) (hs []Handler, err error) {
	var bs []byte
	if bs, err = ioutil.ReadFile(file); err != nil {
		return
	}
	err = yaml.Unmarshal(bs, &hs)
	return
}