summaryrefslogtreecommitdiff
path: root/plugins/CloudFile/src/Services
diff options
context:
space:
mode:
authoraunsane <aunsane@gmail.com>2018-03-11 21:29:25 +0300
committeraunsane <aunsane@gmail.com>2018-03-11 21:30:54 +0300
commit5532fd4e1374c15c13e203a89b7cd060c7e15499 (patch)
treec96b03c78f6c9be6dd365e8c83f30090c58e6e22 /plugins/CloudFile/src/Services
parent30b4b861bc47eab24d3236879d50791871d5c90a (diff)
CloudFile should create Miranda NG folder instead of placing files into Yandex Disk root (fixed #1184)
Diffstat (limited to 'plugins/CloudFile/src/Services')
-rw-r--r--plugins/CloudFile/src/Services/dropbox_service.cpp95
-rw-r--r--plugins/CloudFile/src/Services/dropbox_service.h12
-rw-r--r--plugins/CloudFile/src/Services/google_api.h2
-rw-r--r--plugins/CloudFile/src/Services/google_service.cpp100
-rw-r--r--plugins/CloudFile/src/Services/google_service.h10
-rw-r--r--plugins/CloudFile/src/Services/microsoft_service.cpp110
-rw-r--r--plugins/CloudFile/src/Services/microsoft_service.h10
-rw-r--r--plugins/CloudFile/src/Services/yandex_api.h8
-rw-r--r--plugins/CloudFile/src/Services/yandex_service.cpp85
-rw-r--r--plugins/CloudFile/src/Services/yandex_service.h10
10 files changed, 197 insertions, 245 deletions
diff --git a/plugins/CloudFile/src/Services/dropbox_service.cpp b/plugins/CloudFile/src/Services/dropbox_service.cpp
index a18ba4f32c..dfbdc0f799 100644
--- a/plugins/CloudFile/src/Services/dropbox_service.cpp
+++ b/plugins/CloudFile/src/Services/dropbox_service.cpp
@@ -33,7 +33,7 @@ int CDropboxService::GetIconId() const
bool CDropboxService::IsLoggedIn()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
if (!token || token[0] == 0)
return false;
return true;
@@ -98,7 +98,7 @@ unsigned CDropboxService::RevokeAccessTokenThread(void *param)
{
CDropboxService *service = (CDropboxService*)param;
- ptrA token(db_get_sa(NULL, service->GetAccountName(), "TokenSecret"));
+ ptrA token(service->getStringA("TokenSecret"));
DropboxAPI::RevokeAccessTokenRequest request(token);
NLHR_PTR response(request.Send(service->m_hConnection));
@@ -114,53 +114,50 @@ void CDropboxService::HandleJsonError(JSONNode &node)
}
}
-void CDropboxService::UploadFile(const char *data, size_t size, CMStringA &path)
+auto CDropboxService::UploadFile(const char *data, size_t size, const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
BYTE strategy = db_get_b(NULL, MODULE, "ConflictStrategy", OnConflict::REPLACE);
- DropboxAPI::UploadFileRequest request(token, path, data, size, (OnConflict)strategy);
+ DropboxAPI::UploadFileRequest request(token, path.c_str(), data, size, (OnConflict)strategy);
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- path = root["path_lower"].as_string().c_str();
+ return root["path_lower"].as_string();
}
-void CDropboxService::CreateUploadSession(const char *chunk, size_t chunkSize, CMStringA &sessionId)
+auto CDropboxService::CreateUploadSession(const char *chunk, size_t chunkSize)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
DropboxAPI::CreateUploadSessionRequest request(token, chunk, chunkSize);
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- sessionId = root["session_id"].as_string().c_str();
+ return root["session_id"].as_string();
}
-void CDropboxService::UploadFileChunk(const char *chunk, size_t chunkSize, const char *sessionId, size_t offset)
+void CDropboxService::UploadFileChunk(const std::string &sessionId, const char *chunk, size_t chunkSize, size_t offset)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- DropboxAPI::UploadFileChunkRequest request(token, sessionId, offset, chunk, chunkSize);
+ ptrA token(getStringA("TokenSecret"));
+ DropboxAPI::UploadFileChunkRequest request(token, sessionId.c_str(), offset, chunk, chunkSize);
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
}
-void CDropboxService::CommitUploadSession(const char *data, size_t size, const char *sessionId, size_t offset, CMStringA &path)
+auto CDropboxService::CommitUploadSession(const std::string &sessionId, const char *data, size_t size, size_t offset, const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
BYTE strategy = db_get_b(NULL, MODULE, "ConflictStrategy", OnConflict::REPLACE);
- DropboxAPI::CommitUploadSessionRequest request(token, sessionId, offset, path, data, size, (OnConflict)strategy);
+ DropboxAPI::CommitUploadSessionRequest request(token, sessionId.c_str(), offset, path.c_str(), data, size, (OnConflict)strategy);
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- path = root["path_lower"].as_string().c_str();
+ return root["path_lower"].as_string();
}
-void CDropboxService::CreateFolder(const char *path)
+void CDropboxService::CreateFolder(const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- DropboxAPI::CreateFolderRequest request(token, path);
+ ptrA token(getStringA("TokenSecret"));
+ DropboxAPI::CreateFolderRequest request(token, path.c_str());
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
@@ -172,10 +169,10 @@ void CDropboxService::CreateFolder(const char *path)
GetJsonResponse(response);
}
-void CDropboxService::CreateSharedLink(const char *path, CMStringA &url)
+auto CDropboxService::CreateSharedLink(const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- DropboxAPI::CreateSharedLinkRequest shareRequest(token, path);
+ ptrA token(getStringA("TokenSecret"));
+ DropboxAPI::CreateSharedLinkRequest shareRequest(token, path.c_str());
NLHR_PTR response(shareRequest.Send(m_hConnection));
if (response == nullptr)
@@ -195,22 +192,21 @@ void CDropboxService::CreateSharedLink(const char *path, CMStringA &url)
JSONNode error = root.at("error");
if (error.isnull()) {
JSONNode link = root.at("url");
- url = link.as_string().c_str();
- return;
+ return link.as_string();
}
json_string tag = error.at(".tag").as_string();
if (tag != "shared_link_already_exists")
throw Exception(tag.c_str());
- DropboxAPI::GetSharedLinkRequest getRequest(token, path);
+ DropboxAPI::GetSharedLinkRequest getRequest(token, path.c_str());
response = getRequest.Send(m_hConnection);
root = GetJsonResponse(response);
JSONNode links = root.at("links").as_array();
JSONNode link = links[(size_t)0].at("url");
- url = link.as_string().c_str();
+ return link.as_string();
}
UINT CDropboxService::Upload(FileTransferParam *ftp)
@@ -222,14 +218,11 @@ UINT CDropboxService::Upload(FileTransferParam *ftp)
if (ftp->IsFolder()) {
T2Utf folderName(ftp->GetFolderName());
- CMStringA path;
- PreparePath(folderName, path);
+ auto path = PreparePath(folderName);
CreateFolder(path);
- CMStringA link;
- CreateSharedLink(path, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(path);
+ ftp->AddSharedLink(link.c_str());
}
ftp->FirstFile();
@@ -241,31 +234,29 @@ UINT CDropboxService::Upload(FileTransferParam *ftp)
size_t chunkSize = ftp->GetCurrentFileChunkSize();
mir_ptr<char>chunk((char*)mir_calloc(chunkSize));
- CMStringA path;
- const wchar_t *serverFolder = ftp->GetServerFolder();
+ std::string path;
+ auto serverFolder = ftp->GetServerFolder();
if (serverFolder) {
char serverPath[MAX_PATH] = { 0 };
mir_snprintf(serverPath, "%s\\%s", T2Utf(serverFolder), fileName);
- PreparePath(serverPath, path);
+ path = PreparePath(serverPath);
}
- else PreparePath(fileName, path);
+ else
+ path = PreparePath(fileName);
- if (chunkSize == fileSize)
- {
+ if (chunkSize == fileSize) {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
- UploadFile(chunk, size, path);
+ path = UploadFile(chunk, size, path);
ftp->Progress(size);
}
- else
- {
+ else {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
- CMStringA sessionId;
- CreateUploadSession(chunk, size, sessionId);
+ auto sessionId = CreateUploadSession(chunk, size);
ftp->Progress(size);
@@ -275,7 +266,7 @@ UINT CDropboxService::Upload(FileTransferParam *ftp)
ftp->CheckCurrentFile();
size = ftp->ReadCurrentFile(chunk, chunkSize);
- UploadFileChunk(chunk, size, sessionId, offset);
+ UploadFileChunk(sessionId, chunk, size, offset);
offset += size;
ftp->Progress(size);
@@ -286,21 +277,19 @@ UINT CDropboxService::Upload(FileTransferParam *ftp)
? ftp->ReadCurrentFile(chunk, fileSize - offset)
: 0;
- CommitUploadSession(chunk, size, sessionId, offset, path);
+ path = CommitUploadSession(sessionId, chunk, size, offset, path);
ftp->Progress(size);
}
if (!ftp->IsFolder()) {
- CMStringA link;
- CreateSharedLink(path, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(path);
+ ftp->AddSharedLink(link.c_str());
}
} while (ftp->NextFile());
}
catch (Exception &ex) {
- Netlib_Logf(m_hConnection, "%s: %s", MODULE, ex.what());
+ debugLogA("%s: %s", GetModuleName(), ex.what());
ftp->SetStatus(ACKRESULT_FAILED);
return ACKRESULT_FAILED;
}
diff --git a/plugins/CloudFile/src/Services/dropbox_service.h b/plugins/CloudFile/src/Services/dropbox_service.h
index 6c55bf2bc7..fa481183c3 100644
--- a/plugins/CloudFile/src/Services/dropbox_service.h
+++ b/plugins/CloudFile/src/Services/dropbox_service.h
@@ -9,12 +9,12 @@ private:
void HandleJsonError(JSONNode &node) override;
- void UploadFile(const char *data, size_t size, CMStringA &path);
- void CreateUploadSession(const char *chunk, size_t chunkSize, CMStringA &sessionId);
- void UploadFileChunk(const char *chunk, size_t chunkSize, const char *sessionId, size_t offset);
- void CommitUploadSession(const char *chunk, size_t chunkSize, const char *sessionId, size_t offset, CMStringA &path);
- void CreateFolder(const char *path);
- void CreateSharedLink(const char *path, CMStringA &url);
+ auto UploadFile(const char *data, size_t size, const std::string &path);
+ auto CreateUploadSession(const char *chunk, size_t chunkSize);
+ void UploadFileChunk(const std::string &sessionId, const char *chunk, size_t chunkSize, size_t offset);
+ auto CommitUploadSession(const std::string &sessionId, const char *chunk, size_t chunkSize, size_t offset, const std::string &path);
+ void CreateFolder(const std::string &path);
+ auto CreateSharedLink(const std::string &path);
public:
CDropboxService(const char *protoName, const wchar_t *userName);
diff --git a/plugins/CloudFile/src/Services/google_api.h b/plugins/CloudFile/src/Services/google_api.h
index d67ef82012..2f5baa8eb3 100644
--- a/plugins/CloudFile/src/Services/google_api.h
+++ b/plugins/CloudFile/src/Services/google_api.h
@@ -9,7 +9,7 @@ namespace GDriveAPI
#define GDRIVE_API_VER "/v3"
#define GDRIVE_API GOOGLE_API "/drive" GDRIVE_API_VER "/files"
#define GDRIVE_UPLOAD GOOGLE_API "/upload/drive" GDRIVE_API_VER "/files"
-#define GDRIVE_SHARE "https://drive.google.com/open?id=%s"
+#define GDRIVE_SHARE "https://drive.google.com/open?id="
#define GOOGLE_APP_ID "528761318515-h1etlccvk5vjsbjuuj8i73cud8do4adi.apps.googleusercontent.com"
#include "../../../miranda-private-keys/Google/client_secret.h"
diff --git a/plugins/CloudFile/src/Services/google_service.cpp b/plugins/CloudFile/src/Services/google_service.cpp
index dcf4e0725d..dde5fc4b39 100644
--- a/plugins/CloudFile/src/Services/google_service.cpp
+++ b/plugins/CloudFile/src/Services/google_service.cpp
@@ -33,18 +33,18 @@ int CGDriveService::GetIconId() const
bool CGDriveService::IsLoggedIn()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
if (!token || token[0] == 0)
return false;
time_t now = time(nullptr);
- time_t expiresIn = db_get_dw(NULL, GetAccountName(), "ExpiresIn");
+ time_t expiresIn = getDword("ExpiresIn");
return now < expiresIn;
}
void CGDriveService::Login()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- ptrA refreshToken(db_get_sa(NULL, GetAccountName(), "RefreshToken"));
+ ptrA token(getStringA("TokenSecret"));
+ ptrA refreshToken(getStringA("RefreshToken"));
if (token && refreshToken && refreshToken[0]) {
GDriveAPI::RefreshTokenRequest request(refreshToken);
NLHR_PTR response(request.Send(m_hConnection));
@@ -52,11 +52,11 @@ void CGDriveService::Login()
JSONNode root = GetJsonResponse(response);
JSONNode node = root.at("access_token");
- db_set_s(NULL, GetAccountName(), "TokenSecret", node.as_string().c_str());
+ setString("TokenSecret", node.as_string().c_str());
node = root.at("expires_in");
time_t expiresIn = time(nullptr) + node.as_int();
- db_set_dw(NULL, GetAccountName(), "ExpiresIn", expiresIn);
+ setDword("ExpiresIn", expiresIn);
return;
}
@@ -146,21 +146,19 @@ void CGDriveService::HandleJsonError(JSONNode &node)
}
}
-void CGDriveService::UploadFile(const char *parentId, const char *name, const char *data, size_t size, CMStringA &fileId)
+auto CGDriveService::UploadFile(const std::string &parentId, const std::string &fileName, const char *data, size_t size)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- GDriveAPI::UploadFileRequest request(token, parentId, name, data, size);
+ ptrA token(getStringA("TokenSecret"));
+ GDriveAPI::UploadFileRequest request(token, parentId.c_str(), fileName.c_str(), data, size);
NLHR_PTR response(request.Send(m_hConnection));
-
JSONNode root = GetJsonResponse(response);
- if (root)
- fileId = root["id"].as_string().c_str();
+ return root["id"].as_string();
}
-void CGDriveService::CreateUploadSession(const char *parentId, const char *name, CMStringA &uploadUri)
+auto CGDriveService::CreateUploadSession(const std::string &parentId, const std::string &fileName)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- GDriveAPI::CreateUploadSessionRequest request(token, parentId, name);
+ ptrA token(getStringA("TokenSecret"));
+ GDriveAPI::CreateUploadSessionRequest request(token, parentId.c_str(), fileName.c_str());
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
@@ -169,57 +167,56 @@ void CGDriveService::CreateUploadSession(const char *parentId, const char *name,
for (int i = 0; i < response->headersCount; i++) {
if (mir_strcmpi(response->headers[i].szName, "Location"))
continue;
-
- uploadUri = response->headers[i].szValue;
- return;
+ return std::string(response->headers[i].szValue);
}
}
HttpResponseToError(response);
}
-void CGDriveService::UploadFileChunk(const char *uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize, CMStringA &fileId)
+auto CGDriveService::UploadFileChunk(const std::string &uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize)
{
- GDriveAPI::UploadFileChunkRequest request(uploadUri, chunk, chunkSize, offset, fileSize);
+ GDriveAPI::UploadFileChunkRequest request(uploadUri.c_str(), chunk, chunkSize, offset, fileSize);
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
if (response->resultCode == HTTP_CODE_PERMANENT_REDIRECT)
- return;
+ return std::string();
if (HTTP_CODE_SUCCESS(response->resultCode)) {
JSONNode root = GetJsonResponse(response);
- if (root)
- fileId = root["id"].as_string().c_str();
+ return root["id"].as_string();
}
- else HttpResponseToError(response);
+
+ HttpResponseToError(response);
}
-void CGDriveService::CreateFolder(const char *path, CMStringA &folderId)
+auto CGDriveService::CreateFolder(const char *path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
GDriveAPI::CreateFolderRequest request(token, path);
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- folderId = root["id"].as_string().c_str();
+ return root["id"].as_string();
}
-void CGDriveService::CreateSharedLink(const char *itemId, CMStringA &url)
+auto CGDriveService::CreateSharedLink(const std::string &itemId)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- GDriveAPI::GrantPermissionsRequest request(token, itemId);
+ ptrA token(getStringA("TokenSecret"));
+ GDriveAPI::GrantPermissionsRequest request(token, itemId.c_str());
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
if (HTTP_CODE_SUCCESS(response->resultCode)) {
- CMStringA sharedUrl(CMStringDataFormat::FORMAT, GDRIVE_SHARE, itemId);
- url = sharedUrl;
+ std::string url = GDRIVE_SHARE;
+ url += itemId;
+ return url;
}
- else HttpResponseToError(response);
+
+ HttpResponseToError(response);
}
UINT CGDriveService::Upload(FileTransferParam *ftp)
@@ -233,61 +230,50 @@ UINT CGDriveService::Upload(FileTransferParam *ftp)
return ACKRESULT_FAILED;
}
- CMStringA folderId;
+ std::string folderId;
if (ftp->IsFolder()) {
- CreateFolder(T2Utf(ftp->GetFolderName()), folderId);
-
- CMStringA link;
- CreateSharedLink(folderId, link);
-
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ folderId = CreateFolder(T2Utf(ftp->GetFolderName()));
+ auto link = CreateSharedLink(folderId);
+ ftp->AddSharedLink(link.c_str());
}
ftp->FirstFile();
do {
- CMStringA fileName(T2Utf(ftp->GetCurrentRelativeFilePath()).get());
+ std::string fileName = T2Utf(ftp->GetCurrentRelativeFilePath()).get();
uint64_t fileSize = ftp->GetCurrentFileSize();
size_t chunkSize = ftp->GetCurrentFileChunkSize();
mir_ptr<char>chunk((char*)mir_calloc(chunkSize));
- CMStringA fileId;
+ std::string fileId;
if (chunkSize == fileSize) {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
-
- UploadFile(folderId, fileName, chunk, size, fileId);
-
+ fileId = UploadFile(folderId, fileName, chunk, size);
ftp->Progress(size);
}
else {
- CMStringA uploadUri;
- CreateUploadSession(uploadUri, folderId, fileName);
+ auto uploadUri = CreateUploadSession(folderId, fileName);
uint64_t offset = 0;
double chunkCount = ceil(double(fileSize) / chunkSize);
while (chunkCount > 0) {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
-
- UploadFileChunk(uploadUri, chunk, size, offset, fileSize, fileId);
-
+ auto fileId = UploadFileChunk(uploadUri, chunk, size, offset, fileSize);
offset += size;
ftp->Progress(size);
}
}
if (!ftp->IsFolder()) {
- CMStringA link;
- CreateSharedLink(fileId, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(fileId);
+ ftp->AddSharedLink(link.c_str());
}
} while (ftp->NextFile());
}
catch (Exception &ex) {
- Netlib_Logf(m_hConnection, "%s: %s", MODULE, ex.what());
+ debugLogA("%s: %s", GetAccountName(), ex.what());
ftp->SetStatus(ACKRESULT_FAILED);
return ACKRESULT_FAILED;
}
diff --git a/plugins/CloudFile/src/Services/google_service.h b/plugins/CloudFile/src/Services/google_service.h
index 3bb4466e57..9da2f14e5f 100644
--- a/plugins/CloudFile/src/Services/google_service.h
+++ b/plugins/CloudFile/src/Services/google_service.h
@@ -9,11 +9,11 @@ private:
void HandleJsonError(JSONNode &node) override;
- void UploadFile(const char *parentId, const char *name, const char *data, size_t size, CMStringA &fileId);
- void CreateUploadSession(const char *parentId, const char *name, CMStringA &uploadUri);
- void UploadFileChunk(const char *uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize, CMStringA &fileId);
- void CreateFolder(const char *path, CMStringA &folderId);
- void CreateSharedLink(const char *itemId, CMStringA &url);
+ auto UploadFile(const std::string &parentId, const std::string &fileName, const char *data, size_t size);
+ auto CreateUploadSession(const std::string &parentId, const std::string &fileName);
+ auto UploadFileChunk(const std::string &uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize);
+ auto CreateFolder(const char *path);
+ auto CreateSharedLink(const std::string &itemId);
public:
CGDriveService(const char *protoName, const wchar_t *userName);
diff --git a/plugins/CloudFile/src/Services/microsoft_service.cpp b/plugins/CloudFile/src/Services/microsoft_service.cpp
index 53b3a8574e..af4ebe2439 100644
--- a/plugins/CloudFile/src/Services/microsoft_service.cpp
+++ b/plugins/CloudFile/src/Services/microsoft_service.cpp
@@ -33,18 +33,18 @@ int COneDriveService::GetIconId() const
bool COneDriveService::IsLoggedIn()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
if (!token || token[0] == 0)
return false;
time_t now = time(nullptr);
- time_t expiresIn = db_get_dw(NULL, GetAccountName(), "ExpiresIn");
+ time_t expiresIn = getWord("ExpiresIn");
return now < expiresIn;
}
void COneDriveService::Login()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- ptrA refreshToken(db_get_sa(NULL, GetAccountName(), "RefreshToken"));
+ ptrA token(getStringA("TokenSecret"));
+ ptrA refreshToken(getStringA("RefreshToken"));
if (token && refreshToken && refreshToken[0]) {
OneDriveAPI::RefreshTokenRequest request(refreshToken);
NLHR_PTR response(request.Send(m_hConnection));
@@ -56,7 +56,7 @@ void COneDriveService::Login()
node = root.at("expires_in");
time_t expiresIn = time(nullptr) + node.as_int();
- db_set_dw(NULL, GetAccountName(), "ExpiresIn", expiresIn);
+ setDword("ExpiresIn", expiresIn);
return;
}
@@ -67,9 +67,9 @@ void COneDriveService::Login()
void COneDriveService::Logout()
{
- db_unset(NULL, GetAccountName(), "TokenSecret");
- db_unset(NULL, GetAccountName(), "ExpiresIn");
- db_unset(NULL, GetAccountName(), "RefreshToken");
+ delSetting("ExpiresIn");
+ delSetting("TokenSecret");
+ delSetting("RefreshToken");
}
unsigned COneDriveService::RequestAccessTokenThread(void *owner, void *param)
@@ -112,14 +112,14 @@ unsigned COneDriveService::RequestAccessTokenThread(void *owner, void *param)
}
node = root.at("access_token");
- db_set_s(NULL, service->GetAccountName(), "TokenSecret", node.as_string().c_str());
+ service->setString("TokenSecret", node.as_string().c_str());
node = root.at("expires_in");
time_t expiresIn = time(nullptr) + node.as_int();
- db_set_dw(NULL, service->GetAccountName(), "ExpiresIn", expiresIn);
+ service->setDword("ExpiresIn", expiresIn);
node = root.at("refresh_token");
- db_set_s(NULL, service->GetAccountName(), "RefreshToken", node.as_string().c_str());
+ service->setString("RefreshToken", node.as_string().c_str());
SetDlgItemTextA(hwndDlg, IDC_OAUTH_CODE, "");
@@ -137,72 +137,70 @@ void COneDriveService::HandleJsonError(JSONNode &node)
}
}
-void COneDriveService::UploadFile(const char *parentId, const char *name, const char *data, size_t size, CMStringA &fileId)
+auto COneDriveService::UploadFile(const std::string &parentId, const std::string &fileName, const char *data, size_t size)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
BYTE strategy = db_get_b(NULL, MODULE, "ConflictStrategy", OnConflict::REPLACE);
- OneDriveAPI::UploadFileRequest *request = mir_strlen(parentId)
- ? new OneDriveAPI::UploadFileRequest(token, parentId, name, data, size, (OnConflict)strategy)
- : new OneDriveAPI::UploadFileRequest(token, name, data, size, (OnConflict)strategy);
+ OneDriveAPI::UploadFileRequest *request = !parentId.empty()
+ ? new OneDriveAPI::UploadFileRequest(token, parentId.c_str(), fileName.c_str(), data, size, (OnConflict)strategy)
+ : new OneDriveAPI::UploadFileRequest(token, fileName.c_str(), data, size, (OnConflict)strategy);
NLHR_PTR response(request->Send(m_hConnection));
delete request;
JSONNode root = GetJsonResponse(response);
- fileId = root["id"].as_string().c_str();
+ return root["id"].as_string();
}
-void COneDriveService::CreateUploadSession(const char *parentId, const char *name, CMStringA &uploadUri)
+auto COneDriveService::CreateUploadSession(const std::string &parentId, const std::string &fileName)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
BYTE strategy = db_get_b(NULL, MODULE, "ConflictStrategy", OnConflict::REPLACE);
- OneDriveAPI::CreateUploadSessionRequest *request = mir_strlen(parentId)
- ? new OneDriveAPI::CreateUploadSessionRequest(token, parentId, name, (OnConflict)strategy)
- : new OneDriveAPI::CreateUploadSessionRequest(token, name, (OnConflict)strategy);
+ OneDriveAPI::CreateUploadSessionRequest *request = !parentId.empty()
+ ? new OneDriveAPI::CreateUploadSessionRequest(token, parentId.c_str(), fileName.c_str(), (OnConflict)strategy)
+ : new OneDriveAPI::CreateUploadSessionRequest(token, fileName.c_str(), (OnConflict)strategy);
NLHR_PTR response(request->Send(m_hConnection));
delete request;
JSONNode root = GetJsonResponse(response);
- uploadUri = root["uploadUrl"].as_string().c_str();
+ return root["uploadUrl"].as_string();
}
-void COneDriveService::UploadFileChunk(const char *uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize, CMStringA &fileId)
+auto COneDriveService::UploadFileChunk(const std::string &uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize)
{
- OneDriveAPI::UploadFileChunkRequest request(uploadUri, chunk, chunkSize, offset, fileSize);
+ OneDriveAPI::UploadFileChunkRequest request(uploadUri.c_str(), chunk, chunkSize, offset, fileSize);
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
if (response->resultCode == HTTP_CODE_ACCEPTED)
- return;
+ return std::string();
if (HTTP_CODE_SUCCESS(response->resultCode)) {
JSONNode root = GetJsonResponse(response);
- if (root)
- fileId = root["id"].as_string().c_str();
+ return root["id"].as_string();
}
- else HttpResponseToError(response);
+
+ HttpResponseToError(response);
}
-void COneDriveService::CreateFolder(const char *path, CMStringA &folderId)
+auto COneDriveService::CreateFolder(const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- OneDriveAPI::CreateFolderRequest request(token, path);
+ ptrA token(getStringA("TokenSecret"));
+ OneDriveAPI::CreateFolderRequest request(token, path.c_str());
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- folderId = root["id"].as_string().c_str();
+ return root["id"].as_string();
}
-void COneDriveService::CreateSharedLink(const char *itemId, CMStringA &url)
+auto COneDriveService::CreateSharedLink(const std::string &itemId)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- OneDriveAPI::CreateSharedLinkRequest request(token, itemId);
+ ptrA token(getStringA("TokenSecret"));
+ OneDriveAPI::CreateSharedLinkRequest request(token, itemId.c_str());
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- url = root["link"]["webUrl"].as_string().c_str();
+ return root["link"]["webUrl"].as_string();
}
UINT COneDriveService::Upload(FileTransferParam *ftp)
@@ -216,40 +214,36 @@ UINT COneDriveService::Upload(FileTransferParam *ftp)
return ACKRESULT_FAILED;
}
- CMStringA folderId;
+ std::string folderId;
if (ftp->IsFolder()) {
- CMStringA folderName(T2Utf(ftp->GetFolderName()).get());
+ T2Utf folderName(ftp->GetFolderName());
- CMStringA path;
- PreparePath(folderName, path);
- CreateFolder(path, folderId);
+ auto path = PreparePath(folderName);
+ folderId = CreateFolder(path);
- CMStringA link;
- CreateSharedLink(path, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(path);
+ ftp->AddSharedLink(link.c_str());
}
ftp->FirstFile();
do {
- T2Utf fileName(ftp->GetCurrentRelativeFilePath());
+ std::string fileName = T2Utf(ftp->GetCurrentRelativeFilePath());
uint64_t fileSize = ftp->GetCurrentFileSize();
size_t chunkSize = ftp->GetCurrentFileChunkSize();
mir_ptr<char>chunk((char*)mir_calloc(chunkSize));
- CMStringA fileId;
+ std::string fileId;
if (chunkSize == fileSize) {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
- UploadFile(folderId, T2Utf(fileName), chunk, size, fileId);
+ fileId = UploadFile(folderId, fileName, chunk, size);
ftp->Progress(size);
}
else {
- char uploadUri[1024];
- CreateUploadSession(uploadUri, T2Utf(fileName), folderId);
+ auto uploadUri = CreateUploadSession(folderId, fileName);
uint64_t offset = 0;
double chunkCount = ceil(double(fileSize) / chunkSize);
@@ -257,7 +251,7 @@ UINT COneDriveService::Upload(FileTransferParam *ftp)
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
- UploadFileChunk(uploadUri, chunk, size, offset, fileSize, fileId);
+ fileId = UploadFileChunk(uploadUri, chunk, size, offset, fileSize);
offset += size;
ftp->Progress(size);
@@ -265,15 +259,13 @@ UINT COneDriveService::Upload(FileTransferParam *ftp)
}
if (!ftp->IsFolder()) {
- CMStringA link;
- CreateSharedLink(fileId, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(fileId);
+ ftp->AddSharedLink(link.c_str());
}
} while (ftp->NextFile());
}
catch (Exception &ex) {
- Netlib_Logf(m_hConnection, "%s: %s", MODULE, ex.what());
+ debugLogA("%s: %s", GetAccountName(), ex.what());
ftp->SetStatus(ACKRESULT_FAILED);
return ACKRESULT_FAILED;
}
diff --git a/plugins/CloudFile/src/Services/microsoft_service.h b/plugins/CloudFile/src/Services/microsoft_service.h
index 03c3ea2585..f9e41a93c5 100644
--- a/plugins/CloudFile/src/Services/microsoft_service.h
+++ b/plugins/CloudFile/src/Services/microsoft_service.h
@@ -8,11 +8,11 @@ private:
void HandleJsonError(JSONNode &node) override;
- void UploadFile(const char *parentId, const char *name, const char *data, size_t size, CMStringA &fileId);
- void CreateUploadSession(const char *parentId, const char *name, CMStringA &uploadUri);
- void UploadFileChunk(const char *uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize, CMStringA &fileId);
- void CreateFolder(const char *path, CMStringA &folderId);
- void CreateSharedLink(const char *itemId, CMStringA &url);
+ auto UploadFile(const std::string &parentId, const std::string &fileName, const char *data, size_t size);
+ auto CreateUploadSession(const std::string &parentId, const std::string &fileName);
+ auto UploadFileChunk(const std::string &uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize);
+ auto CreateFolder(const std::string &path);
+ auto CreateSharedLink(const std::string &itemId);
public:
COneDriveService(const char *protoName, const wchar_t *userName);
diff --git a/plugins/CloudFile/src/Services/yandex_api.h b/plugins/CloudFile/src/Services/yandex_api.h
index 6c2d97448b..19c76d61f9 100644
--- a/plugins/CloudFile/src/Services/yandex_api.h
+++ b/plugins/CloudFile/src/Services/yandex_api.h
@@ -59,7 +59,7 @@ namespace YandexAPI
HttpRequest(REQUEST_GET, YADISK_API "/upload")
{
AddOAuthHeader(token);
- AddUrlParameter("path=%s", ptrA(mir_urlEncode(path)));
+ AddUrlParameter("path=app:%s", ptrA(mir_urlEncode(path)));
if (strategy == OnConflict::REPLACE)
AddUrlParameter("overwrite=true");
}
@@ -97,7 +97,7 @@ namespace YandexAPI
HttpRequest(REQUEST_PUT, YADISK_API)
{
AddOAuthHeader(token);
- AddUrlParameter("path=%s", ptrA(mir_urlEncode(path)));
+ AddUrlParameter("path=app:%s", ptrA(mir_urlEncode(path)));
}
};
@@ -108,7 +108,7 @@ namespace YandexAPI
HttpRequest(REQUEST_PUT, YADISK_API "/publish")
{
AddOAuthHeader(token);
- AddUrlParameter("path=%s", ptrA(mir_urlEncode(path)));
+ AddUrlParameter("path=app:%s", ptrA(mir_urlEncode(path)));
}
};
@@ -119,7 +119,7 @@ namespace YandexAPI
HttpRequest(REQUEST_GET, YADISK_API)
{
AddOAuthHeader(token);
- AddUrlParameter("path=%s", ptrA(mir_urlEncode(path)));
+ AddUrlParameter("path=app:%s", ptrA(mir_urlEncode(path)));
}
};
};
diff --git a/plugins/CloudFile/src/Services/yandex_service.cpp b/plugins/CloudFile/src/Services/yandex_service.cpp
index a40dd0d862..662659f8c6 100644
--- a/plugins/CloudFile/src/Services/yandex_service.cpp
+++ b/plugins/CloudFile/src/Services/yandex_service.cpp
@@ -33,18 +33,18 @@ int CYandexService::GetIconId() const
bool CYandexService::IsLoggedIn()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
if (!token || token[0] == 0)
return false;
time_t now = time(nullptr);
- time_t expiresIn = db_get_dw(NULL, GetAccountName(), "ExpiresIn");
+ time_t expiresIn = getDword("ExpiresIn");
return now < expiresIn;
}
void CYandexService::Login()
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- ptrA refreshToken(db_get_sa(NULL, GetAccountName(), "RefreshToken"));
+ ptrA token(getStringA("TokenSecret"));
+ ptrA refreshToken(getStringA("RefreshToken"));
if (token && refreshToken && refreshToken[0]) {
YandexAPI::RefreshTokenRequest request(refreshToken);
NLHR_PTR response(request.Send(m_hConnection));
@@ -52,14 +52,14 @@ void CYandexService::Login()
JSONNode root = GetJsonResponse(response);
JSONNode node = root.at("access_token");
- db_set_s(NULL, GetAccountName(), "TokenSecret", node.as_string().c_str());
+ setString("TokenSecret", node.as_string().c_str());
node = root.at("expires_in");
time_t expiresIn = time(nullptr) + node.as_int();
- db_set_dw(NULL, GetAccountName(), "ExpiresIn", expiresIn);
+ setDword("ExpiresIn", expiresIn);
node = root.at("refresh_token");
- db_set_s(NULL, GetAccountName(), "RefreshToken", node.as_string().c_str());
+ setString("RefreshToken", node.as_string().c_str());
return;
}
@@ -113,14 +113,14 @@ unsigned CYandexService::RequestAccessTokenThread(void *owner, void *param)
}
node = root.at("access_token");
- db_set_s(NULL, service->GetAccountName(), "TokenSecret", node.as_string().c_str());
+ service->setString("TokenSecret", node.as_string().c_str());
node = root.at("expires_in");
time_t expiresIn = time(nullptr) + node.as_int();
- db_set_dw(NULL, service->GetAccountName(), "ExpiresIn", expiresIn);
+ service->setDword("ExpiresIn", expiresIn);
node = root.at("refresh_token");
- db_set_s(NULL, service->GetAccountName(), "RefreshToken", node.as_string().c_str());
+ service->setString("RefreshToken", node.as_string().c_str());
SetDlgItemTextA(hwndDlg, IDC_OAUTH_CODE, "");
@@ -148,21 +148,20 @@ void CYandexService::HandleJsonError(JSONNode &node)
}
}
-void CYandexService::CreateUploadSession(const char *path, CMStringA &uploadUri)
+auto CYandexService::CreateUploadSession(const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
+ ptrA token(getStringA("TokenSecret"));
BYTE strategy = db_get_b(NULL, MODULE, "ConflictStrategy", OnConflict::REPLACE);
- YandexAPI::GetUploadUrlRequest request(token, path, (OnConflict)strategy);
+ YandexAPI::GetUploadUrlRequest request(token, path.c_str(), (OnConflict)strategy);
NLHR_PTR response(request.Send(m_hConnection));
JSONNode root = GetJsonResponse(response);
- if (root)
- uploadUri = root["href"].as_string().c_str();
+ return root["href"].as_string();
}
-void CYandexService::UploadFile(const char *uploadUri, const char *data, size_t size)
+void CYandexService::UploadFile(const std::string &uploadUri, const char *data, size_t size)
{
- YandexAPI::UploadFileRequest request(uploadUri, data, size);
+ YandexAPI::UploadFileRequest request(uploadUri.c_str(), data, size);
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
@@ -173,9 +172,9 @@ void CYandexService::UploadFile(const char *uploadUri, const char *data, size_t
HttpResponseToError(response);
}
-void CYandexService::UploadFileChunk(const char *uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize)
+void CYandexService::UploadFileChunk(const std::string &uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize)
{
- YandexAPI::UploadFileChunkRequest request(uploadUri, chunk, chunkSize, offset, fileSize);
+ YandexAPI::UploadFileChunkRequest request(uploadUri.c_str(), chunk, chunkSize, offset, fileSize);
NLHR_PTR response(request.Send(m_hConnection));
HandleHttpError(response);
@@ -187,29 +186,28 @@ void CYandexService::UploadFileChunk(const char *uploadUri, const char *chunk, s
HttpResponseToError(response);
}
-void CYandexService::CreateFolder(const char *path)
+void CYandexService::CreateFolder(const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- YandexAPI::CreateFolderRequest request(token, path);
+ ptrA token(getStringA("TokenSecret"));
+ YandexAPI::CreateFolderRequest request(token, path.c_str());
NLHR_PTR response(request.Send(m_hConnection));
GetJsonResponse(response);
}
-void CYandexService::CreateSharedLink(const char *path, CMStringA &url)
+auto CYandexService::CreateSharedLink(const std::string &path)
{
- ptrA token(db_get_sa(NULL, GetAccountName(), "TokenSecret"));
- YandexAPI::PublishRequest publishRequest(token, path);
+ ptrA token(getStringA("TokenSecret"));
+ YandexAPI::PublishRequest publishRequest(token, path.c_str());
NLHR_PTR response(publishRequest.Send(m_hConnection));
GetJsonResponse(response);
- YandexAPI::GetResourcesRequest resourcesRequest(token, path);
+ YandexAPI::GetResourcesRequest resourcesRequest(token, path.c_str());
response = resourcesRequest.Send(m_hConnection);
JSONNode root = GetJsonResponse(response);
- if (root)
- url = root["public_url"].as_string().c_str();
+ return root["public_url"].as_string();
}
UINT CYandexService::Upload(FileTransferParam *ftp)
@@ -226,14 +224,11 @@ UINT CYandexService::Upload(FileTransferParam *ftp)
if (ftp->IsFolder()) {
T2Utf folderName(ftp->GetFolderName());
- CMStringA path;
- PreparePath(folderName, path);
+ auto path = PreparePath(folderName);
CreateFolder(path);
- CMStringA link;
- CreateSharedLink(path, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(path);
+ ftp->AddSharedLink(link.c_str());
}
ftp->FirstFile();
@@ -242,11 +237,8 @@ UINT CYandexService::Upload(FileTransferParam *ftp)
T2Utf fileName(ftp->GetCurrentRelativeFilePath());
uint64_t fileSize = ftp->GetCurrentFileSize();
- CMStringA path;
- PreparePath(fileName, path);
-
- CMStringA uploadUri;
- CreateUploadSession(path, uploadUri);
+ auto path = PreparePath(fileName);
+ auto uploadUri = CreateUploadSession(path);
size_t chunkSize = ftp->GetCurrentFileChunkSize();
mir_ptr<char>chunk((char*)mir_calloc(chunkSize));
@@ -254,36 +246,29 @@ UINT CYandexService::Upload(FileTransferParam *ftp)
if (chunkSize == fileSize) {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
-
UploadFile(uploadUri, chunk, size);
-
ftp->Progress(size);
}
- else
- {
+ else {
uint64_t offset = 0;
double chunkCount = ceil(double(fileSize) / chunkSize);
while (chunkCount > 0) {
ftp->CheckCurrentFile();
size_t size = ftp->ReadCurrentFile(chunk, chunkSize);
-
UploadFileChunk(uploadUri, chunk, size, offset, fileSize);
-
offset += size;
ftp->Progress(size);
}
}
if (!ftp->IsFolder()) {
- CMStringA link;
- CreateSharedLink(path, link);
- ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link)));
- ftp->AddSharedLink(link);
+ auto link = CreateSharedLink(path);
+ ftp->AddSharedLink(link.c_str());
}
} while (ftp->NextFile());
}
catch (Exception &ex) {
- Netlib_Logf(m_hConnection, "%s: %s", MODULE, ex.what());
+ debugLogA("%s: %s", GetAccountName(), ex.what());
ftp->SetStatus(ACKRESULT_FAILED);
return ACKRESULT_FAILED;
}
diff --git a/plugins/CloudFile/src/Services/yandex_service.h b/plugins/CloudFile/src/Services/yandex_service.h
index 2971d6beec..76a12feef0 100644
--- a/plugins/CloudFile/src/Services/yandex_service.h
+++ b/plugins/CloudFile/src/Services/yandex_service.h
@@ -9,11 +9,11 @@ private:
void HandleJsonError(JSONNode &node) override;
- void CreateUploadSession(const char *path, CMStringA &uploadUri);
- void UploadFile(const char *uploadUri, const char *data, size_t size);
- void UploadFileChunk(const char *uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize);
- void CreateFolder(const char *path);
- void CreateSharedLink(const char *path, CMStringA &url);
+ auto CreateUploadSession(const std::string &path);
+ void UploadFile(const std::string &uploadUri, const char *data, size_t size);
+ void UploadFileChunk(const std::string &uploadUri, const char *chunk, size_t chunkSize, uint64_t offset, uint64_t fileSize);
+ void CreateFolder(const std::string &path);
+ auto CreateSharedLink(const std::string &path);
public:
CYandexService(const char *protoName, const wchar_t *userName);