From b0629be093fb832e68191e283045a6bc5b3b40de Mon Sep 17 00:00:00 2001
From: Jordan Hotmann <jhot@protonmail.com>
Date: Thu, 14 Dec 2023 09:57:43 -0700
Subject: [PATCH] Allow gokapi upload with a byte array

---
 pkg/gokapi/api.go | 59 ++++++++++++++++++++++++++++-------------------
 1 file changed, 35 insertions(+), 24 deletions(-)

diff --git a/pkg/gokapi/api.go b/pkg/gokapi/api.go
index cf6c555..848d461 100644
--- a/pkg/gokapi/api.go
+++ b/pkg/gokapi/api.go
@@ -1,9 +1,12 @@
 package gokapi
 
 import (
+	"bytes"
+	"encoding/base64"
 	"fmt"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"code.jhot.me/jhot/hats/internal/util"
 	"github.com/gabriel-vasile/mimetype"
@@ -64,12 +67,36 @@ type UploadOptions struct {
 	FilenameOverride string
 }
 
+func (c *GokapiClient) UploadBytes(data []byte, opts *UploadOptions) (GokapiFile, error) {
+	mimeType := mimetype.Detect(data)
+	if opts.FilenameOverride == "" {
+		opts.FilenameOverride = strings.TrimRight(base64.StdEncoding.EncodeToString(data), "=") + mimeType.Extension()
+	}
+
+	req := c.restClient.R().
+		SetFileReader("file", opts.FilenameOverride, bytes.NewReader(data)).
+		SetMultipartFormData(map[string]string{
+			"allowedDownloads": fmt.Sprintf("%d", opts.AllowedDownloads),
+			"expiryDays":       fmt.Sprintf("%d", opts.ExpiryDays),
+			"password":         opts.Password,
+		})
+
+	var returnData struct {
+		Result   string     `json:"Result"`
+		FileInfo GokapiFile `json:"FileInfo"`
+	}
+
+	_, err := util.CheckSuccess(req.SetResult(&returnData).Post("files/add"))
+	if err != nil {
+		return GokapiFile{}, err
+	}
+
+	return returnData.FileInfo, nil
+}
+
 func (c *GokapiClient) UploadFile(filePath string, opts *UploadOptions) (GokapiFile, error) {
-	var fileName string
-	if opts.FilenameOverride != "" {
-		fileName = opts.FilenameOverride
-	} else {
-		fileName = filepath.Base(filePath)
+	if opts.FilenameOverride == "" {
+		opts.FilenameOverride = filepath.Base(filePath)
 	}
 
 	f, err := os.Open(filePath)
@@ -78,27 +105,11 @@ func (c *GokapiClient) UploadFile(filePath string, opts *UploadOptions) (GokapiF
 	}
 	defer f.Close()
 
-	mimeType, _ := mimetype.DetectFile(filePath)
-
-	req := c.restClient.R().
-		SetMultipartField("file", fileName, mimeType.String(), f).
-		SetMultipartFormData(map[string]string{
-			"allowedDownloads": fmt.Sprintf("%d", opts.AllowedDownloads),
-			"expiryDays":       fmt.Sprintf("%d", opts.ExpiryDays),
-			"password":         opts.Password,
-		})
-
-	var data struct {
-		Result   string     `json:"Result"`
-		FileInfo GokapiFile `json:"FileInfo"`
-	}
-
-	_, err = util.CheckSuccess(req.SetResult(&data).Post("files/add"))
+	data, err := os.ReadFile(filePath)
 	if err != nil {
-		return GokapiFile{}, err
+		return GokapiFile{}, fmt.Errorf("error reading file: %w", err)
 	}
-
-	return data.FileInfo, nil
+	return c.UploadBytes(data, opts)
 }
 
 func (c *GokapiClient) GetDownloadUrl(f GokapiFile) string {