pg_power 1.0.0

data/lib/pg_power/schema_dumper/foreigner_methods.rb

@@ -0,0 +1,58 @@
+# Provides methods to extend {ActiveRecord::SchemaDumper} to dump
+# foreign keys.
+module PgPower::SchemaDumper::ForeignerMethods
+  # Hooks {ActiveRecord::SchemaDumper#table} method to dump foreign keys.
+  def tables_with_foreign_keys(stream)
+    tables_without_foreign_keys(stream)
+
+    table_names = @connection.tables.sort
+    table_names += get_non_public_schema_table_names.sort
+
+    table_names.sort.each do |table|
+      next if ['schema_migrations', ignore_tables].flatten.any? do |ignored|
+        case ignored
+        when String; table == ignored
+        when Regexp; table =~ ignored
+        else
+          raise StandardError, 'ActiveRecord::SchemaDumper.ignore_tables accepts an array of String and / or Regexp values.'
+        end
+      end
+      foreign_keys(table, stream)
+    end
+  end
+
+
+  # Finds all foreign keys on passed table and writes appropriated
+  # statements to stream.
+  def foreign_keys(table_name, stream)
+    if (foreign_keys = @connection.foreign_keys(table_name)).any?
+      add_foreign_key_statements = foreign_keys.map do |foreign_key|
+        statement_parts = [ ('add_foreign_key ' + foreign_key.from_table.inspect) ]
+        statement_parts << foreign_key.to_table.inspect
+        statement_parts << (':name => ' + foreign_key.options[:name].inspect)
+
+        if foreign_key.options[:column] != "#{foreign_key.to_table.singularize}_id"
+          statement_parts << (':column => ' + foreign_key.options[:column].inspect)
+        end
+        if foreign_key.options[:primary_key] != 'id'
+          statement_parts << (':primary_key => ' + foreign_key.options[:primary_key].inspect)
+        end
+        if foreign_key.options[:dependent].present?
+          statement_parts << (':dependent => ' + foreign_key.options[:dependent].inspect)
+        end
+
+        # Always exclude the index
+        #  If an index was created in a migration, it will get dumped to the schema
+        #  separately from the foreign key.  This will raise an exception if
+        #  add_foreign_key is run without :exclude_index => true.
+        statement_parts << (':exclude_index => true')
+
+        '  ' + statement_parts.join(', ')
+      end
+
+      stream.puts add_foreign_key_statements.sort.join("\n")
+      stream.puts
+    end
+  end
+  private :foreign_keys
+end

data/lib/pg_power/schema_dumper/schema_methods.rb

@@ -0,0 +1,51 @@
+# Extends ActiveRecord::SchemaDumper class to dump schemas other than "public"
+# and tables from those schemas.
+module PgPower::SchemaDumper::SchemaMethods
+  # * Dumps schemas.
+  # * Dumps tables from public schema using native #tables method.
+  # * Dumps tables from schemas other than public.
+  def tables_with_schemas(stream)
+    schemas(stream)
+    tables_without_schemas(stream)
+    non_public_schema_tables(stream)
+  end
+
+  # Generates code to create schemas.
+  def schemas(stream)
+    # Don't create "public" schema since it exists by default.
+    schema_names = PgPower::Tools.schemas - ["public", "information_schema"]
+    schema_names.each do |schema_name|
+      schema(schema_name, stream)
+    end
+    stream << "\n"
+  end
+  private :schemas
+
+  # Generates code to create schema.
+  def schema(schema_name, stream)
+    stream << "  create_schema \"#{schema_name}\"\n"
+  end
+  private :schema
+
+  # Dumps tables from schemas other than public
+  def non_public_schema_tables(stream)
+    get_non_public_schema_table_names.each do |name|
+      table(name, stream)
+    end
+  end
+  private :non_public_schema_tables
+
+  # Returns a sorted list of non-public schema tables
+  # Usage:
+  #   get_non_public_schema_table_names # => ['demography.cities','demography.countries','politics.members']
+  def get_non_public_schema_table_names
+    result = @connection.query(<<-SQL, 'SCHEMA')
+      SELECT schemaname || '.' || tablename
+      FROM pg_tables
+      WHERE schemaname NOT IN ('pg_catalog', 'information_schema', 'public')
+      ORDER BY schemaname, tablename
+    SQL
+    result.flatten
+  end
+  private :get_non_public_schema_table_names
+end

data/lib/pg_power/tools.rb

@@ -0,0 +1,56 @@
+module PgPower
+  # Provides utility methods to work with PostgreSQL databases.
+  # Usage:
+  #   PgPower::Tools.create_schema "services"  # => create new PG schema "services"
+  #   PgPower::Tools.create_schema "nets"
+  #   PgPower::Tools.drop_schema "services"    # => remove the schema
+  #   PgPower::Tools.schemas                   # => ["public", "information_schema", "nets"]
+  #   PgPower::Tools.move_table_to_schema :computers, :nets
+  module Tools
+    extend self
+
+    # Creates PostgreSQL schema
+    def create_schema(schema_name)
+      sql = %{CREATE SCHEMA "#{schema_name}"}
+      connection.execute sql
+    end
+
+    # Drops PostgreSQL schema
+    def drop_schema(schema_name)
+      sql = %{DROP SCHEMA "#{schema_name}"}
+      connection.execute sql
+    end
+
+    # Returns an array of existing schemas.
+    def schemas
+      sql = "SELECT nspname FROM pg_namespace WHERE nspname !~ '^pg_.*' order by nspname"
+      connection.query(sql).flatten
+    end
+
+    # Move table to another schema without loosing data, indexes or constraints.
+    # @param [String] table table name (schema prefix is allowed)
+    # @param [String] new_schema schema where table should be moved to
+    def move_table_to_schema(table, new_schema)
+      schema, table = to_schema_and_table(table)
+      sql = %{ALTER TABLE "#{schema}"."#{table}" SET SCHEMA "#{new_schema}"}
+      connection.execute sql
+    end
+
+
+
+    # Return databas connections
+    def connection
+      ActiveRecord::Base.connection
+    end
+    private :connection
+
+    # Extract schema name and table name from qualified table name
+    # @param [String, Symbol] table_name table name
+    # @return [Array[String, String]] schema and table
+    def to_schema_and_table(table_name)
+      table, schema = table_name.to_s.split(".", 2).reverse
+      schema ||= "public"
+      [schema, table]
+    end
+  end
+end

data/lib/pg_power/version.rb

@@ -0,0 +1,4 @@
+module PgPower
+  # Version of pg_power gem.
+  VERSION = "1.0.0"
+end

data/lib/tasks/pg_power_tasks.rake

@@ -0,0 +1,4 @@
+# desc "Explaining what the task does"
+# task :pg_power do
+#   # Task goes here
+# end

metadata

@@ -0,0 +1,213 @@
+--- !ruby/object:Gem::Specification 
+name: pg_power
+version: !ruby/object:Gem::Version 
+  hash: 23
+  prerelease: 
+  segments: 
+  - 1
+  - 0
+  - 0
+  version: 1.0.0
+platform: ruby
+authors: 
+- Potapov Sergey
+- Arthur Shagall
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-09-05 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id001
+  name: pg
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id002
+  name: rspec-rails
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 5
+        segments: 
+        - 3
+        - 1
+        version: "3.1"
+  version_requirements: *id003
+  name: rails
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id004
+  name: rcov
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id005
+  name: yard
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id006
+  name: metrical
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id007
+  name: jeweler
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id008 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id008
+  name: ruby-debug
+  prerelease: false
+description: ActiveRecord extensions for PostgreSQL. Provides useful tools and ability to create/drop schemas in migrations.
+email: 
+- blake131313@gmail.com
+- arthur.shagall@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- README.markdown
+- lib/core_ext/active_record/connection_adapters/abstract/schema_statements.rb
+- lib/core_ext/active_record/connection_adapters/postgresql_adapter.rb
+- lib/core_ext/active_record/schema_dumper.rb
+- lib/pg_power.rb
+- lib/pg_power/connection_adapters.rb
+- lib/pg_power/connection_adapters/abstract_adapter.rb
+- lib/pg_power/connection_adapters/abstract_adapter/comment_methods.rb
+- lib/pg_power/connection_adapters/abstract_adapter/foreigner_methods.rb
+- lib/pg_power/connection_adapters/abstract_adapter/index_methods.rb
+- lib/pg_power/connection_adapters/abstract_adapter/schema_methods.rb
+- lib/pg_power/connection_adapters/foreign_key_definition.rb
+- lib/pg_power/connection_adapters/index_definition.rb
+- lib/pg_power/connection_adapters/postgresql_adapter.rb
+- lib/pg_power/connection_adapters/postgresql_adapter/comment_methods.rb
+- lib/pg_power/connection_adapters/postgresql_adapter/foreigner_methods.rb
+- lib/pg_power/connection_adapters/postgresql_adapter/index_methods.rb
+- lib/pg_power/connection_adapters/postgresql_adapter/schema_methods.rb
+- lib/pg_power/connection_adapters/table.rb
+- lib/pg_power/connection_adapters/table/comment_methods.rb
+- lib/pg_power/connection_adapters/table/foreigner_methods.rb
+- lib/pg_power/engine.rb
+- lib/pg_power/migration.rb
+- lib/pg_power/migration/command_recorder.rb
+- lib/pg_power/migration/command_recorder/comment_methods.rb
+- lib/pg_power/migration/command_recorder/foreigner_methods.rb
+- lib/pg_power/migration/command_recorder/schema_methods.rb
+- lib/pg_power/schema_dumper.rb
+- lib/pg_power/schema_dumper/comment_methods.rb
+- lib/pg_power/schema_dumper/foreigner_methods.rb
+- lib/pg_power/schema_dumper/schema_methods.rb
+- lib/pg_power/tools.rb
+- lib/pg_power/version.rb
+- lib/tasks/pg_power_tasks.rake
+homepage: https://github.com/TMXCredit/pg_power
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: ActiveRecord extensions for PostgreSQL.
+test_files: []
+