twins/config.go

155 lines
3.2 KiB
Go

package main
import (
"crypto/tls"
"errors"
"io/ioutil"
"log"
"net/url"
"os"
"regexp"
"strconv"
"strings"
"github.com/kballard/go-shellquote"
"github.com/yookoala/gofast"
"gopkg.in/yaml.v3"
)
type pathConfig struct {
// Path to match
Path string
// Resource to serve
Root string
Proxy string
Command string
// Request input
Input string
// Request sensitive input
SensitiveInput string
// List directory entries
ListDirectory bool
// Content type
Type string
// FastCGI server address
FastCGI string
r *regexp.Regexp
cmd []string
}
type hostConfig struct {
Cert string
Key string
Paths []*pathConfig
cert *tls.Certificate
}
type serverConfig struct {
Listen string
Hosts map[string]*hostConfig
DisableSize bool
hostname string
port int
fcgiPools map[string]gofast.ConnFactory
}
var config *serverConfig
func readconfig(configPath string) error {
if configPath == "" {
return errors.New("file unspecified")
}
if _, err := os.Stat(configPath); os.IsNotExist(err) {
return errors.New("file not found")
}
configData, err := ioutil.ReadFile(configPath)
if err != nil {
return err
}
var newConfig *serverConfig
err = yaml.Unmarshal(configData, &newConfig)
if err != nil {
return err
}
config = newConfig
if config.Listen == "" {
log.Fatal("listen address must be specified")
}
split := strings.Split(config.Listen, ":")
if len(split) != 2 {
config.hostname = config.Listen
config.Listen += ":1965"
} else {
config.hostname = split[0]
config.port, err = strconv.Atoi(split[1])
if err != nil {
log.Fatalf("invalid port specified: %s", err)
}
}
config.fcgiPools = make(map[string]gofast.ConnFactory)
for hostname, host := range config.Hosts {
if host.Cert == "" || host.Key == "" {
log.Fatal("a certificate must be specified for each domain (gemini requires TLS for all connections)")
}
cert, err := tls.LoadX509KeyPair(host.Cert, host.Key)
if err != nil {
log.Fatalf("failed to load certificate: %s", err)
}
host.cert = &cert
for _, serve := range host.Paths {
if serve.Path == "" {
log.Fatal("path must be specified in serve entry")
} else if (serve.Root != "" && (serve.Proxy != "" || serve.Command != "")) ||
(serve.Proxy != "" && (serve.Root != "" || serve.Command != "")) ||
(serve.Command != "" && (serve.Root != "" || serve.Proxy != "")) {
log.Fatal("only one root, proxy or command resource may specified for a path")
}
if serve.Path[0] == '^' {
serve.r = regexp.MustCompile(serve.Path)
}
if serve.FastCGI != "" {
if serve.Root == "" {
log.Fatalf("root must be specified to use fastcgi resource %s of path %s%s", serve.FastCGI, hostname, serve.Path)
}
if config.fcgiPools[serve.FastCGI] == nil {
f, err := url.Parse(serve.FastCGI)
if err != nil {
log.Fatalf("failed to parse fastcgi resource %s: %s", serve.FastCGI, err)
}
config.fcgiPools[serve.FastCGI] = gofast.SimpleConnFactory(f.Scheme, f.Host+f.Path)
}
} else if serve.Command != "" {
serve.cmd, err = shellquote.Split(serve.Command)
if err != nil {
log.Fatalf("failed to parse command %s: %s", serve.cmd, err)
}
}
}
}
return nil
}