diff options
Diffstat (limited to 'plugins/CloudFile/src/Services/yandex_service.cpp')
| -rw-r--r-- | plugins/CloudFile/src/Services/yandex_service.cpp | 219 |
1 files changed, 219 insertions, 0 deletions
diff --git a/plugins/CloudFile/src/Services/yandex_service.cpp b/plugins/CloudFile/src/Services/yandex_service.cpp new file mode 100644 index 0000000000..03f34a432e --- /dev/null +++ b/plugins/CloudFile/src/Services/yandex_service.cpp @@ -0,0 +1,219 @@ +#include "..\stdafx.h" +#include "yandex_api.h" + +CYandexService::CYandexService(HNETLIBUSER hConnection) + : CCloudService(hConnection) +{ +} + +const char* CYandexService::GetModule() const +{ + return "Yandex"; +} + +const wchar_t* CYandexService::GetText() const +{ + return L"Яндекс.Диск"; +} + +HANDLE CYandexService::GetIcon() const +{ + return NULL; +} + +bool CYandexService::IsLoggedIn() +{ + ptrA token(db_get_sa(NULL, GetModule(), "TokenSecret")); + return token != NULL; +} + +void CYandexService::Login() +{ + COAuthDlg(this, YANDEX_OAUTH "/authorize?response_type=code&client_id=" YANDEX_APP_ID, RequestAccessTokenThread).DoModal(); +} + +void CYandexService::Logout() +{ + mir_forkthreadex(RevokeAccessTokenThread, this); +} + +unsigned CYandexService::RequestAccessTokenThread(void *owner, void *param) +{ + HWND hwndDlg = (HWND)param; + CYandexService *service = (CYandexService*)owner; + + if (service->IsLoggedIn()) + service->Logout(); + + char requestToken[128]; + GetDlgItemTextA(hwndDlg, IDC_OAUTH_CODE, requestToken, _countof(requestToken)); + + YandexAPI::GetAccessTokenRequest request(requestToken); + NLHR_PTR response(request.Send(service->hConnection)); + + if (response == NULL || response->resultCode != HTTP_CODE_OK) { + const char *error = response->dataLength + ? response->pData + : service->HttpStatusToError(response->resultCode); + + Netlib_Logf(service->hConnection, "%s: %s", service->GetModule(), error); + ShowNotification(TranslateT("server does not respond"), MB_ICONERROR); + return 0; + } + + JSONNode root = JSONNode::parse(response->pData); + if (root.empty()) { + Netlib_Logf(service->hConnection, "%s: %s", service->GetModule(), service->HttpStatusToError(response->resultCode)); + ShowNotification(TranslateT("server does not respond"), MB_ICONERROR); + return 0; + } + + JSONNode node = root.at("error_description"); + if (!node.isnull()) { + ptrW error_description(mir_a2u_cp(node.as_string().c_str(), CP_UTF8)); + Netlib_Logf(service->hConnection, "%s: %s", service->GetModule(), service->HttpStatusToError(response->resultCode)); + ShowNotification((wchar_t*)error_description, MB_ICONERROR); + return 0; + } + + node = root.at("access_token"); + db_set_s(NULL, service->GetModule(), "TokenSecret", node.as_string().c_str()); + ProtoBroadcastAck(MODULE, NULL, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)ID_STATUS_OFFLINE, (WPARAM)ID_STATUS_ONLINE); + + SetDlgItemTextA(hwndDlg, IDC_OAUTH_CODE, ""); + + EndDialog(hwndDlg, 1); + + return 0; +} + +unsigned CYandexService::RevokeAccessTokenThread(void *param) +{ + CYandexService *service = (CYandexService*)param; + + ptrA token(db_get_sa(NULL, service->GetModule(), "TokenSecret")); + YandexAPI::RevokeAccessTokenRequest request(token); + NLHR_PTR response(request.Send(service->hConnection)); + + return 0; +} + +void CYandexService::HandleJsonError(JSONNode &node) +{ + JSONNode error = node.at("error"); + if (!error.isnull()) { + json_string tag = error.at(".tag").as_string(); + throw Exception(tag.c_str()); + } +} + +void CYandexService::GetUploadUrl(char *path, char *url) +{ + ptrA token(db_get_sa(NULL, GetModule(), "TokenSecret")); + YandexAPI::GetUploadUrlRequest request(token, path); + NLHR_PTR response(request.Send(hConnection)); + + JSONNode root = GetJsonResponse(response); + JSONNode node = root.at("href"); + mir_strcpy(url, node.as_string().c_str()); +} + +void CYandexService::UploadFile(const char *url, const char *data, size_t size) +{ + ptrA token(db_get_sa(NULL, GetModule(), "TokenSecret")); + YandexAPI::UploadFileRequest request(token, url, data, size); + NLHR_PTR response(request.Send(hConnection)); + + if (response == NULL) + throw Exception(HttpStatusToError()); + + if (response->resultCode >= HTTP_CODE_OK && + response->resultCode <= HTTP_CODE_MULTIPLE_CHOICES) { + return; + } + + if (response->dataLength) + throw Exception(response->pData); + throw Exception(HttpStatusToError(response->resultCode)); +} + +void CYandexService::CreateFolder(const char *path) +{ + ptrA token(db_get_sa(NULL, GetModule(), "TokenSecret")); + YandexAPI::CreateFolderRequest request(token, path); + NLHR_PTR response(request.Send(hConnection)); + + GetJsonResponse(response); +} + +void CYandexService::CreateSharedLink(const char *path, char *url) +{ + ptrA token(db_get_sa(NULL, GetModule(), "TokenSecret")); + YandexAPI::PublishRequest request(token, path); + NLHR_PTR response(request.Send(hConnection)); + + JSONNode root = GetJsonResponse(response); + JSONNode link = root.at("href"); + mir_strcpy(url, link.as_string().c_str()); +} + +UINT CYandexService::Upload(FileTransferParam *ftp) +{ + if (!IsLoggedIn()) + Login(); + + if (!IsLoggedIn()) { + ftp->SetStatus(ACKRESULT_FAILED); + return ACKRESULT_FAILED; + } + + try { + const wchar_t *folderName = ftp->GetFolderName(); + if (folderName) { + char path[MAX_PATH], link[MAX_PATH]; + PreparePath(folderName, path); + CreateFolder(path); + CreateSharedLink(path, link); + ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(link))); + } + + ftp->FirstFile(); + do + { + const wchar_t *fileName = ftp->GetCurrentRelativeFilePath(); + uint64_t fileSize = ftp->GetCurrentFileSize(); + + char path[MAX_PATH]; + const wchar_t *serverFolder = ftp->GetServerFolder(); + if (serverFolder) { + wchar_t serverPath[MAX_PATH] = { 0 }; + mir_snwprintf(serverPath, L"%s\\%s", serverFolder, fileName); + PreparePath(serverPath, path); + } + else + PreparePath(fileName, path); + char url[MAX_PATH]; + GetUploadUrl(path, url); + + mir_ptr<char>data((char*)mir_calloc(fileSize)); + size_t size = ftp->ReadCurrentFile(data, fileSize); + UploadFile(url, data, size); + + ftp->Progress(size); + + if (!wcschr(fileName, L'\\')) { + char url[MAX_PATH]; + CreateSharedLink(path, url); + ftp->AppendFormatData(L"%s\r\n", ptrW(mir_utf8decodeW(url))); + } + } while (ftp->NextFile()); + } + catch (Exception &ex) { + Netlib_Logf(hConnection, "%s: %s", MODULE, ex.what()); + ftp->SetStatus(ACKRESULT_FAILED); + return ACKRESULT_FAILED; + } + + ftp->SetStatus(ACKRESULT_SUCCESS); + return ACKRESULT_SUCCESS; +} |