Add log level options

README.md

@@ -168,6 +168,14 @@ type: Boolean
 description: Allow uppercase within the name of the module or environment passed to the API.
 default: true
 
+### Logging Settings
+
+#### `loglevel`
+
+type: String
+description: Set the Logging level for the application. Valid values are `debug`, `error`, `warn`, `fatal`, and `info`.
+default `warn`
+
 ### Configure the webhook
 
 This API provides two endpoints that need to be called:

app/controllers/application_controller.rb

@@ -10,7 +10,11 @@ class ApplicationController < Sinatra::Base
     set :database, "sqlite3:db/#{ENV['SINATRA_ENV']}.sqlite3"
     set :public_folder, 'public'
     set :views, 'app/views'
-    set :logger, Logger.new(STDOUT)
+
+    # Logging settings
+    logger = Logger.new(STDOUT)
+    logger.level = LOGLEVEL
+    set :logger, logger
   end
 
   not_found do

config/config.yml.example

@@ -12,4 +12,7 @@ development:
   ignore_environments: []
   allow_uppercase: true
 
+  # Logging
+  loglevel: debug
+
 test: &development

config/environment.rb

@@ -7,14 +7,15 @@
 
 config_path = ENV['WEBHOOK_CONFDIR'] || 'config/config.yml'
 
+require './lib/puppet_webhook'
 config = YAML.load_file(config_path)
 APP_CONFIG = OpenStruct.new(config[ENV['SINATRA_ENV']])
+LOGLEVEL = PuppetWebhook.loglevel
 
 ActiveRecord::Base.establish_connection(
   adapter: 'sqlite3',
   database: "db/#{ENV['SINATRA_ENV']}.sqlite3"
 )
 
-require './lib/puppet_webhook'
 require './app/controllers/application_controller'
 require_all 'app'

lib/puppet_webhook.rb

@@ -6,4 +6,19 @@
 
 module PuppetWebhook
   VERSION = '2.1.2'
+
+  def self.loglevel
+    case APP_CONFIG.loglevel
+    when 'debug'
+      Logger::DEBUG
+    when 'info'
+      Logger::INFO
+    when 'error'
+      Logger::ERROR
+    when 'fatal'
+      Logger::FATAL
+    else
+      Logger::WARN
+    end
+  end
 end