116 lines
2.2 KiB
Go
Executable File
116 lines
2.2 KiB
Go
Executable File
package config
|
|
|
|
import (
|
|
"local/rproxy3/storage/packable"
|
|
"log"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func GetPort() string {
|
|
v := packable.NewString()
|
|
conf.Get(nsConf, flagPort, v)
|
|
return ":" + strings.TrimPrefix(v.String(), ":")
|
|
}
|
|
|
|
func GetRoutes() map[string]string {
|
|
v := packable.NewString()
|
|
conf.Get(nsConf, flagRoutes, v)
|
|
m := make(map[string]string)
|
|
for _, v := range strings.Split(v.String(), ",") {
|
|
if len(v) == 0 {
|
|
return m
|
|
}
|
|
from := v[:strings.Index(v, ":")]
|
|
to := v[strings.Index(v, ":")+1:]
|
|
m[from] = to
|
|
}
|
|
return m
|
|
}
|
|
|
|
func GetTCP() (string, bool) {
|
|
v := packable.NewString()
|
|
conf.Get(nsConf, flagTCP, v)
|
|
tcpAddr := v.String()
|
|
return tcpAddr, notEmpty(tcpAddr)
|
|
}
|
|
|
|
func GetSSL() (string, string, bool) {
|
|
v := packable.NewString()
|
|
conf.Get(nsConf, flagCert, v)
|
|
certPath := v.String()
|
|
conf.Get(nsConf, flagKey, v)
|
|
keyPath := v.String()
|
|
return certPath, keyPath, notEmpty(certPath, keyPath)
|
|
}
|
|
|
|
func GetAuth() (string, string, bool) {
|
|
v := packable.NewString()
|
|
conf.Get(nsConf, flagUser, v)
|
|
user := v.String()
|
|
conf.Get(nsConf, flagPass, v)
|
|
pass := v.String()
|
|
return user, pass, notEmpty(user, pass)
|
|
}
|
|
|
|
func notEmpty(s ...string) bool {
|
|
for i := range s {
|
|
if s[i] == "" || s[i] == "/dev/null" {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func GetRate() (int, int) {
|
|
r := packable.NewString()
|
|
conf.Get(nsConf, flagRate, r)
|
|
b := packable.NewString()
|
|
conf.Get(nsConf, flagBurst, b)
|
|
|
|
rate, err := strconv.Atoi(r.String())
|
|
if err != nil {
|
|
log.Printf("illegal rate: %v", err)
|
|
rate = 5
|
|
}
|
|
burst, _ := strconv.Atoi(b.String())
|
|
if err != nil {
|
|
log.Printf("illegal burst: %v", err)
|
|
burst = 5
|
|
}
|
|
|
|
return rate, burst
|
|
}
|
|
|
|
func GetTimeout() int {
|
|
t := packable.NewString()
|
|
conf.Get(nsConf, flagTimeout, t)
|
|
|
|
timeout, err := strconv.Atoi(t.String())
|
|
if err != nil || timeout == 5 {
|
|
return 5
|
|
}
|
|
|
|
return timeout
|
|
}
|
|
|
|
func GetRewrites(hostMatch string) map[string]string {
|
|
v := packable.NewString()
|
|
conf.Get(nsConf, flagRewrites, v)
|
|
m := make(map[string]string)
|
|
for _, v := range strings.Split(v.String(), ",") {
|
|
vs := strings.Split(v, ":")
|
|
if len(v) < 3 {
|
|
continue
|
|
}
|
|
host := vs[0]
|
|
if host != hostMatch {
|
|
continue
|
|
}
|
|
from := vs[1]
|
|
to := strings.Join(vs[2:], ":")
|
|
m[from] = to
|
|
}
|
|
return m
|
|
}
|