triggerhappy 0.0.6 → 0.1.0

data/README

@@ -2,6 +2,8 @@
 
 Library which adds SQL Triggers to Rails. Adds create_trigger and drop_trigger to the ActiveRecord::ConnectionAdapters::AbstractAdapter (which makes them available to migrations) and adds support for dumping triggers in the ActiveRecord::SchemaDumper.
 
+Now supports creation / removal of stored procedures in migrations!
+
 == Installation
 
 To install:
@@ -69,4 +71,4 @@ If you find any issues please send an email to stewbawka@gmail.com .
 
 == Contributing
 
-If you would like to implement trigger support for other adapters then please drop me an email. Better yet, write up the adapter modifications and send them to me. :-)
+If you would like to implement trigger support for other adapters then please drop me an email. Better yet, write up the adapter modifications and send them to me. :-)

data/lib/core_ext/module.rb

@@ -0,0 +1,11 @@
+# This is required for 1.1.6 support
+class Module
+  def alias_method_chain(target, feature)
+    # Strip out punctuation on predicates or bang methods since
+    # e.g. target?_without_feature is not a valid method name.
+    aliased_target, punctuation = target.to_s.sub(/([?!=])$/, ''), $1
+    yield(aliased_target, punctuation) if block_given?
+    alias_method "#{aliased_target}_without_#{feature}#{punctuation}", target
+    alias_method target, "#{aliased_target}_with_#{feature}#{punctuation}"
+  end
+end

data/lib/triggerhappy/connection_adapters/abstract/schema_definitions.rb

@@ -12,5 +12,16 @@ module Triggerhappy
         @sql = sql
       end
     end
+
+    class ProcDefinition
+      attr_accessor :name, :sql
+      
+      def initialize(base, name, sql)
+        @base = base
+        @name = name
+        @sql = sql
+      end
+    end
+
   end
-end
+end

data/lib/triggerhappy/connection_adapters/abstract/schema_statements.rb

@@ -24,6 +24,31 @@ module Triggerhappy
           execute drop_sql
         end
       end
+
+      # Create a proc.
+      def create_proc(name, sql, options={})
+        if supports_procs?
+          proc_definition = ProcDefinition.new(self, name, sql)
+
+          if options[:force]
+            drop_proc(name) rescue nil
+          end
+
+          create_sql = "CREATE PROCEDURE "
+          create_sql << "#{name}"
+          create_sql << proc_definition.sql
+          execute create_sql
+        end
+      end
+
+      # Drop a proc.
+      def drop_proc(name, options={})
+        if supports_procs?
+          drop_sql = "DROP PROCEDURE #{name}"
+          execute drop_sql
+        end
+      end
+
     end
   end
-end
+end

data/lib/triggerhappy/connection_adapters/abstract_adapter.rb

@@ -26,6 +26,21 @@ module ActiveRecord
         raise NotImplementedError, "trigger_events is an abstract method"
       end
 
+      # Subclasses should override and return true if they support procs.
+      def supports_procs?
+        return false
+      end
+      
+      # Get a list of all procs for the current database
+      def procs(name = nil)
+        raise NotImplementedError, "procs is an abstract method"
+      end
+      
+      # Get the code for the specified proc
+      def proc_code(proc, name=nil)
+        raise NotImplementedError, "proc_code is an abstract method"
+      end
+
     end
   end
-end
+end

data/lib/triggerhappy/connection_adapters/sqlserver_adapter.rb

@@ -1,16 +1,26 @@
 module ActiveRecord
   module ConnectionAdapters
     class SQLServerAdapter
-      # Returns true as this adapter supports views.
       def supports_triggers?
         true
       end
+
+      def supports_procs?
+        true
+      end
+
       
       # Returns all the trigger names from the currently connected schema.
       def triggers(name = nil)
         select_values("SELECT name FROM sys.triggers", name)
       end
 
+      # Returns all the proc names from the currently connected schema.
+      def procs(name = nil)
+        select_values("SELECT name FROM sysobjects WHERE type = 'P'", name)
+      end
+
+
       def trigger_sp_helptext(trigger, name=nil)
        q =<<-ENDSQL
 SELECT OBJECT_DEFINITION(OBJECT_ID('#{trigger}'));
@@ -43,7 +53,27 @@ ENDSQL
         trigger_def = trigger_sp_helptext(trigger, name)
         trigger_def.sub(/^CREATE.* AS /, '')       
       end 
-     
+
+      def proc_sp_helptext(proc, name=nil)
+       q =<<-ENDSQL
+SELECT OBJECT_DEFINITION(OBJECT_ID('#{proc}'));
+ENDSQL
+
+       proc_def = select_value(q, name)
+ 
+       if !proc_def.blank?
+          return proc_def
+        else
+          raise "No proc called #{proc} found"
+        end
+ 
+      end
+
+      def proc_code(proc, name=nil)
+        proc_def = proc_sp_helptext(proc, name)
+        proc_def.sub(/^CREATE PROCEDURE \w+\s+/, '')      
+      end 
+  
     end
   end
-end
+end

data/lib/triggerhappy/schema_dumper.rb

@@ -5,20 +5,25 @@ module ActiveRecord
     # Acceptable values are strings as well as regexp.
     # This setting is only used if ActiveRecord::Base.schema_format == :ruby
     cattr_accessor :ignore_triggers 
+    cattr_accessor :ignore_procs
     @@ignore_triggers = []
+    @@ignore_procs = []
     
     def trailer_with_triggers(stream)
       # do nothing...we'll call this later
     end
     alias_method_chain :trailer, :triggers
     
-    # Add triggers to the end of the dump stream
+    # Add triggers/procs to the end of the dump stream
     def dump_with_triggers(stream)
       dump_without_triggers(stream)
       begin
         if @connection.supports_triggers?
           triggers(stream)
         end
+        if @connection.supports_procs?
+          procs(stream)
+        end
       rescue => e
         ActiveRecord::Base.logger.error "Unable to dump triggers: #{e}"
       end
@@ -35,12 +40,28 @@ module ActiveRecord
           when String: t == ignored
           when Regexp: t =~ ignored
           else
-            raise StandardError, 'ActiveRecord::SchemaDumper.ignore_views accepts an array of String and / or Regexp values.'
+            raise StandardError, 'ActiveRecord::SchemaDumper.ignore_triggers accepts an array of String and / or Regexp values.'
           end
         end 
         trigger(t, stream)
       end
     end
+
+    # Add procs to the stream
+    def procs(stream)
+      @connection.procs.sort.each do |p|
+        next if [ignore_procs].flatten.any? do |ignored|
+          case ignored
+          when String: p == ignored
+          when Regexp: p =~ ignored
+          else
+            raise StandardError, 'ActiveRecord::SchemaDumper.ignore_procs accepts an array of String and / or Regexp values.'
+          end
+        end 
+        proc(p, stream)
+      end
+    end
+
     
     # Add the specified trigger to the stream
     def trigger(trigger, stream)
@@ -64,5 +85,26 @@ module ActiveRecord
       
       stream
     end
+
+    # Add the specified proc to the stream
+    def proc(proc, stream)
+      begin
+        t = StringIO.new
+        t.print "  proc_code = <<EOSQL\n"
+        t.print @connection.proc_code(proc)
+        t.print "EOSQL\n"
+        t.print "  create_proc(#{proc.to_sym.inspect}, "
+        t.print "proc_code, "
+        t.print ":force => true)\n"
+        t.rewind
+        stream.print t.read
+      rescue => e
+        stream.puts "# Could not dump proc #{proc.inspect} because of following #{e.class}"
+        stream.puts "#   #{e.message}"
+        stream.puts "#   #{e.trace}"
+      end
+      
+      stream
+    end
   end
-end
+end

data/lib/triggerhappy/version.rb

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

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: triggerhappy
 version: !ruby/object:Gem::Version 
-  hash: 19
+  hash: 27
   prerelease: 
   segments: 
   - 0
+  - 1
   - 0
-  - 6
-  version: 0.0.6
+  version: 0.1.0
 platform: ruby
 authors: 
 - Stuart Wade
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-13 00:00:00 -03:00
+date: 2011-08-11 00:00:00 -03:00
 default_executable: 
 dependencies: []
 
@@ -28,20 +28,18 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
-- lib/triggerhappy/connection_adapters/abstract/schema_definitions.rb
-- lib/triggerhappy/connection_adapters/abstract/schema_statements.rb
-- lib/triggerhappy/connection_adapters/abstract_adapter.rb
+- lib/core_ext/module.rb
+- lib/triggerhappy.rb
+- lib/triggerhappy/version.rb
+- lib/triggerhappy/schema_dumper.rb
 - lib/triggerhappy/connection_adapters/mysql_adapter.rb
 - lib/triggerhappy/connection_adapters/sqlserver_adapter.rb
-- lib/triggerhappy/schema_dumper.rb
-- lib/triggerhappy/version.rb
-- lib/triggerhappy.rb
+- lib/triggerhappy/connection_adapters/abstract_adapter.rb
+- lib/triggerhappy/connection_adapters/abstract/schema_statements.rb
+- lib/triggerhappy/connection_adapters/abstract/schema_definitions.rb
 - CHANGELOG
 - Rakefile
 - README
-- triggerhappy-0.0.4.gem
-- triggerhappy-0.0.5.gem
-- triggerhappy.gemspec
 has_rdoc: true
 homepage: http://blog.stewbawka.com
 licenses: []
@@ -74,7 +72,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: triggerhappy
-rubygems_version: 1.5.2
+rubygems_version: 1.5.3
 signing_key: 
 specification_version: 3
 summary: sql triggers in migrations

data/triggerhappy.gemspec

@@ -1,14 +0,0 @@
-Gem::Specification.new do |s|
-  s.name        = "triggerhappy"
-  s.version     = "0.0.6"
-  s.author      = "Stuart Wade"
-  s.email       = "stewbawka@gmail.com"
-  s.homepage    = "http://blog.stewbawka.com"
-  s.summary = "sql triggers in migrations"
-  s.description = "Provide create_trigger and drop_trigger in migrations. Also extends SchemaDumper to clue in about triggers"
-  s.files        = Dir["{lib,test}/**/*"] + Dir["[A-Z]*"]
-  s.require_path = "lib"
-  
-  s.rubyforge_project = s.name
-  s.required_rubygems_version = ">= 1.3.4"
-end