121 lines
5.3 KiB
Go
121 lines
5.3 KiB
Go
package config
|
|
|
|
import (
|
|
"flag"
|
|
"regexp"
|
|
|
|
"github.com/gorilla/securecookie"
|
|
|
|
"golang.org/x/oauth2"
|
|
oauth2FB "golang.org/x/oauth2/facebook"
|
|
oauth2Github "golang.org/x/oauth2/github"
|
|
)
|
|
|
|
const (
|
|
PWDHostnameRegex = "[0-9]{1,3}-[0-9]{1,3}-[0-9]{1,3}-[0-9]{1,3}"
|
|
PortRegex = "[0-9]{1,5}"
|
|
AliasnameRegex = "[0-9|a-z|A-Z|-]*"
|
|
AliasSessionRegex = "[0-9|a-z|A-Z]{8}"
|
|
AliasGroupRegex = "(" + AliasnameRegex + ")-(" + AliasSessionRegex + ")"
|
|
PWDHostPortGroupRegex = "^.*ip(" + PWDHostnameRegex + ")(?:-?(" + PortRegex + "))?(?:\\..*)?$"
|
|
AliasPortGroupRegex = "^.*pwd" + AliasGroupRegex + "(?:-?(" + PortRegex + "))?\\..*$"
|
|
)
|
|
|
|
var NameFilter = regexp.MustCompile(PWDHostPortGroupRegex)
|
|
var AliasFilter = regexp.MustCompile(AliasPortGroupRegex)
|
|
|
|
var PortNumber, Key, Cert, SessionsFile, PWDContainerName, L2ContainerName, L2Subdomain, HashKey, SSHKeyPath, L2RouterIP, DindVolumeSize, CookieHashKey, CookieBlockKey, DefaultDinDImage, DefaultSessionDuration string
|
|
var UseLetsEncrypt, ExternalDindVolume, NoWindows bool
|
|
var LetsEncryptCertsDir string
|
|
var MaxLoadAvg float64
|
|
var ForceTLS bool
|
|
var Providers map[string]*oauth2.Config
|
|
var SecureCookie *securecookie.SecureCookie
|
|
var AdminToken string
|
|
|
|
var GithubClientID, GithubClientSecret string
|
|
var FacebookClientID, FacebookClientSecret string
|
|
var DockerClientID, DockerClientSecret string
|
|
|
|
var PlaygroundDomain string
|
|
|
|
func ParseFlags() {
|
|
flag.StringVar(&LetsEncryptCertsDir, "letsencrypt-certs-dir", "/certs", "Path where let's encrypt certs will be stored")
|
|
flag.BoolVar(&UseLetsEncrypt, "letsencrypt-enable", false, "Enabled let's encrypt tls certificates")
|
|
flag.BoolVar(&ForceTLS, "tls", false, "Use TLS to connect to docker daemons")
|
|
flag.StringVar(&PortNumber, "port", "3000", "Port number")
|
|
flag.StringVar(&Key, "tls-server-key", "./pwd/server-key.pem", "Server key for SSL")
|
|
flag.StringVar(&Cert, "tls-cert", "./pwd/server.pem", "Give a SSL cert")
|
|
flag.StringVar(&SessionsFile, "save", "./pwd/sessions", "Tell where to store sessions file")
|
|
flag.StringVar(&PWDContainerName, "name", "pwd", "Container name used to run PWD (used to be able to connect it to the networks it creates)")
|
|
flag.StringVar(&L2ContainerName, "l2", "l2", "Container name used to run L2 Router")
|
|
flag.StringVar(&L2RouterIP, "l2-ip", "", "Host IP address for L2 router ping response")
|
|
flag.StringVar(&L2Subdomain, "l2-subdomain", "direct", "Subdomain to the L2 Router")
|
|
flag.StringVar(&HashKey, "hash_key", "salmonrosado", "Hash key to use for cookies")
|
|
flag.StringVar(&DindVolumeSize, "dind-volume-size", "5G", "Dind volume folder size")
|
|
flag.BoolVar(&NoWindows, "win-disable", false, "Disable windows instances")
|
|
flag.BoolVar(&ExternalDindVolume, "dind-external-volume", false, "Use external dind volume though XFS volume driver")
|
|
flag.Float64Var(&MaxLoadAvg, "maxload", 100, "Maximum allowed load average before failing ping requests")
|
|
flag.StringVar(&SSHKeyPath, "ssh_key_path", "", "SSH Private Key to use")
|
|
flag.StringVar(&CookieHashKey, "cookie-hash-key", "", "Hash key to use to validate cookies")
|
|
flag.StringVar(&CookieBlockKey, "cookie-block-key", "", "Block key to use to encrypt cookies")
|
|
flag.StringVar(&DefaultDinDImage, "default-dind-image", "franela/dind", "Default DinD image to use if not specified otherwise")
|
|
flag.StringVar(&DefaultSessionDuration, "default-session-duration", "4h", "Default session duration if not specified otherwise")
|
|
|
|
flag.StringVar(&GithubClientID, "oauth-github-client-id", "", "Github OAuth Client ID")
|
|
flag.StringVar(&GithubClientSecret, "oauth-github-client-secret", "", "Github OAuth Client Secret")
|
|
|
|
flag.StringVar(&FacebookClientID, "oauth-facebook-client-id", "", "Facebook OAuth Client ID")
|
|
flag.StringVar(&FacebookClientSecret, "oauth-facebook-client-secret", "", "Facebook OAuth Client Secret")
|
|
|
|
flag.StringVar(&DockerClientID, "oauth-docker-client-id", "", "Docker OAuth Client ID")
|
|
flag.StringVar(&DockerClientSecret, "oauth-docker-client-secret", "", "Docker OAuth Client Secret")
|
|
|
|
flag.StringVar(&PlaygroundDomain, "playground-domain", "localhost", "Domain to use for the playground")
|
|
flag.StringVar(&AdminToken, "admin-token", "", "Token to validate admin user for admin endpoints")
|
|
|
|
flag.Parse()
|
|
|
|
SecureCookie = securecookie.New([]byte(CookieHashKey), []byte(CookieBlockKey))
|
|
|
|
registerOAuthProviders()
|
|
}
|
|
|
|
func registerOAuthProviders() {
|
|
Providers = map[string]*oauth2.Config{}
|
|
if GithubClientID != "" && GithubClientSecret != "" {
|
|
conf := &oauth2.Config{
|
|
ClientID: GithubClientID,
|
|
ClientSecret: GithubClientSecret,
|
|
Scopes: []string{"user:email"},
|
|
Endpoint: oauth2Github.Endpoint,
|
|
}
|
|
|
|
Providers["github"] = conf
|
|
}
|
|
if FacebookClientID != "" && FacebookClientSecret != "" {
|
|
conf := &oauth2.Config{
|
|
ClientID: FacebookClientID,
|
|
ClientSecret: FacebookClientSecret,
|
|
Scopes: []string{"email", "public_profile"},
|
|
Endpoint: oauth2FB.Endpoint,
|
|
}
|
|
|
|
Providers["facebook"] = conf
|
|
}
|
|
if DockerClientID != "" && DockerClientSecret != "" {
|
|
oauth2.RegisterBrokenAuthHeaderProvider(".id.docker.com")
|
|
conf := &oauth2.Config{
|
|
ClientID: DockerClientID,
|
|
ClientSecret: DockerClientSecret,
|
|
Scopes: []string{"openid"},
|
|
Endpoint: oauth2.Endpoint{
|
|
AuthURL: "https://id.docker.com/id/oauth/authorize/",
|
|
TokenURL: "https://id.docker.com/id/oauth/token",
|
|
},
|
|
}
|
|
|
|
Providers["docker"] = conf
|
|
}
|
|
}
|