Merge branch 'master' of git://github.com/jeremydurham/rails-sqlserver-adapter

Although Jeremy didn't comment it, he also changed over to using SCOPE_IDENTITY instead of @@IDENTITY, which is a Good Thing when you have triggers, etc.

Author: cjheath

lib/active_record/connection_adapters/sqlserver_adapter.rb

@@ -89,7 +89,7 @@ def self.construct_finder_sql_for_association_limiting(options, join_dependency)
 
   module ConnectionAdapters
     class SQLServerColumn < Column# :nodoc:
-      attr_reader :identity, :is_special
+      attr_reader :identity, :is_special, :is_utf8
 
       def initialize(info)
         if info[:type] =~ /numeric|decimal/i
@@ -100,6 +100,7 @@ def initialize(info)
         super(info[:name], info[:default_value], type, info[:is_nullable])
         @identity = info[:is_identity]
         @is_special = ["text", "ntext", "image"].include?(info[:type])
+        @is_utf8 = type =~ /nvarchar|ntext/i
         # TODO: check ok to remove @scale = scale_value
         @limit = nil unless limitable?(type)
       end
@@ -386,10 +387,12 @@ def empty_insert_statement(table_name)
       end
 
       def insert_sql(sql, name = nil, pk = nil, id_value = nil, sequence_name = nil)
+        set_utf8_values!(sql)
         super || select_value("SELECT SCOPE_IDENTITY() AS Ident")
       end
 
       def update_sql(sql, name = nil)
+        set_utf8_values!(sql)
         auto_commiting = @connection["AutoCommit"]
         begin
           begin_db_transaction if auto_commiting
@@ -716,7 +719,7 @@ def set_identity_insert(table_name, enable = true)
 
         def get_table_name(sql)
           if sql =~ /^\s*insert\s+into\s+([^\(\s]+)\s*|^\s*update\s+([^\(\s]+)\s*/i
-            $1
+            $1 || $2
           elsif sql =~ /from\s+([^\(\s]+)\s*/i
             $1
           else
@@ -768,6 +771,40 @@ def repair_special_columns(sql)
           end
           sql
         end
+        
+        def get_utf8_columns(table_name)
+          utf8 = []
+          @table_columns ||= []
+          @table_columns[table_name] ||= columns(table_name)
+          @table_columns[table_name].each do |col|
+            utf8 << col.name if col.is_utf8
+          end
+          utf8
+        end
+        
+        def set_utf8_values!(sql)
+          utf8_cols = get_utf8_columns(get_table_name(sql))
+          if sql =~ /^\s*UPDATE/i            
+            utf8_cols.each do |col|
+              sql.gsub!("[#{col.to_s}] = '", "[#{col.to_s}] = N'")
+            end
+          elsif sql =~ /^\s*INSERT/i
+            # TODO This code should be simplified
+            # Get columns and values, split them into arrays, and store the original_values for when we need to replace them
+            columns_and_values = sql.scan(/\((.*?)\)/).flatten
+            columns = columns_and_values.first.split(',')
+            values =  columns_and_values[1].split(',')
+            original_values = values.dup
+            # Iterate columns that should be UTF8, and append an N to the value, if the value is not NULL
+            utf8_cols.each do |col|
+              columns.each_with_index do |column, idx|
+                values[idx] = " N#{values[idx].gsub(/^ /, '')}" if column =~ /\[#{col}\]/ and values[idx] !~ /^NULL$/
+              end
+            end
+            # Replace (in place) the SQL
+            sql.gsub!(original_values.join(','), values.join(','))
+          end
+        end
 
     end #class SQLServerAdapter < AbstractAdapter
   end #module ConnectionAdapters