Smack/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/OutgoingJingleFileOffer.java

/**
 *
 * Copyright 2017 Paul Schaub
 *
 * 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.
 */
package org.jivesoftware.smackx.jingle_filetransfer;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.jivesoftware.smack.SmackException;
import org.jivesoftware.smack.XMPPConnection;
import org.jivesoftware.smack.XMPPException;
import org.jivesoftware.smack.packet.IQ;
import org.jivesoftware.smackx.bytestreams.BytestreamSession;
import org.jivesoftware.smackx.jingle.JingleManager;
import org.jivesoftware.smackx.jingle.JingleTransportMethodManager;
import org.jivesoftware.smackx.jingle.Role;
import org.jivesoftware.smackx.jingle.element.Jingle;
import org.jivesoftware.smackx.jingle.element.JingleContent;
import org.jivesoftware.smackx.jingle.transports.JingleTransportInitiationCallback;
import org.jivesoftware.smackx.jingle.transports.JingleTransportManager;
import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransfer;

import org.jxmpp.jid.FullJid;

/**
 * We are the initiator and we are the sender.
 */
public class OutgoingJingleFileOffer extends JingleFileTransferSession {

    private static final Logger LOGGER = Logger.getLogger(OutgoingJingleFileOffer.class.getName());

    private Thread sendingThread;
    private File source;


    public OutgoingJingleFileOffer(XMPPConnection connection, FullJid responder, String sid) {
        super(connection, connection.getUser().asFullJidOrThrow(), responder, Role.initiator, sid, Type.offer);
    }

    public OutgoingJingleFileOffer(XMPPConnection connection, FullJid recipient) {
        this(connection, recipient, JingleManager.randomSid());
    }

    public void sendFile(JingleFileTransfer file, JingleContent.Creator creator, String name) throws InterruptedException, XMPPException.XMPPErrorException, SmackException.NotConnectedException, SmackException.NoResponseException {
        if (getState() == State.fresh) {
            transportManager = JingleTransportMethodManager.getInstanceFor(connection)
                    .getBestAvailableTransportManager();

            if (transportManager == null) {
                throw new IllegalStateException("There must be at least one workable transport method.");
            }

            transport = transportManager.createTransport();

            jutil.sendSessionInitiateFileOffer(getResponder(), getSessionId(), creator, name, file, transport);
        }
    }

    @Override
    public IQ handleSessionAccept(Jingle sessionAccept) {
        setState(State.active);

        transportManager.initiateOutgoingSession(transport, new JingleTransportInitiationCallback() {
            @Override
            public void onSessionInitiated(final BytestreamSession session) {
                sendingThread = new SendingThread(session, source);
                sendingThread.run();
            }
        });

        return jutil.createAck(sessionAccept);
    }

    @Override
    public IQ handleSessionTerminate(Jingle sessionTerminate) {

        if (sendingThread != null && !sendingThread.isInterrupted()) {
            sendingThread.interrupt();
        }

        setState(State.terminated);
        return jutil.createAck(sessionTerminate);
    }

    @Override
    public IQ handleTransportReplace(Jingle transportReplace)
            throws InterruptedException, XMPPException.XMPPErrorException,
            SmackException.NotConnectedException, SmackException.NoResponseException {
        JingleTransportManager<?> replacementManager = JingleTransportMethodManager.getInstanceFor(connection)
                .getTransportManager(transportReplace);

        if (replacementManager != null) {
            jutil.sendTransportAccept(transportReplace.getFrom().asFullJidOrThrow(),
                    transportReplace.getInitiator(), transportReplace.getSid(), creator, name,
                    replacementManager.createTransport());
        }

        return jutil.createAck(transportReplace);
    }

    private static class SendingThread extends Thread {

        private final BytestreamSession session;
        private final File source;

        public SendingThread(BytestreamSession session, File source) {
            this.session = session;
            this.source = source;
        }

        @Override
        public void run() {
            InputStream inputStream;
            OutputStream outputStream;

            try {
                inputStream = new FileInputStream(source);
                outputStream = session.getOutputStream();

                byte[] filebuf = new byte[(int) source.length()];
                int r = inputStream.read(filebuf);

                if (r < 0) {
                    throw new IOException("Read returned -1");
                }

                outputStream.write(filebuf);
            }
            catch (IOException e) {
                LOGGER.log(Level.SEVERE, "Could not send file: " + e, e);
            }
            finally {
                try {
                    session.close();
                } catch (IOException e) {
                    LOGGER.log(Level.SEVERE, "Could not close session.", e);
                }
            }
        }

        @Override
        public void interrupt() {
            try {
                session.close();
            } catch (IOException e) {
                LOGGER.log(Level.SEVERE, "Could not close session.", e);
            }
            super.interrupt();
        }
    }
}