cloud-agents context-aware client (#801)

Author: yoonsio

pkg/cloudagents/auth.go

@@ -15,6 +15,7 @@
 package cloudagents
 
 import (
+	"context"
 	"fmt"
 	"io"
 	"net/http"
@@ -23,9 +24,9 @@ import (
 	lksdk "github.com/livekit/server-sdk-go/v2"
 )
 
-// newRequest creates a new HTTP request with the auth and version headers.
-func (c *Client) newRequest(method, url string, body io.Reader) (*http.Request, error) {
-	req, err := http.NewRequest(method, url, body)
+// newRequestWithContext creates a new HTTP request with the auth and version headers.
+func (c *Client) newRequestWithContext(ctx context.Context, method, url string, body io.Reader) (*http.Request, error) {
+	req, err := http.NewRequestWithContext(ctx, method, url, body)
 	if err != nil {
 		return nil, err
 	}

pkg/cloudagents/build.go

@@ -34,7 +34,7 @@ func (c *Client) build(ctx context.Context, id string, writer io.Writer) error {
 	params := url.Values{}
 	params.Add("agent_id", id)
 	fullUrl := fmt.Sprintf("%s/build?%s", c.agentsURL, params.Encode())
-	req, err := c.newRequest("POST", fullUrl, nil)
+	req, err := c.newRequestWithContext(ctx, "POST", fullUrl, nil)
 	if err != nil {
 		return err
 	}

pkg/cloudagents/logs.go

@@ -40,7 +40,7 @@ func (c *Client) StreamLogs(ctx context.Context, logType, agentID string, writer
 	params.Add("agent_id", agentID)
 	params.Add("log_type", logType)
 	fullUrl := fmt.Sprintf("%s/logs?%s", c.getAgentsURL(serverRegion), params.Encode())
-	req, err := c.newRequest("GET", fullUrl, nil)
+	req, err := c.newRequestWithContext(ctx, "GET", fullUrl, nil)
 	if err != nil {
 		return err
 	}