From edee07826f9dcb02e46e0b6a8831579810c08328 Mon Sep 17 00:00:00 2001
From: hedge-sparrow <sparrow@spooky.academy>
Date: Thu, 12 Sep 2024 15:46:57 +0100
Subject: [PATCH 1/2] download command

---
 cli/download.go | 123 ++++++++++++++++++++++++++++++++++++++++++++++++
 cli/root.go     |   1 +
 2 files changed, 124 insertions(+)
 create mode 100644 cli/download.go

diff --git a/cli/download.go b/cli/download.go
new file mode 100644
index 0000000..3c3abc0
--- /dev/null
+++ b/cli/download.go
@@ -0,0 +1,123 @@
+package cli
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"os"
+	"path"
+
+	"github.com/pkg/errors"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+func DownloadCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "download",
+		Short: "Download bundle from url",
+		Long:  "Download bundle from url",
+		Args:  cobra.MaximumNArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			v := viper.GetViper()
+
+			// This only works with generated config, so let's make sure we don't mess up user's real files.
+			bundleLocation := v.GetString("support-bundle-location")
+			if len(args) > 0 && args[0] != "" {
+				bundleLocation = args[0]
+			}
+			if bundleLocation == "" {
+				return errors.New("support-bundle-location is required")
+			}
+
+			token := v.GetString("token")
+			if token == "" {
+				return errors.New("token is required when downloading bundle")
+			}
+
+			fmt.Println("Downloading bundle...")
+
+			file, err := downloadBundleToDisk(bundleLocation, token)
+			if err != nil {
+				return err
+			}
+			fmt.Println(file)
+
+			return nil
+		},
+	}
+
+	return cmd
+}
+
+func downloadBundleToDisk(bundleUrl string, token string) (string, error) {
+	parsedUrl, err := url.Parse(bundleUrl)
+	if err != nil {
+		return "", errors.Wrap(err, "failed to parse url")
+	}
+
+	_, slug := path.Split(parsedUrl.Path)
+	if slug == "" {
+		return "", errors.New("failed to extract slug from URL")
+	}
+	sbEndpoint := fmt.Sprintf("https://api.replicated.com/vendor/v3/supportbundle/%s", slug)
+	req, err := http.NewRequest("GET", sbEndpoint, nil)
+	if err != nil {
+		return "", errors.Wrap(err, "failed to create HTTP request")
+	}
+
+	req.Header.Add("Authorization", token)
+	req.Header.Add("Content-Type", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", errors.Wrap(err, "failed to execute request")
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", errors.Wrap(err, "failed to read GQL response")
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return "", errors.Errorf("unexpected status code: %v", resp.StatusCode)
+	}
+
+	bundleObj := struct {
+		Bundle struct {
+			SignedUri string `json:"signedUri"`
+		} `json:"bundle"`
+	}{}
+	err = json.Unmarshal(body, &bundleObj)
+	if err != nil {
+		return "", errors.Wrapf(err, "failed to unmarshal response: %s", body)
+	}
+
+	resp, err = http.Get(bundleObj.Bundle.SignedUri)
+	if err != nil {
+		return "", errors.Wrap(err, "failed to execute signed URL request")
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return "", errors.Errorf("unexpected status code: %v", resp.StatusCode)
+	}
+
+	sbFile, err := os.Create("support-bundle.tgz")
+	if err != nil {
+		return "", errors.Wrap(err, "failed to create file")
+	}
+	defer sbFile.Close()
+
+	_, err = io.Copy(sbFile, resp.Body)
+	if err != nil {
+		return "", errors.Wrap(err, "failed to copy bundle to file")
+	}
+
+	sbFile.Close()
+
+	return sbFile.Name(), nil
+}
diff --git a/cli/root.go b/cli/root.go
index 650aadb..0e1fb1f 100644
--- a/cli/root.go
+++ b/cli/root.go
@@ -26,6 +26,7 @@ func RootCmd() *cobra.Command {
 
 	cmd.AddCommand(ServeCmd())
 	cmd.AddCommand(ShellCmd())
+	cmd.AddCommand(DownloadCmd())
 
 	viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
 

From 0a54e6b6ec9b1a8763c6916b8400cb423f4cf2b9 Mon Sep 17 00:00:00 2001
From: hedge-sparrow <sparrow@spooky.academy>
Date: Wed, 18 Sep 2024 15:29:59 +0100
Subject: [PATCH 2/2] Remove pointless .Close()

---
 cli/download.go | 2 --
 1 file changed, 2 deletions(-)

diff --git a/cli/download.go b/cli/download.go
index 3c3abc0..fbb04ac 100644
--- a/cli/download.go
+++ b/cli/download.go
@@ -117,7 +117,5 @@ func downloadBundleToDisk(bundleUrl string, token string) (string, error) {
 		return "", errors.Wrap(err, "failed to copy bundle to file")
 	}
 
-	sbFile.Close()
-
 	return sbFile.Name(), nil
 }