squawk/qomemo/bundle.cpp

/*
 * Created by victoria on 2021-05-12.
 */

#include "bundle.h"

#include <QXmppPubSubIq.h>

#include <QDebug>

#include "shared/qxmppfactories.h"

using namespace QXmpp::Factories;

QXmppElement QXmpp::Omemo::PreKey::toXml() const {
    auto pk = createElement("preKeyPublic");
    pk.setAttribute("preKeyId", QString::number(id));
    // TODO: Base64
    pk.setValue(data);

    return pk;
}

void QXmpp::Omemo::PreKey::fromXml(const QXmppElement &element) {
    if (!elementMatches(element, "preKeyPublic"))
        return;

    id = element.attribute("preKeyId").toInt();
    // TODO: Base64
    data = element.value();
}

QXmppElement QXmpp::Omemo::Bundle::toXml() const {
    auto spkNode = createElement("signedPreKeyPublic");
    spkNode.setAttribute("signedPreKeyId", QString::number(spkId));
    spkNode.setValue(spk);

    auto spksNode = createElement("signedPreKeySignature");
    spksNode.setValue(spks);

    auto ikNode = createElement("identityKey");
    ikNode.setValue(ik);

    auto prekeysNode = createElement("prekeys");
    for (const auto &pk : prekeys) {
        prekeysNode.appendChild(pk.toXml());
    }

    auto result = createElement("bundle", "eu.siacs.conversations.axolotl");
    result.appendChild(spkNode);
    result.appendChild(spksNode);
    result.appendChild(ikNode);
    result.appendChild(prekeysNode);

    return result;
}

QXmppIq QXmpp::Omemo::Bundle::toIq(int deviceId) const {
    QXmppIq iq{};

    iq.setType(QXmppIq::Set);

    auto item = createElement("item");
    item.appendChild(toXml());

    auto publish = createElement("publish");
    publish.setAttribute(
            "node",
            QString("eu.siacs.conversations.axolotl.bundles:%s").arg(deviceId));
    publish.appendChild(item);

    auto pubSub = createElement("pubsub", "http://jabber.org/protocol/pubsub");
    pubSub.appendChild(publish);
    pubSub.appendChild(createOpenPublishOptions());

    iq.extensions().push_back(pubSub);

    return iq;
}

void QXmpp::Omemo::Bundle::fromXml(const QXmppElement &element) {
    if (!elementMatches(element, "bundle", "eu.siacs.conversations.axolotl"))
        return;

    auto spkNode = element.firstChildElement("spk");
    if (spkNode.isNull()) {
        qWarning() << "'bundle': missing 'spk'";
        return;
    }
    spk = spkNode.value();
    spkId = spkNode.attribute("id").toInt();

    auto spksNode = element.firstChildElement("spks");
    if (spksNode.isNull()) {
        qWarning() << "'bundle': missing 'spks'";
        return;
    }
    spks = spksNode.value();

    auto ikNode = element.firstChildElement("ik");
    if (ikNode.isNull()) {
        qWarning() << "'bundle': missing 'ik'";
        return;
    }
    ik = ikNode.value();

    auto prekeysNode = element.firstChildElement("prekeys");
    auto pkNode = prekeysNode.firstChildElement("pk");

    prekeys.clear();
    while (!pkNode.isNull()) {
        PreKey pk{};
        pk.fromXml(pkNode);
        prekeys.push_back(pk);
    }
}

QXmppPubSubIq QXmpp::Omemo::Bundle::fetchDeviceBundleIq(int deviceId) {
    QXmppPubSubIq iq{};
    iq.setType(QXmppIq::Get);
    iq.setQueryNode(
            QString("eu.siacs.conversations.axolotl.bundles:%1").arg(deviceId));

    QXmppPubSubItem item{};
    item.setId(QString::number(deviceId));
    iq.setItems({item});

    return iq;
}