org.apacheextras.camel.component.jcifs.SmbClient Maven / Gradle / Ivy

Go to download
/**************************************************************************************
 Copyright (C) 2010 Redpill Linpro AB
 http://code.google.com/a/apache-extras.org/p/camel-extra

 This program is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public License
 as published by the Free Software Foundation; either version 2
 of the License, or (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.


 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 02110-1301, USA.

 http://www.gnu.org/licenses/gpl-2.0-standalone.html
 ***************************************************************************************/
package org.apacheextras.camel.component.jcifs;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.MalformedURLException;
import java.util.ArrayList;
import java.util.List;

import jcifs.smb.NtlmPasswordAuthentication;
import jcifs.smb.SmbException;
import jcifs.smb.SmbFile;
import jcifs.smb.SmbFileOutputStream;

import org.apache.camel.util.IOHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *  
 */
public class SmbClient {

	private NtlmPasswordAuthentication authentication;
	private SmbApiFactory smbApiFactory = new JcifsSmbApiFactory();
	

	protected final transient Logger log = LoggerFactory.getLogger(getClass());
	
	/**
	 * Allows for overriding the default {@link SmbApiFactory} implementation.
	 * 
	 * @param smbFileFactory a {@link SmbApiFactory} to use instead of the default implementation.
	 */
	public void setSmbApiFactory(SmbApiFactory smbApiFactory) {
		this.smbApiFactory = smbApiFactory;
	}

	/** 
	 * Creates the internal NtlmPasswordAuthentication, that is used for authentication, from the provided credentials. 
	 * 
	 * @param domain User domain to use at login
	 * @param username User name to use at login
	 * @param password The password for the provided user
	 */
	public void login(String domain, String username, String password) {
		if (log.isDebugEnabled()) {
			log.debug("login() domain[" + domain + "] username[" + username + "] password[***]");
		}
		setAuthentication(new NtlmPasswordAuthentication(domain, username, password));
	}

	/**
	 * 
	 * @param url
	 * @param out
	 * @return
	 * @throws IOException
	 * @throws MalformedURLException
	 */
	public boolean retrieveFile(String url, OutputStream out) throws IOException, MalformedURLException {
		if (log.isDebugEnabled()) {
			log.debug("retrieveFile() path[" + url + "]");
	    }
		SmbFile smbFile = smbApiFactory.createSmbFile(url, authentication);
		IOHelper.copyAndCloseInput(smbFile.getInputStream(), out);
		return true;
	}

	public boolean createDirs(String url) {
		if (log.isDebugEnabled()) {
			log.debug("createDirs() path[" + url + "]");
		}
		SmbFile smbFile;
		try {
			smbFile = smbApiFactory.createSmbFile(url, authentication);
			if (!smbFile.exists()) {
				smbFile.mkdirs();
			}
		} catch (MalformedURLException e) {
			return false;
		} catch (SmbException e) {
			return false;
		}
		return true;

	}

	public InputStream getInputStream(String url) throws IOException {
		if (log.isDebugEnabled()) {
			log.debug("getInputStream() path[" + url + "]");
		}
		SmbFile smbFile = smbApiFactory.createSmbFile(url, authentication);
		return smbFile.getInputStream();
	}

	public boolean storeFile(String url, InputStream inputStream, boolean append) throws IOException {
		if (log.isDebugEnabled())
			log.debug("storeFile path[" + url + "]");
		SmbFile smbFile = smbApiFactory.createSmbFile(url, authentication);
		SmbFileOutputStream smbout = smbApiFactory.createSmbFileOutputStream(smbFile, append);
		byte[] buf = new byte[512 * 1024];
		int numRead;
		while ( (numRead = inputStream.read(buf)) >= 0)
			smbout.write(buf, 0, numRead);
		smbout.close();
		return true;
	}

	public List listFiles(String url) throws SmbException, MalformedURLException  {
		List fileList = new ArrayList();
		SmbFile dir = smbApiFactory.createSmbFile(url, authentication);
		for (SmbFile f : dir.listFiles()){
			fileList.add(f);
		}
		return fileList;
	}	

	public boolean isExist(String url) throws Exception {
		SmbFile sFile = smbApiFactory.createSmbFile(url, authentication);
		return sFile.exists();
	}

	public boolean delete(String url) throws Exception {
		SmbFile sFile = smbApiFactory.createSmbFile(url, authentication);
		try {
			sFile.delete();
		} catch(SmbException e) {
			return false;
		}
		return true;
	}

	public boolean rename(String fromUrl, String toUrl) throws Exception {
		SmbFile sFile = smbApiFactory.createSmbFile(fromUrl, authentication);
		SmbFile renamedFile = smbApiFactory.createSmbFile(toUrl, authentication);
		try {
			sFile.renameTo(renamedFile);
		} catch (SmbException e) {
			return false;
		}
		return true;
	}

	public NtlmPasswordAuthentication getAuthentication() {
		return authentication;
	}

	public void setAuthentication(NtlmPasswordAuthentication authentication) {
		this.authentication = authentication;
	}
}