chore: possibility to set env variables for S3 endpoints

highcard-dev · Jan 29, 2025 · df10417 · df10417
1 parent 9455f7f
commit df10417
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/cmd/backup.go b/cmd/backup.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"fmt"
+	"os"
 
 	"github.com/highcard-dev/daemon/internal/core/ports"
 	"github.com/highcard-dev/daemon/internal/core/services"
@@ -52,9 +53,15 @@ var BackupCmd = &cobra.Command{
 }
 
 func init() {
+
+	accessKeyEnv := os.Getenv("DRUID_S3_ACCESS_KEY")
+	secretKeyEnv := os.Getenv("DRUID_S3_SECRET_KEY")
+	bucketEnv := os.Getenv("DRUID_S3_BUCKET")
+	endpointEnv := os.Getenv("DRUID_S3_ENDPOINT")
+
 	BackupCmd.Flags().StringVarP(&tgzTempDir, "tgz-temp-dir", "", "", "Temporary location for the backup tgz file")
-	BackupCmd.Flags().StringVarP(&s3AccessKey, "s3-access-key", "", "", "S3 access key")
-	BackupCmd.Flags().StringVarP(&s3SecretKey, "s3-secret-key", "", "", "S3 secret key")
-	BackupCmd.Flags().StringVarP(&s3Bucket, "s3-bucket", "", "", "S3 bucket")
-	BackupCmd.Flags().StringVarP(&s3Endpoint, "s3-endpoint", "", "", "S3 endpoint")
+	BackupCmd.Flags().StringVarP(&s3AccessKey, "s3-access-key", "", accessKeyEnv, "S3 access key. You can also set the DRUID_S3_ACCESS_KEY environment variable")
+	BackupCmd.Flags().StringVarP(&s3SecretKey, "s3-secret-key", "", secretKeyEnv, "S3 secret key. You can also set the DRUID_S3_SECRET_KEY environment variable")
+	BackupCmd.Flags().StringVarP(&s3Bucket, "s3-bucket", "", bucketEnv, "S3 bucket. You can also set the DRUID_S3_BUCKET environment variable")
+	BackupCmd.Flags().StringVarP(&s3Endpoint, "s3-endpoint", "", endpointEnv, "S3 endpoint. You can also set the DRUID_S3_ENDPOINT environment variable")
 }