Classes in this File | Line Coverage | Branch Coverage | Complexity | ||||
UploadKeys |
|
| 1.0;1 |
1 | /** | |
2 | * Copyright (c) 2014-2015, Thindeck.com | |
3 | * All rights reserved. | |
4 | * | |
5 | * Redistribution and use in source and binary forms, with or without | |
6 | * modification, are permitted provided that the following conditions | |
7 | * are met: 1) Redistributions of source code must retain the above | |
8 | * copyright notice, this list of conditions and the following | |
9 | * disclaimer. 2) Redistributions in binary form must reproduce the above | |
10 | * copyright notice, this list of conditions and the following | |
11 | * disclaimer in the documentation and/or other materials provided | |
12 | * with the distribution. 3) Neither the name of the thindeck.com nor | |
13 | * the names of its contributors may be used to endorse or promote | |
14 | * products derived from this software without specific prior written | |
15 | * permission. | |
16 | * | |
17 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
18 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT | |
19 | * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND | |
20 | * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL | |
21 | * THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, | |
22 | * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES | |
23 | * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
24 | * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
25 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, | |
26 | * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | |
27 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED | |
28 | * OF THE POSSIBILITY OF SUCH DAMAGE. | |
29 | */ | |
30 | package com.thindeck.bosses; | |
31 | ||
32 | import com.google.common.base.Joiner; | |
33 | import com.jcabi.ssh.Shell; | |
34 | import com.thindeck.agents.Remote; | |
35 | import com.thindeck.api.Boss; | |
36 | import com.thindeck.api.Deck; | |
37 | import java.io.ByteArrayInputStream; | |
38 | import java.io.ByteArrayOutputStream; | |
39 | import java.io.IOException; | |
40 | import java.nio.charset.Charset; | |
41 | import org.apache.commons.io.IOUtils; | |
42 | ||
43 | /** | |
44 | * Upload SSH keys to the servers. | |
45 | * | |
46 | * @author Yegor Bugayenko (yegor@teamed.io) | |
47 | * @version $Id$ | |
48 | */ | |
49 | 0 | public final class UploadKeys implements Boss { |
50 | ||
51 | @Override | |
52 | public void exec(final Iterable<Deck> decks) throws IOException { | |
53 | 0 | final Shell shell = new Remote("t1.thindeck.com"); |
54 | 0 | new Shell.Plain(shell).exec("mkdir -p ~/.ssh"); |
55 | 0 | shell.exec( |
56 | "cat > ~/.ssh/id_rsa", | |
57 | IOUtils.toInputStream( | |
58 | IOUtils.toString( | |
59 | this.getClass().getResourceAsStream("id_rsa") | |
60 | ).replaceAll("\n\\s+", "\n") | |
61 | ), | |
62 | new ByteArrayOutputStream(), | |
63 | new ByteArrayOutputStream() | |
64 | ); | |
65 | 0 | shell.exec( |
66 | "cat > ~/.ssh/config", | |
67 | new ByteArrayInputStream( | |
68 | Joiner.on('\n').join( | |
69 | "Host github.com", | |
70 | "\tStrictHostKeyChecking no" | |
71 | ).getBytes(Charset.defaultCharset()) | |
72 | ), | |
73 | new ByteArrayOutputStream(), | |
74 | new ByteArrayOutputStream() | |
75 | ); | |
76 | 0 | new Shell.Plain(shell).exec("chmod 700 ~/.ssh"); |
77 | 0 | new Shell.Plain(shell).exec("chmod -R 600 ~/.ssh/*"); |
78 | 0 | } |
79 | ||
80 | } |