sqlite_backups 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5befa74d4c48692314c28f8ca62ba631a87faaff37c3ee83ee792267127bad8b
-  data.tar.gz: 04e12d29cc5c7285487fd63374d598d5d2e687c113f2dd560a556701d4a80d40
+  metadata.gz: a3496932c9ee0f9e5a993e464fe56ac638b856731709154ef5ba5fd46b2c9523
+  data.tar.gz: a106d4690c2825d8c0333bc88f35b6eb8b187c2e5ba845a5d8311c2ebc44b205
 SHA512:
-  metadata.gz: 6f610817967e118afb908977f1d14c06cacf0400ed9cb202792914181326430f6a8f219f4f8eb74990d04430c3d09d29035ca66ca9c03ce1972564dbbcd0a9b1
-  data.tar.gz: 6bd671346cc80c2a21a160d32e74b2d09a20a2b91645416edbeea89490979edd1b72e9177d36631b006d39d999e1054c43078e551b18f77b6c8d861851a5f207
+  metadata.gz: 287eb0b1e1953cbe7c053e5bb95c012ec1721f0a4707e9c88689d98cc35d5381eb15195a74e305c3f0dc76cb2339afab02322071b7d9ee4aa37ee82f57d1c307
+  data.tar.gz: a253b37d81bbbc79bcdc4ee47ce4c8d0ea7d5c147b423fee9247b186db9fcc30592149d184f5c7540383b3eb1fa074fb8374386d1e21591d29b12155665806f7

data/app/controllers/backups/backups_controller.rb

@@ -4,7 +4,7 @@ module Backups
 
     def show
       files = Backup.where(database: params[:name]).map do
-        { date: it.created_at, key: it.file.key }
+        { date: it.created_at, key: it.file.key, size: it.file.byte_size }
       end
 
       render json: { name: params[:name], files: }

data/lib/backups/restore.rb

@@ -10,19 +10,58 @@ module Backups
       raise StandardError, "No backups found" if backups["files"].blank?
       raise StandardError, "File not found" unless service.exist?(key)
 
-      File.open(path, "wb") do |file|
-        file.write(ActiveSupport::Gzip.decompress(service.download(key)))
-      end
+      download
+      decompress
+
+      File.delete(path) if File.exist?(path)
+      FileUtils.mv(download_path, path)
+      File.delete("#{download_path}.gz") if File.exist?("#{download_path}.gz")
+      File.delete(download_path) if File.exist?(download_path)
+
+      puts "Restore complete!"
     end
 
     private
 
     attr_reader :name
 
+    def download
+      return if File.exist?("#{download_path}.gz") || File.exist?(download_path)
+
+      downloaded_total = 0
+      File.open("#{download_path}.gz", "wb") do |file|
+        service.download(key) do |chunk|
+          file.write(chunk)
+          downloaded_total += chunk.bytesize
+          print "\rDownloading: #{((downloaded_total / size.to_f) * 100.0).to_i}%"
+        end
+      end
+      puts
+    end
+
+    def decompress
+      return if File.exist?(download_path)
+
+      puts "Decompressing..."
+      File.open("#{download_path}.gz", "rb") do |file|
+        File.open(download_path, "wb") do |out|
+          Zlib::GzipReader.wrap(file) do |gz|
+            out.write(gz.read)
+          end
+        end
+      end
+
+      File.delete("#{download_path}.gz")
+    end
+
     def path
       Backups.databases(env_name: Rails.env)[name.to_s]
     end
 
+    def download_path
+      Rails.root.join("tmp/#{key}")
+    end
+
     def service
       ActiveStorage::Blob.services.fetch(Backups.storage_service)
     end
@@ -32,14 +71,22 @@ module Backups
     end
 
     def key
-      @key ||=
+      backup_file[:key]
+    end
+
+    def size
+      backup_file[:size]
+    end
+
+    def backup_file
+      @backup_file ||=
         if backups["files"].count == 1
-          backups["files"].first["key"]
+          backups["files"].first
         else
           CLI::UI.ask("Pick a backup to restore", options:).then do |date|
-            backups["files"].find { it["date"].to_s == date }["key"]
+            backups["files"].find { it["date"].to_s == date }
           end
-        end
+        end.to_h.with_indifferent_access
     end
 
     def options

data/lib/backups/version.rb

@@ -1,3 +1,3 @@
 module Backups
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sqlite_backups
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Nick Pezza