org.jreleaser.sdk.ssh.SftpArtifactUploader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jreleaser-ssh-java-sdk Show documentation
Java SDK for SCP/SFTP upload/download
The newest version!
/*
 * SPDX-License-Identifier: Apache-2.0
 *
 * Copyright 2020-2024 The JReleaser authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jreleaser.sdk.ssh;

import net.schmizz.sshj.SSHClient;
import net.schmizz.sshj.sftp.SFTPClient;
import org.jreleaser.bundle.RB;
import org.jreleaser.model.internal.JReleaserContext;
import org.jreleaser.model.internal.common.Artifact;
import org.jreleaser.model.internal.upload.SftpUploader;
import org.jreleaser.model.spi.upload.UploadException;
import org.jreleaser.sdk.commons.AbstractArtifactUploader;

import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Set;

import static org.jreleaser.sdk.ssh.SshUtils.close;
import static org.jreleaser.sdk.ssh.SshUtils.createDirectories;
import static org.jreleaser.sdk.ssh.SshUtils.createSFTPClient;
import static org.jreleaser.sdk.ssh.SshUtils.createSSHClient;
import static org.jreleaser.sdk.ssh.SshUtils.disconnect;

/**
 * @author Andres Almiray
 * @since 1.1.0
 */
public class SftpArtifactUploader extends AbstractArtifactUploader {
    private SftpUploader uploader;

    public SftpArtifactUploader(JReleaserContext context) {
        super(context);
    }

    @Override
    public SftpUploader getUploader() {
        return uploader;
    }

    @Override
    public void setUploader(SftpUploader uploader) {
        this.uploader = uploader;
    }

    @Override
    public String getType() {
        return org.jreleaser.model.api.upload.SftpUploader.TYPE;
    }

    @Override
    public void upload(String name) throws UploadException {
        Set artifacts = collectArtifacts();
        if (artifacts.isEmpty()) {
            context.getLogger().info(RB.$("artifacts.no.match"));
        }

        SSHClient ssh = createSSHClient(context, uploader);
        SFTPClient sftp = createSFTPClient(uploader, ssh);

        try {
            for (Artifact artifact : artifacts) {
                Path path = artifact.getEffectivePath(context);
                context.getLogger().info(" - {}", path.getFileName());

                if (!context.isDryrun()) {
                    try {
                        String uploadPath = uploader.getResolvedPath(context, artifact);
                        context.getLogger().debug("   " + RB.$("uploader.uploading.to", uploadPath));
                        createDirectories(context, uploader, ssh, Paths.get(uploadPath).getParent());
                        sftp.put(path.toAbsolutePath().toString(), uploadPath);
                    } catch (IOException e) {
                        context.getLogger().trace(e);
                        throw new UploadException(RB.$("ERROR_unexpected_upload",
                            context.getBasedir().relativize(path)), e);
                    }
                }
            }
        } finally {
            close(uploader, sftp);
            disconnect(uploader, ssh);
        }
    }
}