Source code: it/rabellino/toska/FileDistribution.java

   package it.rabellino.toska;
   
   import javax.xml.parsers.SAXParser;
   import javax.xml.parsers.SAXParserFactory;
   
   import java.io.File;
   import java.io.FileOutputStream;
   import java.io.IOException;
   import java.util.Iterator;
  import java.util.Vector;
  
  import org.xml.sax.ContentHandler;
  import org.xml.sax.InputSource;
  import org.xml.sax.SAXException;
  import org.xml.sax.XMLReader;
  
  /**
   * @version   1.0
   * @author <a href="mailto:Nito@Qindel.ES">Nito Martinez</a>
   */
  public class FileDistribution {
    
    private String sourceDir = "/tmp";
    private String targetDir = "/tmp";
    private ConfigHandler handler;
    
    public void setSourceDir(String path) {
      this.sourceDir = path;
    }
  
    public void setTargetDir(String target) {
      this.targetDir = target;
    }
    
    public void setConfigHandler(ConfigHandler handler) {
      this.handler = handler;
    }
    
    
    public void scpFiles() throws ScpException {
      
      Iterator hosts = handler.getHosts().values().iterator();
      
      while (hosts.hasNext()) {
        Host host = (Host) hosts.next();
        Iterator users = host.getUsers().values().iterator();
        while (users.hasNext()) {
          User user = (User) users.next();
    String sourceFile = sourceDir + File.separator +
        host.getName() + File.separator + 
        user.getName();
          File userFile = new File(sourceFile);
    
          if (!userFile.exists()) {
        System.err.println("A supposed created file during the deployment is" +
               " no longer there(?): <" + sourceFile + ">"); 
        throw new ScpException("Source file <" + sourceFile + 
             "> does not exists");
    }
  
    String target = host.getName() + ":" + targetDir;
    doScp(sourceFile, target);
        }
      }  
      
    }
    
    /*
     * doSCP
     * Input: 
     * - source. The source string in the same way as it is done in the normal scp command
     *           Usually something like "host/*" or "host/key"
     * - target. The target string in the same way as scp accepts it. 
     *           In this environment usually "host:/etc/ssh2/keys"
     *
     * Dependencies. scp has to be in the default PATH
     */
    private void doScp(String source, String target) throws ScpException {
      Process p;
      Runtime r;
      
      String commandLine = "scp -q " + source + " " + target;
      
      try {
    r = Runtime.getRuntime();
    p = r.exec(commandLine);
    p.waitFor();
    if (p.exitValue() == 0) {
        System.out.println("Transferred " + source + " to " + target);
    } else {
        throw new ScpException("Error transferring " + source + " to " + target 
         + ". The command line was: <" + commandLine + ">");
    }
      } catch (SecurityException e) {
    throw new ScpException("SecurityException: executing <" + commandLine + "> :" + e);
      } catch (IOException e) {
    throw new ScpException("IOException: executing <" + commandLine + "> :" + e);
      } catch (InterruptedException e) {
    throw new ScpException("InterruptedException: problem during wait while executing <" + 
              commandLine + "> :" + e);
     }
   }
 
 }