/*******************************************************************************
 * Australian National University Data Commons
 * Copyright (C) 2013  The Australian National University
 * 
 * This file is part of Australian National University Data Commons.
 * 
 * Australian National University Data Commons 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 3 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, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/

package au.edu.anu.datacommons.collectionrequest;

import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import com.sun.jersey.api.view.Viewable;

/**
 * An instance of this class can be included in the model object passed to instances of {@link Viewable} so the JSP page
 * can display messages included in the instance of this class.
 * 
 * @author Rahul Khanna
 * 
 */
public class PageMessages {
	public enum MessageType {
		ERROR, WARNING, INFO, SUCCESS
	};

	private Set<String> errors = null;
	private Set<String> warnings = null;
	private Set<String> infos = null;
	private Set<String> successes = null;

	public void add(MessageType msgType, String message, Map<String, Object> model) {
		switch (msgType) {
		case ERROR:
			if (errors == null)
				errors = new HashSet<String>();
			errors.add(message);
			break;

		case WARNING:
			if (warnings == null)
				warnings = new HashSet<String>();
			warnings.add(message);
			break;

		case INFO:
			if (infos == null)
				infos = new HashSet<String>();
			infos.add(message);
			break;

		case SUCCESS:
			if (successes == null)
				successes = new HashSet<String>();
			successes.add(message);
			break;

		default:
			throw new IllegalArgumentException("Invalid MessageType.");
		}

		// Add this instance to the model if not added already.
		if (!model.containsValue(this))
			model.put("messages", this);
	}

	public Set<String> getErrors() {
		return errors;
	}

	public Set<String> getWarnings() {
		return warnings;
	}

	public Set<String> getInfos() {
		return infos;
	}

	public Set<String> getSuccesses() {
		return successes;
	}

	public void clear(MessageType msgType) {
		switch (msgType) {
		case ERROR:
			if (errors != null)
				errors.clear();
			break;

		case WARNING:
			if (warnings != null)
				warnings.clear();
			break;

		case INFO:
			if (infos != null)
				infos.clear();
			break;

		case SUCCESS:
			if (successes != null)
				successes.clear();
			break;

		default:
			throw new IllegalArgumentException("Invalid MessageType.");
		}
	}

	public void clear() {
		clear(MessageType.ERROR);
		clear(MessageType.WARNING);
		clear(MessageType.INFO);
		clear(MessageType.SUCCESS);
	}
}