openfire聊天记录插件

package com.sqj.openfire.chat.logs;

import java.io.File;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import org.apache.commons.lang.StringUtils;
import org.dom4j.DocumentFactory;
import org.dom4j.Element;
import org.jivesoftware.openfire.OfflineMessageStore;
import org.jivesoftware.openfire.XMPPServer;
import org.jivesoftware.openfire.container.Plugin;
import org.jivesoftware.openfire.container.PluginManager;
import org.jivesoftware.openfire.interceptor.InterceptorManager;
import org.jivesoftware.openfire.interceptor.PacketInterceptor;
import org.jivesoftware.openfire.interceptor.PacketRejectedException;
import org.jivesoftware.openfire.session.Session;
import org.jivesoftware.openfire.user.UserManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.xmpp.packet.IQ;
import org.xmpp.packet.JID;
import org.xmpp.packet.Message;
import org.xmpp.packet.Packet;
import org.xmpp.packet.Presence;

import com.sqj.openfire.chat.logs.entity.Chat;
import com.sqj.openfire.chat.logs.entity.XmlToMap;

/** <b>function:</b> 聊天记录插件
 *  @author shm
 *  @createDate 2016-3-14
 *  @version 1.0
 */
public class ChatLogsPlugin implements PacketInterceptor, Plugin {
      private static final Logger log = LoggerFactory.getLogger(ChatLogsPlugin.class);
      private static PluginManager pluginManager;
      private static DbChatLogsManager logsManager;
      private InterceptorManager interceptorManager;

      public ChatLogsPlugin()
      {
        this.interceptorManager = InterceptorManager.getInstance();
        logsManager = DbChatLogsManager.getInstance();
      }

      /**
       *  * <b>function:</b> 拦截消息核心方法，Packet就是拦截消息对象
       *  * @author shm
       *  * @createDate 2016-3-14
       *  */
      @Override
      public void interceptPacket(Packet packet, Session session, boolean incoming, boolean processed) throws PacketRejectedException {
          if (session != null) {
              //debug(packet, incoming, processed, session);
          }
          JID recipient = packet.getTo();
          if (recipient != null) {
              String username = recipient.getNode();
              // 广播消息或是不存在/没注册的用户.
              if (username == null || !UserManager.getInstance().isRegisteredUser(recipient)) {
                  return;
              } else if (!XMPPServer.getInstance().getServerInfo().getXMPPDomain().equals(recipient.getDomain())) {
                  // 非当前openfire服务器信息
                  return;
              } else if ("".equals(recipient.getResource())) {
              }
          }
          this.doAction(packet, incoming, processed, session);
     }

      /**
       * <b>function:</b> 执行保存/分析聊天记录动作
       * @author shm
       * @createDate 2016-3-14
       * @param packet 数据包
       * @param incoming true表示发送方
       * @param session 当前用户session
       */
      private void doAction(Packet packet, boolean incoming, boolean processed, Session session) {
          Packet copyPacket = packet.createCopy();
          if (packet instanceof Message) {
              Message message = (Message) copyPacket;
              // 一对一聊天，单人模式
              if (message.getType() == Message.Type.chat) {
                  log.info("单人聊天信息：{}", message.toXML());
                  //debug("单人聊天信息：" + message.toXML());
                  // 程序执行中；是否为结束或返回状态（是否是当前session用户发送消息）
                  if (processed || !incoming) {
                      return;
                  }         

                  java.sql.Timestamp date = new java.sql.Timestamp(new Date().getTime());

                  //在原有的扩展xml中添加发送日期子节点
                  Element chatInfoElement = message.getElement().element("chatinfo");
                  Element timeElement = DocumentFactory.getInstance().createDocument().addElement("sendtime");
                  timeElement.setText(String.valueOf(date.getTime()));
                  chatInfoElement.add(timeElement);

                  //logsManager.add(this.get(packet, incoming, session));
                  logsManager.addByMap(this.getMsgInfo(packet, incoming, session, date)); 

                  //保存到离线消息表，客户端收到后调用删除离线消息功能，这样可确保即使网络突然掉线或不好的情况下消息丢失的问题
                  OfflineMessageStore offlineMessageStore = new OfflineMessageStore();
                  offlineMessageStore.addMessage(message);

                  //消息回执
                  Message receiptMessage = new Message();
                  receiptMessage.setTo(message.getFrom());
                  receiptMessage.setType(Message.Type.normal);
                  Element received = receiptMessage.addChildElement("received", "urn:xmpp:receipts");
                  received.addAttribute("id", message.getID());
                  received.addAttribute("type", "normal");
                  log.info("回执内容：" + receiptMessage);

                  // 判断接受者是否在线，2代表离线状态，1代表在线状态，0代表用戶不存在
                  //注意一定要修改Constant类中的url,否则url错误或导致消息发送很慢，因为url不对时查找用户是否在线会浪费很多时间
                  /*if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 1) {
                      received.addAttribute("status", "1");
                  } else if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 0) {
                      received.addAttribute("status", "0");
                  } else if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 2) {
                      received.addAttribute("status", "2");
                  }*/

                  try {
                      XMPPServer.getInstance().getPacketDeliverer().deliver(receiptMessage);
                      log.info("服务端回执成功！");
                  } catch (Exception e) {
                      e.printStackTrace();
                  }

              // 群聊天，多人模式
              } else if (message.getType() ==  Message.Type.groupchat) {
                  List<?> els = message.getElement().elements("x");
                  if (els != null && !els.isEmpty()) {
                      log.info("群聊天信息：{}", message.toXML());
                      debug("群聊天信息：" + message.toXML());
                  } else {
                      log.info("群系统信息：{}", message.toXML());
                        debug("群系统信息：" + message.toXML());
                  }
              // 其他信息
              }else {
                  log.info("其他信息：{}", message.toXML());
                  //debug("其他信息：" + message.toXML());
              }
          } else if (packet instanceof IQ) {
              IQ iq = (IQ) copyPacket;
              if (iq.getType() == IQ.Type.set && iq.getChildElement() != null && "session".equals(iq.getChildElement().getName())) {
                  log.info("用户登录成功：{}", iq.toXML());
                  //debug("用户登录成功：" + iq.toXML());
              }
          } else if (packet instanceof Presence) {
              Presence presence = (Presence) copyPacket;
              if (presence.getType() == Presence.Type.unavailable) {
                  log.info("用户退出服务器成功：{}", presence.toXML());
                  //debug("用户退出服务器成功：" + presence.toXML());
              }
         }
     }

/**
       * <b>function:</b> 创建一个聊天记录实体对象，并设置相关数据
       * @author shm
       * @createDate 2016-3-14
       * @param packet 数据包
       * @param incoming 如果为ture就表明是发送者
       * @param session 当前用户session
       * @return 聊天实体
   */
    @SuppressWarnings("unchecked")
    private Chat get(Packet packet, boolean incoming, Session session) {
          Message message = (Message) packet;
          JID jid = session.getAddress();         

         Map<String, Object> map = XmlToMap.xmlToHashMap(message.toXML());
         Chat chat = new Chat();
         if(map.containsKey("zid") && !"null".equals(map.get("zid")) && !StringUtils.isEmpty(map.get("zid")+"")){
             chat.setId(map.get("zid")+"");
         }else{
             chat.setId(UUID.randomUUID().toString().replace("-", ""));
         }
         chat.setCaseId(map.get("caseid")+"");
         chat.setContent(message.getBody());
         chat.setCreator(jid.getNode());
         chat.setSender(jid.getNode());
         chat.setModifier(jid.getNode());
         chat.setSendee(message.getTo().getNode());
         chat.setIdentity(map.get("identity")+"");
         chat.setzType(map.get("ztype")+"");
         chat.setSenderId(map.get("senderid")+"");
         chat.setSendeeId(map.get("sendeeid")+"");

         if(map.get("msglength") != null && !"".equals(map.get("msglength")))
         {
             chat.setMsgLength(Integer.parseInt(map.get("msglength")+""));
         }
         return chat;
      }

    @SuppressWarnings({ "unchecked", "rawtypes" })
    private Map getMsgInfo(Packet packet, boolean incoming, Session session, java.sql.Timestamp date) {
       Message message = (Message) packet;
       Map<String, Object> map = XmlToMap.xmlToHashMap(message.toXML());
       map.put("content", message.getBody());
       map.put("sendtime", date);
       log.info("tag：{}", map.get("tag"));
       //debug("tag：" + map.get("tag"));
       return map;
      }

      /**
       * * <b>function:</b> 调试信息
       * @author shm
       * @createDate 2016-3-14
       * @param packet 数据包
       * @param incoming 如果为ture就表明是发送者
       * @param processed 执行
       * @param session 当前用户session
       */
      private void debug(Packet packet, boolean incoming, boolean processed, Session session) {
          //String info = "[ packetID: " + packet.getID() + ", to: " + packet.getTo() + ", from: " + packet.getFrom() + ", incoming: " + incoming + ", processed: " + processed + " ]";
          StringBuilder info = new StringBuilder();
          info.append("[ packetID: ").append(packet.getID()).append(", to: ").append(packet.getTo()).append(", from: ").append(packet.getFrom()).append(", incoming: ").append(incoming).append(", processed: ").append(processed).append(" ]");

          long timed = System.currentTimeMillis();
          debug("################### start ###################" + timed);
          debug("id:" + session.getStreamID() + ", address: " + session.getAddress());
          debug("info: " + info);         debug("xml: " + packet.toXML());
          debug("################### end #####################" + timed);
          log.info("id:" + session.getStreamID() + ", address: " + session.getAddress());
          log.info("info: {}", info);
          log.info("plugin Name: " + pluginManager.getName(this) + ", xml: " + packet.toXML());
      }        

      private void debug(Object message) {
          if (true) {
              System.out.println(message);
          }
     }        

     @Override
     public void destroyPlugin() {
         interceptorManager.removeInterceptor(this);
         debug("销毁聊天记录插件成功！");
     }     

     @Override
     public void initializePlugin(PluginManager manager, File pluginDirectory) {
         interceptorManager.addInterceptor(this);
         pluginManager = manager;
         debug("安装聊天记录插件成功！");
     }

    /* // 判断接受者是否在线，2代表离线状态，1代表在线状态，0代表用戶不存在
     if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 2) {
         // 离线時，向offline表写数据
         OfflineMessageStore offlineMessageStore = new OfflineMessageStore();
         offlineMessageStore.addMessage(message);
         // 向客户端发回执
         Message receiptMessage = new Message();
         receiptMessage.setTo(message.getFrom());
         receiptMessage.setType(Message.Type.normal);
         Element received = receiptMessage.addChildElement("received", "urn:xmpp:receipts");
         received.setAttributeValue("id", message.getID());
         System.out.println("0000000000回执内容" + receiptMessage);
         try {
             XMPPServer.getInstance().getPacketDeliverer().deliver(receiptMessage);
             System.out.println("服务端回执成功！");
         } catch (Exception e) {
             e.printStackTrace();
         }
     }*/
}