USER.java

/**
 *
 */
package fr.cedrik.email.pop3.commands;

import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Iterator;
import java.util.StringTokenizer;

import org.apache.commons.lang3.StringUtils;

import fr.cedrik.email.pop3.Context;
import fr.cedrik.email.pop3.POP3Command;
import fr.cedrik.email.pop3.ResponseStatus;
import fr.cedrik.email.pop3.State;
import fr.cedrik.util.IteratorChain;

/**
 * @author Cédrik LIME
 */
public class USER extends BasePOP3Command implements POP3Command {

	public USER() {
	}

	@Override
	public boolean isValid(Context context) {
		return context.state == State.AUTHORIZATION || StringUtils.isBlank(context.userName);
	}

	@Override
	public State nextState(Context context) {
		return (StringUtils.isBlank(context.userName) || StringUtils.isBlank(context.userPassword)) ? State.AUTHORIZATION : State.TRANSACTION;
	}

	@Override
	public Iterator<String> call(Context context) throws IOException {
		if (StringUtils.isBlank(context.inputArgs)) {
			return new IteratorChain<String>(ResponseStatus.NEGATIVE.toString());
		}
		// The login should be user@https://webmail.example.com or user@domain.example@https://webmail.example.com
		StringTokenizer tokenizer = new StringTokenizer(context.inputArgs, "@", false);
		if (tokenizer.countTokens() != 2 && tokenizer.countTokens() != 3) {
			return new IteratorChain<String>(ResponseStatus.NEGATIVE.toString("[AUTH] bad user format; should be user@https://webmail.example.com"));
		}
		context.userName = tokenizer.nextToken();
		if (tokenizer.countTokens() == 2) {
			context.userName += "@" + tokenizer.nextToken();
		}
		String serverURL = tokenizer.nextToken();
		try {
			URL url = new URL(serverURL);
			context.remoteSession.setServerAddress(url);
		} catch (MalformedURLException e) {
			context.userName = null;
			return new IteratorChain<String>(ResponseStatus.NEGATIVE.toString("[AUTH] bad user format; should be user@https://webmail.example.com"));
		}
		if (context.isLocked()) {
			logger.warn("An attempt was made to authenticate the locked user \"{}\"", context.userName);
			context.userName = null;
			return new IteratorChain<String>(ResponseStatus.NEGATIVE.toString("[AUTH] user is locked"));
		}
		return new IteratorChain<String>(ResponseStatus.POSITIVE.toString(context.userName + '@' + serverURL));
	}

}