jadex.platform.service.filetransfer.FileTransferService Maven / Gradle / Ivy

Go to download
package jadex.platform.service.filetransfer;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;

import jadex.bridge.IExternalAccess;
import jadex.bridge.IInputConnection;
import jadex.bridge.IOutputConnection;
import jadex.bridge.service.annotation.Service;
import jadex.bridge.service.annotation.ServiceComponent;
import jadex.bridge.service.search.SServiceProvider;
import jadex.bridge.service.types.context.IContextService;
import jadex.bridge.service.types.filetransfer.FileData;
import jadex.bridge.service.types.filetransfer.IFileTransferService;
import jadex.commons.future.DefaultResultListener;
import jadex.commons.future.Future;
import jadex.commons.future.IFuture;
import jadex.commons.future.ISubscriptionIntermediateFuture;
import jadex.commons.future.SubscriptionIntermediateDelegationFuture;
import jadex.commons.future.SubscriptionIntermediateFuture;
import jadex.commons.future.TerminableIntermediateDelegationResultListener;

/**
 *  Service for deployment files on file system.
 */
@Service
public class FileTransferService implements IFileTransferService
{
	/** The agent. */
	@ServiceComponent
	protected IExternalAccess agent; 
	
	/**
	 *  Put a file.
	 *  @param file The file data.
	 *  @param path The target path.
	 *  @return True, when the file has been copied.
	 */
	public ISubscriptionIntermediateFuture uploadFile(IInputConnection con, String path, String name)
	{
//		TerminableIntermediateFuture ret = new TerminableIntermediateFuture();
//		ret.setFinished();
//		return ret;
//		System.out.println("uploadFile: "+Thread.currentThread());
		try
		{
			return con.writeToOutputStream(new FileOutputStream(path+File.separator+name), agent);
		}
		catch(Exception e)
		{
			return new SubscriptionIntermediateFuture(e);
		}
	}
	
	/**
	 *  Download a file.
	 *  @param file The file data.
	 *  @return True, when the file has been copied.
	 */
	public ISubscriptionIntermediateFuture downloadFile(IOutputConnection con, String name)
	{
		SubscriptionIntermediateDelegationFuture ret = new SubscriptionIntermediateDelegationFuture();
		
		try
		{
			File f = new File(name);
			if(f.exists())
			{
//				System.out.println("src size: "+f.length());
				ret.addIntermediateResult(f.length());
				FileInputStream fis = new FileInputStream(f);
				ISubscriptionIntermediateFuture fut = con.writeFromInputStream(fis, agent);
				TerminableIntermediateDelegationResultListener lis = new TerminableIntermediateDelegationResultListener(ret, fut);
				fut.addResultListener(lis);
			}
			else
			{
				ret.setException(new RuntimeException("File does not exist: "+name));
			}
		}
		catch(Exception e)
		{
			ret.setException(e);
		}
		
		return ret;
	}
	
	/**
	 *  Rename a file.
	 *  @param path The target path.
	 *  @return True, if rename was successful.
	 */
	public IFuture renameFile(String path, String name)
	{
		Future ret = new Future();
		try
		{
			File file = new File(path);
			String newname = file.getParent()+"/"+name;
			if(file.renameTo(new File(newname)))
			{
				ret.setResult(name);
			}
			else
			{
				ret.setException(new RuntimeException());
			}
		}
		catch(Exception e)
		{
			ret.setException(e);
		}
		return ret;
	}
	
	/**
	 *  Delete a file.
	 *  @param path The target path.
	 *  @return True, if delete was successful.
	 */
	public IFuture deleteFile(String path)
	{
		Future ret = new Future();
		try
		{
			// file.toPath().delete(); since 1.7 throws Exception
			File file = new File(path);
			if(file.delete())
			{
				ret.setResult(null);
			}
			else
			{
				ret.setException(new RuntimeException());
			}
		}
		catch(Exception e)
		{
			ret.setException(e);
		}
		return ret;
	}
	
	/**
	 *  Get the root devices.
	 *  @return The root device files.
	 */
	public IFuture getRoots()
	{
		File[] roots = File.listRoots();
		return new Future(FileData.convertToRemoteFiles(roots));
	}
	
	/**
	 *  Execute a file.
	 *  @param path The filename to execute.
	 */
	public IFuture openFile(final String path)
	{
		final Future ret = new Future();
		SServiceProvider.getService(agent, IContextService.class)
			.addResultListener(new DefaultResultListener()
		{
			public void resultAvailable(IContextService cs)
			{
				try
				{
					cs.openFile(path);
				}
				catch (IOException e)
				{
					e.printStackTrace();
					ret.setException(e);
				}
			}
		});

		// exec produces strange exceptions?!
		// Runtime.getRuntime().exec(path);
		ret.setResult(null);
		return ret;
	}
	
	/**
	 *  List the contents of a directory.
	 *  @param dir The directory, null for current directory.
	 *  @return The contained files.
	 */
	public IFuture listDirectory(String dir)
	{
		Future ret = new Future();
		File file = dir==null? new File("."): new File(dir);
		if(!file.exists())
		{
			ret.setException(new RuntimeException("Directory does not exist: "+dir));
			return ret;
		}
		if(!file.isDirectory())
		{
			ret.setException(new RuntimeException("File is not directory: "+dir));
			return ret;
		}
		File[] files = file.listFiles();
		return new Future(FileData.convertToRemoteFiles(files));
	}
	
	/**
	 * 
	 */
	public static void main(String[] args)
	{
		try
		{
			Runtime.getRuntime().exec("notepad.exe");
		}
		catch(Exception e)
		{
			e.printStackTrace();
		}
	}
}