pg_tasks 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c836d8fe98ae36885760913a6f2b714b9aab5357
+  data.tar.gz: de74fe7101993253af9ff86ed5006e8b724f7a71
+SHA512:
+  metadata.gz: 62c6bdea7f5012861c484d295f05dc57df2332145be0457e4c8920dbda1904bad6525bb07e5fb314816d00f7c1b9271748b15804c76c4cab8d5d52a44427c0f8
+  data.tar.gz: 62a5064c34861ceff2a6726ac1e828bf8feb3f6149c47782022c9df75a8765bc5f1914fe009f3b47f924270498d3fdd23bddafb5e3fea35dff2068ace6dd3421

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2015 Thomas Schank
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,19 @@
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+
+require 'rdoc/task'
+
+RDoc::Task.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'PgTasks'
+  rdoc.options << '--line-numbers'
+  rdoc.rdoc_files.include('README.rdoc')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+
+Bundler::GemHelper.install_tasks
+

data/lib/pg_tasks/railtie.rb

@@ -0,0 +1,10 @@
+require 'pg_tasks'
+require 'rails'
+module PgTasks
+  class Railtie < Rails::Railtie
+    railtie_name :pg_tasks
+    rake_tasks do
+      load 'tasks/pg_tasks.rake'
+    end
+  end
+end

data/lib/pg_tasks/version.rb

@@ -0,0 +1,3 @@
+module PgTasks
+  VERSION = "1.0.0"
+end

data/lib/pg_tasks.rb

@@ -0,0 +1,136 @@
+require 'active_record/tasks/database_tasks'
+require 'active_record/tasks/postgresql_database_tasks'
+
+module PgTasks
+
+  require 'pg_tasks/railtie' if defined?(Rails)
+
+  DEFAULT_BINARY_DATA_FILE_NAME = 'data.pgbin'
+  DEFAULT_BINARY_STRUCTURE_AND_DATA_FILE_NAME = 'structure_and_data.pgbin'
+
+  class << self
+
+    def reload!
+      load Rails.root.join(__FILE__)
+    end
+
+    %w(data_dump data_restore).each do |method_name|
+      define_method method_name  do |filename = nil|
+        ActiveRecord::Tasks::DatabaseTasks \
+          .perform_pg_db_task_for_config_and_filename \
+          method_name,  current_config,
+          filename_or_default_binary_data_file(filename)
+      end
+    end
+
+    %w(structure_and_data_dump structure_and_data_restore).each do |method_name|
+      define_method method_name  do |filename = nil|
+      ActiveRecord::Tasks::DatabaseTasks \
+        .perform_pg_db_task_for_config_and_filename \
+        method_name, current_config,
+        filename_or_default_binary_structure_and_data_file(filename)
+      end
+    end
+
+    def truncate_tables
+      ActiveRecord::Base.connection.tap do |connection|
+        connection.tables.reject { |tn| tn == 'schema_migrations' }
+        .join(', ').tap do |tables|
+          connection.execute " TRUNCATE TABLE #{tables} CASCADE; "
+        end
+      end
+    end
+
+    private
+
+    def current_config
+      ActiveRecord::Tasks::DatabaseTasks.current_config
+    end
+
+    def filename_or_default_binary_data_file(filename)
+      (filename.present? && filename) || \
+        File.join(ActiveRecord::Tasks::DatabaseTasks.db_dir,
+                  DEFAULT_BINARY_DATA_FILE_NAME)
+    end
+
+    def filename_or_default_binary_structure_and_data_file(filename)
+      (filename.present? && filename) || \
+        File.join(ActiveRecord::Tasks::DatabaseTasks.db_dir,
+                  DEFAULT_BINARY_STRUCTURE_AND_DATA_FILE_NAME)
+    end
+
+  end
+end
+
+module ActiveRecord
+  module Tasks
+    class PostgreSQLDatabaseTasks
+
+      def data_dump(filename)
+        set_psql_env
+        command = 'pg_dump -F c -a -T schema_migrations -x -O -f ' \
+          "#{Shellwords.escape(filename)} " \
+          "#{Shellwords.escape(configuration['database'])}"
+        raise 'Error during data_dump' unless Kernel.system(command)
+        $stdout.puts "The data of '#{configuration['database']} " \
+          "has been dumped to '#{filename}'"
+      end
+
+      def data_restore(filename)
+        set_psql_env
+        command = 'pg_restore --disable-triggers -a -x -O -d ' \
+          "#{Shellwords.escape(configuration['database'])} " \
+          "#{Shellwords.escape(filename)}"
+        raise 'Error during data_restore ' unless Kernel.system(command)
+        $stdout.puts "Data from '#{filename}' has been restored to \
+                        '#{configuration['database']}'"
+      end
+
+      def structure_and_data_dump(filename)
+        set_psql_env
+        command = "pg_dump -F c -x -O -f \
+        #{Shellwords.escape(filename)} \
+        #{Shellwords.escape(configuration['database'])}"
+        unless Kernel.system(command)
+          raise 'Error during structure_and_data_dump'
+        else
+          $stdout.puts 'Structure and data of ' \
+            "'#{configuration['database']}' has been dumped to '#{filename}'"
+        end
+      end
+
+      def structure_and_data_restore(filename)
+        set_psql_env
+        command = 'pg_restore --disable-triggers -x -O -d ' \
+          "#{Shellwords.escape(configuration['database'])} " \
+          "#{Shellwords.escape(filename)}"
+        unless Kernel.system(command)
+          raise 'Error during structure_and_data_restore '
+        else
+          $stdout.puts "Structure and data of '#{configuration['database']}' " \
+            "has been restored to '#{filename}'"
+        end
+      end
+
+    end
+  end
+end
+
+module ActiveRecord
+  module Tasks
+    module DatabaseTasks
+
+      def perform_pg_db_task_for_config_and_filename(task_name, *arguments)
+        configuration = arguments.first
+        filename = arguments.delete_at 1
+        class_for_adapter(configuration['adapter']) \
+          .new(*arguments).send task_name, filename
+      rescue ActiveRecord::NoDatabaseError
+        $stderr.puts "Database '#{configuration['database']}' does not exist"
+      rescue Exception => error
+        $stderr.puts error, *(error.backtrace)
+      end
+
+    end
+  end
+end

data/lib/tasks/pg_tasks.rake

@@ -0,0 +1,37 @@
+require 'pg_tasks'
+
+namespace :db do
+
+  namespace :pg do
+
+    task truncate_tables: [:environment, :load_config] do
+      PgTasks.truncate_tables
+    end
+
+    namespace :data do
+
+      task dump: [:environment, :load_config] do
+        PgTasks.data_dump ENV['FILE']
+      end
+
+      task restore: [:environment, :load_config] do
+        PgTasks.data_restore ENV['FILE']
+      end
+
+    end
+
+    namespace :structure_and_data do
+
+      task dump: [:environment, :load_config] do
+        PgTasks.structure_and_data_dump ENV['FILE']
+      end
+
+      task restore: [:environment, :load_config] do
+        PgTasks.structure_and_data_restore ENV['FILE']
+      end
+
+    end
+
+  end
+
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification
+name: pg_tasks
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Thomas Schank
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-01-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 4.1.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 5.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 4.1.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 5.0.0
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- DrTom@schank.ch
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- Rakefile
+- lib/pg_tasks.rb
+- lib/pg_tasks/railtie.rb
+- lib/pg_tasks/version.rb
+- lib/tasks/pg_tasks.rake
+homepage: https://github.com/DrTom/rails_pg-tasks
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: PostgreSQL Tasks and Functions for Ruby on Rails
+test_files: []
+has_rdoc: