pg_schema 0.0.1

data/.gitignore

@@ -0,0 +1,8 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+InstalledFiles
+_yardoc
+coverage
+doc/

data/.rvmrc

@@ -0,0 +1 @@
+rvm use 1.9.3@postgres_multi_tenancy --create

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in qbwc.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2012 
+
+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,3 @@
+require "bundler/gem_tasks"
+
+Bundler::GemHelper.install_tasks

data/lib/generators/pg_schema/install_generator.rb

@@ -0,0 +1,16 @@
+require 'rails/generators'
+
+module PgSchema
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      namespace "pg_schema:install"
+      desc "Copy schema support files"
+      source_root File.expand_path('../templates', __FILE__)
+      
+      def copy_rake_tasks
+         template('lib/pg_schema.rake', "lib/tasks/pg_schema.rake")
+      end
+      
+    end
+  end
+end

data/lib/generators/pg_schema/templates/lib/pg_schema.rake

@@ -0,0 +1,51 @@
+# https://github.com/rails/rails/blob/6c2a0675f11a9b5b8e88ed7dbccd65cb51be8029/activerecord/lib/active_record/railties/databases.rake
+#
+namespace :db do
+
+  desc 'Create the database defined in config/database.yml for the current Rails.env - also makes test database if in development mode'
+  task :create_with_schema => :load_config do
+    # Make the test database at the same time as the development one, if it exists
+    if Rails.env.development? && ActiveRecord::Base.configurations['test']
+      create_database(ActiveRecord::Base.configurations['test'])
+    end
+    create_database(ActiveRecord::Base.configurations[Rails.env])
+  end
+  
+  def create_database(config)
+    if valid_connection?(config)
+      $stderr.puts "#{config['database']} already exists"
+    else
+      case config['adapter']
+      when 'postgresql'
+        encoding = config[:encoding] || ENV['CHARSET'] || 'utf8'
+        schema_search_path = config.delete('schema_search_path') || 'public'
+        default_schema = schema_search_path.split(',').first.strip
+
+        begin
+          unless valid_connection?(config)
+            ActiveRecord::Base.establish_connection(config.merge('database' => 'postgres', 'schema_search_path' => 'public'))
+            ActiveRecord::Base.connection.create_database(config['database'], config.merge('encoding' => encoding))
+            ActiveRecord::Base.establish_connection(config)
+            $stderr.puts "Database #{config['database']} has been created."
+          end
+          ActiveRecord::Base.connection.create_schema(default_schema, config['username'])
+          $stderr.puts "Schema #{default_schema} has been created in #{config['database']}."
+        rescue
+          $stderr.puts $!, *($!.backtrace)
+          $stderr.puts "Couldn't create database #{config.inspect}."
+        end
+      else
+        $stderr.puts "Only PostgreSQL is supported."
+      end
+    end
+  end
+
+
+  def valid_connection?(config)
+    ActiveRecord::Base.establish_connection(config)
+    ActiveRecord::Base.connection && true
+  rescue 
+    false
+  end
+
+end

data/lib/pg_schema/postgresql_adapter.rb

@@ -0,0 +1,20 @@
+class ActiveRecord::ConnectionAdapters::PostgreSQLAdapter < ActiveRecord::ConnectionAdapters::AbstractAdapter
+    # Example:
+    #   create_schema('products', 'postgres')
+    def create_schema(schema_name, pg_username)
+      execute("CREATE SCHEMA \"#{schema_name}\" AUTHORIZATION \"#{pg_username}\"")
+    end
+
+    # Drops a schema
+    #
+    # Example:
+    #   drop_schema('products', 'postgres')
+    def drop_schema(schema_name)
+      execute("DROP SCHEMA \"#{schema_name}\"")
+    end
+
+    # Returns an array of all schemas in the database
+    def all_schemas
+      query('SELECT schema_name FROM information_schema.schemata').flatten
+    end
+end

data/lib/pg_schema/version.rb

@@ -0,0 +1,3 @@
+module PgSchema
+  VERSION = "0.0.1"
+end

data/lib/pg_schema.rb

@@ -0,0 +1,37 @@
+require 'pg_schema/postgresql_adapter'
+
+module PgSchema
+  module Loader
+    extend ActiveSupport::Concern
+    
+    module ClassMethods
+
+      # Let ActiveRecord do its magic and then include our overrides once a
+      # model inherits from ActiveRecord::Base
+      #
+      def inherited(base)
+        base.send(:include, PgSchema::Base)
+        super
+      end
+    end
+  end
+
+  module Base
+    extend ActiveSupport::Concern
+
+    included do
+       self.table_name = "#{schema_name}.#{table_name}"
+    end
+
+    module ClassMethods
+      def schema_name
+        @schema_name ||= ActiveRecord::Base.connection.schema_search_path.split.first
+      end
+    end
+
+  end
+end
+
+class ActiveRecord::Base
+  include PgSchema::Loader
+end

data/pg_schema.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "pg_schema/version"
+
+Gem::Specification.new do |s|
+  s.name        = "pg_schema"
+  s.version     = PgSchema::VERSION
+  s.authors     = ["Alex Skryl"]
+  s.email       = ["rut216@gmail.com"]
+  s.homepage    = "http://skryl.org"
+  s.summary     = %q{Postgres schema support}
+  s.description = %q{Support for postgres schema}
+
+  s.rubyforge_project = "pg_schema"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+
+end

metadata

@@ -0,0 +1,56 @@
+--- !ruby/object:Gem::Specification
+name: pg_schema
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Alex Skryl
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-19 00:00:00.000000000 Z
+dependencies: []
+description: Support for postgres schema
+email:
+- rut216@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rvmrc
+- Gemfile
+- LICENSE.txt
+- Rakefile
+- lib/generators/pg_schema/install_generator.rb
+- lib/generators/pg_schema/templates/lib/pg_schema.rake
+- lib/pg_schema.rb
+- lib/pg_schema/postgresql_adapter.rb
+- lib/pg_schema/version.rb
+- pg_schema.gemspec
+homepage: http://skryl.org
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: pg_schema
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Postgres schema support
+test_files: []