train-core 1.4.4
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +7 -0
- data/CHANGELOG.md +780 -0
- data/Gemfile +36 -0
- data/LICENSE +201 -0
- data/README.md +197 -0
- data/lib/train.rb +158 -0
- data/lib/train/errors.rb +32 -0
- data/lib/train/extras.rb +11 -0
- data/lib/train/extras/command_wrapper.rb +137 -0
- data/lib/train/extras/stat.rb +132 -0
- data/lib/train/file.rb +151 -0
- data/lib/train/file/local.rb +75 -0
- data/lib/train/file/local/unix.rb +96 -0
- data/lib/train/file/local/windows.rb +63 -0
- data/lib/train/file/remote.rb +36 -0
- data/lib/train/file/remote/aix.rb +21 -0
- data/lib/train/file/remote/linux.rb +19 -0
- data/lib/train/file/remote/qnx.rb +41 -0
- data/lib/train/file/remote/unix.rb +106 -0
- data/lib/train/file/remote/windows.rb +94 -0
- data/lib/train/options.rb +80 -0
- data/lib/train/platforms.rb +84 -0
- data/lib/train/platforms/common.rb +34 -0
- data/lib/train/platforms/detect.rb +12 -0
- data/lib/train/platforms/detect/helpers/os_common.rb +145 -0
- data/lib/train/platforms/detect/helpers/os_linux.rb +75 -0
- data/lib/train/platforms/detect/helpers/os_windows.rb +120 -0
- data/lib/train/platforms/detect/scanner.rb +84 -0
- data/lib/train/platforms/detect/specifications/api.rb +15 -0
- data/lib/train/platforms/detect/specifications/os.rb +578 -0
- data/lib/train/platforms/detect/uuid.rb +34 -0
- data/lib/train/platforms/family.rb +26 -0
- data/lib/train/platforms/platform.rb +101 -0
- data/lib/train/plugins.rb +40 -0
- data/lib/train/plugins/base_connection.rb +169 -0
- data/lib/train/plugins/transport.rb +49 -0
- data/lib/train/transports/local.rb +232 -0
- data/lib/train/version.rb +7 -0
- data/train-core.gemspec +27 -0
- 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.
|
data/README.md
ADDED
@@ -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.
|
data/lib/train.rb
ADDED
@@ -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
|