module TreasureData
module Logger
module Agent
  module Rails

    def self.init_model
      unless defined?(::ActiveRecord)
        # disable model extension if Rails > 3 and
        # ActiveRecord is not loaded (other ORM is used)
        if ::Rails.respond_to?(:version) && ::Rails.version =~ /^3/
          return
        end
        require 'active_record'
      end
      ::ActiveRecord::Base.send(:include, ModelExtension)
    end

    module ModelExtension
      if defined?(ActiveSupport::Concern)
        # Rails 2
        extend ActiveSupport::Concern
      else
        def self.included(mod)
          im = InstanceMethods
          cm = ClassMethods
          mod.class_eval do
            include im
            extend cm
          end
        end
      end

      module InstanceMethods
      end

      module ClassMethods
        def td_enable_model_tracer(tag, options={})
          only = nil
          except = nil
          static = {}

          if o = options[:only]
            only = case o
                   when Array
                     o
                   else
                     [o]
                   end.map {|e| e.to_s }
          end

          if o = options[:except]
            except = case o
                     when Array
                       o
                     else
                       [o]
                     end.map {|e| e.to_s }
          end

          if o = options[:static]
            o.each_pair {|k,v|
              static[k.to_s] = v
            }
          end

          if defined?(after_commit)
            # Rails 3
            m = :after_commit
          else
            # Rails 2
            m = :after_save
          end

          __send__(m) do |record|
            data = {}
            record.attribute_names.each {|name|
              name = name.to_s
              if (!only || only.include?(name)) && (!except || !except.include?(name))
                data[name] = record.read_attribute(name)
              end
            }
            static.each_pair {|k,v|
              data[k] = v
            }
            if time = data['updated_at'] && time.is_a?(Time)
              data['time'] = time.to_i
              data.delete('updated_at')
            end
            TreasureData.log(tag, data)
          end
        end
      end
    end

  end
end
end
end