osmosis 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8a003b88c32b0dcd67c6ef409c0ac5c560d923882faa7436123f21a642e6def5
-  data.tar.gz: f81d8284cda6f0cd05f64c25ca3eb5aef1bf700ffe2932aed84604017ceb9ef5
+  metadata.gz: '033500944d7bcb9c17fb602c3381e842b060f17233096a6286884cc2bdb72c54'
+  data.tar.gz: f05ea8a885a2560d4ec01fec4d21082fc29f665d714b30cfeeab9c7b634b299f
 SHA512:
-  metadata.gz: 74e79675ce1f5575c27a4972d9f48732e4d02d90710c71705b73ce0c50fe392da830d428a6889e0fc866a0b596b7fda5ec9faafe053bf27e9e74c07ee890de98
-  data.tar.gz: d53f62ee450a0746bd36f523df58eeafb4039f2247a627dff92ccc14c4f8d78171cfef3dde806147c0f179b4a8dcada243ebce3a16ebf3014cdad84d28849e2f
+  metadata.gz: 271e0df3804c9cc4ec4d66cd77c32f4538c09da2e9e9390bae58422715eb9c61b5ad201e51f148ddf4586fca8403a3e429a264aac6c980c0c3cb95e41bdf65cd
+  data.tar.gz: f47ce3ddd08b8a025bf7da08267edb6e2a23edfe7ec37cd392feb3908a2fdf22e9e771d86cbc62ab2209fe2c591bd8fddeefccbbdbdf38321571f56d78099758

data/README.md

@@ -16,9 +16,21 @@ If bundler is not being used to manage dependencies, install the gem by executin
 
 Osmosis provides you with different rake tasks that simplify the management of your database.
 
-### Database restore
+### Database import
 
+The import command takes the `.dump` file provided and restores it to the PostgreSQL database. Any pre-existing data is dropped.
 
+```bash
+bundle exec rails db:import <filename>
+```
+
+### Database export
+
+The export command exports the PostgreSQL database into a `.dump` file. If the file already exists, it is overwritten.
+
+```bash
+bundle exec rails db:export <filename>
+```
 
 ## Development
 
@@ -28,7 +40,7 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/osmosis.
+Bug reports and pull requests are welcome on GitHub at https://github.com/code-fabrik/osmosis.
 
 ## License
 

data/lib/osmosis/command.rb

@@ -1,35 +1,51 @@
+require 'expect'
 require 'ostruct'
+require 'pty'
 
 module Osmosis
   class Command
-    def self.restore(argv)
-      files = Dir.glob("./*.dump")
-      case files.size
-      when 0
-        return [
-          OpenStruct.new(type: :print, command: "No backups found for *.dump")
-        ]
-      when 1
-        file = files.first
-        host = ActiveRecord::Base.connection_db_config.host
-        database = ActiveRecord::Base.connection_db_config.database
-        username = ActiveRecord::Base.connection_db_config.configuration_hash[:username]
-        password = ActiveRecord::Base.connection_db_config.configuration_hash[:password]
-        cmd = "PGPASSWORD=#{password} pg_restore -U #{username} -h #{host} -d #{database} -cO --if-exists -w < #{file}"
-        Rake::Task["db:drop"].invoke
-        Rake::Task["db:create"].invoke
-        return [
-          OpenStruct.new(type: :rake, command: 'db:drop'),
-          OpenStruct.new(type: :rake, command: 'db:create'),
-          OpenStruct.new(type: :cmd, command: cmd),
-          OpenStruct.new(type: :print, command: "Restored database '#{database}'"),
-        ]
-      else
-        return [
-          OpenStruct.new(type: :print, command: "Too many backups found for *.dump:"),
-          OpenStruct.new(type: :print, command: ' ' + files.join("\n ")),
-        ]
+    def self.import(argv)
+      file = argv[0]
+      if file.nil?
+        return [OpenStruct.new(type: :print, command: "Please provide a filename parameter: rails db:import <filename>")]
       end
+      if !File.exist?(file)
+        return [OpenStruct.new(type: :print, command: "File #{file} does not exist")]
+      end
+      host = ActiveRecord::Base.connection_db_config.host
+      database = ActiveRecord::Base.connection_db_config.database
+      username = ActiveRecord::Base.connection_db_config.configuration_hash[:username]
+      password = ActiveRecord::Base.connection_db_config.configuration_hash[:password]
+      cmd = "PGPASSWORD=#{password} pg_restore -U #{username} -h #{host} -d #{database} -cO --if-exists -w < #{file}"
+      return [
+        OpenStruct.new(type: :rake, command: 'db:drop'),
+        OpenStruct.new(type: :rake, command: 'db:create'),
+        OpenStruct.new(type: :cmd, command: cmd),
+        OpenStruct.new(type: :print, command: "Imported database '#{database}'"),
+      ]
+    end
+
+    def self.export(argv)
+      file = argv[0]
+      if file.nil?
+        return [OpenStruct.new(type: :print, command: "Please provide a filename parameter: rails db:export <filename>")]
+      end
+      host = ActiveRecord::Base.connection_db_config.host
+      database = ActiveRecord::Base.connection_db_config.database
+      username = ActiveRecord::Base.connection_db_config.configuration_hash[:username]
+      password = ActiveRecord::Base.connection_db_config.configuration_hash[:password]
+      cmd = "PGPASSWORD=#{password} pg_dump -Fc --no-acl --no-owner -h #{host} -U #{username} -w #{database} > #{file}"
+      return [
+        OpenStruct.new(type: :cmd, command: cmd),
+        OpenStruct.new(type: :print, command: "Exported database '#{database}'"),
+      ]
+    end
+
+    def self.connect(argv)
+      cmd = "PGPASSWORD=#{password} psql -U #{username} -h #{host} -d #{database}"
+      return [
+        OpenStruct.new(type: :pty, command: cmd)
+      ]
     end
 
     def self.run(commands)
@@ -40,6 +56,25 @@ module Osmosis
           system cmd.command
         elsif cmd.type == :print
           puts cmd.command
+        elsif cmd.type == :pty
+          PTY.spawn(cmd.command) do |stdout, stdin, pid|
+            trap("INT") { Process.kill("INT", pid) }
+
+            begin
+              loop do
+                # Read output from psql
+                stdout.each { |line| puts line}
+
+                # Check if there's any input from the user
+                if IO.select([STDIN], nil, nil, 0.1)
+                  user_input = STDIN.gets.chomp
+                  stdin.puts user_input
+                end
+              end
+            rescue Errno::EIO
+              puts "psql session ended"
+            end
+          end
         else
           raise StandardError, "Unknown command type #{cmd.type}"
         end

data/lib/osmosis/tasks/db/connect.rake

@@ -0,0 +1,7 @@
+namespace :db do
+  desc "Connects to the application database"
+  task connect: :environment do
+    commands = Osmosis::Command.connect(ARGV)
+    Osmosis::Command.run(commands)
+  end
+end

data/lib/osmosis/tasks/db/import.rake

@@ -0,0 +1,13 @@
+namespace :db do
+  desc "Imports a database from a backup"
+  task import: :environment do
+    commands = Osmosis::Command.import(ARGV)
+    Osmosis::Command.run(commands)
+  end
+
+  desc "Exports a database to a backup"
+  task import: :environment do
+    commands = Osmosis::Command.export(ARGV)
+    Osmosis::Command.run(commands)
+  end
+end

data/lib/osmosis/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Osmosis
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: osmosis
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Lukas_Skywalker
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-20 00:00:00.000000000 Z
+date: 2024-10-08 00:00:00.000000000 Z
 dependencies: []
 description: This powerful database utility library simplifies and automates common
   database tasks, including exports, imports, and data migrations. With its intuitive
@@ -28,7 +28,8 @@ files:
 - lib/osmosis.rb
 - lib/osmosis/command.rb
 - lib/osmosis/railtie.rb
-- lib/osmosis/tasks/db/restore.rake
+- lib/osmosis/tasks/db/connect.rake
+- lib/osmosis/tasks/db/import.rake
 - lib/osmosis/version.rb
 - sig/osmosis.rbs
 homepage: https://github.com/code-fabrik/osmosis

data/lib/osmosis/tasks/db/restore.rake

@@ -1,7 +0,0 @@
-namespace :db do
-  desc "Restores the database from a backup"
-  task restore: :environment do
-    commands = Osmosis::Command.restore(ARGV)
-    Osmosis::Command.run(commands)
-  end
-end