diff options
Diffstat (limited to 'Src/Plugins/Library/ml_downloads/DownloadViewCallback.cpp')
| -rw-r--r-- | Src/Plugins/Library/ml_downloads/DownloadViewCallback.cpp | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/Src/Plugins/Library/ml_downloads/DownloadViewCallback.cpp b/Src/Plugins/Library/ml_downloads/DownloadViewCallback.cpp new file mode 100644 index 00000000..ffd2ca52 --- /dev/null +++ b/Src/Plugins/Library/ml_downloads/DownloadViewCallback.cpp @@ -0,0 +1,125 @@ +#include "Main.h" + +//#include "../Components/wac_downloads/wac_downloads_download_manager.h" + +using namespace Nullsoft::Utility; + +DownloadViewCallback::DownloadViewCallback() +{} + +void DownloadViewCallback::OnInit( DownloadToken token ) +{ + // ---- Inform the download status service of our presence---- + downloadStatus.AddDownloadThread( token, WAC_API_DOWNLOADMANAGER->GetSource( token ), WAC_API_DOWNLOADMANAGER->GetTitle( token ), WAC_API_DOWNLOADMANAGER->GetLocation( token ) ); +} + +void DownloadViewCallback::OnConnect( DownloadToken token ) +{} + +void DownloadViewCallback::OnData( DownloadToken token, void *data, size_t datalen ) +{ + if ( token == NULL ) + return; + + api_httpreceiver *http = WAC_API_DOWNLOADMANAGER->GetReceiver( token ); + size_t totalSize = http->content_length(); + size_t downloaded = (size_t)WAC_API_DOWNLOADMANAGER->GetBytesDownloaded( token ); + + // if killswitch is turned on, then cancel download + downloadStatus.UpdateStatus( token, downloaded, totalSize ); + dirty++; +} + +void DownloadViewCallback::OnCancel( DownloadToken token ) +{ + api_httpreceiver *http = WAC_API_DOWNLOADMANAGER->GetReceiver( token ); + + DownloadedFile *data = new DownloadedFile( AutoWide( http->get_url() ), WAC_API_DOWNLOADMANAGER->GetLocation( token ), WAC_API_DOWNLOADMANAGER->GetSource( token ), WAC_API_DOWNLOADMANAGER->GetTitle( token ), DownloadedFile::DOWNLOAD_CANCELED, _time64( 0 ) ); + + data->totalSize = http->content_length(); + data->bytesDownloaded = (size_t)WAC_API_DOWNLOADMANAGER->GetBytesDownloaded( token ); + + { + AutoLock lock( downloadedFiles ); + downloadedFiles.downloadList.push_back( *data ); + DownloadsUpdated( token, &downloadedFiles.downloadList[ downloadedFiles.downloadList.size() - 1 ] ); + } + + downloadStatus.DownloadThreadDone( token ); + delete data; + dirty++; +} + +void DownloadViewCallback::OnError( DownloadToken token, int error ) +{ + api_httpreceiver *http = WAC_API_DOWNLOADMANAGER->GetReceiver( token ); + + DownloadedFile *data = new DownloadedFile( AutoWide( http->get_url() ), WAC_API_DOWNLOADMANAGER->GetLocation( token ), WAC_API_DOWNLOADMANAGER->GetSource( token ), WAC_API_DOWNLOADMANAGER->GetTitle( token ), DownloadedFile::DOWNLOAD_FAILURE, _time64( 0 ) ); + + data->totalSize = http->content_length(); + data->bytesDownloaded = (size_t)WAC_API_DOWNLOADMANAGER->GetBytesDownloaded( token ); + + { + AutoLock lock( downloadedFiles ); + downloadedFiles.downloadList.push_back( *data ); + DownloadsUpdated( token, &downloadedFiles.downloadList[ downloadedFiles.downloadList.size() - 1 ] ); + } + + downloadStatus.DownloadThreadDone( token ); + delete data; + dirty++; +} + +void DownloadViewCallback::OnFinish( DownloadToken token ) +{ + api_httpreceiver *http = WAC_API_DOWNLOADMANAGER->GetReceiver( token ); + + DownloadedFile *data = new DownloadedFile( AutoWide( http->get_url() ), WAC_API_DOWNLOADMANAGER->GetLocation( token ), WAC_API_DOWNLOADMANAGER->GetSource( token ), WAC_API_DOWNLOADMANAGER->GetTitle( token ), DownloadedFile::DOWNLOAD_SUCCESS, _time64( 0 ) ); + + data->totalSize = http->content_length(); + data->bytesDownloaded = (size_t)WAC_API_DOWNLOADMANAGER->GetBytesDownloaded( token ); + + { + AutoLock lock( downloadedFiles ); + downloadedFiles.downloadList.push_back( *data ); + + //AddDownloadData(*data); + + DownloadsUpdated( token, &downloadedFiles.downloadList[ downloadedFiles.downloadList.size() - 1 ] ); + } + + downloadStatus.DownloadThreadDone( token ); + delete data; + dirty++; +} + + +size_t DownloadViewCallback::AddRef() +{ + return this->_ref_count.fetch_add( 1 ); +} + +size_t DownloadViewCallback::Release() +{ + std::size_t l_ref_countr = this->_ref_count.fetch_sub( 1 ); + if ( l_ref_countr == 0 ) + delete( this ); + + return l_ref_countr; +} + +DownloadViewCallback::~DownloadViewCallback() +{} + +#define CBCLASS DownloadViewCallback +START_DISPATCH; +VCB( IFC_DOWNLOADMANAGERCALLBACK_ONFINISH, OnFinish ) +VCB( IFC_DOWNLOADMANAGERCALLBACK_ONCANCEL, OnCancel ) +VCB( IFC_DOWNLOADMANAGERCALLBACK_ONERROR, OnError ) +VCB( IFC_DOWNLOADMANAGERCALLBACK_ONDATA, OnData ) +VCB( IFC_DOWNLOADMANAGERCALLBACK_ONCONNECT, OnConnect ) +VCB( IFC_DOWNLOADMANAGERCALLBACK_ONINIT, OnInit ) +CB( ADDREF, AddRef ) +CB( RELEASE, Release ) +END_DISPATCH; +#undef CBCLASS
\ No newline at end of file |