Simon Marsh
166234fa89
- Add golang templates to move HTML rendering out of the go code where possible - Add an endpoint for serving static files - Add URL escaping for servers and targets
102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// binary data
|
|
//go:generate go-bindata -prefix bindata -o bindata.go bindata/...
|
|
|
|
type settingType struct {
|
|
servers []string
|
|
domain string
|
|
proxyPort int
|
|
whoisServer string
|
|
listen string
|
|
dnsInterface string
|
|
netSpecificMode string
|
|
titleBrand string
|
|
navBarBrand string
|
|
}
|
|
|
|
var setting settingType
|
|
|
|
func main() {
|
|
var settingDefault = settingType{
|
|
servers: []string{""},
|
|
proxyPort: 8000,
|
|
whoisServer: "whois.verisign-grs.com",
|
|
listen: ":5000",
|
|
dnsInterface: "asn.cymru.com",
|
|
titleBrand: "Bird-lg Go",
|
|
navBarBrand: "Bird-lg Go",
|
|
}
|
|
|
|
if env := os.Getenv("BIRDLG_SERVERS"); env != "" {
|
|
settingDefault.servers = strings.Split(env, ",")
|
|
}
|
|
if env := os.Getenv("BIRDLG_DOMAIN"); env != "" {
|
|
settingDefault.domain = env
|
|
}
|
|
if env := os.Getenv("BIRDLG_PROXY_PORT"); env != "" {
|
|
var err error
|
|
if settingDefault.proxyPort, err = strconv.Atoi(env); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
if env := os.Getenv("BIRDLG_WHOIS"); env != "" {
|
|
settingDefault.whoisServer = env
|
|
}
|
|
if env := os.Getenv("BIRDLG_LISTEN"); env != "" {
|
|
settingDefault.listen = env
|
|
}
|
|
if env := os.Getenv("BIRDLG_DNS_INTERFACE"); env != "" {
|
|
settingDefault.dnsInterface = env
|
|
}
|
|
if env := os.Getenv("BIRDLG_NET_SPECIFIC_MODE"); env != "" {
|
|
settingDefault.netSpecificMode = env
|
|
}
|
|
if env := os.Getenv("BIRDLG_TITLE_BRAND"); env != "" {
|
|
settingDefault.titleBrand = env
|
|
settingDefault.navBarBrand = env
|
|
}
|
|
if env := os.Getenv("BIRDLG_NAVBAR_BRAND"); env != "" {
|
|
settingDefault.navBarBrand = env
|
|
}
|
|
|
|
serversPtr := flag.String("servers", strings.Join(settingDefault.servers, ","), "server name prefixes, separated by comma")
|
|
domainPtr := flag.String("domain", settingDefault.domain, "server name domain suffixes")
|
|
proxyPortPtr := flag.Int("proxy-port", settingDefault.proxyPort, "port bird-lgproxy is running on")
|
|
whoisPtr := flag.String("whois", settingDefault.whoisServer, "whois server for queries")
|
|
listenPtr := flag.String("listen", settingDefault.listen, "address bird-lg is listening on")
|
|
dnsInterfacePtr := flag.String("dns-interface", settingDefault.dnsInterface, "dns zone to query ASN information")
|
|
netSpecificModePtr := flag.String("net-specific-mode", settingDefault.netSpecificMode, "network specific operation mode, [(none)|dn42]")
|
|
titleBrandPtr := flag.String("title-brand", settingDefault.titleBrand, "prefix of page titles in browser tabs")
|
|
navBarBrandPtr := flag.String("navbar-brand", settingDefault.navBarBrand, "brand to show in the navigation bar")
|
|
flag.Parse()
|
|
|
|
if *serversPtr == "" {
|
|
panic("no server set")
|
|
} else if *domainPtr == "" {
|
|
panic("no base domain set")
|
|
}
|
|
|
|
setting = settingType{
|
|
strings.Split(*serversPtr, ","),
|
|
*domainPtr,
|
|
*proxyPortPtr,
|
|
*whoisPtr,
|
|
*listenPtr,
|
|
*dnsInterfacePtr,
|
|
strings.ToLower(*netSpecificModePtr),
|
|
*titleBrandPtr,
|
|
*navBarBrandPtr,
|
|
}
|
|
|
|
ImportTemplates()
|
|
webServerStart()
|
|
}
|