stratagem 0.2.0 → 0.2.2

data/lib/stratagem/{framework_extensions → instrumentation}/method_invocation.rb

@@ -1,4 +1,4 @@
-module Stratagem::ApplicationExtensions
+module Stratagem::Instrumentation
   class MethodInvocation
     attr_accessor :method, :controller_path, :controller_action, :line_number, :model_instance, :model_class, :stack_trace, :args, :type
 

data/lib/stratagem/instrumentation/models.rb

@@ -0,0 +1,25 @@
+module Stratagem::Instrumentation::Models; end
+module Stratagem::Instrumentation::Models::Adapters; end
+
+require 'stratagem/instrumentation/models/mocking'
+require 'stratagem/instrumentation/models/metadata'
+require 'stratagem/instrumentation/models/tracing'
+require 'stratagem/instrumentation/models/annotations'
+require 'stratagem/instrumentation/models/detect'
+
+require 'stratagem/instrumentation/models/persistence'
+require 'stratagem/instrumentation/models/authentication'
+require 'stratagem/instrumentation/models/support_libraries'
+
+# base = File.join(File.dirname(__FILE__), 'models', 'persistence', 'util')
+# Dir.entries(base).select {|s| s =~ /\.rb$/}.each {|helper|
+#   require File.join(base, helper.gsub(/\.rb/, ''))
+# }
+# 
+# base = File.join(File.dirname(__FILE__), 'models', 'persistence')
+# Dir.entries(base).select {|s| s !~ /^\./ && s != 'util' }.each {|adapter_dir|
+#   require File.join(base, adapter_dir, 'detect')
+#   require File.join(base, adapter_dir, 'tracing')
+#   require File.join(base, adapter_dir, 'metadata')
+#   require File.join(base, adapter_dir, 'extensions')
+# }

data/lib/stratagem/instrumentation/models/annotations.rb

@@ -0,0 +1,114 @@
+# Defines the stratagem namespace attached to the model
+module Stratagem::Instrumentation::Models
+  ValidatorDefinition = Struct.new(:validation, :field, :args, :model_class)
+
+  class InstanceAnnotations
+    include Mocking
+
+    def initialize(object)
+      @object = object
+    end
+
+    def method_missing(method, *args, &block)
+      @object.class.stratagem.send(method, *args, &block)
+    end
+  end
+
+  class Annotations
+    include Metadata
+    include Tracing
+
+    attr_reader :model
+
+    AdapterDescriptor = Struct.new(:metadata, :detector, :tracing, :instrumentation)
+
+    class << self
+      def configure(model)
+        puts "configuring #{model.name}"
+        
+        # add the stratagem namespace
+        model.class_eval do 
+          def self.stratagem
+            # one stratagem instance per subclass
+            @@stratagem ||= {}
+            @@stratagem[self] ||= Stratagem::Instrumentation::Models::Annotations.new(self)
+          end
+
+
+          def stratagem
+            @stratagem ||= Stratagem::Instrumentation::Models::InstanceAnnotations.new(self)
+          end
+        end
+      
+        # connect the adapters
+        detect_adapters(model).each {|adapter|
+          if adapter.detector.supports?(model)
+            puts "#{model.name} supports #{adapter.tracing.name}"
+            model.send(:include, adapter.tracing) 
+          end
+        }
+      end
+
+      def detect_adapters(model)
+        Detect.sg_subclasses.map do |detector|
+          namespace = detector.name.split('::')
+          namespace.pop
+          namespace = namespace.join('::')
+          instrumentation = module_class_ref(namespace+"::Instrumentation")
+          tracing = module_class_ref(namespace+"::Tracing")
+          metadata = module_class_ref(namespace+"::Metadata")
+          metadata = metadata.new(model) if metadata
+          AdapterDescriptor.new(metadata, detector, tracing, instrumentation)
+        end
+      end
+
+      def module_class_ref(name)
+        begin
+          module_eval(name)
+        rescue Exception
+          nil
+        end
+      end
+
+
+    end
+
+    def initialize(model)
+      puts "initializing stratagem for #{model.name} - #{self.class.name}"
+      @model = model
+      self.class.detect_adapters(model).each do |adapter|
+        if adapter.detector.supports?(model)
+          # puts "\t#{model.name} supports #{adapter.detector.name}"
+          instrument_model(adapter)
+        # else
+        #   puts "#{model.name} does not support #{adapter.detector.name}"
+        end
+      end
+    rescue
+      puts $!.message
+      puts $!.backtrace
+    end
+
+    def instrument_model(adapter)
+      instrument_model_with(adapter.tracing)
+      instrument_model_with(adapter.instrumentation)
+    end
+
+    def instrument_model_with(instrumentation)
+      return if instrumentation.nil?
+      
+      unless model.ancestors.include?(instrumentation)
+        puts "including #{instrumentation.name} in #{model.name}"
+        model.send(:include, instrumentation)
+      end
+    end
+
+    def adapters
+      # supported adapters may change throughout the lifecycle of a class / object
+      @adapters ||= self.class.detect_adapters(model)
+    end
+
+
+  end
+
+end

data/lib/stratagem/instrumentation/models/association.rb

@@ -0,0 +1,40 @@
+module Stratagem::Instrumentation::Models
+  class Association
+    attr_reader :from_class, :name, :foreign_key, :macro, :options
+
+    def initialize(from_class, name, foreign_key, klass, macro, options)
+      @from_class = from_class
+      @name = name
+      @foreign_key = foreign_key
+      @klass = klass
+      @macro = macro
+      @options = options
+    end
+    
+    def klass
+      if (@klass.kind_of?(String))
+        if (options[:polymorphic])
+          @poly_count ||= 0
+          
+          # all possibilities
+          possible_classes = ActiveRecord::Base.sg_subclasses.sort {|a,b| a.name <=> b.name }.select do |sc|
+            sc.stratagem.relations(:has_many).find {|r| r.klass == from_class } != nil
+          end
+
+          if (possible_classes.size > 0)
+            @poly_count += 1
+            @poly_count = 1 if @poly_count > possible_classes.size
+            return possible_classes[@poly_count-1]
+          else
+            return nil
+          end
+        else
+          # unknown why the class could not be loaded
+          return nil
+        end
+      else
+        return @klass
+      end
+    end
+  end
+end

data/lib/stratagem/instrumentation/models/authentication.rb

@@ -0,0 +1,7 @@
+module Stratagem::Instrumentation::Models::Authentication; end
+
+pattern = File.join(File.dirname(__FILE__), 'authentication', '**', '*.rb')
+Dir.glob(pattern).each {|filename|
+  require filename
+}
+

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/authentication}/authlogic/detect.rb

@@ -1,5 +1,5 @@
-module Stratagem::ApplicationExtensions::Models::Adapters::Authlogic
-  class Detect < Stratagem::ApplicationExtensions::Models::Detect
+module Stratagem::Instrumentation::Models::Authentication::Authlogic
+  class Detect < Stratagem::Instrumentation::Models::Detect
     def self.supports?(model)
       begin
         model.ancestors.include?(::Authlogic::ActsAsAuthentic::MagicColumns::Methods)

data/lib/stratagem/instrumentation/models/authentication/authlogic/instrumentation.rb

@@ -0,0 +1,13 @@
+module Stratagem::Instrumentation::Models::Authentication::Authlogic
+  module Instrumentation
+    def self.included(model)
+      model.class_eval do
+        before_save { |record| 
+          if record.attribute_names.include?('active') 
+            record.active = true 
+          end
+        }
+      end
+    end
+  end
+end

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/authentication}/authlogic/metadata.rb

@@ -1,9 +1,7 @@
-module Stratagem::ApplicationExtensions::Models::Adapters::Authlogic
+module Stratagem::Instrumentation::Models::Authentication::Authlogic
 
   # prefix method names with  to avoid collision
   class Metadata
-    include Stratagem::ApplicationExtensions::Models::Adapters::Common::AuthenticationMetadata
-
     VIRTUAL_COLUMNS = [:password, :password_confirmation]
 
     def authenticates?

data/lib/stratagem/instrumentation/models/authentication/devise/detect.rb

@@ -0,0 +1,11 @@
+module Stratagem::Instrumentation::Models::Authentication::Devise
+  class Detect < Stratagem::Instrumentation::Models::Detect
+    def self.supports?(model)
+      begin
+        model.ancestors.find {|a| a.name.include? 'Devise::Models' } != nil
+      rescue
+        false
+      end
+    end
+  end
+end

data/lib/stratagem/instrumentation/models/authentication/devise/instrumentation.rb

@@ -0,0 +1,18 @@
+module Stratagem::Instrumentation::Models::Authentication::Devise
+  module Instrumentation
+    def self.included(model)
+      model.class_eval do
+        before_save { |record| 
+          if record.attribute_names.include?('active') 
+            puts "activating user"
+            record.active = true 
+          end
+        }
+        
+        after_save {|record|
+          p record
+        }
+      end
+    end
+  end
+end

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/authentication}/devise/metadata.rb

@@ -1,9 +1,13 @@
-module Stratagem::ApplicationExtensions::Models::Adapters::Devise
+module Stratagem::Instrumentation::Models::Authentication::Devise
 
   # prefix method names with  to avoid collision
   class Metadata
-    include Stratagem::ApplicationExtensions::Models::Adapters::Common::AuthenticationMetadata
-
+    begin
+      include Stratagem::Instrumentation::Models::Authentication::Metadata
+    rescue
+      puts $!.message
+    end
+    
     VIRTUAL_COLUMNS = [:password, :password_confirmation]
 
     def authenticates?

data/lib/stratagem/{framework_extensions/models/adapters/util/authentication_metadata.rb → instrumentation/models/authentication/metadata.rb}

@@ -1,7 +1,7 @@
-module Stratagem::ApplicationExtensions::Models::Adapters::Common
+module Stratagem::Instrumentation::Models::Authentication
 
   # prefix method names with  to avoid collision
-  module AuthenticationMetadata
+  module Metadata
     def exclude_attributes_for_mocking
       # open id
       attrs = @model.stratagem.attribute_names.select {|a|

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/authentication}/restful_authentication/detect.rb

@@ -1,7 +1,5 @@
-
-
-module Stratagem::ApplicationExtensions::Models::Adapters::RestfulAuthentication
-  class Detect < Stratagem::ApplicationExtensions::Models::Detect
+module Stratagem::Instrumentation::Models::Authentication::RestfulAuthentication
+  class Detect < Stratagem::Instrumentation::Models::Detect
     def self.supports?(model)
       begin
         model.ancestors.include?(::Authentication::ByPassword)

data/lib/stratagem/{framework_extensions/models/adapters/restful_authentication/extensions.rb → instrumentation/models/authentication/restful_authentication/instrumentation.rb}

@@ -1,5 +1,5 @@
 if (defined?(Authentication))
-  Authentication.subclasses.each do |model|
+  Authentication.sg_subclasses.each do |model|
     model.class_eval do
       # add support for aasm models
       before_save { |record| 

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/authentication}/restful_authentication/metadata.rb

@@ -1,8 +1,8 @@
-module Stratagem::ApplicationExtensions::Models::Adapters::RestfulAuthentication
+module Stratagem::Instrumentation::Models::Authentication::RestfulAuthentication
 
   # prefix method names with  to avoid collision
   class Metadata
-    include Stratagem::ApplicationExtensions::Models::Adapters::Common::AuthenticationMetadata
+    include Stratagem::Instrumentation::Models::Authentication::Metadata
 
     VIRTUAL_COLUMNS = [:password, :password_confirmation]
 

data/lib/stratagem/{framework_extensions → instrumentation}/models/detect.rb

@@ -1,4 +1,4 @@
-module Stratagem::ApplicationExtensions::Models
+module Stratagem::Instrumentation::Models
   class Detect
     def self.supports?(model)
       false

data/lib/stratagem/{framework_extensions → instrumentation}/models/metadata.rb

@@ -1,8 +1,6 @@
 
-module Stratagem::ApplicationExtensions::Models
+module Stratagem::Instrumentation::Models
   module Metadata
-    StratagemAssociation = Struct.new(:name, :foreign_key, :klass, :macro, :options)
-
     INSTANCE_ENUMERATION_METHODS = [:relations, :attribute_names, :ignore_attributes, :internal_attributes, :unaccessible_attributes, :invalid_columns, :exclude_attributes_for_mocking]
     INSTANCE_ENTITY_METHODS = [:attribute_type, :column_from_error, :authenticates?, :whitelists_attributes?, :blacklists_attributes?]
 
@@ -27,6 +25,10 @@ module Stratagem::ApplicationExtensions::Models
 
     # Convenience methods
 
+    def subclasses?
+      model.sg_subclasses().size > 0
+    end
+
     def foreign_keys
       relations(:belongs_to).map {|relation| relation.foreign_key }
     end
@@ -75,7 +77,7 @@ module Stratagem::ApplicationExtensions::Models
             memory
           rescue
             puts "error running callbacks: #{$!.message}"
-            puts $!.backtrace
+            #puts $!.backtrace
           end
         }
         (results || []).flatten.compact.uniq

data/lib/stratagem/{framework_extensions → instrumentation}/models/mocking.rb

@@ -1,4 +1,4 @@
-module Stratagem::ApplicationExtensions::Models
+module Stratagem::Instrumentation::Models
   module Mocking
     def mock_attributes
       @attributes ||= {}

data/lib/stratagem/instrumentation/models/persistence.rb

@@ -0,0 +1,9 @@
+module Stratagem::Instrumentation::Models::Persistence; end
+
+base = File.join(File.dirname(__FILE__), 'persistence')
+Dir.entries(base).select {|s| s !~ /^\./ && s != 'util' }.each {|adapter_dir|
+  require File.join(base, adapter_dir, 'detect')
+  require File.join(base, adapter_dir, 'tracing')
+  require File.join(base, adapter_dir, 'metadata')
+  require File.join(base, adapter_dir, 'extensions')
+}

data/lib/stratagem/instrumentation/models/persistence/active_record/detect.rb

@@ -0,0 +1,18 @@
+module Stratagem::Instrumentation::Models::Persistence::ActiveRecord
+  class Detect < Stratagem::Instrumentation::Models::Detect
+    IGNORE_CLASSES = ['ldap']
+
+    def self.supports?(model)
+      if model.ancestors.include?(ActiveRecord::Base)
+        if IGNORE_CLASSES.find {|ic| model.name.downcase.include?(ic) }
+          puts "Ignoring class: #{model.name}"
+          false
+        else
+          true
+        end
+      else
+        false
+      end
+    end
+  end
+end

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/persistence}/active_record/extensions.rb

@@ -1,5 +1,9 @@
 class ActiveRecord::Base
   class << self
+    def existing_instance_ids
+      find_by_sql("select id from #{table_name}").map {|i| i.id }
+    end
+
     def removed_methods=(methods)
       @@removed_methods = methods
     end
@@ -13,7 +17,7 @@ class ActiveRecord::Base
     end
   end
 
-  Stratagem::ApplicationExtensions::Models::Annotations.configure(self)
+  Stratagem::Instrumentation::Models::Annotations.configure(self)
 
 end
 

data/lib/stratagem/{framework_extensions/models/adapters → instrumentation/models/persistence}/active_record/metadata.rb

@@ -1,20 +1,27 @@
-module Stratagem::ApplicationExtensions::Models::Adapters::ActiveRecord
-
-  # prefix method names with  to avoid collision
+module Stratagem::Instrumentation::Models::Persistence::ActiveRecord
   class Metadata
-
     attr_reader :model, :instance
 
     def initialize(model)
       @model = model
-      @instance = @model.new unless (@model == ActiveRecord::Base)
+      begin
+        @instance = @model.new unless (@model == ActiveRecord::Base)
+      rescue
+        puts "ERROR: #{@model.name} could not be instantiated: #{$!.message}"
+      end
     end
     
     def relations(relation_type=nil) # :belongs_to, :has_many
       @relations ||= {}
       @relations[relation_type || :all] ||= model.reflect_on_all_associations(relation_type).map {|a|
+        klass = begin
+          a.klass # may fail if it's a polymorphic association
+        rescue
+          a.class_name
+        end
+
         begin
-          Stratagem::ApplicationExtensions::Models::Metadata::StratagemAssociation.new(a.name.to_sym, a.association_foreign_key.to_sym, a.klass, a.macro, a.options)
+          Stratagem::Instrumentation::Models::Association.new(model, a.name.to_sym, a.association_foreign_key.to_sym, klass, a.macro, a.options)
         rescue
           puts "ERROR: #{$!.message}"
         end
@@ -34,9 +41,18 @@ module Stratagem::ApplicationExtensions::Models::Adapters::ActiveRecord
     # this is typically a not null enforced by the database but not
     # by the model
     def column_from_error(database_error)
-      if (database_error.kind_of?(Mysql::Error) || database_error.kind_of?(::ActiveRecord::StatementInvalid))
-        database_error.message =~ /Column '(.*)?' cannot/
-        $1 ? $1.to_sym : nil
+      puts "COLUMN FROM ERROR:"
+      puts database_error.message
+      puts database_error.class
+      if (database_error.message.downcase.include?('column ') || database_error.kind_of?(::ActiveRecord::StatementInvalid))
+        database_error.message =~ /column (.*?\s)/i
+        if ($1)
+          column = $1
+          column.gsub!(/^.*?\./, '').gsub!(/\s.*/, '')
+          column
+        else
+          nil
+        end
       else
         puts database_error.class.name
         nil