-
Notifications
You must be signed in to change notification settings - Fork 0
/
httpserver.go
83 lines (76 loc) · 1.77 KB
/
httpserver.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
package main
import (
"errors"
"io"
"log"
"net/http"
"os/exec"
"time"
"github.com/Natolumin/ircmap/formatters"
"github.com/Natolumin/ircmap/irctree"
)
type protoHandler struct {
UpstreamURL string
Timeout time.Duration
}
func (p protoHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
w.WriteHeader(404)
return
}
if r.URL.Path == "/" {
w.Header().Add("Location", "ircmap.html")
w.WriteHeader(301)
return
}
var (
ircmap *irctree.Servers
err error
)
if ircmap, err = GetMap(p.UpstreamURL, p.Timeout); err != nil {
log.Printf("Could not get map from %s, got error %s\n", p.UpstreamURL, err)
w.WriteHeader(500)
return
}
switch r.URL.Path {
case "/map.json":
w.Header().Set("Content-Type", "application/json")
io.WriteString(w, formatters.BuildJson(ircmap, displayAll))
case "/map.dot":
w.Header().Set("Content-Type", "text/vnd.graphviz")
io.WriteString(w, formatters.BuildDot(ircmap.Slice(), displayAll).String())
case "/map.png":
if !doPng {
w.WriteHeader(404)
return
}
w.Header().Set("Content-Type", "image/png")
cmd := exec.Command("dot", dotOptions...)
out, err := formatters.BuildPNG(formatters.BuildDot(ircmap.Slice(), displayAll), cmd)
if err != nil {
w.WriteHeader(500)
} else {
w.Write(out)
}
case "/map.txt":
io.WriteString(w, ircmap.String())
default:
w.WriteHeader(404)
}
}
func GetMap(url string, timeout time.Duration) (*irctree.Servers, error) {
client := http.Client{
CheckRedirect: func(_ *http.Request, _ []*http.Request) error {
return errors.New("Unexpected Redirect")
},
Jar: nil,
Timeout: timeout,
}
resp, err := client.Get(url)
if err != nil {
return nil, err
}
servs, err := decodeMap(resp.Body)
resp.Body.Close()
return servs, err
}