vizzuality-sequel-rails 0.3.4

data/lib/generators/sequel/migration/migration_generator.rb

@@ -0,0 +1,45 @@
+require 'generators/sequel'
+
+module Sequel
+  module Generators
+
+    class MigrationGenerator < Base
+
+      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"
+      class_option :id, :type => :numeric, :desc => "The id to be used in this migration"
+
+      def create_migration_file
+        set_local_assigns!
+        migration_template "migration.rb", "db/migrate/#{file_name}.rb"
+      end
+
+      attr_reader :migration_action, :table_action, :column_action, :use_change
+
+      protected
+      def set_local_assigns!
+        if file_name =~ /^(create|drop)_(.*)$/
+          @table_action   = $1
+          @table_name     = $2.pluralize
+          @column_action  = 'add'
+          @use_change     = @table_action == 'create' ? true : false
+        elsif file_name =~ /^(add|drop|remove)_.*_(?:to|from)_(.*)/
+          @table_action   = 'alter'
+          @table_name     = $2.pluralize
+          @column_action  = $1 == 'add' ? 'add' : 'drop'
+          @use_change     = @column_action == 'add' ? true : false
+        else
+          @table_action   = 'alter'
+          if file_name =~ /^(alter)_(.*)/
+            @table_name   = $2.pluralize
+          else
+            @table_name   = file_name.pluralize
+          end
+          @use_change     = false
+          @column_action  = 'add'
+        end
+      end
+
+    end
+
+  end
+end

data/lib/generators/sequel/migration/templates/migration.rb

@@ -0,0 +1,48 @@
+Sequel.migration do
+  <%- if use_change -%>
+  change do
+    <%= table_action %>_table :<%= table_name %> do
+      <%- if table_action == 'create' -%>
+      primary_key :id
+      <%- end -%>
+      <%- attributes.each do |attribute| -%>
+      <%- if table_action == 'create' -%>
+      <%= attribute.type_class %> :<%= attribute.name %>
+      <%- else -%>
+      <%= column_action %>_column :<%= attribute.name %><% if column_action == 'add' %>, <%= attribute.type_class %><% end %>
+      <%- end -%>
+      <%- end -%>
+    end
+  end
+  <%- else -%>
+  up do
+    <%- if table_action == 'drop' -%>
+    drop_table :<%= table_name %>
+    <%- else -%>
+    <%= table_action %>_table :<%= table_name %> do
+      <%- attributes.each do |attribute| -%>
+      <%- if table_action == 'create' -%>
+      <%= attribute.type_class %> :<%= attribute.name %>
+      <%- else -%>
+      <%= column_action %>_column :<%= attribute.name %><% if column_action == 'add' %>, <%= attribute.type_class %><% end %>
+      <%- end -%>
+      <%- end -%>
+    end
+    <%- end -%>
+  end
+  
+  down do
+    <%- alter_table_action = (table_action == 'drop') ? 'create' : table_action -%>
+    <%- alter_column_action = (column_action == 'add') ? 'drop' : 'add' -%>
+    <%= alter_table_action %>_table :<%= table_name %> do
+      <%- attributes.each do |attribute| -%>
+      <%- if alter_table_action == 'create' -%>
+      <%= attribute.type_class %> :<%= attribute.name %>
+      <%- else -%>
+      <%= alter_column_action %>_column :<%= attribute.name %><% if alter_column_action == 'add' %>, <%= attribute.type_class %><% end %>
+      <%- end -%>
+      <%- end -%>
+    end
+  end
+  <%- end -%>
+end

data/lib/generators/sequel/model/model_generator.rb

@@ -0,0 +1,29 @@
+require 'generators/sequel'
+
+module Sequel
+  module Generators
+
+    class ModelGenerator < Base
+      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"
+
+      check_class_collision
+
+      class_option :migration,  :type => :boolean
+      class_option :timestamps, :type => :boolean
+      class_option :parent,     :type => :string, :desc => "The parent class for the generated model"
+
+      def create_migration_file
+        return unless options[:migration]
+        migration_template "migration.rb", "db/migrate/create_#{table_name}.rb"
+      end
+
+      def create_model_file
+        template 'model.rb', File.join('app/models', class_path, "#{file_name}.rb")
+      end
+
+      hook_for :test_framework
+
+    end
+
+  end
+end

data/lib/generators/sequel/model/templates/migration.rb

@@ -0,0 +1,16 @@
+Sequel.migration do 
+  change do
+
+    create_table :<%= table_name %> do
+      primary_key :id
+      <%- if options[:timestamps] -%>
+      DateTime :created_at
+      DateTime :updated_at
+      <%- end -%>
+      <%- attributes.each do |attribute| -%>
+      <%= attribute.type_class %> :<%= attribute.name %>
+      <%- end -%>
+    end
+
+  end
+end

data/lib/generators/sequel/model/templates/model.rb

@@ -0,0 +1,6 @@
+class <%= class_name %><%= options[:parent] ? " < #{options[:parent].classify}" : " < Sequel::Model" %>
+  <%- if options[:timestamps] -%>
+  plugin :timestamps
+  <%- end -%>
+  
+end

data/lib/generators/sequel/observer/observer_generator.rb

@@ -0,0 +1,19 @@
+require 'generators/sequel'
+
+module Sequel
+  module Generators
+
+    class ObserverGenerator < Base
+
+      check_class_collision :suffix => "Observer"
+
+      def create_observer_file
+        template 'observer.rb', File.join('app/models', class_path, "#{file_name}_observer.rb")
+      end
+
+      hook_for :test_framework
+
+    end
+
+  end
+end

data/lib/generators/sequel/observer/templates/observer.rb

@@ -0,0 +1,7 @@
+class <%= class_name %>Observer
+
+  include Sequel::Observer
+
+  observe <%= class_name %>
+
+end

data/lib/sequel-rails.rb

@@ -0,0 +1,7 @@
+require 'sequel-rails/railtie'
+
+module Rails
+  module Sequel
+  end
+end
+

data/lib/sequel-rails/configuration.rb

@@ -0,0 +1,63 @@
+require 'active_support/core_ext/hash/except'
+require 'active_support/core_ext/class/attribute_accessors'
+
+module Rails
+  module Sequel
+
+    mattr_accessor :configuration
+
+    class Configuration
+
+      def self.for(root, database_yml_hash)
+        Rails::Sequel.configuration ||= new(root, database_yml_hash)
+      end
+
+      attr_reader :root, :raw
+      attr_accessor :logger
+      attr_accessor :migration_dir
+
+      def environment_for(name)
+        environments[name.to_s] || environments[name.to_sym]
+      end
+
+      def environments
+        @environments ||= @raw.inject({}) do |normalized, environment|
+          name, config = environment.first, environment.last
+          normalized[name] = normalize_repository_config(config)
+          normalized
+        end
+      end
+
+    private
+
+      def initialize(root, database_yml_hash)
+        @root, @raw = root, database_yml_hash
+      end
+
+      def normalize_repository_config(hash)
+        config = {}
+        hash.each do |key, value|
+          config[key.to_s] = 
+            if key.to_s == 'port'
+              value.to_i
+            elsif key.to_s == 'adapter' && value == 'sqlite3'
+              'sqlite'
+            elsif key.to_s == 'database' && (hash['adapter'] == 'sqlite3' || 
+                                             hash['adapter'] == 'sqlite'  ||
+                                             hash[:adapter]  == 'sqlite3' ||
+                                             hash[:adapter]  == 'sqlite')
+              value == ':memory:' ? value : File.expand_path((hash['database'] || hash[:database]), root)
+            elsif key.to_s == 'adapter' && value == 'postgresql'
+              'postgres'
+            else
+              value
+            end
+        end
+        
+        config
+      end
+
+    end
+
+  end
+end

data/lib/sequel-rails/migrations.rb

@@ -0,0 +1,30 @@
+require 'sequel/extensions/migration'
+
+module Rails
+  module Sequel
+    class Migrations
+
+      class << self
+        
+        def migrate_up!(version=nil)
+          opts = {}
+          opts[:target] = version.to_i if version
+          
+          
+          
+          ::Sequel::Migrator.run(::Sequel::Model.db, "db/migrate", opts)
+        end
+        
+        def migrate_down!(version=nil)
+          opts = {}
+          opts[:target] = version.to_i if version
+
+          ::Sequel::Migrator.run(::Sequel::Model.db, "db/migrate", opts)
+        end
+        
+      end
+      
+      
+    end
+  end
+end

data/lib/sequel-rails/railtie.rb

@@ -0,0 +1,93 @@
+require 'sequel'
+
+require 'rails'
+require 'active_model/railtie'
+
+# Comment taken from active_record/railtie.rb
+#
+# For now, action_controller must always be present with
+# rails, so let's make sure that it gets required before
+# here. This is needed for correctly setting up the middleware.
+# In the future, this might become an optional require.
+require 'action_controller/railtie'
+
+require 'sequel-rails/setup'
+require 'sequel-rails/railties/log_subscriber'
+require 'sequel-rails/railties/i18n_support'
+
+
+module Rails
+  module Sequel
+
+    class Railtie < Rails::Railtie
+
+      ::Sequel::Railties::LogSubscriber.attach_to :sequel
+
+      config.app_generators.orm :sequel, :migration => true
+      config.rails_fancy_pants_logging = true
+
+      config.action_dispatch.rescue_responses.merge!(
+        'Sequel::Plugins::RailsExtensions::ModelNotFound' => :not_found,
+        'Sequel::ValidationFailed'                        => :unprocessable_entity,
+        'Sequel::NoExistingObject'                        => :unprocessable_entity
+      )
+
+      rake_tasks do
+        load 'sequel-rails/railties/database.rake'
+      end
+
+      initializer 'sequel.configuration' do |app|
+        configure_sequel(app)
+      end
+
+      initializer 'sequel.logger' do |app|
+        setup_logger(app, Rails.logger)
+      end
+
+      initializer 'sequel.i18n_support' do |app|
+        setup_i18n_support(app)
+      end
+
+      # Expose database runtime to controller for logging.
+      initializer 'sequel.log_runtime' do |app|
+        setup_controller_runtime(app)
+      end
+
+      initializer 'sequel.connect' do |app|
+        Rails::Sequel.setup(Rails.env)
+      end
+
+      # Run setup code after_initialize to make sure all config/initializers
+      # are in effect once we setup the connection. This is especially necessary
+      # for the cascaded adapter wrappers that need to be declared before setup.
+      config.after_initialize do |app|
+        ::Sequel::Model.plugin :active_model
+        ::Sequel::Model.plugin :validation_helpers
+        ::Sequel::Model.plugin :rails_extensions
+        ::Sequel::Model.raise_on_save_failure = false
+      end
+
+      # Support overwriting crucial steps in subclasses
+      def configure_sequel(app)
+        app.config.sequel = Rails::Sequel::Configuration.for(
+          Rails.root, app.config.database_configuration
+        )
+      end
+
+      def setup_i18n_support(app)
+        ::Sequel::Model.send :include, Rails::Sequel::I18nSupport
+      end
+
+      def setup_controller_runtime(app)
+        require 'sequel-rails/railties/controller_runtime'
+        ActionController::Base.send :include, Rails::Sequel::Railties::ControllerRuntime
+      end
+
+      def setup_logger(app, logger)
+        app.config.sequel.logger=logger
+      end
+
+    end
+
+  end
+end

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

@@ -0,0 +1,23 @@
+module Sequel
+  module Adapters
+    module Benchmarking
+
+      %w[ create read update delete ].each do |method|
+        class_eval <<-RUBY, __FILE__, __LINE__
+          def #{method}(*args, &block)
+            result = nil
+            @runtime ||= 0
+            @runtime += Benchmark.ms { result = super(*args, &block) }
+            result
+          end
+        RUBY
+      end
+
+      def reset_runtime
+        rt, @runtime = @runtime, 0
+        rt.to_f
+      end
+
+    end
+  end
+end

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

@@ -0,0 +1,43 @@
+require 'active_support/core_ext/module/attr_internal'
+
+module Rails
+  module Sequel
+    module Railties
+
+      module ControllerRuntime
+
+        extend ActiveSupport::Concern
+
+        protected
+
+        attr_internal :db_runtime
+
+        def cleanup_view_runtime
+          db_rt_before_render = ::Rails::Sequel.reset_runtime
+          runtime = super
+          db_rt_after_render = ::Rails::Sequel.reset_runtime
+          self.db_runtime = db_rt_before_render + db_rt_after_render
+          runtime - db_rt_after_render
+        end
+
+        def append_info_to_payload(payload)
+          super
+          payload[:db_runtime] = db_runtime
+        end
+
+
+        module ClassMethods
+
+          def log_process_action(payload)
+            messages, db_runtime = super, payload[:db_runtime]
+            messages << ("Models: %.1fms" % db_runtime.to_f) if db_runtime
+            messages
+          end
+
+        end
+
+      end
+
+    end
+  end
+end