jhot
/
hats
1
0
Fork 0
Code Actions

Allow gokapi upload with a byte array

main v0.18.1
Jordan Hotmann 2023-12-14 09:57:43 -07:00
parent 66963053f5
commit b0629be093
No known key found for this signature in database
GPG Key ID: 01B504170C2A2EA3
1 changed files with 35 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
pkg/gokapi/api.go
View File

@ -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 {