feat: add PickerItem.Stream function

Author: andresperezl

client.go

@@ -66,12 +66,17 @@ func (c *Cobalt) Post(ctx context.Context, params PostRequest, headers ...string
 		return nil, CobaltAPIError(*media)
 	}
 
+	for i := range media.Picker {
+		media.Picker[i].client = c.client
+	}
+
 	return media, nil
 }
 
 // Stream is a helper utility that will return an [io.ReadCloser] using the [PostResponse.URL] from this media object
 // The returned [io.ReadCloser] is the Body of [*http.Response] and must be closed when you are done with the stream.
 // When the [PostResponse.Status] == [ResponseStatusPicker] it will stream the first item from the [PostResponse.Picker] array.
+// Use [PickerItem.Stream] instead to stream an specific item from the picker list.
 func (m *PostResponse) Stream(ctx context.Context) (io.ReadCloser, error) {
 	if m.Status != ResponseStatusTunnel && m.Status != ResponseStatusRedirect && m.Status != ResponseStatusPicker {
 		return nil, fmt.Errorf("unstreamable response type %s", m.Status)
@@ -81,6 +86,19 @@ func (m *PostResponse) Stream(ctx context.Context) (io.ReadCloser, error) {
 	if m.Status == ResponseStatusPicker && len(m.Picker) > 0 {
 		url = m.Picker[0].URL
 	}
+	return stream(ctx, m.client, url)
+}
+
+// Stream is a helper utility that will return an [io.ReadCloser] using the [PickerItem.URL] from this media object
+// The returned [io.ReadCloser] is the Body of [*http.Response] and must be closed when you are done with the stream.
+func (m PickerItem) Stream(ctx context.Context) (io.ReadCloser, error) {
+	return stream(ctx, m.client, m.URL)
+}
+
+func stream(ctx context.Context, client *http.Client, url string) (io.ReadCloser, error) {
+	if client == nil {
+		client = http.DefaultClient
+	}
 	if len(url) == 0 {
 		return nil, fmt.Errorf("url is empty, nothing to stream")
 	}
@@ -90,7 +108,7 @@ func (m *PostResponse) Stream(ctx context.Context) (io.ReadCloser, error) {
 		return nil, err
 	}
 
-	resp, err := m.client.Do(req)
+	resp, err := client.Do(req)
 	if err != nil {
 		return nil, err
 	}

examples/basic/README.md

@@ -0,0 +1,11 @@
+# Basic Example
+
+This example is a simple program that will fetch a media URL, and in case of multiple options,
+it will fetch the first element that is a video.
+
+## Run
+
+```bash
+go run main.go <your_url_here> > my_file.mp4
+```
+

examples/basic/main.go

@@ -0,0 +1,43 @@
+package main
+
+import (
+	"context"
+	"io"
+	"log"
+	"os"
+
+	"github.com/andresperezl/gobalt/v2"
+)
+
+func main() {
+	if len(os.Args) < 2 {
+		log.Fatal("Need to provide an URL as argument")
+	}
+
+	client := gobalt.NewCobaltWithAPI("http://localhost:9000")
+	resp, err := client.Post(context.Background(), gobalt.PostRequest{
+		URL: os.Args[1],
+	})
+	if err != nil {
+		log.Fatal(err)
+	}
+	var stream io.ReadCloser
+
+	if resp.Status == gobalt.ResponseStatusPicker {
+		for _, pi := range resp.Picker {
+			if pi.Type == gobalt.PickerItemTypeVideo {
+				stream, err = pi.Stream(context.Background())
+			}
+		}
+	} else {
+		stream, err = resp.Stream(context.Background())
+	}
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer stream.Close()
+
+	if _, err := os.Stdout.ReadFrom(stream); err != nil {
+		log.Fatal(err)
+	}
+}

go.mod

@@ -1,3 +1,3 @@
 module github.com/andresperezl/gobalt/v2
 
-go 1.23.0
+go 1.24.3

types.go

@@ -206,6 +206,8 @@ type PostResponse struct {
 }
 
 type PickerItem struct {
+	client *http.Client
+
 	Type  PickerItemType `json:"type"`
 	URL   string         `json:"url"`
 	Thumb string         `json:"thumb"`