package org.openslx.satellitedaemon.ftp;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;

import org.apache.ftpserver.ConnectionConfigFactory;
import org.apache.ftpserver.FtpServer;
import org.apache.ftpserver.FtpServerFactory;
import org.apache.ftpserver.ftplet.Authority;
import org.apache.ftpserver.ftplet.FtpException;
import org.apache.ftpserver.ftplet.UserManager;
import org.apache.ftpserver.listener.ListenerFactory;
import org.apache.ftpserver.usermanager.PropertiesUserManagerFactory;
import org.apache.ftpserver.usermanager.SaltedPasswordEncryptor;
import org.apache.ftpserver.usermanager.impl.BaseUser;
import org.apache.ftpserver.usermanager.impl.WritePermission;

/**
 * FTP-Test-Utility, basierend auf Apache FtpServer: {@link http
 * ://www.jarvana.com
 * /jarvana/view/org/apache/ftpserver/ftpserver-core/1.0.6/ftpserver
 * -core-1.0.6-javadoc.jar!/org/apache/ftpserver/FtpServer.html}
 */
public class FtpTestUtil {
	/**
	 * Erzeuge FTP-Server.
	 * 
	 * @param ftpPort
	 *            FTP-Port, z.B. 2121
	 * @param ftpHomeDir
	 *            FTP-Verzeichnis, z.B. "target/FtpHome"
	 * @param readUserName
	 *            leseberechtigter Benutzer: Name
	 * @param readUserPwd
	 *            leseberechtigter Benutzer: Passwort
	 * @param writeUserName
	 *            schreibberechtigter Benutzer: Name
	 * @param writeUserPwd
	 *            schreibberechtigter Benutzer: Passwort
	 * @param ftpUsersPropsFile
	 *            kann null sein, oder z.B. "target/FtpUsers.properties"
	 * @param maxLogins
	 *            maximale Anzahl von Logins (0 fuer Defaultwert)
	 */
	public static FtpServer createFtpServer(int ftpPort, String ftpHomeDir,
			String readUserName, String readUserPwd, String writeUserName,
			String writeUserPwd) throws FtpException, IOException {
		return createFtpServer(ftpPort, ftpHomeDir, readUserName, readUserPwd,
				writeUserName, writeUserPwd, null, 0);
	}

	public static FtpServer createFtpServer(int ftpPort, String ftpHomeDir,
			String readUserName, String readUserPwd, String writeUserName,
			String writeUserPwd, String ftpUsersPropsFile, int maxLogins)
			throws FtpException, IOException {
		return createFtpServer(ftpPort, ftpHomeDir, readUserName, readUserPwd,
				writeUserName, writeUserPwd, ftpUsersPropsFile, maxLogins, 0);
	}

	public static FtpServer createFtpServer(int ftpPort, String ftpHomeDir,
			String readUserName, String readUserPwd, String writeUserName,
			String writeUserPwd, String ftpUsersPropsFile, int maxLogins,
			int maxIdleTimeSec) throws FtpException, IOException {
		File fhd = new File(ftpHomeDir);
		if (!fhd.exists())
			fhd.mkdirs();

		ListenerFactory listenerFactory = new ListenerFactory();
		listenerFactory.setPort(ftpPort);

		PropertiesUserManagerFactory userManagerFactory = new PropertiesUserManagerFactory();
		userManagerFactory.setPasswordEncryptor(new SaltedPasswordEncryptor());
		if (ftpUsersPropsFile != null && ftpUsersPropsFile.trim().length() > 0) {
			File upf = new File(ftpUsersPropsFile);
			if (!upf.exists())
				upf.createNewFile();
			userManagerFactory.setFile(upf);
		}

		// Einen Nur-Lese-User und einen User mit Schreibberechtigung anlegen:
		UserManager userManager = userManagerFactory.createUserManager();
		BaseUser userRd = new BaseUser();
		BaseUser userWr = new BaseUser();
		userRd.setName(readUserName);
		userRd.setPassword(readUserPwd);
		userRd.setHomeDirectory(ftpHomeDir);
		userWr.setName(writeUserName);
		userWr.setPassword(writeUserPwd);
		userWr.setHomeDirectory(ftpHomeDir);
		if (maxIdleTimeSec > 0) {
			userRd.setMaxIdleTime(maxIdleTimeSec);
			userWr.setMaxIdleTime(maxIdleTimeSec);
		}
		ArrayList<Authority> authorities = new ArrayList<Authority>();
		authorities.add(new WritePermission());
		userWr.setAuthorities(authorities);
		userManager.save(userRd);
		userManager.save(userWr);

		FtpServerFactory serverFactory = new FtpServerFactory();
		serverFactory.addListener("default", listenerFactory.createListener());
		serverFactory.setUserManager(userManager);
		if (maxLogins > 0) {
			ConnectionConfigFactory ccf = new ConnectionConfigFactory();
			ccf.setMaxLogins(maxLogins);
			serverFactory.setConnectionConfig(ccf.createConnectionConfig());
		}
		return serverFactory.createServer();
	}
}