package org.openslx.imagemaster.server;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

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.Ftplet;
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;
import org.apache.log4j.Logger;

public class MasterFtpServer implements Runnable {
	private static Logger log = Logger.getLogger( MasterFtpServer.class );
	private FtpServer server;
	private UserManager userManager;
	
	public MasterFtpServer(int port, String adminUsername, String adminPassword, String ftproot) {
		FtpServerFactory serverFactory = new FtpServerFactory();
		ListenerFactory factory = new ListenerFactory();
		// set the port of the listener
		factory.setPort(port);
		
		// replace the default listener
		serverFactory.addListener("default", factory.createListener());
		
		// create user manager
	    PropertiesUserManagerFactory userManagerFactory = new PropertiesUserManagerFactory();
	    userManagerFactory.setFile(new File("src/main/properties/ftp.properties"));
	    userManagerFactory.setPasswordEncryptor(new SaltedPasswordEncryptor());
	    userManager = userManagerFactory.createUserManager();
		
		// create new admin user
	    addUser(adminUsername, adminPassword, ftproot, true);
		serverFactory.setUserManager(userManager);
		
		// add the Ftplet
		HashMap<String, Ftplet> map = new HashMap<String, Ftplet>();
		map.put("Ftplet1", new MasterFtplet());
		serverFactory.setFtplets(map);
		
		// start the server
		server = serverFactory.createServer();
	}

	public boolean addUser(final String username, final String password, final String ftproot, final boolean writeAccess) {
	    // TODO: enable SSL
		boolean result = true;

	    BaseUser user = new BaseUser();
	    user.setName(username);
	    user.setPassword(password);
	    user.setHomeDirectory(ftproot);

	    List<Authority> authorities = new ArrayList<Authority>();
	    if (writeAccess)  authorities.add(new WritePermission());
	    user.setAuthorities(authorities);

	    try {
	    	userManager.save(user);
	    } catch (FtpException e) {
	        result = false;
	    }
	    
	    return result;
	}
	
	public boolean removeUser(final String username) {
		boolean result = true;
		
		try {
			userManager.delete(username);
		} catch (FtpException e) {
			result = false;
		}
		
		return result;
	}

	@Override
	public void run() {
		try {
			log.info( "Starting FTP Sever" );
			server.start();
		} catch (FtpException e1) {
			e1.printStackTrace();
		}
	}
}