winamp/Src/Plugins/Portable/pmp_wifi/SongDownloader.cpp
2024-09-24 14:54:57 +02:00

89 lines
2.1 KiB
C++

#include "api.h"
#include "SongDownloader.h"
#include "main.h"
#include "../../..\Components\wac_network\wac_network_http_receiver_api.h"
#include <strsafe.h>
SongDownloader::SongDownloader(const wchar_t *filename, HANDLE done_event, void (*callback)(void *callbackContext, wchar_t *status), void *context)
: done_event(done_event), callback(callback), context(context)
{
hFile = CreateFile(filename, GENERIC_WRITE, 0, 0, CREATE_ALWAYS, 0, 0);
content_length=0;
bytes_downloaded=0;
}
SongDownloader::~SongDownloader()
{
if (hFile != INVALID_HANDLE_VALUE)
CloseHandle(hFile);
}
void SongDownloader::OnInit(DownloadToken token)
{
api_httpreceiver *jnet = WAC_API_DOWNLOADMANAGER->GetReceiver(token);
if (jnet)
{
jnet->AddHeaderValue("X-Winamp-ID", winamp_id_str);
jnet->AddHeaderValue("X-Winamp-Name", winamp_name);
}
}
void SongDownloader::OnData(DownloadToken token, void *data, size_t datalen)
{
if (!content_length)
{
api_httpreceiver *jnet = WAC_API_DOWNLOADMANAGER->GetReceiver(token);
if (jnet)
{
const char *header = jnet->getheader("content-length");
if (header)
content_length = _strtoui64(header, 0, 10);
}
}
DWORD written = 0;
WriteFile(hFile, data, (DWORD)datalen, &written, 0);
bytes_downloaded+=written;
if (content_length && callback)
{
wchar_t status[128] = {0};
StringCbPrintf(status, sizeof(status), L"Transferring (%d%%)", (int)(100ULL * bytes_downloaded / content_length));
callback(context,status);
}
}
void SongDownloader::OnCancel(DownloadToken token)
{
wchar_t status[128] = {0};
// TODO: lang pack
StringCbCopy(status, sizeof(status), L"Cancelled");
callback(context,status);
SetEvent(done_event);
this->Release();
}
void SongDownloader::OnError(DownloadToken token, int error)
{
wchar_t status[128] = {0};
// TODO: lang pack
StringCbCopy(status, sizeof(status), L"Failed");
callback(context,status);
SetEvent(done_event);
this->Release();
}
void SongDownloader::OnFinish(DownloadToken token)
{
wchar_t status[128] = {0};
// TODO: lang pack
StringCbCopy(status, sizeof(status), L"Done");
callback(context,status);
SetEvent(done_event);
this->Release();
}