qubole_rails_sql_views 0.10.0

data/lib/rails_sql_views/connection_adapters/postgresql_adapter.rb

@@ -0,0 +1,65 @@
+module RailsSqlViews
+  module ConnectionAdapters
+    module PostgreSQLAdapter
+      def self.included(base)
+        base.alias_method_chain :tables, :views_included
+      end
+      # Returns true as this adapter supports views.
+      def supports_views?
+        true
+      end
+      
+      def tables_with_views_included(name = nil)
+        q = <<-SQL
+        SELECT table_name, table_type
+          FROM information_schema.tables
+         WHERE table_schema IN (#{schemas})
+           AND table_type IN ('BASE TABLE', 'VIEW')
+        SQL
+
+        query(q, name).map { |row| row[0] }
+      end
+      
+      def base_tables(name = nil)
+        q = <<-SQL
+        SELECT table_name, table_type
+          FROM information_schema.tables
+         WHERE table_schema IN (#{schemas})
+           AND table_type = 'BASE TABLE'
+        SQL
+        
+        query(q, name).map { |row| row[0] }
+      end
+      alias nonview_tables base_tables
+      
+      def views(name = nil) #:nodoc:
+        q = <<-SQL
+        SELECT table_name, table_type
+          FROM information_schema.tables
+         WHERE table_schema IN (#{schemas})
+           AND table_type = 'VIEW'
+        SQL
+        
+        query(q, name).map { |row| row[0] }
+      end
+
+      def view_select_statement(view, name = nil)
+        q = <<-SQL
+        SELECT view_definition
+          FROM information_schema.views 
+         WHERE table_catalog = (SELECT catalog_name FROM information_schema.information_schema_catalog_name)
+           AND table_schema IN (#{schemas})
+           AND table_name = '#{view}'
+        SQL
+        
+        select_value(q, name) or raise "No view called #{view} found"
+      end
+
+      private
+        
+      def schemas
+        schema_search_path.split(/,/).map { |p| quote(p) }.join(',')
+      end
+    end
+  end
+end

data/lib/rails_sql_views/connection_adapters/postgresql_adapter.rb.orig

@@ -0,0 +1,69 @@
+module RailsSqlViews
+  module ConnectionAdapters
+    module PostgreSQLAdapter
+      def self.included(base)
+<<<<<<< HEAD
+        base.alias_method_chain :tables, :views_included unless method_defined?(:tables_with_views_included)
+=======
+        base.alias_method_chain :tables, :views_included
+>>>>>>> Make tests pass again for PostgreSQL and MySQL adapters.
+      end
+      # Returns true as this adapter supports views.
+      def supports_views?
+        true
+      end
+      
+      def tables_with_views_included(name = nil)
+        q = <<-SQL
+        SELECT table_name, table_type
+          FROM information_schema.tables
+         WHERE table_schema IN (#{schemas})
+           AND table_type IN ('BASE TABLE', 'VIEW')
+        SQL
+
+        query(q, name).map { |row| row[0] }
+      end
+      
+      def base_tables(name = nil)
+        q = <<-SQL
+        SELECT table_name, table_type
+          FROM information_schema.tables
+         WHERE table_schema IN (#{schemas})
+           AND table_type = 'BASE TABLE'
+        SQL
+        
+        query(q, name).map { |row| row[0] }
+      end
+      alias nonview_tables base_tables
+      
+      def views(name = nil) #:nodoc:
+        q = <<-SQL
+        SELECT table_name, table_type
+          FROM information_schema.tables
+         WHERE table_schema IN (#{schemas})
+           AND table_type = 'VIEW'
+        SQL
+        
+        query(q, name).map { |row| row[0] }
+      end
+
+      def view_select_statement(view, name = nil)
+        q = <<-SQL
+        SELECT view_definition
+          FROM information_schema.views 
+         WHERE table_catalog = (SELECT catalog_name FROM information_schema.information_schema_catalog_name)
+           AND table_schema IN (#{schemas})
+           AND table_name = '#{view}'
+        SQL
+        
+        select_value(q, name) or raise "No view called #{view} found"
+      end
+
+      private
+        
+      def schemas
+        schema_search_path.split(/,/).map { |p| quote(p) }.join(',')
+      end
+    end
+  end
+end

data/lib/rails_sql_views/connection_adapters/sqlite_adapter.rb

@@ -0,0 +1,66 @@
+module RailsSqlViews
+  module ConnectionAdapters
+    module SQLiteAdapter
+      def supports_views?
+        true
+      end
+
+      def supports_drop_table_cascade?
+        return false 
+      end
+      
+      def tables(name = nil) #:nodoc:
+        sql = <<-SQL
+          SELECT name
+          FROM sqlite_master
+          WHERE (type = 'table' OR type = 'view') AND NOT name = 'sqlite_sequence'
+        SQL
+
+        execute(sql, name).map do |row|
+          row[0]
+        end
+      end
+
+      def base_tables(name = nil)
+        sql = <<-SQL
+          SELECT name
+          FROM sqlite_master
+          WHERE (type = 'table') AND NOT name = 'sqlite_sequence'
+        SQL
+
+        execute(sql, name).map do |row|
+          row[0]
+        end        
+      end
+      alias nonview_tables base_tables
+      
+      def views(name = nil)
+        sql = <<-SQL
+          SELECT name
+          FROM sqlite_master
+          WHERE type = 'view' AND NOT name = 'sqlite_sequence'
+        SQL
+
+        execute(sql, name).map do |row|
+          row[0]
+        end
+      end
+      
+      # Get the view select statement for the specified table.
+      def view_select_statement(view, name = nil)
+        sql = <<-SQL
+          SELECT sql
+          FROM sqlite_master
+          WHERE name = '#{view}' AND NOT name = 'sqlite_sequence'
+        SQL
+        
+        (select_value(sql, name).gsub("CREATE VIEW #{view} AS ", "")) or raise "No view called #{view} found"
+      end
+      
+      def supports_view_columns_definition?
+        false
+      end
+      
+    end
+  end
+end

data/lib/rails_sql_views/connection_adapters/sqlserver_adapter.rb

@@ -0,0 +1,43 @@
+module RailsSqlViews
+  module ConnectionAdapters
+    module SQLServerAdapter
+      # Returns true as this adapter supports views.
+      def supports_views?
+        true
+      end
+      
+      # Get all of the non-view tables from the currently connected schema
+      def base_tables(name = nil)
+        # this is untested
+        select_values("SELECT table_name FROM information_schema.tables", name)
+      end
+      alias nonview_tables base_tables
+      
+      # Returns all the view names from the currently connected schema.
+      def views(name = nil)
+        select_values("SELECT table_name FROM information_schema.views", name)
+      end
+      
+      # Get the view select statement for the specified view.
+      def view_select_statement(view, name=nil)
+        q =<<-ENDSQL
+          SELECT view_definition FROM information_schema.views
+          WHERE table_name = '#{view}'
+        ENDSQL
+        
+        view_def = select_value(q, name)
+        
+        if view_def
+          return convert_statement(view_def)
+        else
+          raise "No view called #{view} found"
+        end
+      end
+      
+      private
+      def convert_statement(s)
+        s.sub(/^CREATE.* AS (select .*)/i, '\1').gsub(/\n/, '')
+      end
+    end
+  end
+end

data/lib/rails_sql_views/loader.rb

@@ -0,0 +1,18 @@
+
+module RailsSqlViews
+  module Loader
+    SUPPORTED_ADAPTERS = %w( Mysql Mysql2 PostgreSQL SQLServer SQLite OracleEnhanced )
+
+    def self.load_extensions
+      SUPPORTED_ADAPTERS.each do |db|
+        if ActiveRecord::ConnectionAdapters.const_defined?("#{db}Adapter")
+          require "rails_sql_views/connection_adapters/#{db.downcase}_adapter"
+          ActiveRecord::ConnectionAdapters.const_get("#{db}Adapter").class_eval do
+            include RailsSqlViews::ConnectionAdapters::AbstractAdapter
+            include RailsSqlViews::ConnectionAdapters.const_get("#{db}Adapter")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rails_sql_views/schema_dumper.rb

@@ -0,0 +1,112 @@
+module RailsSqlViews
+  module SchemaDumper
+    def self.included(base)
+      base.alias_method_chain :trailer, :views
+      base.alias_method_chain :dump, :views
+      base.alias_method_chain :tables, :views_excluded
+      
+      # A list of views which should not be dumped to the schema. 
+      # Acceptable values are strings as well as regexp.
+      # This setting is only used if ActiveRecord::Base.schema_format == :ruby
+      base.cattr_accessor :ignore_views
+      base.ignore_views = []
+      # Optional: specify the order that in which views are created.
+      # This allows views to depend on and include fields from other views.
+      # It is not necessary to specify all the view names, just the ones that 
+      # need to be created first
+      base.cattr_accessor :view_creation_order
+      base.view_creation_order = []
+    end
+    
+    def trailer_with_views(stream)
+      # do nothing...we'll call this later
+    end
+    
+    # Add views to the end of the dump stream
+    def dump_with_views(stream)
+      dump_without_views(stream)
+      begin
+        if @connection.supports_views?
+          views(stream)
+        end
+      rescue => e
+        if ActiveRecord::Base.logger
+          ActiveRecord::Base.logger.error "Unable to dump views: #{e}"
+        else
+          raise e
+        end
+      end
+      trailer_without_views(stream)
+      stream
+    end
+    
+    # Add views to the stream
+    def views(stream)
+      if view_creation_order.empty?
+        sorted_views = @connection.views.sort
+      else
+        # set union, merge by joining arrays, removing dups
+        # this will float the view name sin view_creation_order to the top
+        # without requiring all the views to be specified
+        sorted_views = view_creation_order | @connection.views
+      end
+      sorted_views.each do |v|
+        next if [ActiveRecord::Migrator.schema_migrations_table_name, ignore_views].flatten.any? do |ignored|
+          case ignored
+          when String then v == ignored
+          when Symbol then v == ignored.to_s
+          when Regexp then v =~ ignored
+          else
+            raise StandardError, 'ActiveRecord::SchemaDumper.ignore_views accepts an array of String and / or Regexp values.'
+          end
+        end 
+        view(v, stream)
+      end
+    end
+    
+    # Add the specified view to the stream
+    def view(view, stream)
+      columns = @connection.columns(view).collect { |c| c.name }
+      begin
+        v = StringIO.new
+
+        v.print "  create_view #{view.inspect}"
+        v.print ", #{@connection.view_select_statement(view).dump}"
+        v.print ", :force => true"
+        v.puts " do |v|"
+
+        columns.each do |column|
+          v.print "    v.column :#{column}"
+          v.puts
+        end
+
+        v.puts "  end"
+        v.puts
+        
+        v.rewind
+        stream.print v.read
+      rescue => e
+        stream.puts "# Could not dump view #{view.inspect} because of following #{e.class}"
+        stream.puts "#   #{e.message}"
+        stream.puts
+      end
+      
+      stream
+    end
+
+    def tables_with_views_excluded(stream)
+      @connection.base_tables.sort.each do |tbl|
+        next if [ActiveRecord::Migrator.schema_migrations_table_name, ignore_tables].flatten.any? do |ignored|
+          case ignored
+          when String then tbl == ignored
+          when Regexp then tbl =~ ignored
+          else
+            raise StandardError, 'ActiveRecord::SchemaDumper.ignore_tables accepts an array of String and / or Regexp values.'
+          end
+        end
+        table(tbl, stream)
+      end
+    end
+
+  end
+end

data/lib/rails_sql_views/version.rb

@@ -0,0 +1,9 @@
+module RailsSqlViews
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 7
+    TINY  = 0
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/lib/rails_sql_views.rb

@@ -0,0 +1,47 @@
+#--
+# Copyright (c) 2006 Anthony Eden
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#++
+
+$:.unshift(File.dirname(__FILE__))
+
+require 'active_record'
+
+require 'core_ext/module'
+
+require 'rails_sql_views/connection_adapters/abstract/schema_definitions'
+require 'rails_sql_views/connection_adapters/abstract/schema_statements'
+require 'rails_sql_views/connection_adapters/abstract_adapter'
+require 'rails_sql_views/schema_dumper'
+require 'rails_sql_views/loader'
+
+ActiveRecord::ConnectionAdapters::AbstractAdapter.class_eval do
+  include RailsSqlViews::ConnectionAdapters::SchemaStatements
+  def self.inherited(sub)
+    RailsSqlViews::Loader.load_extensions
+  end
+end
+
+ActiveRecord::SchemaDumper.class_eval do
+  include RailsSqlViews::SchemaDumper
+end
+
+RailsSqlViews::Loader.load_extensions

data/test/adapter_test.rb

@@ -0,0 +1,82 @@
+require "#{File.dirname(__FILE__)}/test_helper"
+
+class AdapterTest < Test::Unit::TestCase
+  def test_current_database
+    if ActiveRecord::Base.connection.respond_to?(:current_database)
+      assert_equal 'rails_sql_views_unittest', ActiveRecord::Base.connection.current_database
+    end
+  end
+  def test_tables
+    create_view
+    found = ActiveRecord::Base.connection.tables.sort
+    found.delete(ActiveRecord::Migrator.schema_migrations_table_name)
+    assert_equal ["items", "items_people", "people", "people2", "places", "v_people"], found
+  end
+  def test_base_tables
+    create_view
+    found = ActiveRecord::Base.connection.base_tables.sort
+    found.delete(ActiveRecord::Migrator.schema_migrations_table_name)
+    assert_equal ["items", "items_people", "people", "people2", "places"], found
+  end
+  def test_views
+    create_view
+    assert_equal ['v_people'], ActiveRecord::Base.connection.views
+  end
+  def test_columns
+    create_view
+    assert_equal ["f_name", "l_name", "social_security"], ActiveRecord::Base.connection.columns('v_people').collect { |c| c.name }
+  end
+  def test_supports_views
+    assert ActiveRecord::Base.connection.supports_views?
+  end
+  
+  def test_mapped_views
+    create_mapping
+    assert_equal ['v_people'], ActiveRecord::Base.connection.views
+  end
+  def test_mapped_columns
+    create_mapping
+    assert_equal ["f_name", "l_name", "address_id"], ActiveRecord::Base.connection.columns('v_people').collect { |c| c.name }
+  end
+  
+  def test_view_select_statement
+    case ActiveRecord::Base.connection.adapter_name
+    when "MySQL"
+      assert_equal "select `people`.`first_name` AS `f_name`,`people`.`last_name` AS `l_name`,`people`.`ssn` AS `social_security` from `people`", ActiveRecord::Base.connection.view_select_statement('v_people')
+    when "PostgreSQL"
+      assert_equal "SELECT people.first_name AS f_name, people.last_name AS l_name, people.ssn AS social_security FROM people;", ActiveRecord::Base.connection.view_select_statement('v_people')
+    end
+  end
+  
+  def test_old_name_not_found_error_during_mapping
+    assert_raise ActiveRecord::ActiveRecordError do
+      ActiveRecord::Base.connection.create_mapping_view(:people, :v_people, :force => true) do |v|
+        v.map_column :foo, :bar
+      end
+    end
+  end
+ 
+### TODO
+#  def test_only_base_table_triggers_are_dropped_for_disabled_ref_integrity
+#    ActiveRecord::Base.connection.disable_referential_integrity do
+#    end
+#  end
+  
+  private
+  def create_view
+    ActiveRecord::Base.connection.create_view(:v_people, 'select first_name, last_name, ssn from people', :force => true) do |v|
+      v.column :f_name
+      v.column :l_name
+      v.column :social_security
+    end
+  end
+  
+  def create_mapping
+    ActiveRecord::Base.connection.create_mapping_view(:people, :v_people, :force => true) do |v|
+      v.map_column :id, nil
+      v.map_column :first_name, :f_name
+      v.map_column :last_name, :l_name
+      v.map_column :ssn, nil
+    end
+  end
+end

data/test/connection/native_mysql/connection.rb

@@ -0,0 +1,32 @@
+print "Using native MySQL\n"
+
+adapter_name = 'mysql'
+config = YAML.load_file(File.join(File.dirname(__FILE__), '/../../connection.yml'))[adapter_name]
+
+#require 'logger'
+#ActiveRecord::Base.logger = Logger.new("debug.log")
+
+ActiveRecord::Base.silence do
+  ActiveRecord::Base.configurations = {
+    config['database'] => {
+    :adapter  => adapter_name,
+    :username => config['username'],
+    :password => config['password'],
+    :host     => config['host'],
+    :database => config['database'],
+    :encoding => config['encoding'],
+    :schema_file => config['schema_file'],
+  }
+  }
+
+  ActiveRecord::Base.establish_connection config['database']
+  ActiveRecord::Migration.verbose = false
+
+  puts "Resetting database"
+  conn = ActiveRecord::Base.connection
+  conn.recreate_database(conn.current_database)
+  conn.reconnect!
+  lines = open(File.join(File.dirname(__FILE__), ActiveRecord::Base.configurations[config['database']][:schema_file])).readlines
+  lines.join.split(';').each { |line| conn.execute(line) }
+  conn.reconnect!
+end

data/test/connection/native_postgresql/connection.rb

@@ -0,0 +1,31 @@
+print "Using native PostgreSQL\n"
+
+adapter_name = 'postgresql'
+config = YAML.load_file(File.join(File.dirname(__FILE__), '/../../connection.yml'))[adapter_name]
+
+#require 'logger'
+#ActiveRecord::Base.logger = Logger.new("debug.log")
+
+ActiveRecord::Base.silence do
+  ActiveRecord::Base.configurations = {
+  'rails_sql_views_unittest' => {
+    :adapter  => adapter_name,
+    :username => config['username'],
+    :password => config['password'],
+    :host     => config['host'],
+    :database => config['database'],
+    :encoding => config['encoding'],
+    :schema_file => config['schema_file'],
+  }
+  }
+
+  ActiveRecord::Base.establish_connection config['database']
+
+  puts "Resetting database"
+  conn = ActiveRecord::Base.connection
+  #conn.recreate_database(conn.current_database)
+  conn.reconnect!
+  lines = open(File.join(File.dirname(__FILE__), ActiveRecord::Base.configurations[config['database']][:schema_file])).readlines
+  lines.join.split(';').each { |line| conn.execute(line) }
+  conn.reconnect!
+end

data/test/connection/oracle_enhanced/connection.rb

@@ -0,0 +1,29 @@
+print "Using Oracle Enhanced\n"
+
+#require 'logger'
+#ActiveRecord::Base.logger = Logger.new("debug.log")
+
+ActiveRecord::Base.configurations = {
+  'rails_sql_views_unittest' => {
+    :adapter  => :oracle_enhanced,
+    :username => 'rails_sql_views_unittest',
+    :password => 'rails',
+    :host     => 'localhost',
+    :database => 'mydev',
+    :encoding => 'utf8',
+    :procedures_file  => 'procedures.sql',
+    :schema_file      => 'schema.sql',
+  }
+}
+
+ActiveRecord::Base.establish_connection 'rails_sql_views_unittest'
+
+puts "Resetting database"
+conn = ActiveRecord::Base.connection
+#conn.recreate_database(conn.current_database)
+conn.reconnect!
+[:procedures_file, :schema_file].each do |file|
+  lines = open(File.join(File.dirname(__FILE__), ActiveRecord::Base.configurations['rails_sql_views_unittest'][file])).readlines
+  conn.execute(lines.to_s)
+end
+conn.reconnect!

data/test/models/item.rb

@@ -0,0 +1,4 @@
+class Item < ActiveRecord::Base
+  belongs_to :person
+  has_and_belongs_to_many :people
+end

data/test/models/person.rb

@@ -0,0 +1,5 @@
+class Person < ActiveRecord::Base
+  belongs_to :address, :class_name => 'Place', :foreign_key => :address_id
+  has_many :owned_items, :class_name => 'Item'
+  has_and_belongs_to_many :shared_items, :class_name => 'Item'
+end

data/test/models/person2.rb

@@ -0,0 +1,3 @@
+class Person2 < ActiveRecord::Base
+  set_table_name :people2
+end

data/test/models/place.rb

@@ -0,0 +1,2 @@
+class Place < ActiveRecord::Base
+end

data/test/models/v_person.rb

@@ -0,0 +1,4 @@
+require 'active_record/view'
+
+class VPerson < ActiveRecord::View
+end

data/test/models/v_profile.rb

@@ -0,0 +1,3 @@
+class VProfile < ActiveRecord::Base
+  
+end