package args import ( "errors" "flag" "os" "strings" "golang.org/x/oauth2" ) type Arguments struct { DatabasePath string TemplatePath string StaticPath string CloudflareToken string CloudflareZone string Migrate bool Scheduler bool Port int Server bool OauthConfig *oauth2.Config OauthUserInfoURI string Dns bool DnsRecursion []string DnsPort int } func GetArgs() (*Arguments, error) { databasePath := flag.String("database-path", "./hatecomputers.db", "Path to the SQLite database") templatePath := flag.String("template-path", "./templates", "Path to the template directory") staticPath := flag.String("static-path", "./static", "Path to the static directory") scheduler := flag.Bool("scheduler", false, "Run scheduled jobs via cron") migrate := flag.Bool("migrate", false, "Run the migrations") port := flag.Int("port", 8080, "Port to listen on") server := flag.Bool("server", false, "Run the server") dns := flag.Bool("dns", false, "Run DNS resolver") dnsRecursion := flag.String("dns-recursion", "1.1.1.1:53,1.0.0.1:53", "Comma separated list of DNS resolvers") dnsPort := flag.Int("dns-port", 8053, "Port to listen on for DNS resolver") flag.Parse() cloudflareToken := os.Getenv("CLOUDFLARE_TOKEN") cloudflareZone := os.Getenv("CLOUDFLARE_ZONE") oauthClientID := os.Getenv("OAUTH_CLIENT_ID") oauthClientSecret := os.Getenv("OAUTH_CLIENT_SECRET") oauthScopes := os.Getenv("OAUTH_SCOPES") oauthAuthURL := os.Getenv("OAUTH_AUTH_URL") oauthTokenURL := os.Getenv("OAUTH_TOKEN_URL") oauthRedirectURI := os.Getenv("OAUTH_REDIRECT_URI") oauthUserInfoURI := os.Getenv("OAUTH_USER_INFO_URI") envVars := [][]string{ {cloudflareToken, "CLOUDFLARE_TOKEN"}, {cloudflareZone, "CLOUDFLARE_ZONE"}, {oauthClientID, "OAUTH_CLIENT_ID"}, {oauthClientSecret, "OAUTH_CLIENT_SECRET"}, {oauthScopes, "OAUTH_SCOPES"}, {oauthAuthURL, "OAUTH_AUTH_URL"}, {oauthTokenURL, "OAUTH_TOKEN_URL"}, {oauthRedirectURI, "OAUTH_REDIRECT_URI"}, {oauthUserInfoURI, "OAUTH_USER_INFO_URI"}, } for _, envVar := range envVars { if envVar[0] == "" { return nil, errors.New("please set the " + envVar[1] + " environment variable") } } oauthConfig := &oauth2.Config{ ClientID: oauthClientID, ClientSecret: oauthClientSecret, Scopes: strings.Split(oauthScopes, ","), Endpoint: oauth2.Endpoint{ AuthURL: oauthAuthURL, TokenURL: oauthTokenURL, }, RedirectURL: oauthRedirectURI, } arguments := &Arguments{ DatabasePath: *databasePath, TemplatePath: *templatePath, StaticPath: *staticPath, CloudflareToken: cloudflareToken, CloudflareZone: cloudflareZone, Port: *port, Server: *server, Migrate: *migrate, Scheduler: *scheduler, Dns: *dns, DnsRecursion: strings.Split(*dnsRecursion, ","), DnsPort: *dnsPort, OauthConfig: oauthConfig, OauthUserInfoURI: oauthUserInfoURI, } return arguments, nil }