phew big stuff!!!!

internal/config/config.go

@@ -35,6 +35,7 @@ type Config struct {
 	TemplateConfig  *TemplateConfig `yaml:"template"`
 	AccountsConfig  *AccountsConfig `yaml:"accounts"`
 	MediaConfig     *MediaConfig    `yaml:"media"`
+	StorageConfig   *StorageConfig  `yaml:"storage"`
 }
 
 // FromFile returns a new config from a file, or an error if something goes amiss.
@@ -62,6 +63,9 @@ func Empty() *Config {
 	return &Config{
 		DBConfig:       &DBConfig{},
 		TemplateConfig: &TemplateConfig{},
+		AccountsConfig: &AccountsConfig{},
+		MediaConfig:    &MediaConfig{},
+		StorageConfig:  &StorageConfig{},
 	}
 }
 
@@ -147,6 +151,36 @@ func (c *Config) ParseCLIFlags(f KeyedFlags) {
 	if f.IsSet(fn.AccountsRequireApproval) {
 		c.AccountsConfig.RequireApproval = f.Bool(fn.AccountsRequireApproval)
 	}
+
+	// media flags
+	if c.MediaConfig.MaxImageSize == 0 || f.IsSet(fn.MediaMaxImageSize) {
+		c.MediaConfig.MaxImageSize = f.Int(fn.MediaMaxImageSize)
+	}
+
+	if c.MediaConfig.MaxVideoSize == 0 || f.IsSet(fn.MediaMaxVideoSize) {
+		c.MediaConfig.MaxVideoSize = f.Int(fn.MediaMaxVideoSize)
+	}
+
+	// storage flags
+	if c.StorageConfig.Backend == "" || f.IsSet(fn.StorageBackend) {
+		c.StorageConfig.Backend = f.String(fn.StorageBackend)
+	}
+
+	if c.StorageConfig.BasePath == "" || f.IsSet(fn.StorageBasePath) {
+		c.StorageConfig.BasePath = f.String(fn.StorageBasePath)
+	}
+
+	if c.StorageConfig.ServeProtocol == "" || f.IsSet(fn.StorageServeProtocol) {
+		c.StorageConfig.ServeProtocol = f.String(fn.StorageServeProtocol)
+	}
+
+	if c.StorageConfig.ServeHost == "" || f.IsSet(fn.StorageServeHost) {
+		c.StorageConfig.ServeHost = f.String(fn.StorageServeHost)
+	}
+
+	if c.StorageConfig.ServeBasePath == "" || f.IsSet(fn.StorageServeBasePath) {
+		c.StorageConfig.ServeBasePath = f.String(fn.StorageServeBasePath)
+	}
 }
 
 // KeyedFlags is a wrapper for any type that can store keyed flags and give them back.
@@ -166,15 +200,27 @@ type Flags struct {
 	ConfigPath               string
 	Host                     string
 	Protocol                 string
+
 	DbType                   string
 	DbAddress                string
 	DbPort                   string
 	DbUser                   string
 	DbPassword               string
 	DbDatabase               string
+
 	TemplateBaseDir          string
+
 	AccountsOpenRegistration string
 	AccountsRequireApproval  string
+
+	MediaMaxImageSize        string
+	MediaMaxVideoSize        string
+
+	StorageBackend           string
+	StorageBasePath          string
+	StorageServeProtocol     string
+	StorageServeHost         string
+	StorageServeBasePath     string
 }
 
 // GetFlagNames returns a struct containing the names of the various flags used for
@@ -186,15 +232,27 @@ func GetFlagNames() Flags {
 		ConfigPath:               "config-path",
 		Host:                     "host",
 		Protocol:                 "protocol",
+
 		DbType:                   "db-type",
 		DbAddress:                "db-address",
 		DbPort:                   "db-port",
 		DbUser:                   "db-user",
 		DbPassword:               "db-password",
 		DbDatabase:               "db-database",
+
 		TemplateBaseDir:          "template-basedir",
+
 		AccountsOpenRegistration: "accounts-open-registration",
 		AccountsRequireApproval:  "accounts-require-approval",
+
+		MediaMaxImageSize:        "media-max-image-size",
+		MediaMaxVideoSize:        "media-max-video-size",
+
+		StorageBackend:           "storage-backend",
+		StorageBasePath:          "storage-base-path",
+		StorageServeProtocol:     "storage-serve-protocol",
+		StorageServeHost:         "storage-serve-host",
+		StorageServeBasePath:     "storage-serve-base-path",
 	}
 }
 
@@ -207,14 +265,26 @@ func GetEnvNames() Flags {
 		ConfigPath:               "GTS_CONFIG_PATH",
 		Host:                     "GTS_HOST",
 		Protocol:                 "GTS_PROTOCOL",
+
 		DbType:                   "GTS_DB_TYPE",
 		DbAddress:                "GTS_DB_ADDRESS",
 		DbPort:                   "GTS_DB_PORT",
 		DbUser:                   "GTS_DB_USER",
 		DbPassword:               "GTS_DB_PASSWORD",
 		DbDatabase:               "GTS_DB_DATABASE",
+
 		TemplateBaseDir:          "GTS_TEMPLATE_BASEDIR",
+
 		AccountsOpenRegistration: "GTS_ACCOUNTS_OPEN_REGISTRATION",
 		AccountsRequireApproval:  "GTS_ACCOUNTS_REQUIRE_APPROVAL",
+
+		MediaMaxImageSize:        "GTS_MEDIA_MAX_IMAGE_SIZE",
+		MediaMaxVideoSize:        "GTS_MEDIA_MAX_VIDEO_SIZE",
+
+		StorageBackend:           "GTS_STORAGE_BACKEND",
+		StorageBasePath:          "GTS_STORAGE_BASE_PATH",
+		StorageServeProtocol:     "GTS_STORAGE_SERVE_PROTOCOL",
+		StorageServeHost:         "GTS_STORAGE_SERVE_HOST",
+		StorageServeBasePath:     "GTS_STORAGE_SERVE_BASE_PATH",
 	}
 }