talentbox-sequel-rails 0.2.0

data/lib/sequel-rails/railties/i18n_support.rb

@@ -0,0 +1,12 @@
+module Rails
+  module Sequel
+
+    module I18nSupport
+      # Set the i18n scope to overwrite ActiveModel.
+      def i18n_scope #:nodoc:
+        :sequel
+      end
+    end
+
+  end
+end

data/lib/sequel-rails/railties/log_subscriber.rb

@@ -0,0 +1,31 @@
+module Sequel
+  module Railties
+
+    class LogSubscriber < ActiveSupport::LogSubscriber
+
+      def sql(event)
+        name = '%s (%.1fms)' % [event.payload[:name], event.duration]
+        sql  = event.payload[:sql].squeeze(' ')
+
+        if odd?
+          name = color(name, :cyan, true)
+          sql  = color(sql, nil, true)
+        else
+          name = color(name, :magenta, true)
+        end
+
+        debug "  #{name}  #{sql}"
+      end
+
+      def odd?
+        @odd_or_even = !@odd_or_even
+      end
+
+      def logger
+        ::Rails::Sequel.configuration.logger
+      end
+
+    end
+
+  end
+end

data/lib/sequel-rails/runtime.rb

@@ -0,0 +1,14 @@
+module Rails
+  module Sequel
+   
+    class << self
+      def reset_runtime
+        @runtime ||= 0
+        
+        rt, @runtime = @runtime, 0
+        rt
+      end
+    end
+    
+  end
+end

data/lib/sequel-rails/session_store.rb

@@ -0,0 +1,82 @@
+require 'sequel'
+
+# Implements Sequel-specific session store.
+
+module Rails
+  module Sequel
+
+    class SessionStore < ActionDispatch::Session::AbstractStore
+
+      class Session < ::Sequel::Model
+
+        # property :id,         Serial
+        # property :session_id, String,   :required => true, :unique => true, :unique_index => true
+        # property :data,       Object,   :required => true, :default => ActiveSupport::Base64.encode64(Marshal.dump({}))
+        # property :updated_at, DateTime, :required => false, :index => true
+
+        class << self
+          
+          def auto_migrate!
+            self.db.create_table :sessions do
+              primary_key :id
+              column :session_id, String, 
+                     :null    => false, 
+                     :unique  => true, 
+                     :index   => true
+
+              column :data, :text, 
+                     :null => false 
+                     
+              column :updated_at, DateTime, 
+                     :null => true, 
+                     :index => true
+            end
+          end
+          
+        end
+
+        def self.name
+          'session'
+        end
+
+      end
+
+      SESSION_RECORD_KEY = 'rack.session.record'.freeze
+
+      cattr_accessor :session_class
+      self.session_class = Session
+
+      private
+
+      def get_session(env, sid)
+        sid ||= generate_sid
+        session = find_session(sid)
+        env[SESSION_RECORD_KEY] = session
+        [ sid, session.data ]
+      end
+
+      def set_session(env, sid, session_data)
+        session            = get_session_resource(env, sid)
+        session.data       = session_data
+        session.updated_at = Time.now if session.dirty?
+        session.save
+      end
+
+      def get_session_resource(env, sid)
+        if env[ENV_SESSION_OPTIONS_KEY][:id].nil?
+          env[SESSION_RECORD_KEY] = find_session(sid)
+        else
+          env[SESSION_RECORD_KEY] ||= find_session(sid)
+        end
+      end
+
+      def find_session(sid)
+        klass = self.class.session_class
+        
+        klass.where(:session_id => sid).first || klass.new(:session_id => sid)
+      end
+
+    end
+
+  end
+end

data/lib/sequel-rails/setup.rb

@@ -0,0 +1,17 @@
+require 'active_support/core_ext/hash/except'
+
+require 'sequel/extensions/migration'
+
+require 'sequel-rails/configuration'
+require 'sequel-rails/runtime'
+require 'sequel-rails/railties/benchmarking_mixin'
+
+module Rails
+  module Sequel
+
+    def self.setup(environment)
+      ::Sequel.connect({:logger => configuration.logger}.merge(::Rails::Sequel.configuration.environment_for(environment.to_s)))
+    end
+
+  end
+end

data/lib/sequel-rails/storage.rb

@@ -0,0 +1,221 @@
+module Rails
+  module Sequel
+
+    def self.storage
+      Storage
+    end
+
+    class Storage
+      attr_reader :config
+
+      def self.create_all
+        with_local_repositories { |config| create_environment(config) }
+      end
+
+      def self.drop_all
+        with_local_repositories { |config| drop_environment(config) }
+      end
+
+      def self.create_environment(config)
+        new(config).create
+      end
+
+      def self.drop_environment(config)
+        new(config).drop
+      end
+
+      def self.new(config)
+        config = Rails::Sequel.configuration.environments[config.to_s] unless config.kind_of?(Hash)
+        
+        klass = lookup_class(config['adapter'])
+        if klass.equal?(self)
+          super(config)
+        else
+          klass.new(config)
+        end
+      end
+
+      class << self
+      private
+
+        def with_local_repositories
+          Rails::Sequel.configuration.environments.each_value do |config|
+            if config['host'].blank? || %w[ 127.0.0.1 localhost ].include?(config['host'])
+              yield(config)
+            else
+              puts "This task only modifies local databases. #{config['database']} is on a remote host."
+            end
+          end
+        end
+
+        def lookup_class(adapter)
+          klass_name = adapter.camelize.to_sym
+
+          unless Storage.const_defined?(klass_name)
+            raise "Adapter #{adapter} not supported (#{klass_name.inspect})"
+          end
+
+          const_get(klass_name)
+        end
+
+      end
+
+      def initialize(config)
+        @config = config
+      end
+
+      def create
+        _create
+        puts "[sequel] Created database '#{database}'"
+      end
+
+      def drop
+        _drop
+        puts "[sequel] Dropped database '#{database}'"
+      end
+
+      def database
+        @database ||= config['database'] || config['path']
+      end
+
+      def username
+        @username ||= config['username'] || ''
+      end
+
+      def password
+        @password ||= config['password'] || ''
+      end
+
+      def host
+        @host ||= config['host'] || ''
+      end
+
+      def port
+        @port ||= config['port'] || ''
+      end
+
+      def owner
+        @owner ||= config['owner'] || ''
+      end
+        
+      def charset
+        @charset ||= config['charset'] || ENV['CHARSET'] || 'utf8'
+      end
+
+      class Sqlite < Storage
+        def _create
+          return if in_memory?
+          ::Sequel.connect(config.merge('database' => path))
+        end
+
+        def _drop
+          return if in_memory?
+          path.unlink if path.file?
+        end
+
+      private
+
+        def in_memory?
+          database == ':memory:'
+        end
+
+        def path
+          @path ||= Pathname(File.expand_path(database, Rails.root))
+        end
+
+      end
+
+      class Mysql < Storage
+        def _create
+          execute("CREATE DATABASE IF NOT EXISTS `#{database}` DEFAULT CHARACTER SET #{charset} DEFAULT COLLATE #{collation}")
+        end
+
+        def _drop
+          execute("DROP DATABASE IF EXISTS `#{database}`")
+        end
+
+      private
+
+        def execute(statement)
+          commands = 'mysql '
+          commands << "--user=#{username} "     unless username.blank?
+          commands << "--password=#{password} " unless password.blank?
+          commands << "--host=#{host} "         unless host.blank?
+          commands << '-e '
+          commands << statement
+          system(commands)
+        end
+
+        def collation
+          @collation ||= config['collation'] || ENV['COLLATION'] || 'utf8_unicode_ci'
+        end
+
+      end
+
+      class Postgres < Storage
+        def _create
+          commands = "createdb --encoding=#{charset} "
+          commands << "--username=#{username} " unless username.blank?
+          commands << "--owner=#{owner} "       unless owner.blank?
+          commands << "--port=#{port} "         unless port.blank?
+          commands << "--host=#{host} "         unless host.blank?
+          commands << database
+          system(commands)
+        end
+
+        def _drop
+          system(
+            'dropdb',
+            '-U',
+            username,
+            database
+          )
+        end
+      end
+      
+      class Jdbc < Storage
+        
+        def _is_mysql?
+          database.match(/^jdbc:mysql/)
+        end
+        
+        def _root_url
+          database.scan /^jdbc:mysql:\/\/\w*:?\d*/
+        end
+        
+        def db_name
+          database.scan(/^jdbc:mysql:\/\/\w+:?\d*\/(\w+)/).flatten.first
+        end
+        
+        def _params
+          database.scan /\?.*$/
+        end
+        
+        def _create
+          if _is_mysql?
+            ::Sequel.connect("#{_root_url}#{_params}") do |db|
+              db.execute("CREATE DATABASE IF NOT EXISTS `#{db_name}` DEFAULT CHARACTER SET #{charset} DEFAULT COLLATE #{collation}")
+            end
+          end
+        end
+
+        def _drop
+          if _is_mysql?
+            ::Sequel.connect("#{_root_url}#{_params}") do |db|
+              db.execute("DROP DATABASE IF EXISTS `#{db_name}`")
+            end
+          end
+        end
+        
+        private
+        
+        def collation
+          @collation ||= config['collation'] || ENV['COLLATION'] || 'utf8_unicode_ci'
+        end
+        
+        
+      end
+      
+    end
+  end
+end

data/lib/sequel-rails/version.rb

@@ -0,0 +1,5 @@
+module Rails
+  module Sequel
+    VERSION = "0.2.0"
+  end
+end

data/sequel-rails.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+require "sequel-rails/version"
+
+Gem::Specification.new do |s|
+  s.name        = "talentbox-sequel-rails"
+  s.version     = Rails::Sequel::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Brasten Sager (brasten)", "Jonathan TRON"]
+  s.email       = ["brasten@gmail.com", "jonathan.tron@thetalentbox.com"]
+  s.homepage    = "https://github.com/TalentBox/sequel-rails"
+  s.description = "Integrate Sequel with Rails 3"
+  s.summary     = "Use Sequel with Rails 3"
+  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"]
+  s.extra_rdoc_files = ["LICENSE", "README.rdoc"]
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.add_runtime_dependency("sequel", ["~> 3.28.0"])
+  s.add_runtime_dependency("rails", ["~> 3.1.1"])
+
+  s.add_development_dependency("rake", ["~> 0.8.7"])
+  s.add_development_dependency("yard", ["~> 0.5"])
+  s.add_development_dependency("rspec", ["~> 2.7.0"])
+  s.add_development_dependency("autotest", ["~> 4.4.6"])
+  s.add_development_dependency("rcov", ["~> 0.9.11"])
+end

data/spec/rcov.opts

@@ -0,0 +1,6 @@
+--exclude-only "spec\/,^\/"
+--sort coverage
+--callsites
+--xrefs
+--profile
+--text-summary

data/spec/spec.opts

@@ -0,0 +1,4 @@
+--color
+--loadby random
+--format profile
+--backtrace

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require "sequel-rails"
+
+# Load support files
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each { |f| require f }
+
+RSpec.configure do |config|
+end

data/tasks/ci.rake

@@ -0,0 +1 @@
+task :ci => [ 'metrics:all' ]

data/tasks/clean.rake

@@ -0,0 +1,6 @@
+require 'rake/clean'
+
+File.foreach('.gitignore') do |line|
+  line.strip!
+  CLOBBER << line unless line.empty? || line[0, 1] == '#'
+end