// +build !go1.8

// Copyright 2017 Microsoft Corporation

//

//  Licensed under the Apache License, Version 2.0 (the "License");

//  you may not use this file except in compliance with the License.

//  You may obtain a copy of the License at

//

//      http://www.apache.org/licenses/LICENSE-2.0

//

//  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.

package autorest

import (

	"bytes"

	"io/ioutil"

	"net/http"

)

// RetriableRequest provides facilities for retrying an HTTP request.

type RetriableRequest struct {

	req *http.Request

	br  *bytes.Reader

}

// Prepare signals that the request is about to be sent.

func (rr *RetriableRequest) Prepare() (err error) {

	// preserve the request body; this is to support retry logic as

	// the underlying transport will always close the reqeust body

	if rr.req.Body != nil {

		if rr.br != nil {

			_, err = rr.br.Seek(0, 0 /*io.SeekStart*/)

			rr.req.Body = ioutil.NopCloser(rr.br)

		}

		if err != nil {

			return err

		}

		if rr.br == nil {

			// fall back to making a copy (only do this once)

			err = rr.prepareFromByteReader()

		}

	}

	return err

}

func removeRequestBody(req *http.Request) {

	req.Body = nil

	req.ContentLength = 0

}