summaryrefslogtreecommitdiffstats
path: root/examples/oauth/redditclient/redditmodel.cpp
blob: 1b47753ce679d0226e3bc585abfb2f8890f9356a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// Copyright (C) 2017 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause

#include "redditmodel.h"

#include <QJsonArray>

#include <QDesktopServices>

#include <QNetworkAccessManager>
#include <QRestAccessManager>
#include <QRestReply>

#include <QOAuthHttpServerReplyHandler>

using namespace Qt::StringLiterals;

static constexpr auto hotUrl = "https://oauth.reddit.com/hot"_L1;
static constexpr auto authorizationUrl = "https://www.reddit.com/api/v1/authorize"_L1;
static constexpr auto accessTokenUrl = "https://www.reddit.com/api/v1/access_token"_L1;

RedditModel::RedditModel(QObject *parent) : QAbstractTableModel(parent) {}

RedditModel::RedditModel(const QString &clientId, QObject *parent) :
    QAbstractTableModel(parent)
{
    QNetworkAccessManager *qnam = new QNetworkAccessManager(this);
    network = new QRestAccessManager(qnam, qnam);

    redditApi.setBaseUrl(QUrl(hotUrl));

    auto replyHandler = new QOAuthHttpServerReplyHandler(QHostAddress::Any, 1337, this);
    oauth2.setReplyHandler(replyHandler);
    oauth2.setAuthorizationUrl(QUrl(authorizationUrl));
    oauth2.setTokenUrl(QUrl(accessTokenUrl));
    oauth2.setRequestedScopeTokens({"identity", "read"});
    oauth2.setClientIdentifier(clientId);

    QObject::connect(&oauth2, &QAbstractOAuth::granted, this, [this] {
        redditApi.setBearerToken(oauth2.token().toLatin1());
        updateHotThreads();
    });
    connect(&oauth2, &QOAuth2AuthorizationCodeFlow::authorizeWithBrowser, this,
            &QDesktopServices::openUrl);
    oauth2.grant();
}

int RedditModel::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    return threads.size();
}

int RedditModel::columnCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    return threads.size() ? 1 : 0;
}

QVariant RedditModel::data(const QModelIndex &index, int role) const
{
    Q_UNUSED(role);
    if (!index.isValid())
        return QVariant();

    if (role == Qt::DisplayRole) {
        const auto childrenObject = threads.at(index.row());
        Q_ASSERT(childrenObject.value("data"_L1).isObject());
        const auto dataObject = childrenObject.value("data"_L1).toObject();
        return dataObject.value("title"_L1).toString();
    }
    return QVariant();
}

void RedditModel::updateHotThreads()
{
    network->get(redditApi.createRequest(), this, [this](QRestReply &reply) {
        if (!reply.isSuccess()) {
            emit error(reply.errorString());
            return;
        }
        const auto document = reply.readJson();
        Q_ASSERT(document && document->isObject());
        const auto rootObject = document->object();
        Q_ASSERT(rootObject.value("kind"_L1).toString() == "Listing"_L1);
        const auto dataValue = rootObject.value("data"_L1);
        Q_ASSERT(dataValue.isObject());
        const auto dataObject = dataValue.toObject();
        const auto childrenValue = dataObject.value("children"_L1);
        Q_ASSERT(childrenValue.isArray());
        const auto childrenArray = childrenValue.toArray();

        if (childrenArray.isEmpty())
            return;

        beginInsertRows(QModelIndex(), threads.size(), childrenArray.size() + threads.size() - 1);
        for (const auto childValue : std::as_const(childrenArray)) {
            Q_ASSERT(childValue.isObject());
            threads.append(childValue.toObject());
        }
        endInsertRows();
    });
}