OutOfBandData.java

/*
 * Kontalk client common library
 * Copyright (C) 2017 Kontalk Devteam <[email protected]>

 * 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 3 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, see <http://www.gnu.org/licenses/>.
 */

package org.kontalk.client;

import org.jivesoftware.smack.SmackException;
import org.jivesoftware.smack.packet.ExtensionElement;
import org.jivesoftware.smack.provider.ExtensionElementProvider;
import org.xmlpull.v1.XmlPullParser;
import org.xmlpull.v1.XmlPullParserException;

import java.io.IOException;


/**
 * XEP-0066: Out of Band Data
 * http://xmpp.org/extensions/xep-0066.html
 */
public class OutOfBandData implements ExtensionElement {

    public static final String NAMESPACE = "jabber:x:oob";
    public static final String ELEMENT_NAME = "x";

    private final String mUrl;
    private final String mMime;
    private int mWidth;
    private int mHeight;
    private final long mLength;
    private final boolean mEncrypted;

    public OutOfBandData(String url) {
        this(url, null, -1, false);
    }

    public OutOfBandData(String url, String mime, int width, int height, long length, boolean encrypted) {
        mUrl = url;
        mMime = mime;
        mWidth = width;
        mHeight = height;
        mLength = length;
        mEncrypted = encrypted;
    }

    @Override
    public String getElementName() {
        return ELEMENT_NAME;
    }

    @Override
    public String getNamespace() {
        return NAMESPACE;
    }

    public String getUrl() {
        return mUrl;
    }

    public String getMime() {
        return mMime;
    }

    public int getWidth() {
        return mWidth;
    }

    public int getHeight() {
        return mHeight;
    }

    public long getLength() {
        return mLength;
    }

    public boolean isEncrypted() {
        return mEncrypted;
    }

    @Override
    public String toXML() {
        /*
  <x xmlns='jabber:x:oob'>
    <url type='image/png' length='2034782'>http://prime.kontalk.net/media/filename_or_hash</url>
  </x>
         */
        StringBuilder xml = new StringBuilder();
        xml.append(String.format("<%s xmlns='%s'><url", ELEMENT_NAME, NAMESPACE));
        if (mMime != null)
            xml.append(String.format(" type='%s'", mMime));

        if (mLength >= 0)
            xml.append(String.format(" length='%d'", mLength));

        if (mEncrypted)
            xml.append(" encrypted='true'");

        xml
            .append(" width=")
            .append(mWidth)
            .append(" height=")
            .append(mHeight)
            .append(">")
            // TODO should we escape this?
            .append(mUrl)
            .append(String.format("</url></%s>", ELEMENT_NAME));
        return xml.toString();
    }

    public static final class Provider extends ExtensionElementProvider<OutOfBandData> {

        @Override
        public OutOfBandData parse(XmlPullParser parser, int initialDepth) throws XmlPullParserException, IOException, SmackException {
            String url = null, mime = null;
            int width = 0, height = 0;
            long length = -1;
            boolean encrypted = false;
            boolean in_url = false, done = false;

            while (!done)
            {
                int eventType = parser.next();

                if (eventType == XmlPullParser.START_TAG)
                {
                    if ("url".equals(parser.getName())) {
                        in_url = true;
                        mime = parser.getAttributeValue(null, "type");
                        String _width = parser.getAttributeValue(null, "width");
                        try {
                            width = Integer.parseInt(_width);
                        }
                        catch (Exception e) {
                            // ignored
                        }
                        String _height = parser.getAttributeValue(null, "height");
                        try {
                            height = Integer.parseInt(_height);
                        }
                        catch (Exception e) {
                            // ignored
                        }
                        String _length = parser.getAttributeValue(null, "length");
                        try {
                            length = Long.parseLong(_length);
                        }
                        catch (Exception e) {
                            // ignored
                        }
                        String _encrypted = parser.getAttributeValue(null, "encrypted");
                        encrypted = Boolean.parseBoolean(_encrypted);
                    }

                }
                else if (eventType == XmlPullParser.END_TAG)
                {
                    if ("url".equals(parser.getName())) {
                        done = true;
                    }
                }
                else if (eventType == XmlPullParser.TEXT && in_url) {
                    url = parser.getText();
                }
            }

            if (url != null)
                return new OutOfBandData(url, mime, width, height, length, encrypted);
            else
                return null;
        }

    }

}