refactor: env vars for override of file locations

Author: iPromKnight

src/internal/config/config.go

@@ -4,6 +4,7 @@ import (
 	"gopkg.in/yaml.v3"
 	"os"
 	"rclone-manager/internal/constants"
+	"rclone-manager/internal/environment"
 )
 
 type Config struct {
@@ -22,7 +23,8 @@ type Config struct {
 }
 
 func LoadConfig() (*Config, error) {
-	data, err := os.ReadFile(constants.YAMLPath)
+	yamlPath := environment.GetEnvWithFallback(constants.YAMLPathEnvVar, constants.DefaultYAMLPath)
+	data, err := os.ReadFile(yamlPath)
 	if err != nil {
 		return nil, err
 	}

src/internal/constants/constants.go

@@ -28,6 +28,8 @@ const (
 
 // Constants data files
 const (
-	YAMLPath   = "/data/config.yaml"
-	RcloneConf = "/data/rclone.conf"
+	YAMLPathEnvVar    = "RCLONE_MANAGER_CONFIG_YAML"
+	DefaultYAMLPath   = "/data/config.yaml"
+	RcloneConfEnvVar  = "RCLONE_MANAGER_RCLONE_CONF"
+	DefaultRcloneConf = "/data/rclone.conf"
 )

src/internal/environment/environment.go

@@ -25,6 +25,14 @@ func PrepareEnvironment(envVars map[string]string) []string {
 	return mapToSlice(envMap)
 }
 
+func GetEnvWithFallback(key, defaultValue string) string {
+	value := os.Getenv(key)
+	if value == "" {
+		return defaultValue
+	}
+	return value
+}
+
 func splitEnv(env string) []string {
 	parts := make([]string, 2)
 	idx := strings.Index(env, "=")

src/internal/rclone_manager/rclone.go

@@ -4,6 +4,7 @@ import (
 	"github.com/rs/zerolog"
 	"rclone-manager/internal/config"
 	"rclone-manager/internal/constants"
+	"rclone-manager/internal/environment"
 	"rclone-manager/internal/mount_manager"
 	"rclone-manager/internal/serve_manager"
 	"rclone-manager/internal/watcher"
@@ -41,9 +42,12 @@ func InitializeRClone(logger zerolog.Logger) {
 		go mount_manager.InitializeMountEndpoints(conf, logger, &processLock)
 	}
 
+	yamlPath := environment.GetEnvWithFallback(constants.YAMLPathEnvVar, constants.DefaultYAMLPath)
+	rcloneConfPath := environment.GetEnvWithFallback(constants.RcloneConfEnvVar, constants.DefaultRcloneConf)
+
 	filesToWatch := []string{
-		constants.YAMLPath,
-		constants.RcloneConf,
+		yamlPath,
+		rcloneConfPath,
 	}
 
 	watcher.StartNewFileWatcher(filesToWatch, reloadConfig, logger)