com.kixeye.kixmpp.client.module.muc.MucKixmppClientModule.java Source code

Java tutorial

Introduction

Here is the source code for com.kixeye.kixmpp.client.module.muc.MucKixmppClientModule.java

Source

package com.kixeye.kixmpp.client.module.muc;

/*
 * #%L
 * KIXMPP
 * %%
 * Copyright (C) 2014 KIXEYE, Inc
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import io.netty.channel.Channel;

import java.util.Collections;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import org.jdom2.Element;
import org.jdom2.Namespace;
import org.joda.time.DateTime;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.kixeye.kixmpp.KixmppJid;
import com.kixeye.kixmpp.client.KixmppClient;
import com.kixeye.kixmpp.client.module.KixmppClientModule;
import com.kixeye.kixmpp.date.XmppDateUtils;
import com.kixeye.kixmpp.handler.KixmppStanzaHandler;

/**
 * A {@link KixmppClientModule} that deals with MUCs.
 * 
 * @author ebahtijaragic
 */
public class MucKixmppClientModule implements KixmppClientModule {
    private static final Logger logger = LoggerFactory.getLogger(MucKixmppClientModule.class);

    private Set<MucListener<MucMessage>> messageListeners = Collections
            .newSetFromMap(new ConcurrentHashMap<MucListener<MucMessage>, Boolean>());
    private Set<MucListener<MucInvite>> invitationListeners = Collections
            .newSetFromMap(new ConcurrentHashMap<MucListener<MucInvite>, Boolean>());
    private Set<MucListener<MucJoin>> joinListeners = Collections
            .newSetFromMap(new ConcurrentHashMap<MucListener<MucJoin>, Boolean>());

    private KixmppClient client = null;

    /**
     * @param listener the listener to add
     */
    public void addMessageListener(MucListener<MucMessage> listener) {
        messageListeners.add(listener);
    }

    /**
     * @param listener the listener to add
     */
    public void removeMessageListener(MucListener<MucMessage> listener) {
        messageListeners.remove(listener);
    }

    /**
     * @param listener the listener to add
     */
    public void addInviteListener(MucListener<MucInvite> listener) {
        invitationListeners.add(listener);
    }

    /**
     * @param listener the listener to add
     */
    public void removeInviteListener(MucListener<MucInvite> listener) {
        invitationListeners.remove(listener);
    }

    /**
     * @param listener the listener to add
     */
    public void addJoinListener(MucListener<MucJoin> listener) {
        joinListeners.add(listener);
    }

    /**
     * @param listener the listener to add
     */
    public void removeJoinListener(MucListener<MucJoin> listener) {
        joinListeners.remove(listener);
    }

    /**
     * Joins a room.
     * 
     * @param roomJid
     * @param nickname
     */
    public void joinRoom(KixmppJid roomJid, String nickname, Integer maxStanzas, Integer maxChars, Integer seconds,
            DateTime since) {
        Element presence = new Element("presence");
        presence.setAttribute("from", client.getJid().getFullJid());
        presence.setAttribute("to", roomJid + "/" + nickname);

        Element x = new Element("x", "http://jabber.org/protocol/muc");
        presence.addContent(x);

        Element history = new Element("history", "http://jabber.org/protocol/muc");
        if (maxStanzas != null) {
            history.setAttribute("maxstanzas", maxStanzas.toString());
        }
        if (maxChars != null) {
            history.setAttribute("maxchars", maxChars.toString());
        }
        if (seconds != null) {
            history.setAttribute("seconds", seconds.toString());
        }
        if (since != null) {
            history.setAttribute("since", XmppDateUtils.format(since));
        }
        x.addContent(history);

        client.sendStanza(presence);
    }

    /**
     * Joins a room.
     * 
     * @param roomJid
     * @param nickname
     */
    public void joinRoom(KixmppJid roomJid, String nickname) {
        joinRoom(roomJid, nickname, null, null, null, null);
    }

    /**
     * Joins a room with an invitation.
     * 
     * @param invitation
     * @param invitation
     */
    public void joinRoom(MucInvite invitation, String nickname) {
        joinRoom(invitation.getRoomJid(), nickname);
    }

    /**
     * Sends a room message to a room.
     * 
     * @param roomJid
     * @param roomMessage
     */
    public void sendRoomMessage(KixmppJid roomJid, String roomMessage, String nickname) {
        Element message = new Element("message");
        message.setAttribute("from", client.getJid().toString());
        message.setAttribute("to", roomJid.toString());
        message.setAttribute("type", "groupchat");

        Element bodyElement = new Element("body", message.getNamespace());
        bodyElement.setText(roomMessage);
        message.addContent(bodyElement);

        client.sendStanza(message);
    }

    /**
     * @see com.kixeye.kixmpp.client.module.KixmppClientModule#install(com.kixeye.kixmpp.client.KixmppClient)
     */
    public void install(KixmppClient client) {
        this.client = client;

        client.getEventEngine().registerGlobalStanzaHandler("message", mucMessageHandler);
        client.getEventEngine().registerGlobalStanzaHandler("presence", mucPresenceHandler);
    }

    /**
     * @see com.kixeye.kixmpp.client.module.KixmppClientModule#uninstall(com.kixeye.kixmpp.client.KixmppClient)
     */
    public void uninstall(KixmppClient client) {
        client.getEventEngine().unregisterGlobalStanzaHandler("message", mucMessageHandler);
        client.getEventEngine().unregisterGlobalStanzaHandler("presence", mucPresenceHandler);
    }

    private KixmppStanzaHandler mucPresenceHandler = new KixmppStanzaHandler() {
        @Override
        public void handle(Channel channel, Element stanza) {
            Element inX = stanza.getChild("x", Namespace.getNamespace("http://jabber.org/protocol/muc#user"));

            if (inX != null) {
                Element inItem = inX.getChild("item",
                        Namespace.getNamespace("http://jabber.org/protocol/muc#user"));

                if (inItem != null) {
                    KixmppJid from = KixmppJid.fromRawJid(stanza.getAttributeValue("from"));

                    MucJoin message = new MucJoin(from.withoutResource(), from,
                            inItem.getAttributeValue("affiliation"), inItem.getAttributeValue("role"));

                    for (MucListener<MucJoin> listener : joinListeners) {
                        try {
                            listener.handle(message);
                        } catch (Exception e) {
                            logger.error("Exception thrown while executing MucJoin listener", e);
                        }
                    }
                }
            }
        }
    };

    private KixmppStanzaHandler mucMessageHandler = new KixmppStanzaHandler() {
        public void handle(Channel channel, Element stanza) {
            String type = stanza.getAttributeValue("type");

            if (type == null) {
                type = "";
            }

            switch (type) {
            case "chat":
                // ignore
                break;
            case "groupchat":
                String language = null;
                String bodyMessage = null;

                Element body = stanza.getChild("body", stanza.getNamespace());

                if (body != null) {
                    bodyMessage = body.getText();
                    language = body.getAttributeValue("xml:lang");

                    MucMessage message = new MucMessage(KixmppJid.fromRawJid(stanza.getAttributeValue("from")),
                            KixmppJid.fromRawJid(stanza.getAttributeValue("to")), language, bodyMessage);

                    for (MucListener<MucMessage> listener : messageListeners) {
                        try {
                            listener.handle(message);
                        } catch (Exception e) {
                            logger.error("Exception thrown while executing MucInvite listener", e);
                        }
                    }
                }

                break;
            case "":
                // check if invite
                for (Element invitation : stanza.getChildren("x", Namespace.getNamespace("jabber:x:conference"))) {
                    KixmppJid roomJid = KixmppJid.fromRawJid(invitation.getAttributeValue("jid"));
                    KixmppJid toJid = KixmppJid.fromRawJid(stanza.getAttributeValue("to"));
                    KixmppJid fromJid = KixmppJid.fromRawJid(stanza.getAttributeValue("from"));
                    MucInvite invite = new MucInvite(fromJid, toJid, roomJid);
                    for (MucListener<MucInvite> listener : invitationListeners) {
                        try {
                            listener.handle(invite);
                        } catch (Exception e) {
                            logger.error("Exception thrown while executing MucInvite listener", e);
                        }
                    }
                }
                for (Element x : stanza.getChildren("x",
                        Namespace.getNamespace("http://jabber.org/protocol/muc#user"))) {
                    for (Element i : x.getChildren("invite")) {
                        KixmppJid roomJid = KixmppJid.fromRawJid(stanza.getAttributeValue("from"));
                        KixmppJid toJid = KixmppJid.fromRawJid(stanza.getAttributeValue("to"));
                        KixmppJid fromJid = KixmppJid.fromRawJid(i.getAttributeValue("from"));
                        MucInvite invite = new MucInvite(fromJid, toJid, roomJid);
                        for (MucListener<MucInvite> listener : invitationListeners) {
                            try {
                                listener.handle(invite);
                            } catch (Exception e) {
                                logger.error("Exception thrown while executing MucInvite listener", e);
                            }
                        }
                    }
                }
                break;
            }
        }
    };
}