package gateway

import (
	"bytes"
	"context"
	"io"
	"mime/multipart"
	"net/http"
	"strings"

	commonHttp "gitea.redpowerfuture.com/red-future/common/http"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
)

// Upload 以 multipart 方式上传文件字节到 OSS，返回可访问 URL。
// 原签名基于 workflow/model/dto 的 UploadFileBytesReq/Res，抽离时简化为直接传文件名与字节，便于业务侧解耦 workflow。
func Upload(ctx context.Context, fileName string, fileBytes []byte) (string, error) {
	body := &bytes.Buffer{}
	writer := multipart.NewWriter(body)

	part, err := writer.CreateFormFile("file", fileName)
	if err != nil {
		return "", err
	}
	if _, err = part.Write(fileBytes); err != nil {
		return "", err
	}
	if err = writer.Close(); err != nil {
		return "", err
	}

	headers := make(map[string]string)
	if r := g.RequestFromCtx(ctx); r != nil {
		for k, v := range r.Request.Header {
			if len(v) > 0 {
				headers[k] = v[0]
			}
		}
	}
	headers["Content-Type"] = writer.FormDataContentType()

	res := &uploadFileRes{}
	if err = commonHttp.Post(ctx, "oss/file/uploadFile", headers, res, body.Bytes()); err != nil {
		return "", err
	}
	return res.FileURL, nil
}

// uploadFileRes OSS 上传响应的最小字段（原 workflow/model/dto.UploadFileBytesRes 的 URL 部分）。
type uploadFileRes struct {
	FileURL string `json:"fileURL"`
}

// GetFileBytesFromURL 下载远程文件内容（把 filePrefix 前缀替换为 minioPrefix 后经 GoFrame 客户端下载）
func GetFileBytesFromURL(ctx context.Context, fileUrl string) ([]byte, error) {
	newS := strings.ReplaceAll(fileUrl, g.Cfg().MustGet(ctx, "filePrefix").String(), g.Cfg().MustGet(ctx, "minioPrefix").String())
	// 使用 GoFrame 客户端（自带超时、追踪、日志等能力）
	resp, err := g.Client().Get(ctx, newS)
	if err != nil {
		return nil, gerror.Wrapf(err, "failed to request url: %s", newS)
	}
	defer resp.Close()

	// 校验状态码
	if resp.StatusCode != http.StatusOK {
		return nil, gerror.Newf("request failed with status code: %d, url: %s", resp.StatusCode, newS)
	}

	// 读取全部内容
	allBytes, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, gerror.Wrapf(err, "failed to read response body, url: %s", fileUrl)
	}

	return allBytes, nil
}