osmosis 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e26e371d8fc36471c8162c5f15e996bfa801a35009a49ae5a23871582b55a1c5
-  data.tar.gz: 8a087b4531e00b93febd10fcd44684239c4876dee89bbd9d60fa9e4e6d5babe4
+  metadata.gz: '033500944d7bcb9c17fb602c3381e842b060f17233096a6286884cc2bdb72c54'
+  data.tar.gz: f05ea8a885a2560d4ec01fec4d21082fc29f665d714b30cfeeab9c7b634b299f
 SHA512:
-  metadata.gz: 9d8e6d0b9b2fdd4bd7a577aeee586e803310a72ff2a73cc7d4f327fd29f575a61d2553dc017c46b0f1de4f8102bafa65235226682b7b0294130bd195839b3b47
-  data.tar.gz: 2615c4c7a1e7c442a32f779612c29f6b46fb2e419892fee63f90abcc6f100ee78c4687914d3197a03a7e418bfd6dc451cbab20e269f7134ce4bd61650c1f6df6
+  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.2"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,29 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: osmosis
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Lukas_Skywalker
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-20 00:00:00.000000000 Z
-dependencies:
-- !ruby/object:Gem::Dependency
-  name: thor
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '1.3'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '1.3'
+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
   interface and robust feature set, it enables developers to manage databases more
@@ -42,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