acquia_toolbelt 2.3.0 → 2.3.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (154) hide show
  1. data/.gitignore +7 -0
  2. data/.ruby-version +1 -0
  3. data/.travis.yml +7 -0
  4. data/CONTRIBUTING.md +11 -0
  5. data/Gemfile +3 -0
  6. data/Gemfile.lock +55 -0
  7. data/Guardfile +5 -0
  8. data/LICENSE +19 -0
  9. data/README.md +149 -0
  10. data/Rakefile +1 -0
  11. data/acquia_toolbelt.gemspec +34 -0
  12. data/bin/acquia +9 -0
  13. data/lib/acquia_toolbelt/cli/api.rb +164 -0
  14. data/lib/acquia_toolbelt/cli/auth.rb +31 -0
  15. data/lib/acquia_toolbelt/cli/database.rb +237 -0
  16. data/lib/acquia_toolbelt/cli/deploy.rb +36 -0
  17. data/lib/acquia_toolbelt/cli/domain.rb +177 -0
  18. data/lib/acquia_toolbelt/cli/environment.rb +71 -0
  19. data/lib/acquia_toolbelt/cli/file.rb +31 -0
  20. data/lib/acquia_toolbelt/cli/server.rb +67 -0
  21. data/lib/acquia_toolbelt/cli/site.rb +28 -0
  22. data/lib/acquia_toolbelt/cli/ssh.rb +78 -0
  23. data/lib/acquia_toolbelt/cli/svn.rb +73 -0
  24. data/lib/acquia_toolbelt/cli/task.rb +74 -0
  25. data/lib/acquia_toolbelt/cli/ui.rb +44 -0
  26. data/lib/acquia_toolbelt/cli.rb +103 -0
  27. data/lib/acquia_toolbelt/error.rb +4 -0
  28. data/lib/acquia_toolbelt/thor.rb +95 -0
  29. data/lib/acquia_toolbelt/version.rb +3 -0
  30. data/lib/acquia_toolbelt.rb +4 -0
  31. data/lib/vendor/thor/CHANGELOG.md +139 -0
  32. data/lib/vendor/thor/Gemfile +20 -0
  33. data/lib/vendor/thor/LICENSE.md +20 -0
  34. data/lib/vendor/thor/README.md +35 -0
  35. data/lib/vendor/thor/lib/thor/actions/create_file.rb +105 -0
  36. data/lib/vendor/thor/lib/thor/actions/create_link.rb +60 -0
  37. data/lib/vendor/thor/lib/thor/actions/directory.rb +119 -0
  38. data/lib/vendor/thor/lib/thor/actions/empty_directory.rb +137 -0
  39. data/lib/vendor/thor/lib/thor/actions/file_manipulation.rb +317 -0
  40. data/lib/vendor/thor/lib/thor/actions/inject_into_file.rb +109 -0
  41. data/lib/vendor/thor/lib/thor/actions.rb +318 -0
  42. data/lib/vendor/thor/lib/thor/base.rb +654 -0
  43. data/lib/vendor/thor/lib/thor/command.rb +136 -0
  44. data/lib/vendor/thor/lib/thor/core_ext/hash_with_indifferent_access.rb +80 -0
  45. data/lib/vendor/thor/lib/thor/core_ext/io_binary_read.rb +12 -0
  46. data/lib/vendor/thor/lib/thor/core_ext/ordered_hash.rb +100 -0
  47. data/lib/vendor/thor/lib/thor/error.rb +32 -0
  48. data/lib/vendor/thor/lib/thor/group.rb +282 -0
  49. data/lib/vendor/thor/lib/thor/invocation.rb +172 -0
  50. data/lib/vendor/thor/lib/thor/parser/argument.rb +74 -0
  51. data/lib/vendor/thor/lib/thor/parser/arguments.rb +171 -0
  52. data/lib/vendor/thor/lib/thor/parser/option.rb +121 -0
  53. data/lib/vendor/thor/lib/thor/parser/options.rb +218 -0
  54. data/lib/vendor/thor/lib/thor/parser.rb +4 -0
  55. data/lib/vendor/thor/lib/thor/rake_compat.rb +72 -0
  56. data/lib/vendor/thor/lib/thor/runner.rb +322 -0
  57. data/lib/vendor/thor/lib/thor/shell/basic.rb +422 -0
  58. data/lib/vendor/thor/lib/thor/shell/color.rb +148 -0
  59. data/lib/vendor/thor/lib/thor/shell/html.rb +127 -0
  60. data/lib/vendor/thor/lib/thor/shell.rb +88 -0
  61. data/lib/vendor/thor/lib/thor/util.rb +270 -0
  62. data/lib/vendor/thor/lib/thor/version.rb +3 -0
  63. data/lib/vendor/thor/lib/thor.rb +474 -0
  64. data/lib/vendor/thor/spec/actions/create_file_spec.rb +170 -0
  65. data/lib/vendor/thor/spec/actions/create_link_spec.rb +95 -0
  66. data/lib/vendor/thor/spec/actions/directory_spec.rb +169 -0
  67. data/lib/vendor/thor/spec/actions/empty_directory_spec.rb +129 -0
  68. data/lib/vendor/thor/spec/actions/file_manipulation_spec.rb +382 -0
  69. data/lib/vendor/thor/spec/actions/inject_into_file_spec.rb +135 -0
  70. data/lib/vendor/thor/spec/actions_spec.rb +331 -0
  71. data/lib/vendor/thor/spec/base_spec.rb +291 -0
  72. data/lib/vendor/thor/spec/command_spec.rb +80 -0
  73. data/lib/vendor/thor/spec/core_ext/hash_with_indifferent_access_spec.rb +48 -0
  74. data/lib/vendor/thor/spec/core_ext/ordered_hash_spec.rb +115 -0
  75. data/lib/vendor/thor/spec/exit_condition_spec.rb +19 -0
  76. data/lib/vendor/thor/spec/fixtures/application.rb +2 -0
  77. data/lib/vendor/thor/spec/fixtures/app{1}/README +3 -0
  78. data/lib/vendor/thor/spec/fixtures/bundle/execute.rb +6 -0
  79. data/lib/vendor/thor/spec/fixtures/bundle/main.thor +1 -0
  80. data/lib/vendor/thor/spec/fixtures/command.thor +10 -0
  81. data/lib/vendor/thor/spec/fixtures/doc/%file_name%.rb.tt +1 -0
  82. data/lib/vendor/thor/spec/fixtures/doc/COMMENTER +11 -0
  83. data/lib/vendor/thor/spec/fixtures/doc/README +3 -0
  84. data/lib/vendor/thor/spec/fixtures/doc/block_helper.rb +3 -0
  85. data/lib/vendor/thor/spec/fixtures/doc/components/.empty_directory +0 -0
  86. data/lib/vendor/thor/spec/fixtures/doc/config.rb +1 -0
  87. data/lib/vendor/thor/spec/fixtures/doc/config.yaml.tt +1 -0
  88. data/lib/vendor/thor/spec/fixtures/doc/excluding/%file_name%.rb.tt +1 -0
  89. data/lib/vendor/thor/spec/fixtures/enum.thor +10 -0
  90. data/lib/vendor/thor/spec/fixtures/group.thor +128 -0
  91. data/lib/vendor/thor/spec/fixtures/invoke.thor +118 -0
  92. data/lib/vendor/thor/spec/fixtures/path with spaces +0 -0
  93. data/lib/vendor/thor/spec/fixtures/preserve/script.sh +3 -0
  94. data/lib/vendor/thor/spec/fixtures/script.thor +220 -0
  95. data/lib/vendor/thor/spec/fixtures/subcommand.thor +17 -0
  96. data/lib/vendor/thor/spec/group_spec.rb +222 -0
  97. data/lib/vendor/thor/spec/helper.rb +67 -0
  98. data/lib/vendor/thor/spec/invocation_spec.rb +108 -0
  99. data/lib/vendor/thor/spec/parser/argument_spec.rb +53 -0
  100. data/lib/vendor/thor/spec/parser/arguments_spec.rb +66 -0
  101. data/lib/vendor/thor/spec/parser/option_spec.rb +202 -0
  102. data/lib/vendor/thor/spec/parser/options_spec.rb +400 -0
  103. data/lib/vendor/thor/spec/rake_compat_spec.rb +72 -0
  104. data/lib/vendor/thor/spec/register_spec.rb +197 -0
  105. data/lib/vendor/thor/spec/runner_spec.rb +241 -0
  106. data/lib/vendor/thor/spec/shell/basic_spec.rb +330 -0
  107. data/lib/vendor/thor/spec/shell/color_spec.rb +95 -0
  108. data/lib/vendor/thor/spec/shell/html_spec.rb +31 -0
  109. data/lib/vendor/thor/spec/shell_spec.rb +47 -0
  110. data/lib/vendor/thor/spec/subcommand_spec.rb +30 -0
  111. data/lib/vendor/thor/spec/thor_spec.rb +499 -0
  112. data/lib/vendor/thor/spec/util_spec.rb +196 -0
  113. data/lib/vendor/thor/thor.gemspec +24 -0
  114. data/script/release +50 -0
  115. data/script/setup_build +6 -0
  116. data/script/test +23 -0
  117. data/spec/auth_spec.rb +15 -0
  118. data/spec/cassettes/databases/all_databases.json +1 -0
  119. data/spec/cassettes/databases/copy_database_from_dev_to_stage.json +1 -0
  120. data/spec/cassettes/databases/create_a_database_backup.json +1 -0
  121. data/spec/cassettes/databases/create_a_new_database.json +1 -0
  122. data/spec/cassettes/databases/delete_a_database.json +1 -0
  123. data/spec/cassettes/databases/get_all_existing_databases.json +1 -0
  124. data/spec/cassettes/databases/list_all_database_backups.json +1 -0
  125. data/spec/cassettes/databases/view_database_instance_details.json +1 -0
  126. data/spec/cassettes/deploy/release_vcs_branch.json +1 -0
  127. data/spec/cassettes/domains/create_new_domain.json +1 -0
  128. data/spec/cassettes/domains/delete_a_domain.json +1 -0
  129. data/spec/cassettes/domains/get_all_existing_domains.json +1 -0
  130. data/spec/cassettes/domains/list_all_dev_domains.json +1 -0
  131. data/spec/cassettes/domains/move_from_dev_to_stage.json +1 -0
  132. data/spec/cassettes/domains/purge_varnish_cache.json +1 -0
  133. data/spec/cassettes/environments/all_environments.json +1 -0
  134. data/spec/cassettes/environments/disable_live_development.json +1 -0
  135. data/spec/cassettes/environments/enable_live_development.json +1 -0
  136. data/spec/cassettes/files/copy_from_dev_to_stage.json +1 -0
  137. data/spec/cassettes/servers/all_dev_servers.json +1 -0
  138. data/spec/cassettes/servers/all_prod_servers.json +1 -0
  139. data/spec/cassettes/sites/all_sites.json +1 -0
  140. data/spec/cassettes/ssh/all_sshkeys.json +1 -0
  141. data/spec/cassettes/svn/all_svnusers.json +1 -0
  142. data/spec/cassettes/tasks/all_tasks.json +1 -0
  143. data/spec/databases_spec.rb +78 -0
  144. data/spec/deploy_spec.rb +12 -0
  145. data/spec/domains_spec.rb +59 -0
  146. data/spec/environments_spec.rb +35 -0
  147. data/spec/files_spec.rb +11 -0
  148. data/spec/helper.rb +104 -0
  149. data/spec/servers_spec.rb +59 -0
  150. data/spec/sites_spec.rb +19 -0
  151. data/spec/ssh_spec.rb +19 -0
  152. data/spec/svn_spec.rb +11 -0
  153. data/spec/tasks_spec.rb +11 -0
  154. metadata +158 -4
@@ -0,0 +1,218 @@
1
+ class Thor
2
+ class Options < Arguments #:nodoc:
3
+ LONG_RE = /^(--\w+(?:-\w+)*)$/
4
+ SHORT_RE = /^(-[a-z])$/i
5
+ EQ_RE = /^(--\w+(?:-\w+)*|-[a-z])=(.*)$/i
6
+ SHORT_SQ_RE = /^-([a-z]{2,})$/i # Allow either -x -v or -xv style for single char args
7
+ SHORT_NUM = /^(-[a-z])#{NUMERIC}$/i
8
+ OPTS_END = '--'.freeze
9
+
10
+ # Receives a hash and makes it switches.
11
+ def self.to_switches(options)
12
+ options.map do |key, value|
13
+ case value
14
+ when true
15
+ "--#{key}"
16
+ when Array
17
+ "--#{key} #{value.map{ |v| v.inspect }.join(' ')}"
18
+ when Hash
19
+ "--#{key} #{value.map{ |k,v| "#{k}:#{v}" }.join(' ')}"
20
+ when nil, false
21
+ ""
22
+ else
23
+ "--#{key} #{value.inspect}"
24
+ end
25
+ end.join(" ")
26
+ end
27
+
28
+ # Takes a hash of Thor::Option and a hash with defaults.
29
+ #
30
+ # If +stop_on_unknown+ is true, #parse will stop as soon as it encounters
31
+ # an unknown option or a regular argument.
32
+ def initialize(hash_options={}, defaults={}, stop_on_unknown=false)
33
+ @stop_on_unknown = stop_on_unknown
34
+ options = hash_options.values
35
+ super(options)
36
+
37
+ # Add defaults
38
+ defaults.each do |key, value|
39
+ @assigns[key.to_s] = value
40
+ @non_assigned_required.delete(hash_options[key])
41
+ end
42
+
43
+ @shorts, @switches, @extra = {}, {}, []
44
+
45
+ options.each do |option|
46
+ @switches[option.switch_name] = option
47
+
48
+ option.aliases.each do |short|
49
+ name = short.to_s.sub(/^(?!\-)/, '-')
50
+ @shorts[name] ||= option.switch_name
51
+ end
52
+ end
53
+ end
54
+
55
+ def remaining
56
+ @extra
57
+ end
58
+
59
+ def peek
60
+ return super unless @parsing_options
61
+
62
+ result = super
63
+ if result == OPTS_END
64
+ shift
65
+ @parsing_options = false
66
+ super
67
+ else
68
+ result
69
+ end
70
+ end
71
+
72
+ def parse(args)
73
+ @pile = args.dup
74
+ @parsing_options = true
75
+
76
+ while peek
77
+ if parsing_options?
78
+ match, is_switch = current_is_switch?
79
+ shifted = shift
80
+
81
+ if is_switch
82
+ case shifted
83
+ when SHORT_SQ_RE
84
+ unshift($1.split('').map { |f| "-#{f}" })
85
+ next
86
+ when EQ_RE, SHORT_NUM
87
+ unshift($2)
88
+ switch = $1
89
+ when LONG_RE, SHORT_RE
90
+ switch = $1
91
+ end
92
+
93
+ switch = normalize_switch(switch)
94
+ option = switch_option(switch)
95
+ @assigns[option.human_name] = parse_peek(switch, option)
96
+ elsif @stop_on_unknown
97
+ @parsing_options = false
98
+ @extra << shifted
99
+ @extra << shift while peek
100
+ break
101
+ elsif match
102
+ @extra << shifted
103
+ @extra << shift while peek && peek !~ /^-/
104
+ else
105
+ @extra << shifted
106
+ end
107
+ else
108
+ @extra << shift
109
+ end
110
+ end
111
+
112
+ check_requirement!
113
+
114
+ assigns = Thor::CoreExt::HashWithIndifferentAccess.new(@assigns)
115
+ assigns.freeze
116
+ assigns
117
+ end
118
+
119
+ def check_unknown!
120
+ # an unknown option starts with - or -- and has no more --'s afterward.
121
+ unknown = @extra.select { |str| str =~ /^--?(?:(?!--).)*$/ }
122
+ raise UnknownArgumentError, "Unknown switches '#{unknown.join(', ')}'" unless unknown.empty?
123
+ end
124
+
125
+ protected
126
+
127
+ # Check if the current value in peek is a registered switch.
128
+ #
129
+ # Two booleans are returned. The first is true if the current value
130
+ # starts with a hyphen; the second is true if it is a registered switch.
131
+ def current_is_switch?
132
+ case peek
133
+ when LONG_RE, SHORT_RE, EQ_RE, SHORT_NUM
134
+ [true, switch?($1)]
135
+ when SHORT_SQ_RE
136
+ [true, $1.split('').any? { |f| switch?("-#{f}") }]
137
+ else
138
+ [false, false]
139
+ end
140
+ end
141
+
142
+ def current_is_switch_formatted?
143
+ case peek
144
+ when LONG_RE, SHORT_RE, EQ_RE, SHORT_NUM, SHORT_SQ_RE
145
+ true
146
+ else
147
+ false
148
+ end
149
+ end
150
+
151
+ def current_is_value?
152
+ peek && (!parsing_options? || super)
153
+ end
154
+
155
+ def switch?(arg)
156
+ switch_option(normalize_switch(arg))
157
+ end
158
+
159
+ def switch_option(arg)
160
+ if match = no_or_skip?(arg)
161
+ @switches[arg] || @switches["--#{match}"]
162
+ else
163
+ @switches[arg]
164
+ end
165
+ end
166
+
167
+ # Check if the given argument is actually a shortcut.
168
+ #
169
+ def normalize_switch(arg)
170
+ (@shorts[arg] || arg).tr('_', '-')
171
+ end
172
+
173
+ def parsing_options?
174
+ peek
175
+ @parsing_options
176
+ end
177
+
178
+ # Parse boolean values which can be given as --foo=true, --foo or --no-foo.
179
+ #
180
+ def parse_boolean(switch)
181
+ if current_is_value?
182
+ if ["true", "TRUE", "t", "T", true].include?(peek)
183
+ shift
184
+ true
185
+ elsif ["false", "FALSE", "f", "F", false].include?(peek)
186
+ shift
187
+ false
188
+ else
189
+ true
190
+ end
191
+ else
192
+ @switches.key?(switch) || !no_or_skip?(switch)
193
+ end
194
+ end
195
+
196
+ # Parse the value at the peek analyzing if it requires an input or not.
197
+ #
198
+ def parse_peek(switch, option)
199
+ if parsing_options? && (current_is_switch_formatted? || last?)
200
+ if option.boolean?
201
+ # No problem for boolean types
202
+ elsif no_or_skip?(switch)
203
+ return nil # User set value to nil
204
+ elsif option.string? && !option.required?
205
+ # Return the default if there is one, else the human name
206
+ return option.lazy_default || option.default || option.human_name
207
+ elsif option.lazy_default
208
+ return option.lazy_default
209
+ else
210
+ raise MalformattedArgumentError, "No value provided for option '#{switch}'"
211
+ end
212
+ end
213
+
214
+ @non_assigned_required.delete(option)
215
+ send(:"parse_#{option.type}", switch)
216
+ end
217
+ end
218
+ end
@@ -0,0 +1,4 @@
1
+ require 'thor/parser/argument'
2
+ require 'thor/parser/arguments'
3
+ require 'thor/parser/option'
4
+ require 'thor/parser/options'
@@ -0,0 +1,72 @@
1
+ require 'rake'
2
+ require 'rake/dsl_definition'
3
+
4
+ class Thor
5
+ # Adds a compatibility layer to your Thor classes which allows you to use
6
+ # rake package tasks. For example, to use rspec rake tasks, one can do:
7
+ #
8
+ # require 'thor/rake_compat'
9
+ # require 'rspec/core/rake_task'
10
+ #
11
+ # class Default < Thor
12
+ # include Thor::RakeCompat
13
+ #
14
+ # RSpec::Core::RakeTask.new(:spec) do |t|
15
+ # t.spec_opts = ['--options', "./.rspec"]
16
+ # t.spec_files = FileList['spec/**/*_spec.rb']
17
+ # end
18
+ # end
19
+ #
20
+ module RakeCompat
21
+ include Rake::DSL if defined?(Rake::DSL)
22
+
23
+ def self.rake_classes
24
+ @rake_classes ||= []
25
+ end
26
+
27
+ def self.included(base)
28
+ # Hack. Make rakefile point to invoker, so rdoc task is generated properly.
29
+ rakefile = File.basename(caller[0].match(/(.*):\d+/)[1])
30
+ Rake.application.instance_variable_set(:@rakefile, rakefile)
31
+ self.rake_classes << base
32
+ end
33
+ end
34
+ end
35
+
36
+ # override task on (main), for compatibility with Rake 0.9
37
+ self.instance_eval do
38
+ alias rake_namespace namespace
39
+
40
+ def task(*)
41
+ task = super
42
+
43
+ if klass = Thor::RakeCompat.rake_classes.last
44
+ non_namespaced_name = task.name.split(':').last
45
+
46
+ description = non_namespaced_name
47
+ description << task.arg_names.map{ |n| n.to_s.upcase }.join(' ')
48
+ description.strip!
49
+
50
+ klass.desc description, Rake.application.last_description || non_namespaced_name
51
+ Rake.application.last_description = nil
52
+ klass.send :define_method, non_namespaced_name do |*args|
53
+ Rake::Task[task.name.to_sym].invoke(*args)
54
+ end
55
+ end
56
+
57
+ task
58
+ end
59
+
60
+ def namespace(name)
61
+ if klass = Thor::RakeCompat.rake_classes.last
62
+ const_name = Thor::Util.camel_case(name.to_s).to_sym
63
+ klass.const_set(const_name, Class.new(Thor))
64
+ new_klass = klass.const_get(const_name)
65
+ Thor::RakeCompat.rake_classes << new_klass
66
+ end
67
+
68
+ super
69
+ Thor::RakeCompat.rake_classes.pop
70
+ end
71
+ end
72
+
@@ -0,0 +1,322 @@
1
+ require 'thor'
2
+ require 'thor/group'
3
+ require 'thor/core_ext/io_binary_read'
4
+
5
+ require 'fileutils'
6
+ require 'open-uri'
7
+ require 'yaml'
8
+ require 'digest/md5'
9
+ require 'pathname'
10
+
11
+ class Thor::Runner < Thor #:nodoc:
12
+ map "-T" => :list, "-i" => :install, "-u" => :update, "-v" => :version
13
+
14
+ # Override Thor#help so it can give information about any class and any method.
15
+ #
16
+ def help(meth = nil)
17
+ if meth && !self.respond_to?(meth)
18
+ initialize_thorfiles(meth)
19
+ klass, command = Thor::Util.find_class_and_command_by_namespace(meth)
20
+ self.class.handle_no_command_error(command, false) if klass.nil?
21
+ klass.start(["-h", command].compact, :shell => self.shell)
22
+ else
23
+ super
24
+ end
25
+ end
26
+
27
+ # If a command is not found on Thor::Runner, method missing is invoked and
28
+ # Thor::Runner is then responsible for finding the command in all classes.
29
+ #
30
+ def method_missing(meth, *args)
31
+ meth = meth.to_s
32
+ initialize_thorfiles(meth)
33
+ klass, command = Thor::Util.find_class_and_command_by_namespace(meth)
34
+ self.class.handle_no_command_error(command, false) if klass.nil?
35
+ args.unshift(command) if command
36
+ klass.start(args, :shell => self.shell)
37
+ end
38
+
39
+ desc "install NAME", "Install an optionally named Thor file into your system commands"
40
+ method_options :as => :string, :relative => :boolean, :force => :boolean
41
+ def install(name)
42
+ initialize_thorfiles
43
+
44
+ # If a directory name is provided as the argument, look for a 'main.thor'
45
+ # command in said directory.
46
+ begin
47
+ if File.directory?(File.expand_path(name))
48
+ base, package = File.join(name, "main.thor"), :directory
49
+ contents = open(base) {|input| input.read }
50
+ else
51
+ base, package = name, :file
52
+ contents = open(name) {|input| input.read }
53
+ end
54
+ rescue OpenURI::HTTPError
55
+ raise Error, "Error opening URI '#{name}'"
56
+ rescue Errno::ENOENT
57
+ raise Error, "Error opening file '#{name}'"
58
+ end
59
+
60
+ say "Your Thorfile contains:"
61
+ say contents
62
+
63
+ unless options["force"]
64
+ return false if no?("Do you wish to continue [y/N]?")
65
+ end
66
+
67
+ as = options["as"] || begin
68
+ first_line = contents.split("\n")[0]
69
+ (match = first_line.match(/\s*#\s*module:\s*([^\n]*)/)) ? match[1].strip : nil
70
+ end
71
+
72
+ unless as
73
+ basename = File.basename(name)
74
+ as = ask("Please specify a name for #{name} in the system repository [#{basename}]:")
75
+ as = basename if as.empty?
76
+ end
77
+
78
+ location = if options[:relative] || name =~ /^https?:\/\//
79
+ name
80
+ else
81
+ File.expand_path(name)
82
+ end
83
+
84
+ thor_yaml[as] = {
85
+ :filename => Digest::MD5.hexdigest(name + as),
86
+ :location => location,
87
+ :namespaces => Thor::Util.namespaces_in_content(contents, base)
88
+ }
89
+
90
+ save_yaml(thor_yaml)
91
+ say "Storing thor file in your system repository"
92
+ destination = File.join(thor_root, thor_yaml[as][:filename])
93
+
94
+ if package == :file
95
+ File.open(destination, "w") { |f| f.puts contents }
96
+ else
97
+ FileUtils.cp_r(name, destination)
98
+ end
99
+
100
+ thor_yaml[as][:filename] # Indicate success
101
+ end
102
+
103
+ desc "version", "Show Thor version"
104
+ def version
105
+ require 'thor/version'
106
+ say "Thor #{Thor::VERSION}"
107
+ end
108
+
109
+ desc "uninstall NAME", "Uninstall a named Thor module"
110
+ def uninstall(name)
111
+ raise Error, "Can't find module '#{name}'" unless thor_yaml[name]
112
+ say "Uninstalling #{name}."
113
+ FileUtils.rm_rf(File.join(thor_root, "#{thor_yaml[name][:filename]}"))
114
+
115
+ thor_yaml.delete(name)
116
+ save_yaml(thor_yaml)
117
+
118
+ puts "Done."
119
+ end
120
+
121
+ desc "update NAME", "Update a Thor file from its original location"
122
+ def update(name)
123
+ raise Error, "Can't find module '#{name}'" if !thor_yaml[name] || !thor_yaml[name][:location]
124
+
125
+ say "Updating '#{name}' from #{thor_yaml[name][:location]}"
126
+
127
+ old_filename = thor_yaml[name][:filename]
128
+ self.options = self.options.merge("as" => name)
129
+
130
+ if File.directory? File.expand_path(name)
131
+ FileUtils.rm_rf(File.join(thor_root, old_filename))
132
+
133
+ thor_yaml.delete(old_filename)
134
+ save_yaml(thor_yaml)
135
+
136
+ filename = install(name)
137
+ else
138
+ filename = install(thor_yaml[name][:location])
139
+ end
140
+
141
+ unless filename == old_filename
142
+ File.delete(File.join(thor_root, old_filename))
143
+ end
144
+ end
145
+
146
+ desc "installed", "List the installed Thor modules and commands"
147
+ method_options :internal => :boolean
148
+ def installed
149
+ initialize_thorfiles(nil, true)
150
+ display_klasses(true, options["internal"])
151
+ end
152
+
153
+ desc "list [SEARCH]", "List the available thor commands (--substring means .*SEARCH)"
154
+ method_options :substring => :boolean, :group => :string, :all => :boolean, :debug => :boolean
155
+ def list(search="")
156
+ initialize_thorfiles
157
+
158
+ search = ".*#{search}" if options["substring"]
159
+ search = /^#{search}.*/i
160
+ group = options[:group] || "standard"
161
+
162
+ klasses = Thor::Base.subclasses.select do |k|
163
+ (options[:all] || k.group == group) && k.namespace =~ search
164
+ end
165
+
166
+ display_klasses(false, false, klasses)
167
+ end
168
+
169
+ private
170
+
171
+ def self.banner(command, all = false, subcommand = false)
172
+ "thor " + command.formatted_usage(self, all, subcommand)
173
+ end
174
+
175
+ def thor_root
176
+ Thor::Util.thor_root
177
+ end
178
+
179
+ def thor_yaml
180
+ @thor_yaml ||= begin
181
+ yaml_file = File.join(thor_root, "thor.yml")
182
+ yaml = YAML.load_file(yaml_file) if File.exists?(yaml_file)
183
+ yaml || {}
184
+ end
185
+ end
186
+
187
+ # Save the yaml file. If none exists in thor root, creates one.
188
+ #
189
+ def save_yaml(yaml)
190
+ yaml_file = File.join(thor_root, "thor.yml")
191
+
192
+ unless File.exists?(yaml_file)
193
+ FileUtils.mkdir_p(thor_root)
194
+ yaml_file = File.join(thor_root, "thor.yml")
195
+ FileUtils.touch(yaml_file)
196
+ end
197
+
198
+ File.open(yaml_file, "w") { |f| f.puts yaml.to_yaml }
199
+ end
200
+
201
+ def self.exit_on_failure?
202
+ true
203
+ end
204
+
205
+ # Load the Thorfiles. If relevant_to is supplied, looks for specific files
206
+ # in the thor_root instead of loading them all.
207
+ #
208
+ # By default, it also traverses the current path until find Thor files, as
209
+ # described in thorfiles. This look up can be skipped by supplying
210
+ # skip_lookup true.
211
+ #
212
+ def initialize_thorfiles(relevant_to=nil, skip_lookup=false)
213
+ thorfiles(relevant_to, skip_lookup).each do |f|
214
+ Thor::Util.load_thorfile(f, nil, options[:debug]) unless Thor::Base.subclass_files.keys.include?(File.expand_path(f))
215
+ end
216
+ end
217
+
218
+ # Finds Thorfiles by traversing from your current directory down to the root
219
+ # directory of your system. If at any time we find a Thor file, we stop.
220
+ #
221
+ # We also ensure that system-wide Thorfiles are loaded first, so local
222
+ # Thorfiles can override them.
223
+ #
224
+ # ==== Example
225
+ #
226
+ # If we start at /Users/wycats/dev/thor ...
227
+ #
228
+ # 1. /Users/wycats/dev/thor
229
+ # 2. /Users/wycats/dev
230
+ # 3. /Users/wycats <-- we find a Thorfile here, so we stop
231
+ #
232
+ # Suppose we start at c:\Documents and Settings\james\dev\thor ...
233
+ #
234
+ # 1. c:\Documents and Settings\james\dev\thor
235
+ # 2. c:\Documents and Settings\james\dev
236
+ # 3. c:\Documents and Settings\james
237
+ # 4. c:\Documents and Settings
238
+ # 5. c:\ <-- no Thorfiles found!
239
+ #
240
+ def thorfiles(relevant_to=nil, skip_lookup=false)
241
+ thorfiles = []
242
+
243
+ unless skip_lookup
244
+ Pathname.pwd.ascend do |path|
245
+ thorfiles = Thor::Util.globs_for(path).map { |g| Dir[g] }.flatten
246
+ break unless thorfiles.empty?
247
+ end
248
+ end
249
+
250
+ files = (relevant_to ? thorfiles_relevant_to(relevant_to) : Thor::Util.thor_root_glob)
251
+ files += thorfiles
252
+ files -= ["#{thor_root}/thor.yml"]
253
+
254
+ files.map! do |file|
255
+ File.directory?(file) ? File.join(file, "main.thor") : file
256
+ end
257
+ end
258
+
259
+ # Load Thorfiles relevant to the given method. If you provide "foo:bar" it
260
+ # will load all thor files in the thor.yaml that has "foo" e "foo:bar"
261
+ # namespaces registered.
262
+ #
263
+ def thorfiles_relevant_to(meth)
264
+ lookup = [ meth, meth.split(":")[0...-1].join(":") ]
265
+
266
+ files = thor_yaml.select do |k, v|
267
+ v[:namespaces] && !(v[:namespaces] & lookup).empty?
268
+ end
269
+
270
+ files.map { |k, v| File.join(thor_root, "#{v[:filename]}") }
271
+ end
272
+
273
+ # Display information about the given klasses. If with_module is given,
274
+ # it shows a table with information extracted from the yaml file.
275
+ #
276
+ def display_klasses(with_modules=false, show_internal=false, klasses=Thor::Base.subclasses)
277
+ klasses -= [Thor, Thor::Runner, Thor::Group] unless show_internal
278
+
279
+ raise Error, "No Thor commands available" if klasses.empty?
280
+ show_modules if with_modules && !thor_yaml.empty?
281
+
282
+ list = Hash.new { |h,k| h[k] = [] }
283
+ groups = klasses.select { |k| k.ancestors.include?(Thor::Group) }
284
+
285
+ # Get classes which inherit from Thor
286
+ (klasses - groups).each { |k| list[k.namespace.split(":").first] += k.printable_commands(false) }
287
+
288
+ # Get classes which inherit from Thor::Base
289
+ groups.map! { |k| k.printable_commands(false).first }
290
+ list["root"] = groups
291
+
292
+ # Order namespaces with default coming first
293
+ list = list.sort{ |a,b| a[0].sub(/^default/, '') <=> b[0].sub(/^default/, '') }
294
+ list.each { |n, commands| display_commands(n, commands) unless commands.empty? }
295
+ end
296
+
297
+ def display_commands(namespace, list) #:nodoc:
298
+ list.sort!{ |a,b| a[0] <=> b[0] }
299
+
300
+ say shell.set_color(namespace, :blue, true)
301
+ say "-" * namespace.size
302
+
303
+ print_table(list, :truncate => true)
304
+ say
305
+ end
306
+ alias display_tasks display_commands
307
+
308
+ def show_modules #:nodoc:
309
+ info = []
310
+ labels = ["Modules", "Namespaces"]
311
+
312
+ info << labels
313
+ info << [ "-" * labels[0].size, "-" * labels[1].size ]
314
+
315
+ thor_yaml.each do |name, hash|
316
+ info << [ name, hash[:namespaces].join(", ") ]
317
+ end
318
+
319
+ print_table info
320
+ say ""
321
+ end
322
+ end