Subversion Repositories SmartDukaan

package com.spice.profitmandi.service.SshServer;

import com.jcraft.jsch.*;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.util.*;

@Component
public class SSHServiceImpl implements SSHService {

    @Value("172.105.58.16")
    private String host;

    @Value("root")
    private String user;

    @Value("22")
    private int port;

    private static final Logger LOGGER = LogManager.getLogger(SSHService.class);

    @Override
    public void executeCommand(String ipAddress, int port, String username, String password, String commandToExec) {
        try {
            JSch jsch = new JSch();
            Session session = jsch.getSession(username, ipAddress, port);
            session.setPassword(password);
            session.setConfig("StrictHostKeyChecking", "no");
            session.connect();
            // Restart server
            ChannelExec restartChannel = (ChannelExec) session.openChannel("exec");
            restartChannel.setCommand(commandToExec);
            //Reboot server
            restartChannel.connect();
            while (!restartChannel.isClosed()) {
                Thread.sleep(1000);
            }
            restartChannel.disconnect();
            session.disconnect();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private ChannelSftp setupJSch() throws JSchException {
        JSch jsch = new JSch();
        Session JSchSession = jsch.getSession(user, host, port);
        jsch.addIdentity(getClass().getClassLoader().getResource("id_rsa").getPath());
        JSchSession.setConfig("StrictHostKeyChecking", "no");
        JSchSession.connect();
        return (ChannelSftp) JSchSession.openChannel("sftp");
    }

    @Override
    public Map<String, List<String>> readDirectory(String directoryPath) {
        Map<String, List<String>> directoryFileMap = new HashMap<>();
        ChannelSftp channelSftp = null;
        try {
            channelSftp = setupJSch();
            channelSftp.connect();

            readDirectoryRecursively(channelSftp, directoryPath, directoryFileMap);

        } catch (JSchException | SftpException e) {
            e.printStackTrace();
        } finally {
            if (channelSftp != null && channelSftp.isConnected()) {
                channelSftp.disconnect();
            }
        }

        return directoryFileMap;
    }

    private void readDirectoryRecursively(ChannelSftp channelSftp, String directoryPath, Map<String, List<String>> directoryFileMap) throws SftpException {
        channelSftp.cd(directoryPath);

        Vector<ChannelSftp.LsEntry> fileList = channelSftp.ls(directoryPath);
        List<String> fileNames = new ArrayList<>();
        String dirName = directoryPath.substring(directoryPath.lastIndexOf('/') + 1);

        for (ChannelSftp.LsEntry entry : fileList) {
            String fileName = entry.getFilename();

            if (fileName.equals(".") || fileName.equals("..")) {
                continue;
            }

            if (entry.getAttrs().isDir()) {
                readDirectoryRecursively(channelSftp, directoryPath + "/" + fileName, directoryFileMap);
            } else {
                if (fileName.toLowerCase().endsWith(".xml")) {
                    fileNames.add(fileName);
                }
            }
        }

        if (!fileNames.isEmpty()) {
            directoryFileMap.put(dirName, fileNames);
        }
    }

}