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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/* ScummVM - Graphic Adventure Engine
*
* ScummVM is the legal property of its developers, whose names
* are too numerous to list here. Please refer to the COPYRIGHT
* file distributed with this source distribution.
*
* 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
*/
#define FORBIDDEN_SYMBOL_ALLOW_ALL
#include "backends/cloud/onedrive/onedrivetokenrefresher.h"
#include "backends/cloud/onedrive/onedrivestorage.h"
#include "backends/networking/curl/connectionmanager.h"
#include "backends/networking/curl/curljsonrequest.h"
#include "common/config-manager.h"
#include "common/debug.h"
#include "common/json.h"
#include <curl/curl.h>
namespace Cloud {
namespace OneDrive {
OneDriveTokenRefresher::OneDriveTokenRefresher(OneDriveStorage *parent, Networking::JsonCallback callback, const char *url):
CurlJsonRequest(0, url),
_parentStorage(parent),
_innerRequest(
new CurlJsonRequest(
new Common::Callback<OneDriveTokenRefresher, Networking::RequestJsonPair>(this, &OneDriveTokenRefresher::innerRequestCallback),
url
)
), _jsonCallback(callback), _retryId(-1), _started(false) {}
OneDriveTokenRefresher::~OneDriveTokenRefresher() {}
void OneDriveTokenRefresher::innerRequestCallback(Networking::RequestJsonPair pair) {
if (!pair.value) {
//notify user of failure
warning("OneDriveTokenRefresher: got NULL instead of JSON");
ConnMan.getRequestInfo(_id).state = Networking::FINISHED;
if (_jsonCallback) (*_jsonCallback)(Networking::RequestJsonPair(_id, 0));
return;
}
Common::JSONObject result = pair.value->asObject();
if (result.contains("error")) {
//new token needed => request token & then retry original request
ConnMan.getRequestInfo(pair.id).state = Networking::PAUSED;
_retryId = pair.id;
delete pair.value;
_parentStorage->getAccessToken(new Common::Callback<OneDriveTokenRefresher, Storage::RequestBoolPair>(this, &OneDriveTokenRefresher::tokenRefreshed));
return;
}
//notify user of success
ConnMan.getRequestInfo(_id).state = Networking::FINISHED;
if (_jsonCallback) (*_jsonCallback)(Networking::RequestJsonPair(_id, pair.value));
}
void OneDriveTokenRefresher::tokenRefreshed(Storage::RequestBoolPair pair) {
if (!pair.value) {
//failed to refresh token, notify user with NULL in original callback
warning("OneDriveTokenRefresher: failed to refresh token");
ConnMan.getRequestInfo(_id).state = Networking::FINISHED;
if (_jsonCallback) (*_jsonCallback)(Networking::RequestJsonPair(_id, 0));
return;
}
//successfully received refreshed token, can restart the original request now
Networking::RequestInfo &info = ConnMan.getRequestInfo(_retryId);
info.state = Networking::RETRY;
info.retryInSeconds = 1;
//update headers: first change header with token, then pass those to request
for (uint32 i = 0; i < _headers.size(); ++i) {
if (_headers[i].contains("Authorization: bearer ")) {
_headers[i] = "Authorization: bearer " + _parentStorage->accessToken();
}
}
CurlJsonRequest *retryRequest = (CurlJsonRequest *)info.request;
if (retryRequest) retryRequest->setHeaders(_headers);
}
void OneDriveTokenRefresher::handle() {
if (!_started) {
for (uint32 i = 0; i < _headers.size(); ++i)
_innerRequest->addHeader(_headers[i]);
_started = true;
ConnMan.addRequest(_innerRequest);
}
}
void OneDriveTokenRefresher::restart() {
//can't restart as all headers were passed to _innerRequest which is probably dead now
warning("OneDriveTokenRefresher: cannot be restarted");
ConnMan.getRequestInfo(_id).state = Networking::FINISHED;
if (_jsonCallback) (*_jsonCallback)(Networking::RequestJsonPair(_id, 0));
}
Networking::NetworkReadStream *OneDriveTokenRefresher::execute() {
if (!_started) {
for (uint32 i = 0; i < _headers.size(); ++i)
_innerRequest->addHeader(_headers[i]);
_started = true;
} else {
warning("OneDriveTokenRefresher: inner Request is already started");
}
return _innerRequest->execute();
}
} //end of namespace OneDrive
} //end of namespace Cloud
|