cli.rb in backy_rb-0.1.4

- old
+ new
@@ -1,87 +1,175 @@
\ No newline at end of file
+require "thor"
+require "yaml"
+
+module Backy
+  class CLI < Thor
+    def initialize(*args)
+      Logger.log("Initializing Backy CLI...")
+      super
+      Backy.setup
+    end
+
+    desc "dump", "Dump the database"
+    def dump
+      Backy::PgDump.new.call
+      Logger.success("Database dumped successfully.")
+    end
+
+    desc "push", "Push dump to S3 and delete local file"
+    def push
+      file_name = Backy::PgDump.new.call
+      save_to_s3(file_name)
+      File.delete(file_name) if File.exist?(file_name)
+      Logger.sucess("Dump pushed to S3 and local file deleted.")
+    end
+
+    desc "upload", "Upload a specific file to S3"
+    method_option :file, aliases: "-f", desc: "File to upload"
+    def upload
+      file_name = options[:file]
+      validate_file!(file_name)
+      save_to_s3(file_name)
+      Logger.success("File uploaded to S3.")
+    end
+
+    desc "download", "Download a file from S3"
+    method_option :file, aliases: "-f", desc: "File to download"
+    def download
+      file_name = options[:file]
+      validate_file!(file_name, should_exist: false)
+      load_from_s3_if_missing(file_name)
+      Logger.success("File downloaded from S3.")
+    end
+
+    desc "list", "List all dumps"
+    def list
+      list = Backy::List.new.call
+      list.each { |item| say "#{item.local? ? "local" : "     "} #{item.remote? ? "remote" : "      "} : #{item.dump_file}", :yellow }
+      say_instructions_if_any(list)
+    end
+
+    desc "restore", "Restore a database from a dump"
+    method_option :file, aliases: "-f", desc: "File to restore from"
+    def restore
+      file_name = find_file_for_restore(options[:file])
+      pg_restore(load_from_s3_if_missing(file_name)) if file_name
+      Logger.success("Database restored from #{file_name}.")
+    end
+
+    private
+
+    def validate_file!(file_name, should_exist: true)
+      if file_name.nil? || (should_exist ? !File.exist?(file_name) : File.exist?(file_name))
+        Logger.error("Invalid file specified.")
+        exit 1
+      end
+    end
+
+    def find_file_for_restore(template_file_name)
+      file_name = Backy::List.new.call.reverse.find { |item| item.dump_file.starts_with?(template_file_name) }&.dump_file if template_file_name.present?
+      Logger.info "Please set the file to restore from." if file_name.nil?
+      file_name
+    end
+
+    def say_instructions_if_any(list)
+      return unless list.any?
+      Logger.info("\nTo restore a backup, run:")
+      Logger.log("  backy restore --file FILE_NAME")
+    end
+
+    def load_from_s3_if_missing(file_name)
+      Backy::S3Load.new(file_name: file_name).call
+    end
+
+    def save_to_s3(file_name)
+      Backy::S3Save.new(file_name: file_name).call
+    end
+  end
+end