Refactor doIPRanges as DoJSON

I added response header hook support to IPRanges. Previously, it was not possible to encounter an HTTP status error on IPRanges because this condition could never be met:

`if resp.StatusCode < 200 && resp.StatusCode >= 400 {`

Author: brandonc

CHANGELOG.md

@@ -5,6 +5,7 @@
 ## Enhancements
 * Adds `RunPreApplyRunning` and `RunQueuingApply` run statuses by @uk1288 [#712](https://github.com/hashicorp/go-tfe/pull/712)
 * Adds BETA method `Upload` method to `StateVersions` and support for pending state versions by @brandonc [#717](https://github.com/hashicorp/go-tfe/pull/717)
+* Added ContextWithResponseHeaderHook support to `IPRanges` by @brandonc [#717](https://github.com/hashicorp/go-tfe/pull/717)
 
 ## Bug Fixes
 * AgentPool `Update` is not able to remove all allowed workspaces from an agent pool. That operation is now handled by a separate `UpdateAllowedWorkspaces` method using `AgentPoolAllowedWorkspacesUpdateOptions` by @hs26gill [#701](https://github.com/hashicorp/go-tfe/pull/701)

ip_ranges.go

@@ -50,7 +50,7 @@ func (i *ipRanges) Read(ctx context.Context, modifiedSince string) (*IPRange, er
 	}
 
 	ir := &IPRange{}
-	err = req.doIPRanges(ctx, ir)
+	err = req.DoJSON(ctx, ir)
 	if err != nil {
 		return nil, err
 	}

request.go

@@ -78,9 +78,9 @@ func (r ClientRequest) Do(ctx context.Context, model interface{}) error {
 	return unmarshalResponse(resp.Body, model)
 }
 
-// doIPRanges is similar to Do except that The IP ranges API is not returning jsonapi
-// like every other endpoint which means we need to handle it differently.
-func (r *ClientRequest) doIPRanges(ctx context.Context, ir *IPRange) error {
+// DoJSON is similar to Do except that it should be used when a plain JSON response is expected
+// as opposed to json-api.
+func (r *ClientRequest) DoJSON(ctx context.Context, model any) error {
 	// Wait will block until the limiter can obtain a new token
 	// or returns an error if the given context is canceled.
 	if r.limiter != nil {
@@ -92,8 +92,18 @@ func (r *ClientRequest) doIPRanges(ctx context.Context, ir *IPRange) error {
 	// Add the context to the request.
 	contextReq := r.retryableRequest.WithContext(ctx)
 
+	// If the caller provided a response header hook then we'll call it
+	// once we have a response.
+	respHeaderHook := contextResponseHeaderHook(ctx)
+
 	// Execute the request and check the response.
 	resp, err := r.http.Do(contextReq)
+	if resp != nil {
+		// We call the callback whenever there's any sort of response,
+		// even if it's returned in conjunction with an error.
+		respHeaderHook(resp.StatusCode, resp.Header)
+	}
+	defer resp.Body.Close()
 	if err != nil {
 		// If we got an error, and the context has been canceled,
 		// the context's error is probably more useful.
@@ -104,17 +114,21 @@ func (r *ClientRequest) doIPRanges(ctx context.Context, ir *IPRange) error {
 			return err
 		}
 	}
-	defer resp.Body.Close()
 
-	if resp.StatusCode < 200 && resp.StatusCode >= 400 {
-		return fmt.Errorf("error HTTP response while retrieving IP ranges: %d", resp.StatusCode)
-	} else if resp.StatusCode == 304 {
+	if resp.StatusCode < 200 || resp.StatusCode >= 400 {
+		return fmt.Errorf("error HTTP response: %d", resp.StatusCode)
+	}
+
+	// Return here if decoding the response isn't needed.
+	if model == nil {
 		return nil
 	}
 
-	err = json.NewDecoder(resp.Body).Decode(ir)
-	if err != nil {
+	// If v implements io.Writer, write the raw response body.
+	if w, ok := model.(io.Writer); ok {
+		_, err := io.Copy(w, resp.Body)
 		return err
 	}
-	return nil
+
+	return json.NewDecoder(resp.Body).Decode(model)
 }

tfe.go

@@ -198,14 +198,17 @@ type Meta struct {
 	IPRanges IPRanges
 }
 
-func (c *Client) doForeignPUTRequest(ctx context.Context, foreignUrl string, data io.Reader) error {
-	u, err := url.Parse(foreignUrl)
+// doForeignPUTRequest performs a PUT request using the specific data body. The Content-Type
+// header is set to application/octet-stream but no Authentication header is sent. No response
+// body is decoded.
+func (c *Client) doForeignPUTRequest(ctx context.Context, foreignURL string, data io.Reader) error {
+	u, err := url.Parse(foreignURL)
 	if err != nil {
 		return fmt.Errorf("specified URL was not valid: %w", err)
 	}
 
 	reqHeaders := make(http.Header)
-	reqHeaders.Set("Accept", "application/json")
+	reqHeaders.Set("Accept", "application/json, */*")
 	reqHeaders.Set("Content-Type", "application/octet-stream")
 
 	req, err := retryablehttp.NewRequest("PUT", u.String(), data)
@@ -229,7 +232,7 @@ func (c *Client) doForeignPUTRequest(ctx context.Context, foreignUrl string, dat
 		Header:           req.Header,
 	}
 
-	return request.Do(ctx, nil)
+	return request.DoJSON(ctx, nil)
 }
 
 func (c *Client) NewRequest(method, path string, reqAttr any) (*ClientRequest, error) {