/*
 * TencentBlueKing is pleased to support the open source community by making
 * 蓝鲸智云 - 配置平台 (BlueKing - CMDB) available.
 * Copyright (C) 2025 Tencent. All rights reserved.
 * Licensed under the MIT License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at http://opensource.org/licenses/MIT
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 * We undertake not to change the open source license (MIT license) applicable
 * to the current version of the project delivered to anyone in the future.
 */

package transport

import (
	"bytes"
	"fmt"
	"io"
	"net/http"
)

// HTTPIsSuccess method returns ok if HTTP status `code >= 200 and <= 299` otherwise a error.
func HTTPIsSuccess(next http.RoundTripper) http.RoundTripper {
	f := func(req *http.Request) (*http.Response, error) {
		resp, err := next.RoundTrip(req)
		if err != nil {
			return nil, err
		}

		// HTTP status `code >= 200 and <= 299`
		if resp.StatusCode >= 200 && resp.StatusCode <= 299 {
			return resp, nil
		}

		rawBody := resp.Body
		defer rawBody.Close()

		body, err := io.ReadAll(rawBody)
		if err != nil {
			return nil, err
		}
		resp.Body = io.NopCloser(bytes.NewBuffer(body))

		return nil, fmt.Errorf("http status not success, code=%d, body=%s", resp.StatusCode, body)
	}

	return RoundTripFunc(f)
}