train-core 1.4.4

Sign up to get free protection for your applications and to get access to all the features.
Files changed (40) hide show
  1. checksums.yaml +7 -0
  2. data/CHANGELOG.md +780 -0
  3. data/Gemfile +36 -0
  4. data/LICENSE +201 -0
  5. data/README.md +197 -0
  6. data/lib/train.rb +158 -0
  7. data/lib/train/errors.rb +32 -0
  8. data/lib/train/extras.rb +11 -0
  9. data/lib/train/extras/command_wrapper.rb +137 -0
  10. data/lib/train/extras/stat.rb +132 -0
  11. data/lib/train/file.rb +151 -0
  12. data/lib/train/file/local.rb +75 -0
  13. data/lib/train/file/local/unix.rb +96 -0
  14. data/lib/train/file/local/windows.rb +63 -0
  15. data/lib/train/file/remote.rb +36 -0
  16. data/lib/train/file/remote/aix.rb +21 -0
  17. data/lib/train/file/remote/linux.rb +19 -0
  18. data/lib/train/file/remote/qnx.rb +41 -0
  19. data/lib/train/file/remote/unix.rb +106 -0
  20. data/lib/train/file/remote/windows.rb +94 -0
  21. data/lib/train/options.rb +80 -0
  22. data/lib/train/platforms.rb +84 -0
  23. data/lib/train/platforms/common.rb +34 -0
  24. data/lib/train/platforms/detect.rb +12 -0
  25. data/lib/train/platforms/detect/helpers/os_common.rb +145 -0
  26. data/lib/train/platforms/detect/helpers/os_linux.rb +75 -0
  27. data/lib/train/platforms/detect/helpers/os_windows.rb +120 -0
  28. data/lib/train/platforms/detect/scanner.rb +84 -0
  29. data/lib/train/platforms/detect/specifications/api.rb +15 -0
  30. data/lib/train/platforms/detect/specifications/os.rb +578 -0
  31. data/lib/train/platforms/detect/uuid.rb +34 -0
  32. data/lib/train/platforms/family.rb +26 -0
  33. data/lib/train/platforms/platform.rb +101 -0
  34. data/lib/train/plugins.rb +40 -0
  35. data/lib/train/plugins/base_connection.rb +169 -0
  36. data/lib/train/plugins/transport.rb +49 -0
  37. data/lib/train/transports/local.rb +232 -0
  38. data/lib/train/version.rb +7 -0
  39. data/train-core.gemspec +27 -0
  40. metadata +116 -0
data/Gemfile ADDED
@@ -0,0 +1,36 @@
1
+ # encoding: utf-8
2
+ source 'https://rubygems.org'
3
+ gemspec name: 'train'
4
+
5
+ # pin dependency for Ruby 1.9.3 since bundler is not
6
+ # detecting that net-ssh 3 does not work with 1.9.3
7
+ if Gem::Version.new(RUBY_VERSION) <= Gem::Version.new('1.9.3')
8
+ gem 'net-ssh', '~> 2.9'
9
+ end
10
+
11
+ if Gem::Version.new(RUBY_VERSION) < Gem::Version.new('2.2.2')
12
+ gem 'json', '< 2.0'
13
+ gem 'rack', '< 2.0'
14
+ end
15
+
16
+ group :test do
17
+ gem 'bundler', '~> 1.11'
18
+ gem 'minitest', '~> 5.8'
19
+ gem 'rake', '~> 10'
20
+ gem 'rubocop', '~> 0.36.0'
21
+ gem 'simplecov', '~> 0.10'
22
+ gem 'concurrent-ruby', '~> 1.0'
23
+ end
24
+
25
+ group :integration do
26
+ gem 'berkshelf', '~> 4.3'
27
+ gem 'test-kitchen', '~> 1.11'
28
+ gem 'kitchen-vagrant'
29
+ end
30
+
31
+ group :tools do
32
+ gem 'pry', '~> 0.10'
33
+ gem 'rb-readline'
34
+ gem 'license_finder'
35
+ gem 'github_changelog_generator', '~> 1'
36
+ end
data/LICENSE ADDED
@@ -0,0 +1,201 @@
1
+ Apache License
2
+ Version 2.0, January 2004
3
+ http://www.apache.org/licenses/
4
+
5
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
6
+
7
+ 1. Definitions.
8
+
9
+ "License" shall mean the terms and conditions for use, reproduction,
10
+ and distribution as defined by Sections 1 through 9 of this document.
11
+
12
+ "Licensor" shall mean the copyright owner or entity authorized by
13
+ the copyright owner that is granting the License.
14
+
15
+ "Legal Entity" shall mean the union of the acting entity and all
16
+ other entities that control, are controlled by, or are under common
17
+ control with that entity. For the purposes of this definition,
18
+ "control" means (i) the power, direct or indirect, to cause the
19
+ direction or management of such entity, whether by contract or
20
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
21
+ outstanding shares, or (iii) beneficial ownership of such entity.
22
+
23
+ "You" (or "Your") shall mean an individual or Legal Entity
24
+ exercising permissions granted by this License.
25
+
26
+ "Source" form shall mean the preferred form for making modifications,
27
+ including but not limited to software source code, documentation
28
+ source, and configuration files.
29
+
30
+ "Object" form shall mean any form resulting from mechanical
31
+ transformation or translation of a Source form, including but
32
+ not limited to compiled object code, generated documentation,
33
+ and conversions to other media types.
34
+
35
+ "Work" shall mean the work of authorship, whether in Source or
36
+ Object form, made available under the License, as indicated by a
37
+ copyright notice that is included in or attached to the work
38
+ (an example is provided in the Appendix below).
39
+
40
+ "Derivative Works" shall mean any work, whether in Source or Object
41
+ form, that is based on (or derived from) the Work and for which the
42
+ editorial revisions, annotations, elaborations, or other modifications
43
+ represent, as a whole, an original work of authorship. For the purposes
44
+ of this License, Derivative Works shall not include works that remain
45
+ separable from, or merely link (or bind by name) to the interfaces of,
46
+ the Work and Derivative Works thereof.
47
+
48
+ "Contribution" shall mean any work of authorship, including
49
+ the original version of the Work and any modifications or additions
50
+ to that Work or Derivative Works thereof, that is intentionally
51
+ submitted to Licensor for inclusion in the Work by the copyright owner
52
+ or by an individual or Legal Entity authorized to submit on behalf of
53
+ the copyright owner. For the purposes of this definition, "submitted"
54
+ means any form of electronic, verbal, or written communication sent
55
+ to the Licensor or its representatives, including but not limited to
56
+ communication on electronic mailing lists, source code control systems,
57
+ and issue tracking systems that are managed by, or on behalf of, the
58
+ Licensor for the purpose of discussing and improving the Work, but
59
+ excluding communication that is conspicuously marked or otherwise
60
+ designated in writing by the copyright owner as "Not a Contribution."
61
+
62
+ "Contributor" shall mean Licensor and any individual or Legal Entity
63
+ on behalf of whom a Contribution has been received by Licensor and
64
+ subsequently incorporated within the Work.
65
+
66
+ 2. Grant of Copyright License. Subject to the terms and conditions of
67
+ this License, each Contributor hereby grants to You a perpetual,
68
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
69
+ copyright license to reproduce, prepare Derivative Works of,
70
+ publicly display, publicly perform, sublicense, and distribute the
71
+ Work and such Derivative Works in Source or Object form.
72
+
73
+ 3. Grant of Patent License. Subject to the terms and conditions of
74
+ this License, each Contributor hereby grants to You a perpetual,
75
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
76
+ (except as stated in this section) patent license to make, have made,
77
+ use, offer to sell, sell, import, and otherwise transfer the Work,
78
+ where such license applies only to those patent claims licensable
79
+ by such Contributor that are necessarily infringed by their
80
+ Contribution(s) alone or by combination of their Contribution(s)
81
+ with the Work to which such Contribution(s) was submitted. If You
82
+ institute patent litigation against any entity (including a
83
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
84
+ or a Contribution incorporated within the Work constitutes direct
85
+ or contributory patent infringement, then any patent licenses
86
+ granted to You under this License for that Work shall terminate
87
+ as of the date such litigation is filed.
88
+
89
+ 4. Redistribution. You may reproduce and distribute copies of the
90
+ Work or Derivative Works thereof in any medium, with or without
91
+ modifications, and in Source or Object form, provided that You
92
+ meet the following conditions:
93
+
94
+ (a) You must give any other recipients of the Work or
95
+ Derivative Works a copy of this License; and
96
+
97
+ (b) You must cause any modified files to carry prominent notices
98
+ stating that You changed the files; and
99
+
100
+ (c) You must retain, in the Source form of any Derivative Works
101
+ that You distribute, all copyright, patent, trademark, and
102
+ attribution notices from the Source form of the Work,
103
+ excluding those notices that do not pertain to any part of
104
+ the Derivative Works; and
105
+
106
+ (d) If the Work includes a "NOTICE" text file as part of its
107
+ distribution, then any Derivative Works that You distribute must
108
+ include a readable copy of the attribution notices contained
109
+ within such NOTICE file, excluding those notices that do not
110
+ pertain to any part of the Derivative Works, in at least one
111
+ of the following places: within a NOTICE text file distributed
112
+ as part of the Derivative Works; within the Source form or
113
+ documentation, if provided along with the Derivative Works; or,
114
+ within a display generated by the Derivative Works, if and
115
+ wherever such third-party notices normally appear. The contents
116
+ of the NOTICE file are for informational purposes only and
117
+ do not modify the License. You may add Your own attribution
118
+ notices within Derivative Works that You distribute, alongside
119
+ or as an addendum to the NOTICE text from the Work, provided
120
+ that such additional attribution notices cannot be construed
121
+ as modifying the License.
122
+
123
+ You may add Your own copyright statement to Your modifications and
124
+ may provide additional or different license terms and conditions
125
+ for use, reproduction, or distribution of Your modifications, or
126
+ for any such Derivative Works as a whole, provided Your use,
127
+ reproduction, and distribution of the Work otherwise complies with
128
+ the conditions stated in this License.
129
+
130
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
131
+ any Contribution intentionally submitted for inclusion in the Work
132
+ by You to the Licensor shall be under the terms and conditions of
133
+ this License, without any additional terms or conditions.
134
+ Notwithstanding the above, nothing herein shall supersede or modify
135
+ the terms of any separate license agreement you may have executed
136
+ with Licensor regarding such Contributions.
137
+
138
+ 6. Trademarks. This License does not grant permission to use the trade
139
+ names, trademarks, service marks, or product names of the Licensor,
140
+ except as required for reasonable and customary use in describing the
141
+ origin of the Work and reproducing the content of the NOTICE file.
142
+
143
+ 7. Disclaimer of Warranty. Unless required by applicable law or
144
+ agreed to in writing, Licensor provides the Work (and each
145
+ Contributor provides its Contributions) on an "AS IS" BASIS,
146
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
147
+ implied, including, without limitation, any warranties or conditions
148
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
149
+ PARTICULAR PURPOSE. You are solely responsible for determining the
150
+ appropriateness of using or redistributing the Work and assume any
151
+ risks associated with Your exercise of permissions under this License.
152
+
153
+ 8. Limitation of Liability. In no event and under no legal theory,
154
+ whether in tort (including negligence), contract, or otherwise,
155
+ unless required by applicable law (such as deliberate and grossly
156
+ negligent acts) or agreed to in writing, shall any Contributor be
157
+ liable to You for damages, including any direct, indirect, special,
158
+ incidental, or consequential damages of any character arising as a
159
+ result of this License or out of the use or inability to use the
160
+ Work (including but not limited to damages for loss of goodwill,
161
+ work stoppage, computer failure or malfunction, or any and all
162
+ other commercial damages or losses), even if such Contributor
163
+ has been advised of the possibility of such damages.
164
+
165
+ 9. Accepting Warranty or Additional Liability. While redistributing
166
+ the Work or Derivative Works thereof, You may choose to offer,
167
+ and charge a fee for, acceptance of support, warranty, indemnity,
168
+ or other liability obligations and/or rights consistent with this
169
+ License. However, in accepting such obligations, You may act only
170
+ on Your own behalf and on Your sole responsibility, not on behalf
171
+ of any other Contributor, and only if You agree to indemnify,
172
+ defend, and hold each Contributor harmless for any liability
173
+ incurred by, or claims asserted against, such Contributor by reason
174
+ of your accepting any such warranty or additional liability.
175
+
176
+ END OF TERMS AND CONDITIONS
177
+
178
+ APPENDIX: How to apply the Apache License to your work.
179
+
180
+ To apply the Apache License to your work, attach the following
181
+ boilerplate notice, with the fields enclosed by brackets "[]"
182
+ replaced with your own identifying information. (Don't include
183
+ the brackets!) The text should be enclosed in the appropriate
184
+ comment syntax for the file format. We also recommend that a
185
+ file or class name and description of purpose be included on the
186
+ same "printed page" as the copyright notice for easier
187
+ identification within third-party archives.
188
+
189
+ Copyright [yyyy] [name of copyright owner]
190
+
191
+ Licensed under the Apache License, Version 2.0 (the "License");
192
+ you may not use this file except in compliance with the License.
193
+ You may obtain a copy of the License at
194
+
195
+ http://www.apache.org/licenses/LICENSE-2.0
196
+
197
+ Unless required by applicable law or agreed to in writing, software
198
+ distributed under the License is distributed on an "AS IS" BASIS,
199
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200
+ See the License for the specific language governing permissions and
201
+ limitations under the License.
@@ -0,0 +1,197 @@
1
+ # Train - Transport Interface
2
+
3
+ [![Build Status](https://travis-ci.org/chef/train.svg?branch=master)](https://travis-ci.org/chef/train)
4
+ [![Build Status Master](https://ci.appveyor.com/api/projects/status/github/chef/train?branch=master&svg=true&passingText=master%20-%20Ok&pendingText=master%20-%20Pending&failingText=master%20-%20Failing)](https://ci.appveyor.com/project/Chef/train/branch/master)
5
+ [![Gem Version](https://badge.fury.io/rb/train.svg)](https://badge.fury.io/rb/train)
6
+
7
+ Train lets you talk to your local or remote operating systems and APIs with a unified interface.
8
+
9
+ It allows you to:
10
+
11
+ * execute commands via `run_command`
12
+ * interact with files via `file`
13
+ * identify the target operating system via `os`
14
+ * authenticate to API-based services and treat them like a platform
15
+
16
+ Train supports:
17
+
18
+ * Local execution
19
+ * SSH
20
+ * WinRM
21
+ * Docker
22
+ * Mock (for testing and debugging)
23
+ * AWS as an API
24
+ * Azure as an API
25
+
26
+ # Examples
27
+
28
+ ## Setup
29
+
30
+ **Local**
31
+
32
+ ```ruby
33
+ require 'train'
34
+ train = Train.create('local')
35
+ ```
36
+
37
+ **SSH**
38
+
39
+ ```ruby
40
+ require 'train'
41
+ train = Train.create('ssh',
42
+ host: '1.2.3.4', port: 22, user: 'root', key_files: '/vagrant')
43
+ ```
44
+
45
+ If you don't specify the `key_files` and `password` options, SSH agent authentication will be attempted. For example:
46
+
47
+ ```ruby
48
+ require 'train'
49
+ train = Train.create('ssh', host: '1.2.3.4', port: 22, user: 'root')
50
+ ```
51
+
52
+ **WinRM**
53
+
54
+ ```ruby
55
+ require 'train'
56
+ train = Train.create('winrm',
57
+ host: '1.2.3.4', user: 'Administrator', password: '...', ssl: true, self_signed: true)
58
+ ```
59
+
60
+ **Docker**
61
+
62
+ ```ruby
63
+ require 'train'
64
+ train = Train.create('docker', host: 'container_id...')
65
+ ```
66
+
67
+ **AWS**
68
+
69
+ To use AWS API authentication, setup an AWS client profile to store the Access Key ID and Secret Access Key.
70
+
71
+ ```ruby
72
+ require 'train'
73
+ train = Train.create('aws', region: 'us-east-2', profile: 'my-profile')
74
+ ```
75
+
76
+ You may also use the standard AWS CLI environment variables, `AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY` and `AWS_REGION`.
77
+ ```ruby
78
+ require 'train'
79
+ train = Train.create('aws')
80
+ ```
81
+
82
+ ## Configuration
83
+
84
+ To get a list of available options for a plugin:
85
+
86
+ ```ruby
87
+ puts Train.options('ssh')
88
+ ```
89
+ This will provide all configuration options:
90
+
91
+ ```ruby
92
+ {
93
+ :host => { :required => true},
94
+ :port => { :default => 22, :required => true},
95
+ :user => { :default => "root", :required => true},
96
+ :keys => { :default => nil},
97
+ :password => { :default => nil},
98
+ ...
99
+ ```
100
+
101
+ ## Usage
102
+
103
+ ```ruby
104
+ # start or reuse a connection
105
+ conn = train.connection
106
+
107
+ # run a command on Linux/Unix/Mac
108
+ puts conn.run_command('whoami').stdout
109
+
110
+ # get OS info
111
+ puts conn.os[:family]
112
+ puts conn.os[:release]
113
+
114
+ # access files
115
+ puts conn.file('/proc/version').content
116
+
117
+ # access specific API client functionality
118
+ ec2_client = train.connection.aws_client(Aws::EC2::Client)
119
+ puts ec2_client.describe_instances
120
+
121
+ # close the connection
122
+ conn.close
123
+ ```
124
+
125
+ # Testing
126
+
127
+ We perform `unit`, `integration` and `windows` tests.
128
+
129
+ * `unit` tests ensure the intended behaviour of the implementation
130
+ * `integration` tests run against VMs and docker containers
131
+ * `windows` tests that run on appveyor for windows integration tests
132
+
133
+ ## Mac/Linux
134
+
135
+ ```
136
+ bundle exec ruby -W -Ilib:test/unit test/unit/extras/stat_test.rb
137
+ ```
138
+
139
+ ## Windows
140
+
141
+ ```
142
+ # run windows tests
143
+ bundle exec rake test:windows
144
+
145
+ # run single tests
146
+ bundle exec ruby -I .\test\windows\ .\test\windows\local_test.rb
147
+ ```
148
+
149
+
150
+ # Kudos and Contributors
151
+
152
+ Train is heavily based on the work of:
153
+
154
+ * [test-kitchen](https://github.com/test-kitchen/test-kitchen)
155
+
156
+ by [Fletcher Nichol](fnichol@nichol.ca)
157
+ and [a great community of contributors](https://github.com/test-kitchen/test-kitchen/graphs/contributors)
158
+
159
+ * [ohai](https://github.com/chef/ohai)
160
+
161
+ by Adam Jacob, Chef Software Inc.
162
+ and [a great community of contributors](https://github.com/chef/ohai/graphs/contributors)
163
+
164
+
165
+ We also want to thank [halo](https://github.com/halo) who did a great contribution by handing over the `train` gem name.
166
+
167
+ ## Contributing
168
+
169
+ 1. Fork it
170
+ 1. Create your feature branch (git checkout -b my-new-feature)
171
+ 1. Commit your changes (git commit -am 'Add some feature')
172
+ 1. Push to the branch (git push origin my-new-feature)
173
+ 1. Create new Pull Request
174
+
175
+ ## License
176
+
177
+ | **Author:** | Dominik Richter (<drichter@chef.io>)
178
+
179
+ | **Author:** | Christoph Hartmann (<chartmann@chef.io>)
180
+
181
+ | **Copyright:** | Copyright (c) 2015 Chef Software Inc.
182
+
183
+ | **Copyright:** | Copyright (c) 2015 Vulcano Security GmbH.
184
+
185
+ | **License:** | Apache License, Version 2.0
186
+
187
+ Licensed under the Apache License, Version 2.0 (the "License");
188
+ you may not use this file except in compliance with the License.
189
+ You may obtain a copy of the License at
190
+
191
+ http://www.apache.org/licenses/LICENSE-2.0
192
+
193
+ Unless required by applicable law or agreed to in writing, software
194
+ distributed under the License is distributed on an "AS IS" BASIS,
195
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
196
+ See the License for the specific language governing permissions and
197
+ limitations under the License.
@@ -0,0 +1,158 @@
1
+ # encoding: utf-8
2
+ #
3
+ # Author:: Dominik Richter (<dominik.richter@gmail.com>)
4
+
5
+ require 'train/version'
6
+ require 'train/options'
7
+ require 'train/plugins'
8
+ require 'train/errors'
9
+ require 'train/platforms'
10
+ require 'uri'
11
+
12
+ module Train
13
+ # Create a new transport instance, with the plugin indicated by the
14
+ # given name.
15
+ #
16
+ # @param [String] name of the plugin
17
+ # @param [Array] *args list of arguments for the plugin
18
+ # @return [Transport] instance of the new transport or nil
19
+ def self.create(name, *args)
20
+ cls = load_transport(name)
21
+ cls.new(*args) unless cls.nil?
22
+ end
23
+
24
+ # Retrieve the configuration options of a transport plugin.
25
+ #
26
+ # @param [String] name of the plugin
27
+ # @return [Hash] map of default options
28
+ def self.options(name)
29
+ cls = load_transport(name)
30
+ cls.default_options unless cls.nil?
31
+ end
32
+
33
+ # Load the transport plugin indicated by name. If the plugin is not
34
+ # yet found in the plugin registry, it will be attempted to load from
35
+ # `train/transports/plugin_name`.
36
+ #
37
+ # @param [String] name of the plugin
38
+ # @return [Train::Transport] the transport plugin
39
+ def self.load_transport(name)
40
+ res = Train::Plugins.registry[name.to_s]
41
+ return res unless res.nil?
42
+
43
+ # if the plugin wasnt loaded yet:
44
+ require 'train/transports/' + name.to_s
45
+ Train::Plugins.registry[name.to_s]
46
+ rescue LoadError => _
47
+ raise Train::UserError,
48
+ "Can't find train plugin #{name.inspect}. Please install it first."
49
+ end
50
+
51
+ # Resolve target configuration in URI-scheme into
52
+ # all respective fields and merge with existing configuration.
53
+ # e.g. ssh://bob@remote => backend: ssh, user: bob, host: remote
54
+ def self.target_config(config = nil) # rubocop:disable Metrics/AbcSize
55
+ conf = config.nil? ? {} : config.dup
56
+ conf = symbolize_keys(conf)
57
+
58
+ group_keys_and_keyfiles(conf)
59
+
60
+ return conf if conf[:target].to_s.empty?
61
+
62
+ # split up the target's host/scheme configuration
63
+ uri = parse_uri(conf[:target].to_s)
64
+ unless uri.host.nil? and uri.scheme.nil?
65
+ conf[:backend] ||= uri.scheme
66
+ conf[:host] ||= uri.hostname
67
+ conf[:port] ||= uri.port
68
+ conf[:user] ||= uri.user
69
+ conf[:password] ||= uri.password
70
+ conf[:path] ||= uri.path
71
+ end
72
+
73
+ # ensure path is nil, if its empty; e.g. required to reset defaults for winrm
74
+ conf[:path] = nil if !conf[:path].nil? && conf[:path].to_s.empty?
75
+
76
+ # return the updated config
77
+ conf
78
+ end
79
+
80
+ # Takes a map of key-value pairs and turns all keys into symbols. For this
81
+ # to work, only keys are supported that can be turned into symbols.
82
+ # Example: { 'a' => 123 } ==> { a: 123 }
83
+ #
84
+ # @param map [Hash]
85
+ # @return [Hash] new map with all keys being symbols
86
+ def self.symbolize_keys(map)
87
+ map.each_with_object({}) do |(k, v), acc|
88
+ acc[k.to_sym] = v
89
+ acc
90
+ end
91
+ end
92
+ private_class_method :symbolize_keys
93
+
94
+ # Parse a URI. Supports empty URI's with paths, e.g. `mock://`
95
+ #
96
+ # @param string [string] URI string, e.g. `schema://domain.com`
97
+ # @return [URI::Generic] parsed URI object
98
+ def self.parse_uri(string)
99
+ URI.parse(string)
100
+ rescue URI::InvalidURIError => e
101
+ # A use-case we want to catch is parsing empty URIs with a schema
102
+ # e.g. mock://. To do this, we match it manually and fake the hostname
103
+ case string
104
+ when %r{^([a-z]+)://$}
105
+ string += 'dummy'
106
+ when /^([a-z]+):$/
107
+ string += '//dummy'
108
+ else
109
+ raise Train::UserError, e
110
+ end
111
+
112
+ u = URI.parse(string)
113
+ u.host = nil
114
+ u
115
+ end
116
+ private_class_method :parse_uri
117
+
118
+ def self.validate_backend(conf, default = :local)
119
+ return default if conf.nil?
120
+ res = conf[:backend]
121
+
122
+ if (res.nil? || res == 'localhost') && conf[:sudo]
123
+ fail Train::UserError, 'Sudo is only valid when running against a remote host. '\
124
+ 'To run this locally with elevated privileges, run the command with `sudo ...`.'
125
+ end
126
+
127
+ return res if !res.nil?
128
+
129
+ if !conf[:target].nil?
130
+ fail Train::UserError, 'Cannot determine backend from target '\
131
+ "configuration #{conf[:target].inspect}. Valid example: ssh://192.168.0.1."
132
+ end
133
+
134
+ if !conf[:host].nil?
135
+ fail Train::UserError, 'Host configured, but no backend was provided. Please '\
136
+ 'specify how you want to connect. Valid example: ssh://192.168.0.1.'
137
+ end
138
+
139
+ conf[:backend] = default
140
+ end
141
+
142
+ def self.group_keys_and_keyfiles(conf)
143
+ # in case the user specified a key-file, register it that way
144
+ # we will clear the list of keys and put keys and key_files separately
145
+ keys_mixed = conf[:keys]
146
+ return if keys_mixed.nil?
147
+
148
+ conf[:key_files] = []
149
+ conf[:keys] = []
150
+ keys_mixed.each do |key|
151
+ if !key.nil? and File.file?(key)
152
+ conf[:key_files].push(key)
153
+ else
154
+ conf[:keys].push(key)
155
+ end
156
+ end
157
+ end
158
+ end