corn 0.4.2 → 0.4.3

checksums.yaml

@@ -1,7 +1,7 @@
 --- 
 SHA1: 
-  metadata.gz: 3ce59d3b72f284db34d07ce896040fa38633a346
-  data.tar.gz: 7c0f808633dc19752eea4fc865665c6cc67f0f2e
+  metadata.gz: 939489a883d8ffc993fd10af0075cca1a36ecc6d
+  data.tar.gz: 477da0e4d9b6d50f618d9ac2faf7adf9ecf2696f
 SHA512: 
-  metadata.gz: 42f77f91d007c9fce606299e14c0b991313ac4d92af218c95cbc1737848876b3d1845365ac1d87b59d0e67f12bc3440d5f85ed04d7439078d8d5503db6d2d9e6
-  data.tar.gz: 38d9d3f6b32cacf7973bf8c4186c7015ea01348df29d409b9d4a6676c06257981cf3e4cf2e10bdc3a7598151d12f98d0de249393a56d062eec2f01e23c404c00
+  metadata.gz: 29f8ef1199f36c884dc5c53effc485f9fe91691f7a39cd2151e457995ac728242d59bbdfe27be019d011eaf831e99ead332138a2fcae7117635b1cb52f631e59
+  data.tar.gz: f082093672aed933a128c614edfa6b41f0c7fbf1418fd16464e55b606748663ec5c4c846d7802a068fd6848fca6a3c2ef554505d72af27a1adbd04236fc067ed

data/lib/corn.rb

@@ -1,25 +1,37 @@
 require 'rubygems'
-require 'sampling_prof'
-require 'net/http'
-require 'net/https'
-require 'corn/rack'
 require 'logger'
+require 'corn/config'
+require 'corn/rack'
 
 module Corn
-  module_function
-
-  def rack_slow_request_profiler
-    Rack::SlowRequestProfiler
-  end
-
-  def host
-    ENV['CORN_HOST']
-  end
-
-  def client_id
-    ENV['CORN_CLIENT_ID']
-  end
+  include Config
+
+  # Set configurations or get current configurations.
+  #
+  #   host: corn server host
+  #   client_id: corn project client id
+  #   logger: a Logger object, e.g. Rails.logger.
+  #
+  # Configurations for posting data back to Corn server
+  #
+  #   ssl_verify_peer:
+  #     true: set verify_mode to OpenSSL::SSL::VERIFY_PEER
+  #     false: set verify_mode to OpenSSL::SSL::VERIFY_NONE
+  #     default to false
+  #   ssl_ca_file: Net::HTTPSession#ca_file
+  #   ssl_ca_path: Net::HTTPSession#ca_path
+  #   see Ruby Net::HTTP document for details
+  #
+  config({
+           :logger => Logger.new(STDOUT),
+           :host => ENV['CORN_HOST'],
+           :client_id => ENV['CORN_CLIENT_ID'],
+           :ssl_verify_peer => false,
+           :ssl_ca_file => nil,
+           :ssl_ca_path => nil
+         })
 
+  module_function
   def configured?
     !!(host && client_id)
   end
@@ -28,11 +40,7 @@ module Corn
     File.join(host, 'profile_data')
   end
 
-  def logger=(l)
-    @logger = l
-  end
-
-  def logger
-    @logger ||= Logger.new(STDOUT)
+  def rack_slow_request_profiler
+    Rack::SlowRequestProfiler
   end
 end

data/lib/corn/config.rb

@@ -0,0 +1,31 @@
+
+module Corn
+  module Config
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      def config(hash={})
+        @config ||= {}
+        if hash.empty?
+          @config
+        else
+          @config.merge!(hash)
+          hash.each do |key, value|
+            if self.respond_to?(key)
+              next
+            end
+            q = !!value == value ? '?' : ''
+            self.class_eval <<-RUBY, __FILE__, __LINE__
+              def self.#{key}#{q}
+                @config[:#{key}]
+              end
+            RUBY
+          end
+          @config
+        end
+      end
+    end
+  end
+end

data/lib/corn/post.rb

@@ -1,3 +1,5 @@
+require 'net/http'
+require 'net/https'
 require 'thread'
 
 module Corn
@@ -36,7 +38,13 @@ module Corn
       http = Net::HTTP.new(uri.host, uri.port)
       if uri.scheme == 'https'
         http.use_ssl = true
-        http.verify_mode = OpenSSL::SSL::VERIFY_NONE if ENV['CORN_SKIP_SSL_VERIFY']
+        if Corn.ssl_verify_peer
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+          http.ca_file = Corn.ssl_ca_file if Corn.ssl_ca_file
+          http.ca_path = Corn.ssl_ca_path if Corn.ssl_ca_path
+        else
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        end
       end
       res = http.request(req)
       Corn.logger.info("Corn report submitted to #{submit_url}")

data/lib/corn/rack/slow_request_profiler.rb

@@ -1,5 +1,5 @@
-require 'sampling_prof'
 require 'corn/post'
+require 'sampling_prof'
 
 module Corn
   module Rack

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: corn
 version: !ruby/object:Gem::Version 
-  version: 0.4.2
+  version: 0.4.3
 platform: ruby
 authors: 
   - Xiao Li
@@ -35,6 +35,7 @@ extra_rdoc_files: []
 files: 
   - README.md
   - lib/corn.rb
+  - lib/corn/config.rb
   - lib/corn/post.rb
   - lib/corn/rack.rb
   - lib/corn/rack/slow_request_profiler.rb