package common

import (
	"context"

	"gitea.com/bitwsd/document_ai/pkg/constant"
)

type Response struct {
	RequestID string      `json:"request_id"`
	Code      int         `json:"code"`
	Message   string      `json:"message"`
	Data      interface{} `json:"data"`
}

func ErrorResponse(ctx context.Context, code int, message string) *Response {
	return &Response{
		RequestID: ctx.Value(constant.ContextRequestID).(string),
		Code:      code,
		Message:   message,
	}
}

func SuccessResponse(ctx context.Context, data interface{}) *Response {
	return &Response{
		RequestID: ctx.Value(constant.ContextRequestID).(string),
		Code:      CodeSuccess,
		Message:   "success",
		Data:      data,
	}
}