Support configuring replica region and endpoint

Author: MatheusRich

lib/litestream.rb

@@ -33,7 +33,7 @@ def initialize
     end
   end
 
-  mattr_writer :username, :password, :queue, :replica_bucket, :replica_key_id, :replica_access_key, :systemctl_command
+  mattr_writer :username, :password, :queue, :replica_bucket, :replica_region, :replica_endpoint, :replica_key_id, :replica_access_key, :systemctl_command
 
   class << self
     def verify!(database_path)
@@ -77,6 +77,14 @@ def replica_bucket
       @@replica_bucket || configuration.replica_bucket
     end
 
+    def replica_region
+      @@replica_region
+    end
+
+    def replica_endpoint
+      @@replica_endpoint
+    end
+
     def replica_key_id
       @@replica_key_id || configuration.replica_key_id
     end

lib/litestream/commands.rb

@@ -125,6 +125,8 @@ def execute(command, argv = {}, database = nil, async: false, tabled_output: fal
 
       def prepare(command, argv = {}, database = nil)
         ENV["LITESTREAM_REPLICA_BUCKET"] ||= Litestream.replica_bucket
+        ENV["LITESTREAM_REPLICA_REGION"] ||= Litestream.replica_region
+        ENV["LITESTREAM_REPLICA_ENDPOINT"] ||= Litestream.replica_endpoint
         ENV["LITESTREAM_ACCESS_KEY_ID"] ||= Litestream.replica_key_id
         ENV["LITESTREAM_SECRET_ACCESS_KEY"] ||= Litestream.replica_access_key
 

lib/tasks/litestream_tasks.rake

@@ -2,6 +2,8 @@ namespace :litestream do
   desc "Print the ENV variables needed for the Litestream config file"
   task env: :environment do
     puts "LITESTREAM_REPLICA_BUCKET=#{Litestream.replica_bucket}"
+    puts "LITESTREAM_REPLICA_REGION=#{Litestream.replica_region}"
+    puts "LITESTREAM_REPLICA_ENDPOINT=#{Litestream.replica_endpoint}"
     puts "LITESTREAM_ACCESS_KEY_ID=#{Litestream.replica_key_id}"
     puts "LITESTREAM_SECRET_ACCESS_KEY=#{Litestream.replica_access_key}"
 

test/tasks/test_litestream_tasks.rb

@@ -26,6 +26,8 @@ def test_env_task_when_nothing_configured_prints
 
       assert_equal <<~TXT, out
         LITESTREAM_REPLICA_BUCKET=
+        LITESTREAM_REPLICA_REGION=
+        LITESTREAM_REPLICA_ENDPOINT=
         LITESTREAM_ACCESS_KEY_ID=
         LITESTREAM_SECRET_ACCESS_KEY=
       TXT