cn.home1.cloud.config.server.ssh.DeployKey Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of spring-cloud-config-server Show documentation
spring-cloud-config-server
The newest version!
package cn.home1.cloud.config.server.ssh;

import static org.apache.commons.lang3.StringUtils.isNotBlank;
import static org.springframework.util.FileCopyUtils.copyToString;

import cn.home1.cloud.config.server.util.Consts;
import cn.home1.cloud.config.server.util.ResourceUtils;

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;

import org.springframework.cloud.config.server.ssh.SshUriProperties;

import java.io.File;
import java.io.FileReader;

@Slf4j
public class DeployKey {

    private final String privateKeyLocation;

    public DeployKey(final String privateKeyLocation) {
        this.privateKeyLocation = privateKeyLocation;
    }

    public static String getPrivateKeyPath(final String privateKeyLocation) {
        return ResourceUtils.findResourceFile(privateKeyLocation, Consts.DATA_DIRECTORY);
    }

    public void setUp(final SshUriProperties sshUriProperties) {
        final String privateKey = this.getPrivateKey();

        // setup ssh client
        if (sshUriProperties == null) {
            // setup SshSessionFactory as early as possible, requires @Order(HIGHEST_PRECEDENCE) on Configuration
            //SshSessionFactory.setInstance(new CustomJschConfigSessionFactory(file));

            System.setProperty("spring.cloud.config.server.git.private-key", privateKey);
            System.setProperty("spring.cloud.config.server.git.ignore-local-ssh-settings", "true");
        } else {
            log.info("ssh uri properties privateKey is not blank: {}", isNotBlank(sshUriProperties.getPrivateKey()));
            log.info("ssh uri properties is ignore local ssh settings: {}", sshUriProperties.isIgnoreLocalSshSettings());
            log.info("ssh uri properties is strict host key checking: {}", sshUriProperties.isStrictHostKeyChecking());

            sshUriProperties.setPrivateKey(privateKey);
            sshUriProperties.setIgnoreLocalSshSettings(true);
        }
    }

    @SneakyThrows
    public String getPrivateKey() {
        return copyToString(new FileReader(new File(this.getPrivateKeyPath())));
    }

    public String getPrivateKeyPath() {
        return getPrivateKeyPath(this.privateKeyLocation);
    }

    @SneakyThrows
    public String getPublicKey() {
        return copyToString(new FileReader(new File(this.getPublicKeyPath())));
    }

    public String getPublicKeyPath() {
        return this.getPrivateKeyPath() + ".pub";
    }
}