sequel-rails 0.1.8 → 0.4.0.pre

data/lib/sequel_rails/railties/i18n_support.rb

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

data/lib/sequel_rails/railties/log_subscriber.rb

@@ -0,0 +1,56 @@
+module SequelRails
+  module Railties
+
+    class LogSubscriber < ActiveSupport::LogSubscriber
+
+      def self.runtime=(value)
+        Thread.current["sequel_sql_runtime"] = value
+      end
+
+      def self.runtime
+        Thread.current["sequel_sql_runtime"] ||= 0
+      end
+
+      def self.reset_runtime
+        rt, self.runtime = runtime, 0
+        rt
+      end
+
+      def sql(event)
+        self.class.runtime += event.duration
+        return unless logger.debug?
+
+        payload = event.payload
+
+        name = '%s (%.1fms)' % [payload[:name], event.duration]
+        sql  = payload[:sql].squeeze(' ')
+        binds = nil
+
+        unless (payload[:binds] || []).empty?
+          binds = "  " + payload[:binds].map { |col,v|
+            [col.name, v]
+          }.inspect
+        end
+
+        if odd?
+          name = color(name, :cyan, true)
+          sql  = color(sql, nil, true)
+        else
+          name = color(name, :magenta, true)
+        end
+
+        debug "  #{name}  #{sql}#{binds}"
+      end
+
+      def odd?
+        @odd_or_even = !@odd_or_even
+      end
+
+      def logger
+        ::SequelRails.configuration.logger
+      end
+
+    end
+
+  end
+end

data/lib/sequel_rails/sequel/database/active_support_notification.rb

@@ -0,0 +1,28 @@
+require "sequel/database/logging"
+require "active_support/notifications"
+
+module Sequel
+  class Database
+
+    def log_yield(sql, args=nil)
+      sql_for_log = args ? "#{sql}; #{args.inspect}" : sql
+      start = Time.now
+      begin
+        ::ActiveSupport::Notifications.instrument(
+          "sql.sequel",
+          sql: sql,
+          name: self.class,
+          binds: args,
+        ) do
+          yield
+        end
+      rescue => e
+        log_exception(e, sql_for_log) unless @loggers.empty?
+        raise
+      ensure
+        log_duration(Time.now - start, sql_for_log) unless e || @loggers.empty?
+      end
+    end
+
+  end
+end

data/lib/sequel_rails/sequel/plugins/rails_extensions.rb

@@ -0,0 +1,35 @@
+require "sequel"
+
+module Sequel
+  module Plugins
+    # The RailsExtensions plugin adds a single class method to Sequel::Model in
+    # order to make its use in controllers a little more like ActiveRecord's.
+    # The +find!+ method is added which will raise an exception if no object is
+    # found. By adding the following code to a Railtie:
+    #
+    #   config.action_dispatch.rescue_responses.merge!(
+    #    'Sequel::Plugins::RailsExtensions::ModelNotFound' => :not_found
+    #   )
+    #
+    # Usage:
+    #
+    #   # Apply plugin to all models:
+    #   Sequel::Model.plugin :rails_extensions
+    #
+    #   # Apply plugin to a single model:
+    #   Album.plugin :rails_extensions
+    module RailsExtensions
+      class ModelNotFound < Sequel::Error
+      end
+
+      module ClassMethods
+        def find!(args)
+          m = self[args]
+          raise ModelNotFound, "Couldn't find #{self} matching #{args}." unless m
+          m
+        end
+      end
+
+    end
+  end
+end

data/lib/sequel_rails/session_store.rb

@@ -0,0 +1,80 @@
+require "sequel"
+
+# Implements Sequel-specific session store.
+
+module SequelRails
+
+  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

data/lib/sequel_rails/storage.rb

@@ -0,0 +1,58 @@
+require "sequel_rails/storage/abstract"
+require "sequel_rails/storage/sqlite"
+require "sequel_rails/storage/mysql"
+require "sequel_rails/storage/mysql2"
+require "sequel_rails/storage/postgres"
+require "sequel_rails/storage/jdbc"
+
+module SequelRails
+  module Storage
+    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)
+      adapter_for(config).create
+    end
+
+    def self.drop_environment(config)
+      adapter_for(config).drop
+    end
+
+    def self.adapter_for(config_or_env)
+      config = if config_or_env.kind_of? Hash
+                 config_or_env
+               else
+                 ::SequelRails.configuration.environments[config_or_env.to_s]
+               end
+      lookup_class(config['adapter']).new config
+    end
+
+    private
+
+    def self.with_local_repositories
+      ::SequelRails.configuration.environments.each_value do |config|
+        next if config['database'].blank?
+        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 self.lookup_class(adapter)
+      klass_name = adapter.camelize.to_sym
+
+      unless self.const_defined?(klass_name)
+        raise "Adapter #{adapter} not supported (#{klass_name.inspect})"
+      end
+
+      const_get klass_name
+    end
+  end
+end

data/lib/sequel_rails/storage/abstract.rb

@@ -0,0 +1,52 @@
+module SequelRails
+  module Storage
+    class Abstract
+
+      attr_reader :config
+
+      def initialize(config)
+        @config = config
+      end
+
+      def create
+        _create
+        puts "[sequel] Created database '#{database}'"
+      end
+
+      def drop
+        ::Sequel::Model.db.disconnect
+        _drop
+        puts "[sequel] Dropped database '#{database}'"
+      end
+
+      def database
+        @database ||= config['database'] || config['path']
+      end
+
+      def username
+        @username ||= config['username'] || config['user'] || ''
+      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
+
+    end
+  end
+end

data/lib/sequel_rails/storage/jdbc.rb

@@ -0,0 +1,45 @@
+module SequelRails
+  module Storage
+    class Jdbc < Abstract
+
+      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

data/lib/sequel_rails/storage/mysql.rb

@@ -0,0 +1,31 @@
+require 'shellwords'
+
+module SequelRails
+  module Storage
+    class Mysql < Abstract
+      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=#{Shellwords.escape(username)}" unless username.blank?
+        commands << "--password=#{Shellwords.escape(password)}" unless password.blank?
+        commands << "--host=#{host}" unless host.blank?
+        commands << "-e" << statement
+        system(*commands)
+      end
+
+      def collation
+        @collation ||= config['collation'] || ENV['COLLATION'] || 'utf8_unicode_ci'
+      end
+
+    end
+  end
+end

data/lib/sequel_rails/storage/mysql2.rb

@@ -0,0 +1,6 @@
+module SequelRails
+  module Storage
+    class Mysql2 < Mysql
+    end
+  end
+end

data/lib/sequel_rails/storage/postgres.rb

@@ -0,0 +1,22 @@
+module SequelRails
+  module Storage
+    class Postgres < Abstract
+      def _create
+        ENV["PGPASSWORD"] = password unless password.blank?
+        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
+        res = system(*commands)
+        ENV["PGPASSWORD"] = nil
+        res
+      end
+
+      def _drop
+        system("dropdb", "-U", username, database)
+      end
+    end
+  end
+end

data/lib/sequel_rails/storage/sqlite.rb

@@ -0,0 +1,26 @@
+module SequelRails
+  module Storage
+    class Sqlite < Abstract
+      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
+  end
+end