// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
//go:build plus

package nodes

import (
	"bytes"
	"errors"
	"fmt"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
	"github.com/TeaOSLab/EdgeNode/internal/oss"
	"github.com/iwind/TeaGo/types"
	"io"
	"net/http"
)

// 请求OSS源站
func (this *HTTPRequest) doOSSOrigin(origin *serverconfigs.OriginConfig) (resp *http.Response, goNext bool, errorCode string, ossBucketName string, err error) {
	if origin == nil || origin.OSS == nil {
		err = errors.New("'origin' or 'origin.OSS' should not be nil")
		return
	}

	// 只支持少数方法
	var isHeadRequest = this.RawReq.Method != http.MethodGet &&
		this.RawReq.Method != http.MethodPost &&
		this.RawReq.Method != http.MethodPut

	var rangeBytes = this.RawReq.Header.Get("Range")
	if isHeadRequest && len(rangeBytes) == 0 {
		resp, errorCode, ossBucketName, err = oss.SharedManager.Head(this.RawReq, this.ReqHost, origin.OSS)
	} else {
		if len(rangeBytes) > 0 {
			resp, errorCode, ossBucketName, err = oss.SharedManager.GetRange(this.RawReq, this.ReqHost, rangeBytes, origin.OSS)
		} else {
			resp, errorCode, ossBucketName, err = oss.SharedManager.Get(this.RawReq, this.ReqHost, origin.OSS)
		}
	}

	if len(ossBucketName) == 0 {
		this.originAddr = origin.OSS.Type
	} else {
		this.originAddr = origin.OSS.Type + "/" + ossBucketName
	}
	this.tags = append(this.tags, "oss")

	if err != nil {
		if oss.IsNotFound(err) {
			this.write404()
			return nil, false, errorCode, ossBucketName, nil
		}

		if oss.IsTimeout(err) {
			this.writeCode(http.StatusGatewayTimeout, "Read object timeout.", "读取对象超时。")
			return nil, false, errorCode, ossBucketName, nil
		}

		return nil, false, errorCode, ossBucketName, fmt.Errorf("OSS: [%s]: %s: %w", origin.OSS.Type, errorCode, err)
	}

	if isHeadRequest {
		_ = resp.Body.Close()
		resp.Body = io.NopCloser(&bytes.Buffer{})
	}

	// fix Content-Length
	if resp.Header == nil {
		resp.Header = http.Header{}
	}
	if resp.ContentLength > 0 {
		_, ok := resp.Header["Content-Length"]
		if !ok {
			resp.Header.Set("Content-Length", types.String(resp.ContentLength))
		}
	}

	return resp, true, "", ossBucketName, nil
}