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
|
package main
import (
"log"
"net"
"net/http"
"os"
"github.com/anacrolix/tagflag"
)
func setIfGetHeader(w http.ResponseWriter, r *http.Request, set, get string) {
h := r.Header.Get(get)
if h == "" {
return
}
w.Header().Set(set, h)
}
func allowCORS(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if origin := r.Header.Get("Origin"); origin != "" {
w.Header().Set("Access-Control-Allow-Origin", origin)
w.Header().Set("Access-Control-Allow-Credentials", "true")
setIfGetHeader(w, r, "Access-Control-Allow-Methods", "Access-Control-Request-Method")
setIfGetHeader(w, r, "Access-Control-Allow-Headers", "Access-Control-Request-Headers")
w.Header().Set("Access-Control-Expose-Headers", "Content-Type")
}
h.ServeHTTP(w, r)
})
}
func main() {
var flags = struct {
Addr string
}{
Addr: "localhost:8080",
}
tagflag.Parse(&flags)
dir, err := os.Getwd()
if err != nil {
log.Fatal(err)
}
l, err := net.Listen("tcp", flags.Addr)
if err != nil {
log.Fatal(err)
}
defer l.Close()
addr := l.Addr()
log.Printf("serving %q at %s", dir, addr)
log.Fatal(http.Serve(l, allowCORS(http.FileServer(http.Dir(dir)))))
}
|
