amqp-daemon-kit 0.1.8.1
Sign up to get free protection for your applications and to get access to all the features.
- data/.gitignore +5 -0
- data/Configuration.txt +110 -0
- data/Deployment.txt +113 -0
- data/History.txt +131 -0
- data/Logging.txt +96 -0
- data/PostInstall.txt +6 -0
- data/README.rdoc +132 -0
- data/Rakefile +29 -0
- data/RuoteParticipants.txt +113 -0
- data/TODO.txt +27 -0
- data/bin/daemon-kit +18 -0
- data/config/website.yml +2 -0
- data/daemon-kit.gemspec +265 -0
- data/lib/daemon_kit.rb +60 -0
- data/lib/daemon_kit/abstract_logger.rb +249 -0
- data/lib/daemon_kit/application.rb +230 -0
- data/lib/daemon_kit/arguments.rb +165 -0
- data/lib/daemon_kit/commands/console.rb +38 -0
- data/lib/daemon_kit/commands/destroy.rb +10 -0
- data/lib/daemon_kit/commands/generate.rb +10 -0
- data/lib/daemon_kit/config.rb +113 -0
- data/lib/daemon_kit/console_daemon.rb +2 -0
- data/lib/daemon_kit/core_ext.rb +1 -0
- data/lib/daemon_kit/core_ext/configurable.rb +96 -0
- data/lib/daemon_kit/core_ext/string.rb +22 -0
- data/lib/daemon_kit/cron.rb +67 -0
- data/lib/daemon_kit/cucumber/world.rb +38 -0
- data/lib/daemon_kit/deployment/capistrano.rb +516 -0
- data/lib/daemon_kit/dk_amqp.rb +39 -0
- data/lib/daemon_kit/em.rb +43 -0
- data/lib/daemon_kit/error_handlers/base.rb +32 -0
- data/lib/daemon_kit/error_handlers/hoptoad.rb +180 -0
- data/lib/daemon_kit/exceptions.rb +15 -0
- data/lib/daemon_kit/generators.rb +67 -0
- data/lib/daemon_kit/generators/base.rb +60 -0
- data/lib/daemon_kit/initializer.rb +449 -0
- data/lib/daemon_kit/jabber.rb +171 -0
- data/lib/daemon_kit/nanite.rb +7 -0
- data/lib/daemon_kit/nanite/agent.rb +77 -0
- data/lib/daemon_kit/pid_file.rb +61 -0
- data/lib/daemon_kit/ruote_participants.rb +125 -0
- data/lib/daemon_kit/ruote_pseudo_participant.rb +68 -0
- data/lib/daemon_kit/ruote_workitem.rb +187 -0
- data/lib/daemon_kit/safety.rb +84 -0
- data/lib/daemon_kit/tasks.rb +2 -0
- data/lib/daemon_kit/tasks/environment.rake +11 -0
- data/lib/daemon_kit/tasks/framework.rake +123 -0
- data/lib/daemon_kit/tasks/god.rake +62 -0
- data/lib/daemon_kit/tasks/log.rake +8 -0
- data/lib/daemon_kit/tasks/monit.rake +29 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/CHANGELOG.rdoc +89 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/LICENSE +20 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/README.rdoc +297 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/Thorfile +69 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor.rb +244 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/actions.rb +296 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/actions/create_file.rb +103 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/actions/directory.rb +91 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/actions/empty_directory.rb +134 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/actions/file_manipulation.rb +223 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/actions/inject_into_file.rb +104 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/base.rb +540 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/core_ext/file_binary_read.rb +9 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/core_ext/hash_with_indifferent_access.rb +75 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/core_ext/ordered_hash.rb +100 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/error.rb +30 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/group.rb +271 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/invocation.rb +180 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/parser.rb +4 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/parser/argument.rb +67 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/parser/arguments.rb +150 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/parser/option.rb +128 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/parser/options.rb +169 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/rake_compat.rb +66 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/runner.rb +314 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/shell.rb +83 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/shell/basic.rb +239 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/shell/color.rb +108 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/task.rb +102 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/util.rb +224 -0
- data/lib/daemon_kit/vendor/thor-0.13.6/lib/thor/version.rb +3 -0
- data/lib/daemon_kit/xmpp.rb +100 -0
- data/lib/generators/daemon_kit/amqp/USAGE +5 -0
- data/lib/generators/daemon_kit/amqp/amqp_generator.rb +24 -0
- data/lib/generators/daemon_kit/amqp/templates/config/amqp.yml +28 -0
- data/lib/generators/daemon_kit/amqp/templates/config/pre-daemonize/amqp.rb +7 -0
- data/lib/generators/daemon_kit/amqp/templates/libexec/%app_name%-daemon.rb +37 -0
- data/lib/generators/daemon_kit/app/USAGE +7 -0
- data/lib/generators/daemon_kit/app/app_generator.rb +140 -0
- data/lib/generators/daemon_kit/app/templates/Gemfile +12 -0
- data/lib/generators/daemon_kit/app/templates/README.tt +58 -0
- data/lib/generators/daemon_kit/app/templates/Rakefile +6 -0
- data/lib/generators/daemon_kit/app/templates/bin/daemon.tt +7 -0
- data/lib/generators/daemon_kit/app/templates/config/arguments.rb +12 -0
- data/lib/generators/daemon_kit/app/templates/config/boot.rb +70 -0
- data/lib/generators/daemon_kit/app/templates/config/environment.rb.tt +26 -0
- data/lib/generators/daemon_kit/app/templates/config/environments/development.rb +2 -0
- data/lib/generators/daemon_kit/app/templates/config/environments/production.rb +5 -0
- data/lib/generators/daemon_kit/app/templates/config/environments/test.rb +2 -0
- data/lib/generators/daemon_kit/app/templates/config/post-daemonize/readme +5 -0
- data/lib/generators/daemon_kit/app/templates/config/pre-daemonize/readme +12 -0
- data/lib/generators/daemon_kit/app/templates/lib/%app_name%.rb +2 -0
- data/lib/generators/daemon_kit/app/templates/libexec/%app_name%-daemon.rb +18 -0
- data/lib/generators/daemon_kit/app/templates/script/console +3 -0
- data/lib/generators/daemon_kit/app/templates/script/destroy +3 -0
- data/lib/generators/daemon_kit/app/templates/script/generate +3 -0
- data/lib/generators/daemon_kit/capistrano/capistrano_generator.rb +26 -0
- data/lib/generators/daemon_kit/capistrano/templates/Capfile +10 -0
- data/lib/generators/daemon_kit/capistrano/templates/USAGE +10 -0
- data/lib/generators/daemon_kit/capistrano/templates/config/deploy.rb.tt +67 -0
- data/lib/generators/daemon_kit/capistrano/templates/config/deploy/logrotate.erb +13 -0
- data/lib/generators/daemon_kit/capistrano/templates/config/deploy/production.rb.tt +6 -0
- data/lib/generators/daemon_kit/capistrano/templates/config/deploy/staging.rb.tt +6 -0
- data/lib/generators/daemon_kit/capistrano/templates/config/environments/staging.rb +0 -0
- data/lib/generators/daemon_kit/cron/USAGE +5 -0
- data/lib/generators/daemon_kit/cron/cron_generator.rb +24 -0
- data/lib/generators/daemon_kit/cron/templates/config/pre-daemonize/cron.rb +11 -0
- data/lib/generators/daemon_kit/cron/templates/libexec/%app_name%-daemon.rb +48 -0
- data/lib/generators/daemon_kit/cucumber/USAGE +11 -0
- data/lib/generators/daemon_kit/cucumber/cucumber_generator.rb +45 -0
- data/lib/generators/daemon_kit/cucumber/templates/config/environments/cucumber.rb +2 -0
- data/lib/generators/daemon_kit/cucumber/templates/features/step_definitions/.empty_directory +0 -0
- data/lib/generators/daemon_kit/cucumber/templates/features/support/env.rb +7 -0
- data/lib/generators/daemon_kit/cucumber/templates/script/cucumber +7 -0
- data/lib/generators/daemon_kit/cucumber/templates/tasks/cucumber.rake +13 -0
- data/lib/generators/daemon_kit/nanite_agent/USAGE +5 -0
- data/lib/generators/daemon_kit/nanite_agent/nanite_agent_generator.rb +29 -0
- data/lib/generators/daemon_kit/nanite_agent/templates/config/nanite.yml +35 -0
- data/lib/generators/daemon_kit/nanite_agent/templates/config/pre-daemonize/nanite_agent.rb +6 -0
- data/lib/generators/daemon_kit/nanite_agent/templates/lib/actors/sample.rb +11 -0
- data/lib/generators/daemon_kit/nanite_agent/templates/libexec/%app_name%-daemon.rb +31 -0
- data/lib/generators/daemon_kit/rspec/USAGE +5 -0
- data/lib/generators/daemon_kit/rspec/rspec_generator.rb +20 -0
- data/lib/generators/daemon_kit/rspec/templates/spec/%app_name%_spec.rb +11 -0
- data/lib/generators/daemon_kit/rspec/templates/spec/spec.opts +1 -0
- data/lib/generators/daemon_kit/rspec/templates/spec/spec_helper.rb +23 -0
- data/lib/generators/daemon_kit/rspec/templates/tasks/rspec.rake +21 -0
- data/lib/generators/daemon_kit/ruote/USAGE +5 -0
- data/lib/generators/daemon_kit/ruote/ruote_generator.rb +29 -0
- data/lib/generators/daemon_kit/ruote/templates/config/amqp.yml +30 -0
- data/lib/generators/daemon_kit/ruote/templates/config/pre-daemonize/ruote.rb +13 -0
- data/lib/generators/daemon_kit/ruote/templates/config/ruote.yml +23 -0
- data/lib/generators/daemon_kit/ruote/templates/lib/%app_name%.rb +4 -0
- data/lib/generators/daemon_kit/ruote/templates/lib/sample.rb +26 -0
- data/lib/generators/daemon_kit/ruote/templates/libexec/%app_name%-daemon.rb +33 -0
- data/lib/generators/daemon_kit/test_unit/USAGE +5 -0
- data/lib/generators/daemon_kit/test_unit/templates/tasks/test_unit.rake +7 -0
- data/lib/generators/daemon_kit/test_unit/templates/test/%app_name%_test.rb.tt +9 -0
- data/lib/generators/daemon_kit/test_unit/templates/test/test_helper.rb +6 -0
- data/lib/generators/daemon_kit/test_unit/test_unit_generator.rb +20 -0
- data/lib/generators/daemon_kit/xmpp/templates/config/pre-daemonize/xmpp.rb +6 -0
- data/lib/generators/daemon_kit/xmpp/templates/config/xmpp.yml +29 -0
- data/lib/generators/daemon_kit/xmpp/templates/libexec/%app_name%-daemon.rb +27 -0
- data/lib/generators/daemon_kit/xmpp/xmpp_generator.rb +24 -0
- data/script/console +10 -0
- data/script/destroy +14 -0
- data/script/generate +14 -0
- data/script/txt2html +71 -0
- data/spec/abstract_logger_spec.rb +126 -0
- data/spec/argument_spec.rb +70 -0
- data/spec/config_spec.rb +83 -0
- data/spec/configurable_spec.rb +56 -0
- data/spec/daemon_kit_spec.rb +7 -0
- data/spec/error_handlers_spec.rb +23 -0
- data/spec/fixtures/env.yml +15 -0
- data/spec/fixtures/noenv.yml +4 -0
- data/spec/initializer_spec.rb +26 -0
- data/spec/spec.opts +1 -0
- data/spec/spec_helper.rb +27 -0
- data/tasks/cucumber.rake +13 -0
- data/tasks/rspec.rake +20 -0
- data/tasks/tests.rake +6 -0
- data/templates/god/god.erb +69 -0
- data/templates/monit/monit.erb +14 -0
- data/test/test_amqp_generator.rb +48 -0
- data/test/test_cron_generator.rb +45 -0
- data/test/test_daemon-kit_generator.rb +84 -0
- data/test/test_daemon_kit_config.rb +28 -0
- data/test/test_deploy_capistrano_generator.rb +48 -0
- data/test/test_generator_helper.rb +29 -0
- data/test/test_helper.rb +7 -0
- data/test/test_nanite_agent_generator.rb +49 -0
- data/test/test_ruote_generator.rb +51 -0
- data/test/test_test_unit_generator.rb +46 -0
- metadata +302 -0
@@ -0,0 +1,83 @@
|
|
1
|
+
require 'rbconfig'
|
2
|
+
require 'thor/shell/color'
|
3
|
+
|
4
|
+
class Thor
|
5
|
+
module Base
|
6
|
+
# Returns the shell used in all Thor classes. If you are in a Unix platform
|
7
|
+
# it will use a colored log, otherwise it will use a basic one without color.
|
8
|
+
#
|
9
|
+
def self.shell
|
10
|
+
@shell ||= if Config::CONFIG['host_os'] =~ /mswin|mingw/
|
11
|
+
Thor::Shell::Basic
|
12
|
+
else
|
13
|
+
Thor::Shell::Color
|
14
|
+
end
|
15
|
+
end
|
16
|
+
|
17
|
+
# Sets the shell used in all Thor classes.
|
18
|
+
#
|
19
|
+
def self.shell=(klass)
|
20
|
+
@shell = klass
|
21
|
+
end
|
22
|
+
end
|
23
|
+
|
24
|
+
module Shell
|
25
|
+
SHELL_DELEGATED_METHODS = [:ask, :yes?, :no?, :say, :say_status, :print_table]
|
26
|
+
|
27
|
+
# Add shell to initialize config values.
|
28
|
+
#
|
29
|
+
# ==== Configuration
|
30
|
+
# shell<Object>:: An instance of the shell to be used.
|
31
|
+
#
|
32
|
+
# ==== Examples
|
33
|
+
#
|
34
|
+
# class MyScript < Thor
|
35
|
+
# argument :first, :type => :numeric
|
36
|
+
# end
|
37
|
+
#
|
38
|
+
# MyScript.new [1.0], { :foo => :bar }, :shell => Thor::Shell::Basic.new
|
39
|
+
#
|
40
|
+
def initialize(args=[], options={}, config={})
|
41
|
+
super
|
42
|
+
self.shell = config[:shell]
|
43
|
+
self.shell.base ||= self if self.shell.respond_to?(:base)
|
44
|
+
end
|
45
|
+
|
46
|
+
# Holds the shell for the given Thor instance. If no shell is given,
|
47
|
+
# it gets a default shell from Thor::Base.shell.
|
48
|
+
def shell
|
49
|
+
@shell ||= Thor::Base.shell.new
|
50
|
+
end
|
51
|
+
|
52
|
+
# Sets the shell for this thor class.
|
53
|
+
def shell=(shell)
|
54
|
+
@shell = shell
|
55
|
+
end
|
56
|
+
|
57
|
+
# Common methods that are delegated to the shell.
|
58
|
+
SHELL_DELEGATED_METHODS.each do |method|
|
59
|
+
module_eval <<-METHOD, __FILE__, __LINE__
|
60
|
+
def #{method}(*args)
|
61
|
+
shell.#{method}(*args)
|
62
|
+
end
|
63
|
+
METHOD
|
64
|
+
end
|
65
|
+
|
66
|
+
# Yields the given block with padding.
|
67
|
+
def with_padding
|
68
|
+
shell.padding += 1
|
69
|
+
yield
|
70
|
+
ensure
|
71
|
+
shell.padding -= 1
|
72
|
+
end
|
73
|
+
|
74
|
+
protected
|
75
|
+
|
76
|
+
# Allow shell to be shared between invocations.
|
77
|
+
#
|
78
|
+
def _shared_configuration #:nodoc:
|
79
|
+
super.merge!(:shell => self.shell)
|
80
|
+
end
|
81
|
+
|
82
|
+
end
|
83
|
+
end
|
@@ -0,0 +1,239 @@
|
|
1
|
+
require 'tempfile'
|
2
|
+
|
3
|
+
class Thor
|
4
|
+
module Shell
|
5
|
+
class Basic
|
6
|
+
attr_accessor :base, :padding
|
7
|
+
|
8
|
+
# Initialize base and padding to nil.
|
9
|
+
#
|
10
|
+
def initialize #:nodoc:
|
11
|
+
@base, @padding = nil, 0
|
12
|
+
end
|
13
|
+
|
14
|
+
# Sets the output padding, not allowing less than zero values.
|
15
|
+
#
|
16
|
+
def padding=(value)
|
17
|
+
@padding = [0, value].max
|
18
|
+
end
|
19
|
+
|
20
|
+
# Ask something to the user and receives a response.
|
21
|
+
#
|
22
|
+
# ==== Example
|
23
|
+
# ask("What is your name?")
|
24
|
+
#
|
25
|
+
def ask(statement, color=nil)
|
26
|
+
say("#{statement} ", color)
|
27
|
+
$stdin.gets.strip
|
28
|
+
end
|
29
|
+
|
30
|
+
# Say (print) something to the user. If the sentence ends with a whitespace
|
31
|
+
# or tab character, a new line is not appended (print + flush). Otherwise
|
32
|
+
# are passed straight to puts (behavior got from Highline).
|
33
|
+
#
|
34
|
+
# ==== Example
|
35
|
+
# say("I know you knew that.")
|
36
|
+
#
|
37
|
+
def say(message="", color=nil, force_new_line=(message.to_s !~ /( |\t)$/))
|
38
|
+
message = message.to_s
|
39
|
+
message = set_color(message, color) if color
|
40
|
+
|
41
|
+
if force_new_line
|
42
|
+
$stdout.puts(message)
|
43
|
+
else
|
44
|
+
$stdout.print(message)
|
45
|
+
$stdout.flush
|
46
|
+
end
|
47
|
+
end
|
48
|
+
|
49
|
+
# Say a status with the given color and appends the message. Since this
|
50
|
+
# method is used frequently by actions, it allows nil or false to be given
|
51
|
+
# in log_status, avoiding the message from being shown. If a Symbol is
|
52
|
+
# given in log_status, it's used as the color.
|
53
|
+
#
|
54
|
+
def say_status(status, message, log_status=true)
|
55
|
+
return if quiet? || log_status == false
|
56
|
+
spaces = " " * (padding + 1)
|
57
|
+
color = log_status.is_a?(Symbol) ? log_status : :green
|
58
|
+
|
59
|
+
status = status.to_s.rjust(12)
|
60
|
+
status = set_color status, color, true if color
|
61
|
+
say "#{status}#{spaces}#{message}", nil, true
|
62
|
+
end
|
63
|
+
|
64
|
+
# Make a question the to user and returns true if the user replies "y" or
|
65
|
+
# "yes".
|
66
|
+
#
|
67
|
+
def yes?(statement, color=nil)
|
68
|
+
ask(statement, color) =~ is?(:yes)
|
69
|
+
end
|
70
|
+
|
71
|
+
# Make a question the to user and returns true if the user replies "n" or
|
72
|
+
# "no".
|
73
|
+
#
|
74
|
+
def no?(statement, color=nil)
|
75
|
+
!yes?(statement, color)
|
76
|
+
end
|
77
|
+
|
78
|
+
# Prints a table.
|
79
|
+
#
|
80
|
+
# ==== Parameters
|
81
|
+
# Array[Array[String, String, ...]]
|
82
|
+
#
|
83
|
+
# ==== Options
|
84
|
+
# ident<Integer>:: Ident the first column by ident value.
|
85
|
+
#
|
86
|
+
def print_table(table, options={})
|
87
|
+
return if table.empty?
|
88
|
+
|
89
|
+
formats, ident = [], options[:ident].to_i
|
90
|
+
options[:truncate] = terminal_width if options[:truncate] == true
|
91
|
+
|
92
|
+
0.upto(table.first.length - 2) do |i|
|
93
|
+
maxima = table.max{ |a,b| a[i].size <=> b[i].size }[i].size
|
94
|
+
formats << "%-#{maxima + 2}s"
|
95
|
+
end
|
96
|
+
|
97
|
+
formats[0] = formats[0].insert(0, " " * ident)
|
98
|
+
formats << "%s"
|
99
|
+
|
100
|
+
table.each do |row|
|
101
|
+
sentence = ""
|
102
|
+
|
103
|
+
row.each_with_index do |column, i|
|
104
|
+
sentence << formats[i] % column.to_s
|
105
|
+
end
|
106
|
+
|
107
|
+
sentence = truncate(sentence, options[:truncate]) if options[:truncate]
|
108
|
+
$stdout.puts sentence
|
109
|
+
end
|
110
|
+
end
|
111
|
+
|
112
|
+
# Deals with file collision and returns true if the file should be
|
113
|
+
# overwriten and false otherwise. If a block is given, it uses the block
|
114
|
+
# response as the content for the diff.
|
115
|
+
#
|
116
|
+
# ==== Parameters
|
117
|
+
# destination<String>:: the destination file to solve conflicts
|
118
|
+
# block<Proc>:: an optional block that returns the value to be used in diff
|
119
|
+
#
|
120
|
+
def file_collision(destination)
|
121
|
+
return true if @always_force
|
122
|
+
options = block_given? ? "[Ynaqdh]" : "[Ynaqh]"
|
123
|
+
|
124
|
+
while true
|
125
|
+
answer = ask %[Overwrite #{destination}? (enter "h" for help) #{options}]
|
126
|
+
|
127
|
+
case answer
|
128
|
+
when is?(:yes), is?(:force), ""
|
129
|
+
return true
|
130
|
+
when is?(:no), is?(:skip)
|
131
|
+
return false
|
132
|
+
when is?(:always)
|
133
|
+
return @always_force = true
|
134
|
+
when is?(:quit)
|
135
|
+
say 'Aborting...'
|
136
|
+
raise SystemExit
|
137
|
+
when is?(:diff)
|
138
|
+
show_diff(destination, yield) if block_given?
|
139
|
+
say 'Retrying...'
|
140
|
+
else
|
141
|
+
say file_collision_help
|
142
|
+
end
|
143
|
+
end
|
144
|
+
end
|
145
|
+
|
146
|
+
# Called if something goes wrong during the execution. This is used by Thor
|
147
|
+
# internally and should not be used inside your scripts. If someone went
|
148
|
+
# wrong, you can always raise an exception. If you raise a Thor::Error, it
|
149
|
+
# will be rescued and wrapped in the method below.
|
150
|
+
#
|
151
|
+
def error(statement)
|
152
|
+
$stderr.puts statement
|
153
|
+
end
|
154
|
+
|
155
|
+
# Apply color to the given string with optional bold. Disabled in the
|
156
|
+
# Thor::Shell::Basic class.
|
157
|
+
#
|
158
|
+
def set_color(string, color, bold=false) #:nodoc:
|
159
|
+
string
|
160
|
+
end
|
161
|
+
|
162
|
+
protected
|
163
|
+
|
164
|
+
def is?(value) #:nodoc:
|
165
|
+
value = value.to_s
|
166
|
+
|
167
|
+
if value.size == 1
|
168
|
+
/\A#{value}\z/i
|
169
|
+
else
|
170
|
+
/\A(#{value}|#{value[0,1]})\z/i
|
171
|
+
end
|
172
|
+
end
|
173
|
+
|
174
|
+
def file_collision_help #:nodoc:
|
175
|
+
<<HELP
|
176
|
+
Y - yes, overwrite
|
177
|
+
n - no, do not overwrite
|
178
|
+
a - all, overwrite this and all others
|
179
|
+
q - quit, abort
|
180
|
+
d - diff, show the differences between the old and the new
|
181
|
+
h - help, show this help
|
182
|
+
HELP
|
183
|
+
end
|
184
|
+
|
185
|
+
def show_diff(destination, content) #:nodoc:
|
186
|
+
diff_cmd = ENV['THOR_DIFF'] || ENV['RAILS_DIFF'] || 'diff -u'
|
187
|
+
|
188
|
+
Tempfile.open(File.basename(destination), File.dirname(destination)) do |temp|
|
189
|
+
temp.write content
|
190
|
+
temp.rewind
|
191
|
+
system %(#{diff_cmd} "#{destination}" "#{temp.path}")
|
192
|
+
end
|
193
|
+
end
|
194
|
+
|
195
|
+
def quiet? #:nodoc:
|
196
|
+
base && base.options[:quiet]
|
197
|
+
end
|
198
|
+
|
199
|
+
# This code was copied from Rake, available under MIT-LICENSE
|
200
|
+
# Copyright (c) 2003, 2004 Jim Weirich
|
201
|
+
def terminal_width
|
202
|
+
if ENV['THOR_COLUMNS']
|
203
|
+
result = ENV['THOR_COLUMNS'].to_i
|
204
|
+
else
|
205
|
+
result = unix? ? dynamic_width : 80
|
206
|
+
end
|
207
|
+
(result < 10) ? 80 : result
|
208
|
+
rescue
|
209
|
+
80
|
210
|
+
end
|
211
|
+
|
212
|
+
# Calculate the dynamic width of the terminal
|
213
|
+
def dynamic_width
|
214
|
+
@dynamic_width ||= (dynamic_width_stty.nonzero? || dynamic_width_tput)
|
215
|
+
end
|
216
|
+
|
217
|
+
def dynamic_width_stty
|
218
|
+
%x{stty size 2>/dev/null}.split[1].to_i
|
219
|
+
end
|
220
|
+
|
221
|
+
def dynamic_width_tput
|
222
|
+
%x{tput cols 2>/dev/null}.to_i
|
223
|
+
end
|
224
|
+
|
225
|
+
def unix?
|
226
|
+
RUBY_PLATFORM =~ /(aix|darwin|linux|(net|free|open)bsd|cygwin|solaris|irix|hpux)/i
|
227
|
+
end
|
228
|
+
|
229
|
+
def truncate(string, width)
|
230
|
+
if string.length <= width
|
231
|
+
string
|
232
|
+
else
|
233
|
+
( string[0, width-3] || "" ) + "..."
|
234
|
+
end
|
235
|
+
end
|
236
|
+
|
237
|
+
end
|
238
|
+
end
|
239
|
+
end
|
@@ -0,0 +1,108 @@
|
|
1
|
+
require 'thor/shell/basic'
|
2
|
+
|
3
|
+
class Thor
|
4
|
+
module Shell
|
5
|
+
# Inherit from Thor::Shell::Basic and add set_color behavior. Check
|
6
|
+
# Thor::Shell::Basic to see all available methods.
|
7
|
+
#
|
8
|
+
class Color < Basic
|
9
|
+
# Embed in a String to clear all previous ANSI sequences.
|
10
|
+
CLEAR = "\e[0m"
|
11
|
+
# The start of an ANSI bold sequence.
|
12
|
+
BOLD = "\e[1m"
|
13
|
+
|
14
|
+
# Set the terminal's foreground ANSI color to black.
|
15
|
+
BLACK = "\e[30m"
|
16
|
+
# Set the terminal's foreground ANSI color to red.
|
17
|
+
RED = "\e[31m"
|
18
|
+
# Set the terminal's foreground ANSI color to green.
|
19
|
+
GREEN = "\e[32m"
|
20
|
+
# Set the terminal's foreground ANSI color to yellow.
|
21
|
+
YELLOW = "\e[33m"
|
22
|
+
# Set the terminal's foreground ANSI color to blue.
|
23
|
+
BLUE = "\e[34m"
|
24
|
+
# Set the terminal's foreground ANSI color to magenta.
|
25
|
+
MAGENTA = "\e[35m"
|
26
|
+
# Set the terminal's foreground ANSI color to cyan.
|
27
|
+
CYAN = "\e[36m"
|
28
|
+
# Set the terminal's foreground ANSI color to white.
|
29
|
+
WHITE = "\e[37m"
|
30
|
+
|
31
|
+
# Set the terminal's background ANSI color to black.
|
32
|
+
ON_BLACK = "\e[40m"
|
33
|
+
# Set the terminal's background ANSI color to red.
|
34
|
+
ON_RED = "\e[41m"
|
35
|
+
# Set the terminal's background ANSI color to green.
|
36
|
+
ON_GREEN = "\e[42m"
|
37
|
+
# Set the terminal's background ANSI color to yellow.
|
38
|
+
ON_YELLOW = "\e[43m"
|
39
|
+
# Set the terminal's background ANSI color to blue.
|
40
|
+
ON_BLUE = "\e[44m"
|
41
|
+
# Set the terminal's background ANSI color to magenta.
|
42
|
+
ON_MAGENTA = "\e[45m"
|
43
|
+
# Set the terminal's background ANSI color to cyan.
|
44
|
+
ON_CYAN = "\e[46m"
|
45
|
+
# Set the terminal's background ANSI color to white.
|
46
|
+
ON_WHITE = "\e[47m"
|
47
|
+
|
48
|
+
# Set color by using a string or one of the defined constants. If a third
|
49
|
+
# option is set to true, it also adds bold to the string. This is based
|
50
|
+
# on Highline implementation and it automatically appends CLEAR to the end
|
51
|
+
# of the returned String.
|
52
|
+
#
|
53
|
+
def set_color(string, color, bold=false)
|
54
|
+
color = self.class.const_get(color.to_s.upcase) if color.is_a?(Symbol)
|
55
|
+
bold = bold ? BOLD : ""
|
56
|
+
"#{bold}#{color}#{string}#{CLEAR}"
|
57
|
+
end
|
58
|
+
|
59
|
+
protected
|
60
|
+
|
61
|
+
# Overwrite show_diff to show diff with colors if Diff::LCS is
|
62
|
+
# available.
|
63
|
+
#
|
64
|
+
def show_diff(destination, content) #:nodoc:
|
65
|
+
if diff_lcs_loaded? && ENV['THOR_DIFF'].nil? && ENV['RAILS_DIFF'].nil?
|
66
|
+
actual = File.binread(destination).to_s.split("\n")
|
67
|
+
content = content.to_s.split("\n")
|
68
|
+
|
69
|
+
Diff::LCS.sdiff(actual, content).each do |diff|
|
70
|
+
output_diff_line(diff)
|
71
|
+
end
|
72
|
+
else
|
73
|
+
super
|
74
|
+
end
|
75
|
+
end
|
76
|
+
|
77
|
+
def output_diff_line(diff) #:nodoc:
|
78
|
+
case diff.action
|
79
|
+
when '-'
|
80
|
+
say "- #{diff.old_element.chomp}", :red, true
|
81
|
+
when '+'
|
82
|
+
say "+ #{diff.new_element.chomp}", :green, true
|
83
|
+
when '!'
|
84
|
+
say "- #{diff.old_element.chomp}", :red, true
|
85
|
+
say "+ #{diff.new_element.chomp}", :green, true
|
86
|
+
else
|
87
|
+
say " #{diff.old_element.chomp}", nil, true
|
88
|
+
end
|
89
|
+
end
|
90
|
+
|
91
|
+
# Check if Diff::LCS is loaded. If it is, use it to create pretty output
|
92
|
+
# for diff.
|
93
|
+
#
|
94
|
+
def diff_lcs_loaded? #:nodoc:
|
95
|
+
return true if defined?(Diff::LCS)
|
96
|
+
return @diff_lcs_loaded unless @diff_lcs_loaded.nil?
|
97
|
+
|
98
|
+
@diff_lcs_loaded = begin
|
99
|
+
require 'diff/lcs'
|
100
|
+
true
|
101
|
+
rescue LoadError
|
102
|
+
false
|
103
|
+
end
|
104
|
+
end
|
105
|
+
|
106
|
+
end
|
107
|
+
end
|
108
|
+
end
|
@@ -0,0 +1,102 @@
|
|
1
|
+
class Thor
|
2
|
+
class Task < Struct.new(:name, :description, :usage, :options)
|
3
|
+
FILE_REGEXP = /^#{Regexp.escape(File.dirname(__FILE__))}/
|
4
|
+
|
5
|
+
# A dynamic task that handles method missing scenarios.
|
6
|
+
class Dynamic < Task
|
7
|
+
def initialize(name, options=nil)
|
8
|
+
super(name.to_s, "A dynamically-generated task", name.to_s, options)
|
9
|
+
end
|
10
|
+
|
11
|
+
def run(instance, args=[])
|
12
|
+
if (instance.methods & [name.to_s, name.to_sym]).empty?
|
13
|
+
super
|
14
|
+
else
|
15
|
+
instance.class.handle_no_task_error(name)
|
16
|
+
end
|
17
|
+
end
|
18
|
+
end
|
19
|
+
|
20
|
+
def initialize(name, description, usage, options=nil)
|
21
|
+
super(name.to_s, description, usage, options || {})
|
22
|
+
end
|
23
|
+
|
24
|
+
def initialize_copy(other) #:nodoc:
|
25
|
+
super(other)
|
26
|
+
self.options = other.options.dup if other.options
|
27
|
+
end
|
28
|
+
|
29
|
+
# By default, a task invokes a method in the thor class. You can change this
|
30
|
+
# implementation to create custom tasks.
|
31
|
+
def run(instance, args=[])
|
32
|
+
public_method?(instance) ?
|
33
|
+
instance.send(name, *args) : instance.class.handle_no_task_error(name)
|
34
|
+
rescue ArgumentError => e
|
35
|
+
handle_argument_error?(instance, e, caller) ?
|
36
|
+
instance.class.handle_argument_error(self, e) : (raise e)
|
37
|
+
rescue NoMethodError => e
|
38
|
+
handle_no_method_error?(instance, e, caller) ?
|
39
|
+
instance.class.handle_no_task_error(name) : (raise e)
|
40
|
+
end
|
41
|
+
|
42
|
+
# Returns the formatted usage by injecting given required arguments
|
43
|
+
# and required options into the given usage.
|
44
|
+
def formatted_usage(klass, namespace=true)
|
45
|
+
namespace = klass.namespace unless namespace == false
|
46
|
+
|
47
|
+
# Add namespace
|
48
|
+
formatted = if namespace
|
49
|
+
"#{namespace.gsub(/^(default|thor:runner:)/,'')}:"
|
50
|
+
else
|
51
|
+
""
|
52
|
+
end
|
53
|
+
|
54
|
+
# Add usage with required arguments
|
55
|
+
formatted << if klass && !klass.arguments.empty?
|
56
|
+
usage.to_s.gsub(/^#{name}/) do |match|
|
57
|
+
match << " " << klass.arguments.map{ |a| a.usage }.compact.join(' ')
|
58
|
+
end
|
59
|
+
else
|
60
|
+
usage.to_s
|
61
|
+
end
|
62
|
+
|
63
|
+
# Add required options
|
64
|
+
formatted << " #{required_options}"
|
65
|
+
|
66
|
+
# Strip and go!
|
67
|
+
formatted.strip
|
68
|
+
end
|
69
|
+
|
70
|
+
protected
|
71
|
+
|
72
|
+
def not_debugging?(instance)
|
73
|
+
!(instance.class.respond_to?(:debugging) && instance.class.debugging)
|
74
|
+
end
|
75
|
+
|
76
|
+
def required_options
|
77
|
+
@required_options ||= options.map{ |_, o| o.usage if o.required? }.compact.sort.join(" ")
|
78
|
+
end
|
79
|
+
|
80
|
+
# Given a target, checks if this class name is not a private/protected method.
|
81
|
+
def public_method?(instance) #:nodoc:
|
82
|
+
collection = instance.private_methods + instance.protected_methods
|
83
|
+
(collection & [name.to_s, name.to_sym]).empty?
|
84
|
+
end
|
85
|
+
|
86
|
+
def sans_backtrace(backtrace, caller) #:nodoc:
|
87
|
+
saned = backtrace.reject { |frame| frame =~ FILE_REGEXP }
|
88
|
+
saned -= caller
|
89
|
+
end
|
90
|
+
|
91
|
+
def handle_argument_error?(instance, error, caller)
|
92
|
+
not_debugging?(instance) && error.message =~ /wrong number of arguments/ &&
|
93
|
+
sans_backtrace(error.backtrace, caller).empty?
|
94
|
+
end
|
95
|
+
|
96
|
+
def handle_no_method_error?(instance, error, caller)
|
97
|
+
not_debugging?(instance) &&
|
98
|
+
error.message =~ /^undefined method `#{name}' for #{Regexp.escape(instance.to_s)}$/
|
99
|
+
end
|
100
|
+
|
101
|
+
end
|
102
|
+
end
|