litesream fix, add test case

Author: rubys

lib/generators/templates/litestream.rake.erb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 LITESTREAM_CONFIG = ENV["LITESTREAM_CONFIG"] || Rails.root.join("tmp/litestream.yml").to_s
 
 LITESTREAM_TEMPLATE = <<-EOF
@@ -36,7 +37,7 @@ namespace :litestream do
     end
 
     @dbs.each do |db|
-      next if File.exist?(db) or !ENV["BUCKET_NAME"]
+      next if File.exist?(db) || !ENV["BUCKET_NAME"]
       system "litestream restore -config #{LITESTREAM_CONFIG} -if-replica-exists #{db}"
       exit $?.exitstatus unless $?.exitstatus == 0
     end
@@ -45,8 +46,12 @@ namespace :litestream do
   task :run do
     require "shellwords"
 
-    exec(*%w[bundle exec litestream replicate -config],
-      LITESTREAM_CONFIG, "-exec", Shellwords.join(ARGV[1..-1]))
+    if ENV["BUCKET_NAME"]
+      exec(*%w[bundle exec litestream replicate -config],
+        LITESTREAM_CONFIG, "-exec", Shellwords.join(ARGV[1..]))
+    else
+      exec(*ARGV[1..])
+    end
   end
 end
 

test/base.rb

@@ -130,6 +130,16 @@ def check_puma
     assert_equal expected, results
   end
 
+  def check_raketask
+    results = IO.read("lib/tasks/litestream.rake")
+
+    IO.write("#{@results}/litestream.rake", results) if @capture
+
+    expected = IO.read("#{@results}/litestream.rake")
+
+    assert_equal expected, results
+  end
+
   def teardown
     return if ENV["TEST_KEEP"]
     Dir.chdir ".."

test/results/litestream/litestream.rake

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+LITESTREAM_CONFIG = ENV["LITESTREAM_CONFIG"] || Rails.root.join("tmp/litestream.yml").to_s
+
+LITESTREAM_TEMPLATE = <<-EOF
+# This is the configuration file for litestream.
+#
+# For more details, see: https://litestream.io/reference/config/
+#
+dbs:
+<% for db in @dbs -%>
+  - path: <%= db %>
+    replicas:
+      - type: s3
+        endpoint: $AWS_ENDPOINT_URL_S3
+        bucket: $BUCKET_NAME
+        path: storage/<%= File.basename(db) %>
+        access-key-id: $AWS_ACCESS_KEY_ID
+        secret-access-key: $AWS_SECRET_ACCESS_KEY
+<% end -%>
+EOF
+
+namespace :litestream do
+  task prepare: "db:load_config" do
+    require "erubi"
+
+    @dbs =
+      ActiveRecord::Base
+        .configurations
+        .configs_for(env_name: "production", include_hidden: true)
+        .select { |config| [ "sqlite3", "litedb" ].include? config.adapter }
+        .map(&:database)
+
+    result = eval(Erubi::Engine.new(LITESTREAM_TEMPLATE).src)
+
+    unless File.exist?(LITESTREAM_CONFIG) && File.read(LITESTREAM_CONFIG) == result
+      File.write(LITESTREAM_CONFIG, result)
+    end
+
+    @dbs.each do |db|
+      next if File.exist?(db) || !ENV["BUCKET_NAME"]
+      system "litestream restore -config #{LITESTREAM_CONFIG} -if-replica-exists #{db}"
+      exit $?.exitstatus unless $?.exitstatus == 0
+    end
+  end
+
+  task :run do
+    require "shellwords"
+
+    if ENV["BUCKET_NAME"]
+      exec(*%w[bundle exec litestream replicate -config],
+        LITESTREAM_CONFIG, "-exec", Shellwords.join(ARGV[1..]))
+    else
+      exec(*ARGV[1..])
+    end
+  end
+end
+
+namespace :db do
+  task prepare: "litestream:prepare"
+end

test/test_litestream.rb

@@ -14,5 +14,6 @@ def test_litestream
     check_dockerfile
     check_entrypoint
     check_toml
+    check_raketask
   end
 end