/
main.go
124 lines (100 loc) · 2.28 KB
/
main.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
// Sweetdrop
// Zero configuration, self hosted, file sharing.
package main
import (
"flag"
"fmt"
"log"
"math/rand"
"net/http"
"os"
"path/filepath"
"runtime"
"golang.org/x/net/http2"
)
const VERSION = "0.1.0"
var (
help = flag.Bool("h", false, "Show usage")
h1Port = flag.String("h1", ":53370", "HTTP port, set blank '' to disable")
h2Port = flag.String("h2", ":53371", "HTTPS & HTTP2 port, set blank '' to disable")
rootDir = flag.String("root", "", "Share directory root, defaults to current")
accessKey = flag.String("key", "", "Sets access key, defaults to random 32 char")
)
func main() {
setup()
startWebServer()
select {}
}
func randSeq(n int) string {
letters := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890")
b := make([]rune, n)
for i := range b {
b[i] = letters[rand.Intn(len(letters))]
}
return string(b)
}
func startWebServer() {
key := "/" + *accessKey + "/"
mux := http.NewServeMux()
mux.HandleFunc(key+"upload", upload)
mux.Handle(key, http.StripPrefix(key, http.FileServer(assetFS())))
mux.Handle(key+"f/", http.StripPrefix(key+"f/", http.FileServer(http.Dir(*rootDir))))
if *h1Port != "" {
log.Println("http://localhost" + *h1Port + key)
s := &http.Server{
Addr: *h1Port,
Handler: mux,
}
http2.ConfigureServer(s, nil)
go func() {
err := s.ListenAndServe()
if err != nil {
log.Fatal("HTTP server error:", err)
}
}()
}
if *h2Port != "" {
log.Println("https://localhost" + *h2Port + key)
s := &http.Server{
Addr: *h2Port,
Handler: mux,
TLSConfig: CreateTLS(),
}
http2.ConfigureServer(s, nil)
go func() {
err := s.ListenAndServeTLS("", "")
if err != nil {
log.Fatal("HTTPS/2 server error:", err)
}
}()
}
}
func setup() {
runtime.GOMAXPROCS(runtime.NumCPU())
flag.Usage = func() {
fmt.Println("Sweetdrop ", VERSION)
flag.PrintDefaults()
}
flag.Parse()
if *help {
flag.Usage()
os.Exit(0)
}
if *rootDir == "" {
cwd, err := os.Getwd()
if err != nil {
panic(err)
}
rootDir = &cwd
} else {
*rootDir = filepath.Clean(*rootDir)
_, err := os.Stat(*rootDir)
if err != nil {
log.Fatal("Root directory not valid")
}
}
log.Println("Sweetdrop "+VERSION+" sharing", *rootDir)
if *accessKey == "" {
*accessKey = randSeq(32)
}
}