package httpserver

import (
	"encoding/json"
	"net/http"
)

type IHttpResponse interface {
	Code(code int64)
	Msg(msg string)
	Data(data interface{})
	Response()
}

type HttpResponseData struct {
	Code      int64       `json:"code"`
	Data      interface{} `json:"data"`
	Msg       string      `json:"msg"`
	RequestId string      `json:"requestId"`
}

func (d *HttpResponseData) Encode() []byte {
	s, err := json.Marshal(d)
	if err != nil {
		return []byte("response marshal err " + err.Error())
	}
	return s
}

type HttpResponse struct {
	writer   http.ResponseWriter
	respData *HttpResponseData
}

func NewHttpResponse(writer http.ResponseWriter) IHttpResponse {
	return &HttpResponse{
		writer:   writer,
		respData: &HttpResponseData{
			RequestId:"",
		},
	}
}

func (resp *HttpResponse) Code(code int64) {
	resp.respData.Code = code
}

func (resp *HttpResponse) Msg(msg string) {
	resp.respData.Msg = msg
}

func (resp *HttpResponse) Data(data interface{}) {
	resp.respData.Data = data
}

func (resp *HttpResponse) Response() {
	resp.writer.Write(resp.respData.Encode())
}