sqlite_backups 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5befa74d4c48692314c28f8ca62ba631a87faaff37c3ee83ee792267127bad8b
-  data.tar.gz: 04e12d29cc5c7285487fd63374d598d5d2e687c113f2dd560a556701d4a80d40
+  metadata.gz: 8409eedf7805943faaa85ebb184dee513b9ffaaa4c67b1eb1163368b0f63e062
+  data.tar.gz: d8ccbd330bf4214f838b8a7011747001b44d8ecfdf22d738412246b908d945d2
 SHA512:
-  metadata.gz: 6f610817967e118afb908977f1d14c06cacf0400ed9cb202792914181326430f6a8f219f4f8eb74990d04430c3d09d29035ca66ca9c03ce1972564dbbcd0a9b1
-  data.tar.gz: 6bd671346cc80c2a21a160d32e74b2d09a20a2b91645416edbeea89490979edd1b72e9177d36631b006d39d999e1054c43078e551b18f77b6c8d861851a5f207
+  metadata.gz: 9ec82f3d967d7d5f57d4315c83dc3b11f3fdd2aaf9930b0551f2ee98b4b41cbe09856628dfbe929e33fb0f0bce9df9c2916662c4b7a08876598d7ec345f2d380
+  data.tar.gz: 7eb1b2a3841b38ec722d23748d6caf2600244354a916bbebc421cc484cd72fadaeca7251aa5e344d6cb6cfc8c6d9239dcfd242c899d7a3b5782304e5ad56f867

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|
+            gz.each_line { out.write(it) }
+          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.1"
 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.1
 platform: ruby
 authors:
 - Nick Pezza