/
file.go
100 lines (86 loc) · 2.43 KB
/
file.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package utils
import (
"bytes"
"fmt"
"github.com/atotto/clipboard"
"github.com/charmbracelet/bubbles/list"
"io/ioutil"
"net/http"
"os"
"path/filepath"
)
var FileURL string
func (i item) Title() string { return i.title }
func (i item) Description() string { return i.desc }
func (i item) FilterValue() string { return i.title }
type item struct {
title, desc string
}
func formatSize(size int64) string {
const unit = 1024
if size < unit {
return fmt.Sprintf("%d B", size)
}
div, exp := int64(unit), 0
for n := size / unit; n >= unit; n /= unit {
div *= unit
exp++
}
return fmt.Sprintf("%.1f %ciB", float64(size)/float64(div), "KMGTPE"[exp])
}
func GetFiles(dir string) ([]list.Item, error) {
files, err := os.ReadDir(dir)
if err != nil {
return nil, fmt.Errorf("error reading directory: %w", err)
}
var validItems []list.Item
for _, file := range files {
fileInfo, err := file.Info()
if err != nil {
return nil, fmt.Errorf("error getting file info: %w", err)
}
if fileInfo.IsDir() {
if fileInfo.Mode().Perm()&(1<<2) == 0 {
continue
}
innerFiles, err := ioutil.ReadDir(filepath.Join(dir, file.Name()))
if err != nil {
return nil, fmt.Errorf("error reading directory: %w", err)
}
validItems = append(validItems, item{title: file.Name(), desc: fmt.Sprintf("Directory - %d items", len(innerFiles))})
} else {
size := formatSize(fileInfo.Size())
validItems = append(validItems, item{title: file.Name(), desc: fileInfo.Mode().String() + " " + size})
}
}
return validItems, nil
}
func UploadFile(filePath string) error {
conf, err := GetConf()
if err != nil {
return err
}
data, err := ioutil.ReadFile(filePath)
if err != nil {
return fmt.Errorf("error reading file: %w", err)
}
req, err := http.NewRequest("PUT", "http://v0.api.upyun.com/"+conf.Bucket+"/"+filePath, bytes.NewBuffer(data))
if err != nil {
return fmt.Errorf("error creating request: %w", err)
}
req.SetBasicAuth(conf.Operator, conf.Secret)
req.Header.Set("Content-Type", "application/octet-stream")
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return fmt.Errorf("error uploading file: %w", err)
}
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("bad status: %s", resp.Status)
}
FileURL = conf.Bucketurl + "/" + filePath
if err := clipboard.WriteAll(FileURL); err != nil {
return fmt.Errorf("error copying file URL to clipboard: %w", err)
}
return resp.Body.Close()
}