app.rb

# frozen_string_literal: true

require 'sinatra/base'
require 'sinatra/config_file'
require 'sinatra/json'
require 'json'

unless ENV['RACK_ENV'] == 'production'
  require 'sinatra/reloader'
  require 'pry'
end

class GelfProxy < Sinatra::Base
  before do
    content_type :json
  end

  register Sinatra::ConfigFile
  config_file 'config/proxy.yml'
  #############################################################################
  # load configuration from ENV or file or default
  #############################################################################
  APP_NAME = ENV['APP_NAME'] || settings.app_name || 'gelf proxy'
  LOGGER_USERNAME = ENV['LOGGER_USERNAME'] || settings.logger_username || 'gelf'
  LOGGER_PASSWORD = ENV['LOGGER_PASSWORD'] || settings.logger_password || 'fleg'

  #############################################################################
  # production env settings, force auth
  #############################################################################
  if ENV['RACK_ENV'] == 'production'
    use Rack::Auth::Basic, 'Protected Area' do |username, password|
      username == LOGGER_USERNAME && password == LOGGER_PASSWORD
    end
  else
    register Sinatra::Reloader
    also_reload 'lib/*'
  end

  #############################################################################
  # index
  #############################################################################
  get '/' do
    { hello: 'world' }.to_json
  end

  #############################################################################
  # log
  #############################################################################
  post '/log' do
    request.body.rewind
    log = JSON.parse(request.body.read)
    log.store('facility', APP_NAME)
    puts log.to_json
  rescue
    puts request.body.read
  end

  ## Run the app when file called ##
  run! if $PROGRAM_NAME == __FILE__
end