package response

import (
	"fmt"
	"time"

	"github.com/gin-gonic/gin"
)

type Button struct {
	Color string `json:"-"`
	Text  string `json:"text"`
	Url   string `json:"url"`
	Type  string `json:"type"`
}

type Window struct {
	Title string `json:"title"`
	Close bool   `json:"close"`
}

type Data struct {
	Buttons []*Button `json:"buttons"`
	Window  Window    `json:"window"`
}

type Response struct {
	ctx   *gin.Context
	Code  int    `json:"code"`
	Msg   string `json:"message"`
	Trace string `json:"trace"`
	Data  Data   `json:"data"`
}

func New(ctx *gin.Context, code int, message string) *Response {
	var trace string
	var ok bool
	if ctx.Keys != nil {
		if trace, ok = ctx.Keys["req_id"].(string); !ok {
			trace = fmt.Sprintf("%d", time.Now().UnixMicro())
		}
	}

	response := new(Response)
	response.Code = code
	response.Msg = message
	response.Trace = trace
	response.Data.Buttons = make([]*Button, 0)
	response.Data.Window = Window{Title: "温馨提示", Close: true}
	return response
}

func (r *Response) Button(b ...*Button) *Response {
	if len(b) > 1 {
		r.Data.Buttons = append(r.Data.Buttons[:0], b...)
	}

	if len(b) == 1 {
		r.Data.Buttons = append(r.Data.Buttons, b...)
	}

	return r
}

func (r *Response) Window(win *Window) *Response {
	if win.Title != "" {
		r.Data.Window.Title = win.Title
	}

	r.Data.Window.Close = win.Close
	return r
}

func (r *Response) Message(message string) *Response {
	r.Msg = message
	return r
}