Message.java

package br.com.caelum.vraptor.i18n;

import java.text.MessageFormat;
import java.util.Arrays;
import java.util.ResourceBundle;

import javax.enterprise.inject.Vetoed;

@Vetoed
public class Message {

	private static final String[] ZERO = { "0" };
	private static final String[] ONE = { "1" };
	private String key;
	private String[] args;
	private final ResourceBundle bundle;

	public Message(ResourceBundle bundle, String key) {
		this.bundle = bundle;
		this.key = key;
	}

	public Message count(int count) {
		if (count == 0) {
			key += ".zero";
			args = ZERO;
		} else if (count == 1) {
			key += ".one";
			args = ONE;
		} else {
			key += ".other";
			args = new String[] { "" + count };
		}
		return this;
	}

	public Message args(String... parameters) {
		if (this.args == null) {
			this.args = parameters;
			return this;
		}
		int pos = args.length;
		this.args = Arrays.copyOf(this.args, this.args.length
				+ parameters.length);
		for (String parameter : parameters) {
			this.args[pos++] = parameter;
		}
		return this;
	}

	private String getValue() {
		String message = bundle.getString(key);
		if (message.equals("???" + key + "???")) {
			return "<span class='i18n_missing_key'>" + key + "</span>";
		}
		if (this.args == null) {
			return message;
		}
		return MessageFormat.format(message, (Object[])args);
	}

	@Override
	public String toString() {
		return getValue();
	}

}