package com.hotent.openfire.plugin;

import java.io.File;

import java.util.List;

import java.util.Map;

import net.sf.json.JSONArray;

import org.dom4j.DocumentHelper;

import org.dom4j.Element;

import org.dom4j.QName;

import org.jivesoftware.openfire.XMPPServer;

import org.jivesoftware.openfire.auth.DefaultAuthProvider;

import org.jivesoftware.openfire.container.Plugin;

import org.jivesoftware.openfire.container.PluginManager;

import org.jivesoftware.openfire.user.UserManager;

import org.jivesoftware.util.PropertyEventDispatcher;

import org.jivesoftware.util.PropertyEventListener;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.xmpp.component.Component;

import org.xmpp.component.ComponentException;

import org.xmpp.component.ComponentManager;

import org.xmpp.component.ComponentManagerFactory;

import org.xmpp.packet.IQ;

import org.xmpp.packet.JID;

import org.xmpp.packet.Packet;

public class OrgTreePlugin implements Component, Plugin {

    private static final Logger Log = LoggerFactory.getLogger(OrgTreePlugin.class);

    public static final String NAMESPACE_JABBER_IQ_SEARCH = "jabber:iq:loadOrg";

    public static final String SERVICENAME = "plugin.loadOrg.serviceName";

    private UserManager userManager;

    private PluginManager pluginManager;

    private ComponentManager componentManager;

    private String serviceName = "loadOrg";

    private String serverName ;

    private OrgTreeProvider orgTreeProvider;

     // 插件初始化的时候，注入bean。

    public OrgTreePlugin(){

        userManager = UserManager.getInstance();

        serverName = XMPPServer.getInstance().getServerInfo().getXMPPDomain();

        orgTreeProvider = DefaultOrgTreeProvider.getInstance();

    }

    public String getName() {

        return pluginManager.getName(this);

    }

    public String getDescription() {

        return pluginManager.getDescription(this);

    }

    // 初始化插件调用的方法。此时注册插件服务。

    @Override

    public void initializePlugin(PluginManager manager, File pluginDirectory) {

        pluginManager = manager;

        componentManager = ComponentManagerFactory.getComponentManager();

        try {

            //注册服务。

            componentManager.addComponent(serviceName, this);

        } catch (ComponentException e) {

            Log.error("===================OrgTree 注册服务失败！！=====================================");

            throw new RuntimeException("OrgTree 注册服务失败！！",e);

        }

        Log.info("===================OrgTree 注册服务成功！！=====================================");

    }

    public void initialize(JID jid, ComponentManager componentManager) {

    }

    public void start() {

    }

    public void destroyPlugin() {

        pluginManager = null;

        try {

            componentManager.removeComponent(serviceName);

            componentManager = null;

        } catch (Exception e) {

        }

        serviceName = null;

        userManager = null;

    }

    @Override

    public void shutdown() {

    }

    // 当客户端请求该服务时，访问该方法，接下来处理消息包

    //第一次访问注册服务时候， 也会测试性质的访问该服务。如果出现异常，将从服务列表移除

    //

    @Override

    public void processPacket(Packet p) {

        if (!(p instanceof IQ)) {

            return;

        }

        final IQ packet = (IQ) p;

        if (packet.getType().equals(IQ.Type.error) || packet.getType().equals(IQ.Type.result)) {

            return;

        }

        //将消息包处理，并将结果数据，返回给请求者。

        final IQ replyPacket = reply(packet);

        try {

            componentManager.sendPacket(this, replyPacket);

        } catch (ComponentException e) {

        }

    }

    /***

     * 处理消息包，转成回执包

     * @param packet

     * @return

     */

    private IQ reply(IQ packet) {

        /*if (!packet.getType().equals(IQ.Type.get)) {

            throw new IllegalArgumentException("This method only accepts 'get' typed IQ stanzas as an argument.");

        }*/

        String orgId ="";

        final Element element = packet.getChildElement();

        if(element != null){

            Element e = element.element("orgId");

            if(e!= null)orgId = e.attribute("id").getText();

        }

        String authId = packet.getFrom().toString();

        if(authId .contains("@"))

        authId = authId.substring(0, authId.indexOf("@"));

        IQ replyPacket = IQ.createResultIQ(packet);

        Element queryResult = DocumentHelper.createElement(QName.get("query", NAMESPACE_JABBER_IQ_SEARCH));

        //去数据库获取数据

        List<OrgTreeNode> orgTreeList = orgTreeProvider.LoadOrgChildOrgUser(orgId, authId);

        //orgTreeList.remove(orgTreeList.size()-1);

        JSONArray json = JSONArray.fromObject(orgTreeList);

        queryResult.addElement("orgTreeList").addText(json.toString());

        replyPacket.setChildElement(queryResult);

        return replyPacket;

    }

}