eugenekorpan/multitenant-mysql

require_relative './db'

module Multitenant
  module Mysql
    class Tenant
      def self.exists? tenant_name
        return true if tenant_name.blank?
        tenant = Multitenant::Mysql.configs.tenant
        field = Multitenant::Mysql.configs.bucket.field
        if tenant.where(field => tenant_name).blank?
          raise NoTenantRegisteredError.new("No tenant registered: #{tenant_name}")
        end
        true
      end
    end

    class ConnectionSwitcher
      attr_accessor :action_controller, :method

      def initialize(action_controller, method)
        @action_controller = action_controller
        @method            = method
      end

      def self.set_tenant(tenant_name)
        return unless Tenant.exists?(tenant_name)
        DB.establish_connection_for tenant_name
      end

      def execute
        tenant_name = @action_controller.send(method)
        return unless Tenant.exists?(tenant_name)
        DB.establish_connection_for tenant_name
      end
    end

  end
end