influxdb-rails 0.0.1

data/gemfiles/Gemfile.rails-4.0.x

@@ -0,0 +1,8 @@
+source "https://rubygems.org"
+
+gem 'actionpack', '~> 4.0.0'
+gem 'activesupport', '~> 4.0.0'
+gem 'activemodel', '~> 4.0.0'
+gem 'rspec-rails', '>= 2.0'
+
+gemspec :path => '../'

data/gemfiles/Gemfile.rails-4.0.x.lock

@@ -0,0 +1,109 @@
+PATH
+  remote: /Users/todd/Projects/errplane/errplane-ruby
+  specs:
+    errplane (1.0.6)
+      json
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actionpack (4.0.0)
+      activesupport (= 4.0.0)
+      builder (~> 3.1.0)
+      erubis (~> 2.7.0)
+      rack (~> 1.5.2)
+      rack-test (~> 0.6.2)
+    activemodel (4.0.0)
+      activesupport (= 4.0.0)
+      builder (~> 3.1.0)
+    activesupport (4.0.0)
+      i18n (~> 0.6, >= 0.6.4)
+      minitest (~> 4.2)
+      multi_json (~> 1.3)
+      thread_safe (~> 0.1)
+      tzinfo (~> 0.3.37)
+    atomic (1.1.10)
+    builder (3.1.4)
+    coderay (1.0.9)
+    diff-lcs (1.2.4)
+    erubis (2.7.0)
+    fakeweb (1.3.0)
+    ffi (1.9.0)
+    formatador (0.2.4)
+    guard (1.8.1)
+      formatador (>= 0.2.4)
+      listen (>= 1.0.0)
+      lumberjack (>= 1.0.2)
+      pry (>= 0.9.10)
+      thor (>= 0.14.6)
+    guard-rspec (3.0.2)
+      guard (>= 1.8)
+      rspec (~> 2.13)
+    i18n (0.6.4)
+    json (1.8.0)
+    listen (1.2.2)
+      rb-fsevent (>= 0.9.3)
+      rb-inotify (>= 0.9)
+      rb-kqueue (>= 0.2)
+    lumberjack (1.0.4)
+    method_source (0.8.1)
+    minitest (4.7.5)
+    multi_json (1.7.7)
+    pry (0.9.12.2)
+      coderay (~> 1.0.5)
+      method_source (~> 0.8)
+      slop (~> 3.4)
+    rack (1.5.2)
+    rack-test (0.6.2)
+      rack (>= 1.0)
+    railties (4.0.0)
+      actionpack (= 4.0.0)
+      activesupport (= 4.0.0)
+      rake (>= 0.8.7)
+      thor (>= 0.18.1, < 2.0)
+    rake (10.1.0)
+    rb-fsevent (0.9.3)
+    rb-inotify (0.9.0)
+      ffi (>= 0.5.0)
+    rb-kqueue (0.2.0)
+      ffi (>= 0.5.0)
+    rdoc (4.0.1)
+      json (~> 1.4)
+    rspec (2.14.1)
+      rspec-core (~> 2.14.0)
+      rspec-expectations (~> 2.14.0)
+      rspec-mocks (~> 2.14.0)
+    rspec-core (2.14.4)
+    rspec-expectations (2.14.0)
+      diff-lcs (>= 1.1.3, < 2.0)
+    rspec-mocks (2.14.1)
+    rspec-rails (2.14.0)
+      actionpack (>= 3.0)
+      activesupport (>= 3.0)
+      railties (>= 3.0)
+      rspec-core (~> 2.14.0)
+      rspec-expectations (~> 2.14.0)
+      rspec-mocks (~> 2.14.0)
+    slop (3.4.6)
+    thor (0.18.1)
+    thread_safe (0.1.2)
+      atomic
+    tzinfo (0.3.37)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  actionpack (~> 4.0.0)
+  activemodel (~> 4.0.0)
+  activesupport (~> 4.0.0)
+  bundler (>= 1.0.0)
+  errplane!
+  fakeweb
+  guard
+  guard-rspec
+  rake
+  rdoc
+  rspec
+  rspec-rails (>= 2.0)
+  tzinfo

data/generators/influxdb/influxdb_generator.rb

@@ -0,0 +1,48 @@
+class InfluxDBGenerator < Rails::Generator::Base
+  def add_options!(option)
+    option.on("-k", "--api-key=API_KEY", String, "API key for your InfluxDB organization") {|v| options[:api_key] = v}
+    option.on("-a", "--application-id=APP_ID", String, "Your InfluxDB application id (optional)") {|v| options[:application_id] = v}
+  end
+
+  def manifest
+    if options[:api_key].blank?
+      puts "You must provide an API key using -k or --api-key."
+      exit
+    end
+
+    begin
+      puts "Contacting InfluxDB API"
+      application_name = "ApplicationName"
+      api_key = options[:api_key]
+
+      connection = Net::HTTP.new("influxdb.com", 443)
+      connection.use_ssl = true
+      connection.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      url = "/api/v1/applications?api_key=#{api_key}&name=#{application_name}"
+      response = connection.post(url, nil)
+
+      @application = JSON.parse(response.body)
+
+      unless response.is_a?(Net::HTTPSuccess)
+        raise "The InfluxDB API returned an error: #{response.inspect}"
+      end
+    rescue => e
+      puts "We ran into a problem creating your application via the API!"
+      puts "If this issue persists, contact us at support@influxdb.com with the following details:"
+      puts "API Key: #{e.class}: #{options[:api_key]}"
+      puts "#{e.class}: #{e.message}"
+    end
+
+    record do |m|
+      m.template "initializer.rb", "config/initializers/influxdb.rb",
+        :assigns => {
+          :application_id => options[:application_id] || @application["key"] || secure_random.hex(4),
+          :api_key => options[:api_key]
+        }
+    end
+  end
+
+  def secure_random
+    defined?(SecureRandom) ? SecureRandom : ActiveSupport::SecureRandom
+  end
+end

data/generators/influxdb/templates/initializer.rb

@@ -0,0 +1,5 @@
+require "influxdb"
+
+InfluxDB::Rails.configure do |config|
+
+end

data/influxdb-rails.gemspec

@@ -0,0 +1,33 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "influxdb/rails/version"
+
+Gem::Specification.new do |s|
+  s.name        = "influxdb-rails"
+  s.version     = InfluxDB::Rails::VERSION
+  s.authors     = ["Todd Persen"]
+  s.email       = ["todd@influxdb.com"]
+  s.homepage    = "http://influxdb.com"
+  s.summary     = %q{InfluxDB bindings for Ruby on Rails.}
+  s.description = %q{This gem automatically instruments your Ruby on Rails 3.x/4.x applications using InfluxDB for storage.}
+
+  s.rubyforge_project = "influxdb-rails"
+
+  s.files         = Dir.glob('**/*')
+  s.test_files    = Dir.glob('test/**/*') + Dir.glob('spec/**/*') + Dir.glob('features/**/*')
+  s.executables   = Dir.glob('bin/**/*').map {|f| File.basename(f)}
+  s.require_paths = ["lib"]
+
+  s.licenses = ['MIT']
+
+  s.add_runtime_dependency 'influxdb'
+  s.add_runtime_dependency 'railties'
+
+  s.add_development_dependency 'bundler', ['>= 1.0.0']
+  s.add_development_dependency 'fakeweb', ['>= 0']
+  s.add_development_dependency 'rake', ['>= 0']
+  s.add_development_dependency 'rdoc', ['>= 0']
+  s.add_development_dependency 'rspec', ['>= 0']
+  s.add_development_dependency 'rspec-rails', ['>= 0']
+  s.add_development_dependency 'tzinfo', ['>= 0']
+end

data/lib/influxdb-rails.rb

@@ -0,0 +1,82 @@
+require "net/http"
+require "net/https"
+require "rubygems"
+require "socket"
+require "thread"
+
+require "influxdb/rails/version"
+require "influxdb/rails/logger"
+require "influxdb/rails/exception_presenter"
+require "influxdb/rails/configuration"
+require "influxdb/rails/backtrace"
+require "influxdb/rails/rack"
+
+require "influxdb/rails/railtie" if defined?(Rails::Railtie)
+
+module InfluxDB
+  module Rails
+    class << self
+      include InfluxDB::Rails::Logger
+
+      attr_writer :configuration
+      attr_accessor :client
+
+      def configure(silent = false)
+        yield(configuration)
+        self.client = InfluxDB::Client.new
+      end
+
+      def configuration
+        @configuration ||= InfluxDB::Rails::Configuration.new
+      end
+
+      def report_exception_unless_ignorable(e, env = {})
+        report_exception(e, env) unless ignorable_exception?(e)
+      end
+      alias_method :transmit_unless_ignorable, :report_exception_unless_ignorable
+
+      def report_exception(e, env = {})
+        begin
+          env = influxdb_request_data if env.empty? && defined? influxdb_request_data
+          exception_presenter = ExceptionPresenter.new(e, env)
+          log :info, "Exception: #{exception_presenter.to_json[0..512]}..."
+
+          InfluxDB::Rails.client.write_point "rails.exceptions",
+            :context => exception_presenter.context,
+            :dimensions => exception_presenter.dimensions
+
+        rescue => e
+          log :info, "[InfluxDB] Something went terribly wrong. Exception failed to take off! #{e.class}: #{e.message}"
+        end
+      end
+      alias_method :transmit, :report_exception
+
+      def current_timestamp
+        Time.now.utc.to_i
+      end
+
+      def ignorable_exception?(e)
+        configuration.ignore_current_environment? ||
+        !!configuration.ignored_exception_messages.find{ |msg| /.*#{msg}.*/ =~ e.message  } ||
+        configuration.ignored_exceptions.include?(e.class.to_s)
+      end
+
+      def rescue(&block)
+        block.call
+      rescue StandardError => e
+        if configuration.ignore_current_environment?
+          raise(e)
+        else
+          transmit_unless_ignorable(e)
+        end
+      end
+
+      def rescue_and_reraise(&block)
+        block.call
+      rescue StandardError => e
+        transmit_unless_ignorable(e)
+        raise(e)
+      end
+    end
+  end
+end

data/lib/influxdb/rails/air_traffic_controller.rb

@@ -0,0 +1,39 @@
+module InfluxDB
+  module Rails
+    module AirTrafficController
+      def influxdb_request_data
+        unfiltered_params = params.to_hash
+        if respond_to?(:filter_parameters)
+          filtered_params = filter_parameters(unfiltered_params)
+        elsif defined? request.filtered_parameters
+          filtered_params = request.filtered_parameters
+        else
+          filtered_params = unfiltered_params.except(:password, :password_confirmation)
+        end
+
+        {
+          :params => filtered_params,
+          :session_data => influxdb_session_data,
+          :controller => params[:controller],
+          :action => params[:action],
+          :request_url => influxdb_request_url,
+          :user_agent => request.env["HTTP_USER_AGENT"],
+          :remote_ip => request.remote_ip,
+          :referer => request.referer,
+          :current_user => (current_user rescue nil)
+        }
+      end
+
+      private
+      def influxdb_session_data
+        session.respond_to?(:to_hash) ? session.to_hash : session.data
+      end
+
+      def influxdb_request_url
+        url = "#{request.protocol}#{request.host}"
+        url << ":#{request.port}" unless [80, 443].include?(request.port)
+        url << request.fullpath
+      end
+    end
+  end
+end

data/lib/influxdb/rails/backtrace.rb

@@ -0,0 +1,44 @@
+module InfluxDB
+  module Rails
+    class Backtrace
+      class Line
+        FORMAT = %r{^((?:[a-zA-Z]:)?[^:]+):(\d+)(?::in `([^']+)')?$}.freeze
+
+        attr_reader :file
+        attr_reader :number
+        attr_reader :method
+
+        def initialize(line)
+          _, @file, @number, @method = line.match(FORMAT).to_a
+        end
+
+        def to_s
+          "#{file}:#{number} in `#{method}'"
+        end
+
+        def inspect
+          "<Line: #{to_s}>"
+        end
+      end
+
+      attr_reader :lines
+
+      def initialize(backtrace)
+        @lines = Array(backtrace).each.collect do |line|
+          InfluxDB::Rails.configuration.backtrace_filters.each do |filter|
+            line = filter.call(line)
+          end
+          Line.new(line)
+        end
+      end
+
+      def to_a
+        lines.map(&:to_s)
+      end
+
+      def inspect
+        "<Backtrace: " + lines.collect { |line| line.to_s }.join(", ") + ">"
+      end
+    end
+  end
+end

data/lib/influxdb/rails/configuration.rb

@@ -0,0 +1,117 @@
+module InfluxDB
+  module Rails
+    class Configuration
+      attr_accessor :influxdb_host
+      attr_accessor :influxdb_port
+      attr_accessor :application_id
+      attr_accessor :application_name
+      attr_accessor :application_root
+
+      attr_accessor :logger
+      attr_accessor :environment
+      attr_accessor :framework
+      attr_accessor :framework_version
+      attr_accessor :language
+      attr_accessor :language_version
+      attr_accessor :ignored_exceptions
+      attr_accessor :ignored_exception_messages
+      attr_accessor :ignored_reports
+      attr_accessor :ignored_environments
+      attr_accessor :ignored_user_agents
+      attr_accessor :backtrace_filters
+      attr_accessor :aggregated_exception_classes
+      attr_accessor :environment_variables
+      attr_accessor :environment_variable_filters
+
+      attr_accessor :instrumentation_enabled
+      attr_accessor :debug
+      attr_accessor :reraise_global_exceptions
+
+      DEFAULTS = {
+        :influxdb_host => "localhost",
+        :influxdb_port => 8086,
+        :ignored_exceptions => %w{ActiveRecord::RecordNotFound
+                                  ActionController::RoutingError},
+        :ignored_exception_messages => [],
+        :ignored_reports => [],
+        :ignored_environments => %w{test cucumber selenium},
+        :ignored_user_agents => %w{GoogleBot},
+        :environment_variable_filters => [
+          /password/i,
+          /key/i,
+          /secret/i,
+          /ps1/i,
+          /rvm_.*_clr/i,
+          /color/i
+        ],
+        :backtrace_filters => [
+          lambda { |line| line.gsub(/^\.\//, "") },
+          lambda { |line|
+            return line if InfluxDB::Rails.configuration.application_root.to_s.empty?
+            line.gsub(/#{InfluxDB::Rails.configuration.application_root}/, "[APP_ROOT]")
+          },
+          lambda { |line|
+            if defined?(Gem) && !Gem.path.nil? && !Gem.path.empty?
+              Gem.path.each { |path| line = line.gsub(/#{path}/, "[GEM_ROOT]") }
+            end
+            line
+          }
+        ]
+      }
+
+      def initialize
+        @influxdb_host = DEFAULTS[:influxdb_host]
+        @influxdb_port = DEFAULTS[:influxdb_port]
+        @ignored_exceptions = DEFAULTS[:ignored_exceptions].dup
+        @ignored_exception_messages = DEFAULTS[:ignored_exception_messages].dup
+        @ignored_reports = DEFAULTS[:ignored_reports].dup
+        @ignored_environments = DEFAULTS[:ignored_environments].dup
+        @ignored_user_agents = DEFAULTS[:ignored_user_agents].dup
+        @backtrace_filters = DEFAULTS[:backtrace_filters].dup
+        @environment_variable_filters = DEFAULTS[:environment_variable_filters]
+        @aggregated_exception_classes = []
+        @debug = false
+        @rescue_global_exceptions = false
+        @instrumentation_enabled = true
+      end
+
+      def debug?
+        !!@debug
+      end
+
+      def instrumentation_enabled?
+        !!@instrumentation_enabled
+      end
+
+      def reraise_global_exceptions?
+        !!@reraise_global_exceptions
+      end
+
+      def ignore_user_agent?(incoming_user_agent)
+        return false if self.ignored_user_agents.nil?
+        self.ignored_user_agents.any? {|agent| incoming_user_agent =~ /#{agent}/}
+      end
+
+      def ignore_current_environment?
+        self.ignored_environments.include?(self.environment)
+      end
+
+      def define_custom_exception_data(&block)
+        @custom_exception_data_handler = block
+      end
+
+      def add_custom_exception_data(exception_presenter)
+        @custom_exception_data_handler.call(exception_presenter) if @custom_exception_data_handler
+      end
+
+      def database_name
+        @application_id.to_s + @environment.to_s
+      end
+
+      private
+      def initialize_http_connection
+        Net::HTTP.new(@app_host, "80")
+      end
+    end
+  end
+end