141 lines
3.6 KiB
Go
141 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
sr "tildegit.org/tjp/sliderule"
|
|
"tildegit.org/tjp/sliderule/contrib/cgi"
|
|
"tildegit.org/tjp/sliderule/contrib/fs"
|
|
"tildegit.org/tjp/sliderule/gemini/gemtext/atomconv"
|
|
"tildegit.org/tjp/sliderule/logging"
|
|
"tildegit.org/tjp/sliderule/spartan"
|
|
"tildegit.org/tjp/syw"
|
|
)
|
|
|
|
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":
|
|
addSpartanGitRoute(router, route)
|
|
}
|
|
}
|
|
|
|
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
|
|
})
|
|
}
|
|
|
|
func addSpartanGitRoute(router *sr.Router, route RouteDirective) {
|
|
buildAndAddRoute(router, route, func(route RouteDirective) sr.Handler {
|
|
subrouter := syw.SpartanRouter(route.FsPath, route.Modifiers.Templates)
|
|
handler := sr.HandlerFunc(func(ctx context.Context, request *sr.Request) *sr.Response {
|
|
reqclone := cloneRequest(request)
|
|
reqclone.Path = strings.TrimPrefix(reqclone.Path, route.URLPath)
|
|
|
|
handler, params := subrouter.Match(reqclone)
|
|
if handler == nil {
|
|
return nil
|
|
}
|
|
return handler.Handle(context.WithValue(ctx, sr.RouteParamsKey, params), request)
|
|
})
|
|
if route.Modifiers.AutoAtom {
|
|
handler = atomconv.Auto(handler)
|
|
}
|
|
return handler
|
|
})
|
|
}
|