mysql_view_support 0.2.0

data/generators/sql_view_migration/sql_view_migration_generator.rb

@@ -0,0 +1,7 @@
+class SqlViewMigrationGenerator < Rails::Generator::NamedBase
+  def manifest
+    record do |m|
+      m.migration_template 'migration.rb', 'db/views'
+    end
+  end
+end

data/generators/sql_view_migration/templates/migration.rb

@@ -0,0 +1,7 @@
+class Create<%= class_name.underscore.camelize %>Migration < ActiveRecord::Migration
+  def self.up
+    update_view :<%= class_name.underscore.pluralize %> do
+      select_statement = ""
+    end
+  end
+end

data/lib/active_record/connection_adapters/mysql_with_views_adapter.rb

@@ -0,0 +1,55 @@
+require_dependency 'active_record/connection_adapters/mysql_adapter'
+
+module ActiveRecord
+  class Base
+    class << self
+      def mysql_with_views_connection(config) # :nodoc:
+        ConnectionAdapters::MysqlWithViewsAdapter.new(config)
+      end
+    end
+  end
+
+  class ConnectionAdapters::MysqlWithViewsAdapter < DelegateClass(ActiveRecord::ConnectionAdapters::MysqlAdapter)
+    attr_reader :view_options
+    def initialize(config)
+      super ActiveRecord::Base.mysql_connection(config)
+      @view_options = (config.symbolize_keys[:view_options] || {}).symbolize_keys.freeze
+    end
+
+    def create_view(name, column_names = nil, options = {})
+      raise ArgumentError, "You must provide the SQL statement that defines the view." unless block_given?
+      select_statement = yield
+      command = ["CREATE"]
+      command << "OR REPLACE" if options[:force]
+      command << "VIEW #{quote_table_name(name)}"
+      command << "(" + [column_names].flatten.join(", ") + ")" if column_names
+      command << "AS #{select_statement}"
+      execute command.join(" ")
+    end
+
+    def update_view(name, column_names = nil, options = {}, &block)
+      create_view name, column_names, options.merge(:force => true), &block
+    end
+
+    def drop_view(name)
+      execute "DROP VIEW #{quote_table_name(name)}"
+    end
+
+    def views(name = nil) #:nodoc:
+      tables = []
+      result = execute("SHOW FULL TABLES", name)
+      result.each { |field| tables << field[0] if field[1] == 'VIEW' }
+      result.free
+      tables
+    end
+
+    def pk_and_sequence_for(table)
+      if table_key = super(table)
+        table_key
+      elsif views.include?( table.to_s ) and columns(table).any?{|col| col.name == 'id'}
+        view_key = ['id', nil]
+      end
+    end
+
+  end
+end

data/tasks/views.rake

@@ -0,0 +1,62 @@
+unless defined? remove_task
+  Rake::TaskManager.class_eval do
+    def remove_task(task_name)
+      @tasks.delete(task_name.to_s)
+    end
+  end
+
+  def remove_task(task_name)
+    Rake.application.remove_task(task_name)
+  end
+end
+
+alias drop_database_without_view_support drop_database
+def drop_database(config)
+  drop_database_without_view_support(config.merge('adapter' => config['adapter'].gsub(/_with_views/, "")))
+end
+
+alias create_database_without_view_support create_database
+def create_database(config)
+  create_database_without_view_support(config.merge('adapter' => config['adapter'].gsub(/_with_views/, "")))
+end
+
+namespace :db do
+  namespace :views do
+    desc "Run views migration from db/views"
+    task :update => :environment do
+      view_migration_classes = Dir.glob(File.join(Rails.root, "db", "views", "*.rb")).map{|view_migration|
+        if File.basename(view_migration) =~ /[0-9]+_([^.]+).rb/
+          require view_migration
+          view_migration_class = "create_#{$1}_migration".camelize.constantize
+        end
+      }
+
+      view_migration_classes.each{|mc|
+        mc.migrate(:up)
+      }
+    end
+  end
+  namespace :test do
+    original_purge_task = Rake::Task["db:test:purge"]
+    remove_task "db:test:purge"
+
+    desc "Empty the test database"
+    task :purge => :environment do
+      abcs = ActiveRecord::Base.configurations
+      if abcs["test"]["adapter"] == "mysql_with_views"
+        ActiveRecord::Base.establish_connection(:test)
+        ActiveRecord::Base.connection.recreate_database(abcs["test"]["database"], abcs["test"])
+      else
+        original_purge_task.invoke
+      end
+    end
+  end
+
+end
+
+# We need to invoke the db:views:update task IMIDIATELY after the db:migrate task.
+# Failure to do so will cause broken schema dumps to happen.
+Rake::Task["db:migrate"].actions.unshift lambda{Rake::Task["db:views:update"].invoke}
+
+
+

metadata

@@ -0,0 +1,58 @@
+--- !ruby/object:Gem::Specification 
+name: mysql_view_support
+version: !ruby/object:Gem::Version 
+  version: 0.2.0
+platform: ruby
+authors: 
+- Tom Lea
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-01-08 00:00:00 +00:00
+default_executable: 
+dependencies: []
+
+description: 
+email: contrib@tomlea.co.uk
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- tasks/views.rake
+- generators/sql_view_migration/sql_view_migration_generator.rb
+- generators/sql_view_migration/templates/migration.rb
+- lib/active_record/connection_adapters/mysql_with_views_adapter.rb
+has_rdoc: true
+homepage: http://labs.reevoo.com
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Provides view support to rails.
+test_files: []
+