package eu.siacs.conversations.xmpp.stanzas;

import eu.siacs.conversations.xml.Element;

public class IqPacket extends AbstractStanza {

	public static final int TYPE_ERROR = -1;
	public static final int TYPE_SET = 0;
	public static final int TYPE_RESULT = 1;
	public static final int TYPE_GET = 2;

	private IqPacket(final String name) {
		super(name);
	}

	public IqPacket(final int type) {
		super("iq");
		switch (type) {
			case TYPE_SET:
				this.setAttribute("type", "set");
				break;
			case TYPE_GET:
				this.setAttribute("type", "get");
				break;
			case TYPE_RESULT:
				this.setAttribute("type", "result");
				break;
			case TYPE_ERROR:
				this.setAttribute("type", "error");
				break;
			default:
				break;
		}
	}

	public IqPacket() {
		super("iq");
	}

	public Element query() {
		Element query = findChild("query");
		if (query == null) {
			query = addChild("query");
		}
		return query;
	}

	public Element query(final String xmlns) {
		final Element query = query();
		query.setAttribute("xmlns", xmlns);
		return query();
	}

	public int getType() {
		final String type = getAttribute("type");
		switch (type) {
			case "error":
				return TYPE_ERROR;
			case "result":
				return TYPE_RESULT;
			case "set":
				return TYPE_SET;
			case "get":
				return TYPE_GET;
			default:
				return 1000;
		}
	}

	public IqPacket generateResponse(final int type) {
		final IqPacket packet = new IqPacket(type);
		packet.setTo(this.getFrom());
		packet.setId(this.getId());
		return packet;
	}

}