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
|
package main
import (
"context"
"errors"
"fmt"
sr "tildegit.org/tjp/sliderule"
"tildegit.org/tjp/sliderule/contrib/cgi"
"tildegit.org/tjp/sliderule/contrib/fs"
"tildegit.org/tjp/sliderule/gemini"
"tildegit.org/tjp/sliderule/gemini/gemtext/atomconv"
"tildegit.org/tjp/sliderule/logging"
"tildegit.org/tjp/sliderule/spartan"
)
func buildSpartanServers(servers []Server, config *Configuration) ([]sr.Server, error) {
_, info, _, errlog := Loggers(config)
groups := map[string][]*Server{}
for i := range servers {
addr := fmt.Sprintf("%s:%d", servers[i].IP.String(), servers[i].Port)
grp, ok := groups[addr]
if !ok {
groups[addr] = []*Server{&servers[i]}
} else {
groups[addr] = append(grp, &servers[i])
}
}
result := []sr.Server{}
for addr, configs := range groups {
_ = info.Log("msg", "starting spartan server", "addr", addr)
var handler sr.Handler
if len(configs) == 1 {
handler = routes(*configs[0])
} else {
mapping := map[string]sr.Handler{}
for _, config := range configs {
router := routes(*config)
for _, hostname := range config.Hostnames {
mapping[hostname] = router
}
}
handler = sr.VirtualHosts(mapping, sr.HandlerFunc(func(_ context.Context, _ *sr.Request) *sr.Response {
return spartan.ClientError(errors.New("Proxy request refused"))
}))
}
var hostname string
for _, conf := range configs {
if len(conf.Hostnames) > 0 {
hostname = conf.Hostnames[0]
break
}
}
sptnsrv, err := spartan.NewServer(
context.Background(),
hostname,
"tcp",
addr,
logging.LogRequests(info)(handler),
errlog,
)
if err != nil {
return nil, err
}
result = append(result, sptnsrv)
}
return result, nil
}
func addSpartanRoute(router *sr.Router, route RouteDirective) {
switch route.Type {
case "static":
addSpartanStaticRoute(router, route)
case "cgi":
buildAndAddRoute(router, route, func(route RouteDirective) sr.Handler {
return cgi.SpartanCGIDirectory(route.FsPath, route.URLPath, route.Modifiers.ExecCmd)
})
case "git":
//TODO
}
}
func addSpartanStaticRoute(router *sr.Router, route RouteDirective) {
buildAndAddRoute(router, route, func(route RouteDirective) sr.Handler {
handlers := []sr.Handler{}
if route.Modifiers.Exec {
handlers = append(handlers, cgi.SpartanCGIDirectory(route.FsPath, route.URLPath, route.Modifiers.ExecCmd))
}
handlers = append(handlers, fs.SpartanFileHandler(route.FsPath, route.URLPath))
if route.Modifiers.DirDefault != "" {
handlers = append(
handlers,
fs.SpartanDirectoryDefault(route.FsPath, route.URLPath, route.Modifiers.DirDefault),
)
}
if route.Modifiers.DirList {
handlers = append(handlers, fs.SpartanDirectoryListing(route.FsPath, route.URLPath, nil))
}
handler := sr.FallthroughHandler(handlers...)
if route.Modifiers.AutoAtom {
handler = atomconv.Auto(handler)
}
return handler
})
}
|
