source: branches/2.2/jabberit_messenger/java_source/src/nu/fw/jeti/plugins/filetransfer/ibb/IBBExtension.java @ 3102

/* 
 *      Jeti, a Java Jabber client, Copyright (C) 2004 E.S. de Boer  
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *      GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *      For questions, comments etc, 
 *      use the website at http://jeti.jabberstudio.org
 *  or mail me at eric@jeti.tk
 */

// Created on 8-sept-2004
package nu.fw.jeti.plugins.filetransfer.ibb;

import java.util.HashMap;
import java.util.Map;

import nu.fw.jeti.backend.XExecutableExtension;
import nu.fw.jeti.jabber.Backend;
import nu.fw.jeti.jabber.elements.Extension;
import nu.fw.jeti.jabber.elements.IQExtension;
import nu.fw.jeti.jabber.elements.InfoQuery;
import nu.fw.jeti.jabber.elements.Packet;
import nu.fw.jeti.jabber.elements.XMPPError;
import nu.fw.jeti.jabber.elements.XMPPErrorTag;
import nu.fw.jeti.plugins.filetransfer.GetFileWindow;
import nu.fw.jeti.plugins.filetransfer.Plugin;
import nu.fw.jeti.util.Log;

/**
 * @author E.S. de Boer
 *
 */
public class IBBExtension extends Extension implements IQExtension, XExecutableExtension
{
        private boolean close;
        private boolean open;
        private String sid;
        private int sequence;
        private int blockSize=4096;
        //private byte[] data;
        private String base64data;
        private static Map fileWindows = new HashMap(10);

        public IBBExtension(String sid,int blockSize)
        {
                open=true;
                this.blockSize = blockSize;
                this.sid=sid;
        }
        
        public IBBExtension(String sid)
        {
                close=true;
                this.sid=sid;
        }
        
        
        public IBBExtension(String sid,int sequence, String data)
        {
                this.sid = sid;
                this.sequence = sequence;
                base64data = data;
        }
        
        public IBBExtension(boolean open,boolean close, String sid, int blockSize,int sequence, String data)
        {
                this.open=open;
                this.close = close;
                this.sid = sid;
                this.blockSize=blockSize;
                this.sequence = sequence;
                base64data = data;
        }
        
        public String getData()
        {
                return base64data;
        }
        
        public String getSid()
        {
                return sid;
        }
        
        public boolean isOpen()
        {
                return open;
        }
        
        public boolean isClose()
        {
                return close;
        }
        
        
        public void execute(Packet iq, Backend backend)
        {
                //System.out.println(getSid());
                GetFileWindow w= (GetFileWindow)fileWindows.get(sid);
                if(w==null)
                {
                        Log.xmlReceivedError("ibb stream from unknown");
                }
                else
                {
                        IBBReceive r= (IBBReceive)w.getStreamReceive();
                        r.addData(getData());
                }
        }

        public void execute(InfoQuery iq, Backend backend)
        {
                if(iq.getType().equals("set"))
                {
                        //Plugin.ibb(iq.getFrom(),this);
                                                
                        if(isOpen())
                        {
                                GetFileWindow w= Plugin.getGetFile(iq.getFrom(),getSid());
                                if(w!=null)
                                {//file accepted by filetransfer protocol;
                                        //System.out.println(getSid());
                                        w.startDownloading(new IBBReceive(iq.getFrom(),getSid(),backend,w));
                                        fileWindows.put(getSid(),w);
                                        backend.send(new InfoQuery(iq.getFrom(),"result",iq.getID(),null));
                                        Log.xmlReceivedError("ibb stream init from unknown");
                                }
                                else
                                {
                                        XMPPError e = new XMPPError("cancel",501);
                                        e.addError(new XMPPErrorTag("feature-not-implemented"));
                                        backend.send(new InfoQuery(iq.getFrom(),iq.getID(),e));
                                }
                        }
                        if(isClose())
                        {
                                GetFileWindow w=  (GetFileWindow) fileWindows.remove(sid);
                                if(w!=null)((IBBReceive)w.getStreamReceive()).stopDownloading();
                                //System.out.println(" close  " + sid);
                                backend.send(new InfoQuery(iq.getFrom(),"result",iq.getID(),null));
                        }
                }
        }
        
        public void appendToXML(StringBuffer xml)
        {
                if(open)
                {
                        xml.append("<open xmlns= 'http://jabber.org/protocol/ibb'");
                        appendAttribute(xml,"sid",sid);
                        appendAttribute(xml,"block-size",String.valueOf(blockSize));
                        xml.append("/>");
                }
                else if (close)
                {
                        xml.append("<close xmlns= 'http://jabber.org/protocol/ibb'");
                        appendAttribute(xml,"sid",sid);
                        xml.append("/>");
                }
                else
                {
                        xml.append("<data xmlns= 'http://jabber.org/protocol/ibb'");
                        appendAttribute(xml,"sid",sid);
                        appendAttribute(xml,"seq",String.valueOf(sequence));
                        xml.append(">");
                        //addData(data,xml);
                        xml.append(base64data);
                        xml.append("</data>");
                        //TODO change to AMP protocol instead of hardcoding
                        xml.append("<amp xmlns='http://jabber.org/protocol/amp'>");
                        xml.append("<rule condition='deliver-at' value='stored' action='error'/>");
                        xml.append("<rule condition='match-resource' value='exact' action='error'/>");
                        xml.append("</amp>");
                }
        }
}
/*
 * Overrides for emacs
 * Local variables:
 * tab-width: 4
 * End:
 */