package response

import (
	"fmt"
	"lone-services/pkg/utils"
	"net/http"

	jsoniter "github.com/json-iterator/go"
)

type BaseController struct{}

func (b *BaseController) Fail(w http.ResponseWriter) {
	b.Out(w, nil, Fail)
}

func (b *BaseController) Error(w http.ResponseWriter, cm Error) {
	b.Out(w, nil, cm)
}

func (b *BaseController) Out(w http.ResponseWriter, data any, cm Error) {
	out := NewError(cm.GetCode(), cm.GetMsg()).WithData(data)

	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)
	_ = jsoniter.NewEncoder(w).Encode(out)
}

func (b *BaseController) OutPut(w http.ResponseWriter, code int32, data any, msg string) {
	if code == utils.Ok.Code && !utils.GetConfigBool("encrypt.debug") {
		jsonData, err := jsoniter.Marshal(data)
		if err != nil {
			fmt.Printf("response encrypt: json.Marshal failed: %v\n", err)
			msg = utils.ErrorEncryptAesError.Msg
			code = utils.ErrorEncryptAesError.Code
		} else {
			key := utils.GetConfigString("encrypt.encrypt_key")
			res, enErr := utils.Crypto{}.Encrypt(jsonData)
			if enErr != nil {
				fmt.Printf("response encrypt failed: keyLen=%d keyEmpty=%v errCode=%d errMsg=%s\n",
					len(key), key == "", enErr.GetCode(), enErr.GetMsg())
				msg = utils.ErrorEncryptAesError.Msg
				code = utils.ErrorEncryptAesError.Code
			} else {
				data = res
			}
		}
	}

	out := NewError(int(code), msg).WithData(data)

	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)
	_ = jsoniter.NewEncoder(w).Encode(out)
}