rails_multisite 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e45ef95202698ebf603b6442abf02eba8f6a25e2
+  data.tar.gz: c47de25e43b5a8c01d8a3643bcff11592075dc4c
+SHA512:
+  metadata.gz: 87c49b6fdd43d33155168227cf94949d8e114a6b440bf0c53e3233e19c1cd78f42e7057c17d2752920225b050b2effc4b7cf29e2715f452283392bcbf1aca0cd
+  data.tar.gz: 4ef167a5687c24657d69a1744a1d9c18d07593f7fc1546bca98bce24996056fa2b92abe58c843b4070b6e6af95464ea0a2cb0574ed6cff17eeda914632d236e6

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Sam Saffron
+
+MIT License
+
+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/README.md

@@ -0,0 +1,29 @@
+# RailsMultisite
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rails_multisite'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rails_multisite
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/lib/rails_multisite/connection_management.rb

@@ -0,0 +1,168 @@
+module RailsMultisite
+  class ConnectionManagement
+    CONFIG_FILE = 'config/multisite.yml'
+
+    def self.rails4?
+      !!(Rails.version =~ /^4/)
+    end
+
+    def self.establish_connection(opts)
+      if opts[:db] == "default" && (!defined?(@@default_spec) || !@@default_spec)
+        # don't do anything .. handled implicitly
+      else
+        spec = connection_spec(opts) || @@default_spec
+        handler = nil
+        if spec != @@default_spec
+          handler = @@connection_handlers[spec]
+          unless handler
+            handler = ActiveRecord::ConnectionAdapters::ConnectionHandler.new
+            if rails4?
+              handler.establish_connection(ActiveRecord::Base, spec)
+            end
+            @@connection_handlers[spec] = handler
+          end
+        else
+          handler = @@default_connection_handler
+          if rails4? && !@@established_default
+            handler.establish_connection(ActiveRecord::Base, spec)
+            @@established_default = true
+          end
+        end
+
+        ActiveRecord::Base.connection_handler = handler
+
+        unless rails4?
+          handler.establish_connection("ActiveRecord::Base", spec)
+        end
+      end
+    end
+
+    def self.each_connection
+      old = current_db
+      connected = ActiveRecord::Base.connection_pool.connected?
+      all_dbs.each do |db|
+        establish_connection(:db => db)
+        yield db
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+      end
+      establish_connection(:db => old)
+      ActiveRecord::Base.connection_handler.clear_active_connections! unless connected
+    end
+
+    def self.all_dbs
+      ["default"] +
+        if defined?(@@db_spec_cache) && @@db_spec_cache
+          @@db_spec_cache.keys.to_a
+        else
+          []
+        end
+    end
+
+    def self.current_db
+      db = ActiveRecord::Base.connection_pool.spec.config[:db_key] || "default"
+    end
+
+    def self.config_filename=(config_filename)
+      @@config_filename = config_filename
+    end
+
+    def self.config_filename
+      @@config_filename ||= File.absolute_path(Rails.root.to_s + "/" + RailsMultisite::ConnectionManagement::CONFIG_FILE)
+    end
+
+    def self.current_hostname
+      config = ActiveRecord::Base.connection_pool.spec.config
+      config[:host_names].nil? ? config[:host] : config[:host_names].first
+    end
+
+
+    def self.clear_settings!
+      @@db_spec_cache = nil
+      @@host_spec_cache = nil
+      @@default_spec = nil
+    end
+
+    def self.load_settings!
+      spec_klass = rails4? ? ActiveRecord::ConnectionAdapters::ConnectionSpecification : ActiveRecord::Base::ConnectionSpecification
+      configs = YAML::load(File.open(self.config_filename))
+      configs.each do |k,v|
+        raise ArgumentError.new("Please do not name any db default!") if k == "default"
+        v[:db_key] = k
+      end
+
+      @@db_spec_cache = Hash[*configs.map do |k, data|
+        [k, spec_klass::Resolver.new(k, configs).spec]
+      end.flatten]
+
+      @@host_spec_cache = {}
+      configs.each do |k,v|
+        next unless v["host_names"]
+        v["host_names"].each do |host|
+          @@host_spec_cache[host] = @@db_spec_cache[k]
+        end
+      end
+
+      @@default_spec = spec_klass::Resolver.new(Rails.env, ActiveRecord::Base.configurations).spec
+      ActiveRecord::Base.configurations[Rails.env]["host_names"].each do |host|
+        @@host_spec_cache[host] = @@default_spec
+      end
+
+      @@default_connection_handler = ActiveRecord::Base.connection_handler
+
+      # inject our connection_handler pool
+      # WARNING MONKEY PATCH
+      #
+      # see: https://github.com/rails/rails/issues/8344#issuecomment-10800848
+      if ActiveRecord::VERSION::MAJOR == 3
+        ActiveRecord::Base.send :include, NewConnectionHandler
+        ActiveRecord::Base.connection_handler = @@default_connection_handler
+      end
+
+      @@connection_handlers = {}
+      @@established_default = false
+    end
+
+    module NewConnectionHandler
+      def self.included(klass)
+        klass.class_eval do
+          define_singleton_method :connection_handler do
+            Thread.current[:connection_handler] || @connection_handler
+          end
+          define_singleton_method :connection_handler= do |handler|
+            @connection_handler ||= handler
+            Thread.current[:connection_handler] = handler
+          end
+        end
+      end
+    end
+
+
+    def initialize(app, config = nil)
+      @app = app
+    end
+
+    def call(env)
+      request = Rack::Request.new(env)
+      begin
+
+        #TODO: add a callback so users can simply go to a domain to register it, or something
+        return [404, {}, ["not found"]] unless @@host_spec_cache[request.host]
+
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+        self.class.establish_connection(:host => request['__ws'] || request.host)
+        @app.call(env)
+      ensure
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+      end
+    end
+
+    def self.connection_spec(opts)
+      if opts[:host]
+        @@host_spec_cache[opts[:host]]
+      else
+        @@db_spec_cache[opts[:db]]
+      end
+    end
+
+  end
+end

data/lib/rails_multisite/railtie.rb

@@ -0,0 +1,21 @@
+module RailsMultisite
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      Dir[File.join(File.dirname(__FILE__),'../tasks/*.rake')].each { |f| load f }
+    end
+
+    initializer "RailsMultisite.init" do |app|
+      if File.exists?(ConnectionManagement.config_filename)
+        ConnectionManagement.load_settings!
+        app.middleware.insert_after(ActiveRecord::ConnectionAdapters::ConnectionManagement, RailsMultisite::ConnectionManagement)
+        app.middleware.delete(ActiveRecord::ConnectionAdapters::ConnectionManagement)
+        if ENV['RAILS_DB']
+          ConnectionManagement.establish_connection(:db => ENV['RAILS_DB'])
+        end
+      end
+    end
+
+
+  end
+end
+

data/lib/rails_multisite/version.rb

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

data/lib/rails_multisite.rb

@@ -0,0 +1,3 @@
+require "rails_multisite/version"
+require "rails_multisite/railtie"
+require "rails_multisite/connection_management"

metadata

@@ -0,0 +1,50 @@
+--- !ruby/object:Gem::Specification
+name: rails_multisite
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Sam Saffron
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-29 00:00:00.000000000 Z
+dependencies: []
+description: Multi tenancy support for Rails
+email:
+- sam.saffron@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- lib/rails_multisite.rb
+- lib/rails_multisite/connection_management.rb
+- lib/rails_multisite/railtie.rb
+- lib/rails_multisite/version.rb
+homepage: ''
+licenses: []
+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.0
+signing_key: 
+specification_version: 4
+summary: Multi tenancy support for Rails
+test_files: []
+has_rdoc: