package org.argeo.slc.jsch;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.argeo.slc.core.deploy.ResourceSet;
import org.springframework.core.io.Resource;

public class SshFilesDeployment implements Runnable {
	private String targetBase = "";
	private ResourceSet resourceSet;
	private SshTarget sshTarget;

	public void run() {
		JschMultiTasks multiTasks = new JschMultiTasks();
		multiTasks.setSshTarget(sshTarget);

		List<String> subDirs = new ArrayList<String>();
		Map<String, Resource> resources = resourceSet.listResources();
		for (String relPath : resources.keySet()) {
			// Create dir if necessary
			String dir;
			int lastIndexSubDir = relPath.lastIndexOf('/');
			if (lastIndexSubDir > 0)
				dir = targetBase + '/' + relPath.substring(0, lastIndexSubDir);
			else
				dir = targetBase;
			if (!subDirs.contains(dir)) {
				RemoteExec remoteExec = new RemoteExec();
				remoteExec.setCommand("mkdir -p " + dir);
				subDirs.add(dir);
				multiTasks.getTasks().add(remoteExec);
			}

			// Copy resource
			Resource resource = resources.get(relPath);
			ScpTo scpTo = new ScpTo();
			scpTo.setLocalResource(resource);
			scpTo.setRemotePath(targetBase + "/" + relPath);
			multiTasks.getTasks().add(scpTo);
		}

		multiTasks.run();
	}

	public void setTargetBase(String targetBase) {
		this.targetBase = targetBase;
	}

	public void setResourceSet(ResourceSet resourceSet) {
		this.resourceSet = resourceSet;
	}

	public void setSshTarget(SshTarget sshTarget) {
		this.sshTarget = sshTarget;
	}

}