Replace CLI class with a more suitable Application class

This commit is contained in:
2017-10-23 13:58:19 +01:00
parent ee4d363d3a
commit 6812e52dee
8 changed files with 199 additions and 127 deletions

View File

@@ -1,3 +1,6 @@
AllCops:
TargetRubyVersion: 2.3
Documentation:
Enabled: false

View File

@@ -1,4 +1,5 @@
# coding: utf-8
lib = File.expand_path('../lib', __FILE__)
$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
require 'bunnyrun/version'
@@ -24,7 +25,7 @@ Gem::Specification.new do |spec|
spec.add_development_dependency 'bundler', '~> 1.14'
spec.add_development_dependency 'rake', '~> 10.0'
spec.add_development_dependency 'rspec', '~> 3.0'
spec.add_development_dependency 'rubocop', '~> 0.47'
spec.add_development_dependency 'rubocop', '~> 0.50'
spec.add_development_dependency 'byebug'
spec.add_runtime_dependency 'bunny', '~> 2.6'

View File

@@ -1,11 +1,18 @@
#!/usr/bin/env ruby
# frozen_string_literal: true
$LOAD_PATH.unshift(File.expand_path('../../lib', File.realpath(__FILE__)))
require 'bunnyrun'
require 'foobar'
cli = BunnyRun::CLI.new(
name: File.basename(__FILE__),
version: Foobar::VERSION
)
module Foobar
class Application < BunnyRun::Application
name File.basename(__FILE__)
usage '<options> [<path>]'
version Foobar::VERSION
cli.run(ARGV)
option :log_message, 'Message to log after success', type: :string
end
end
Foobar::Application.run(argv: ARGV)

View File

@@ -1,3 +1,5 @@
# frozen_string_literal: true
require 'bunnyrun'
module Foobar
@@ -15,6 +17,7 @@ module Foobar
publish('ping-pong', 'PONG', routing_key: 'pong')
message.ack
logger.info
end
end
end

View File

@@ -1,5 +1,7 @@
#!/usr/bin/env ruby
# frozen_string_literal: true
$LOAD_PATH.unshift(File.expand_path('../../lib', File.realpath(__FILE__)))
require 'bunnyrun'
BunnyRun::CLI.run(ARGV)
BunnyRun::Application.run(argv: ARGV)

View File

@@ -1,4 +1,6 @@
require 'bunnyrun/cli'
# frozen_string_literal: true
require 'bunnyrun/application'
require 'bunnyrun/consumer'
require 'bunnyrun/runner'
require 'bunnyrun/version'

173
lib/bunnyrun/application.rb Normal file
View File

@@ -0,0 +1,173 @@
# frozen_string_literal: true
require 'trollop'
require 'bunnyrun/consumer'
require 'bunnyrun/runner'
require 'bunnyrun/version'
module BunnyRun
class Application
class << self
def run(*args)
new.run(*args)
end
def name(input = nil)
@name = input unless input.nil?
@name
end
def usage(input = nil)
@usage = input unless input.nil?
@usage
end
def version(input = nil)
@version = input unless input.nil?
@version
end
def option(name, description, opts)
options << [name, description, opts]
end
def options
@options ||= []
end
end
def name
self.class.name
end
def usage
self.class.usage
end
def version
self.class.version
end
def run(opts = {})
argv = opts.delete(:argv)
options = parse_args(argv)
options.merge!(opts)
require_files(options[:paths]) if options[:paths].any?
consumers = Consumer.children
runner = Runner.new(options, consumers)
runner.run
end
private
def parse_args(args)
args = args.clone
opts = Trollop.with_standard_exception_handling(parser) do
parser.parse(args)
end
opts[:paths] = args
opts
end
def parser
@parser ||= begin
parser = Trollop::Parser.new
define_version(parser)
define_usage(parser)
define_connection_options(parser)
define_logging_options(parser)
define_application_options(parser)
parser.banner "\nHelp/Version:"
parser
end
end
def define_usage(parser)
parser.banner "Usage: #{name} #{usage}"
end
def define_version(parser)
parser.version "#{name} #{version}"
end
def define_connection_options(parser)
parser.banner "\nRabbitMQ Options:"
parser.opt :url, 'Connection string ' \
'(example: "amqp://guest:guest@127.0.0.1:5672/vhost")',
short: 'U', type: :string,
default: ENV['RABBITMQ_URL'] || nil
parser.opt :host, 'Host',
short: 'H', type: :string,
default: ENV['RABBITMQ_HOST'] || '127.0.0.1'
parser.opt :port, 'Port',
short: 'P', type: :int,
default: (ENV['RABBITMQ_PORT'] || 5672).to_i
parser.opt :ssl, 'Connect using SSL',
short: 's', type: :bool,
default: trufy?(ENV.fetch('RABBITMQ_SSL', false))
parser.opt :vhost, 'Virtual host',
short: 'V', type: :string,
default: ENV['RABBITMQ_VHOST'] || '/'
parser.opt :user, 'Username',
short: 'u', type: :string,
default: ENV['RABBITMQ_USER'] || 'guest'
parser.opt :pass, 'Password',
short: 'p', type: :string,
default: ENV['RABBITMQ_PASS'] || 'guest'
parser.opt :prefetch, 'Default prefetch count',
short: :none, type: :int,
default: (ENV['RABBITMQ_PREFETCH'] || 1).to_i
parser.conflicts :url, :host
parser.conflicts :url, :port
parser.conflicts :url, :ssl
parser.conflicts :url, :vhost
parser.conflicts :url, :user
parser.conflicts :url, :pass
end
def define_logging_options(parser)
parser.banner "\nLogging Options:"
parser.opt :log_target, 'Log target, file path or STDOUT',
short: 't', type: :string,
default: ENV['LOG_TARGET'] || 'STDOUT'
parser.opt :log_level, 'Log level (debug, info, warn, error, fatal)',
short: 'l', type: :string,
default: ENV['LOG_LEVEL'] || 'info'
parser.opt :bunny_log_target, 'Log target used by Bunny',
short: :none, type: :string,
default: ENV['BUNNY_LOG_TARGET'] || 'STDOUT'
parser.opt :bunny_log_level, 'Log level used by Bunny',
short: :none, type: :string,
default: ENV['BUNNY_LOG_LEVEL'] || 'warn'
end
def define_application_options(parser)
return if self.class.options.empty?
parser.banner "\nApplication Options:"
self.class.options.each do |opt_args|
parser.opt(*opt_args)
end
end
def trufy?(input)
%w[true 1 y yes].include?(input)
end
def require_files(paths)
paths.each do |path|
require File.expand_path(path)
end
end
end
end

View File

@@ -1,119 +0,0 @@
require 'trollop'
require 'bunnyrun/consumer'
require 'bunnyrun/runner'
require 'bunnyrun/version'
module BunnyRun
class CLI
def self.run(argv = [])
new.run(argv)
end
def initialize(name: nil, version: nil, usage: nil)
@name = name
@version = version
@usage = usage
end
def name
@name ||= 'bunnyrun'
end
def version
@version ||= BunnyRun::VERSION
end
def usage
@usage ||= '[options] [path ...]'
end
def run(args = [])
options = parse_args(args)
require_files(options[:paths]) if options[:paths].any?
consumers = Consumer.children
runner = Runner.new(options, consumers)
runner.run
end
private
def parse_args(args)
args = args.clone
opts = Trollop.with_standard_exception_handling(parser) do
parser.parse(args)
end
opts[:paths] = args
opts
end
def parser
@parser ||= begin
parser = Trollop::Parser.new
define_version(parser)
define_usage(parser)
parser.banner "\nOptions:"
define_connection_options(parser)
define_logging_options(parser)
parser
end
end
def define_usage(parser)
parser.banner "Usage: #{name} #{usage}"
end
def define_version(parser)
parser.version "#{name} #{version}"
end
def define_connection_options(parser)
parser.opt :url, 'Connection string ' \
'(example: "amqp://guest:guest@127.0.0.1:5672/vhost")',
short: 'U', type: :string
parser.opt :host, 'Host',
short: 'H', type: :string, default: '127.0.0.1'
parser.opt :port, 'Port',
short: 'P', type: :int, default: 5672
parser.opt :ssl, 'Connect using SSL',
short: 's', type: :bool, default: false
parser.opt :vhost, 'Virtual host',
short: 'V', type: :string, default: '/'
parser.opt :user, 'Username',
short: 'u', type: :string, default: 'guest'
parser.opt :pass, 'Password',
short: 'p', type: :string, default: 'guest'
parser.opt :prefetch, 'Default prefetch count',
short: :none, type: :int, default: 1
parser.conflicts :url, :host
parser.conflicts :url, :port
parser.conflicts :url, :ssl
parser.conflicts :url, :vhost
parser.conflicts :url, :user
parser.conflicts :url, :pass
end
def define_logging_options(parser)
parser.opt :log_target, 'Log target, file path or STDOUT',
short: 't', type: :string, default: 'STDOUT'
parser.opt :log_level, 'Log level (debug, info, warn, error, fatal)',
short: 'l', type: :string, default: 'info'
parser.opt :bunny_log_target, 'Log target used by Bunny',
short: :none, type: :string, default: 'STDOUT'
parser.opt :bunny_log_level, 'Log level used by Bunny',
short: :none, type: :string, default: 'warn'
end
def require_files(paths)
paths.each do |path|
require File.expand_path(path)
end
end
end
end