1 files changed, 153 insertions, 0 deletions
diff --git a/Src/Plugins/Library/ml_wire/DownloadStatus.cpp b/Src/Plugins/Library/ml_wire/DownloadStatus.cpp
new file mode 100644
index 00000000..3ccaf419
--- /dev/null
+++ b/Src/Plugins/Library/ml_wire/DownloadStatus.cpp
@@ -0,0 +1,153 @@
+#include "main.h"
+#include "api__ml_wire.h"
+#include "DownloadStatus.h"
+#include "DownloadsDialog.h"
+#include "./navigation.h"
+
+#include <strsafe.h>
+
+DownloadStatus downloadStatus;
+
+using namespace Nullsoft::Utility;
+
+DownloadStatus::Status::Status() 
+{
+	Init();
+}
+
+DownloadStatus::Status::Status( size_t _downloaded, size_t _maxSize, const wchar_t *_channel, const  wchar_t *_item, const wchar_t *_path )
+{
+	Init();
+
+	downloaded = _downloaded;
+	maxSize    = _maxSize;
+	channel    = _wcsdup( _channel );
+	item       = _wcsdup( _item );
+	path       = _wcsdup( _path );
+}
+
+const DownloadStatus::Status &DownloadStatus::Status::operator =( const DownloadStatus::Status &copy )
+{
+	Reset();
+	Init();
+
+	downloaded = copy.downloaded;
+	maxSize    = copy.maxSize;
+	channel    = _wcsdup( copy.channel );
+	item       = _wcsdup( copy.item );
+	path       = _wcsdup( copy.path );
+	killswitch = copy.killswitch;
+
+	return *this;
+}
+
+DownloadStatus::Status::~Status()
+{
+	Reset();
+}
+
+void DownloadStatus::Status::Init()
+{
+	downloaded = 0;
+	maxSize    = 0;
+	killswitch = 0;
+	channel    = 0;
+	item       = 0;
+	path       = 0;
+}
+
+void DownloadStatus::Status::Reset()
+{
+	if ( channel )
+	{
+		free( channel );
+		channel = 0;
+	}
+
+	if ( item )
+	{
+		free( item );
+		item = 0;
+	}
+
+	if ( path )
+	{
+		free( path );
+		path = 0;
+	}
+}
+
+void DownloadStatus::AddDownloadThread(DownloadToken token, const wchar_t *channel, const wchar_t *item, const wchar_t *path)
+{
+	{
+		AutoLock lock(statusLock);
+		downloads[token] = Status(0,0,channel,item,path);
+		DownloadsUpdated(downloads[token],token);
+	}
+
+	Navigation_ShowService(SERVICE_DOWNLOADS, SHOWMODE_AUTO);
+}
+
+void DownloadStatus::DownloadThreadDone(DownloadToken token)
+{
+	{
+		AutoLock lock(statusLock);
+		downloads.erase(token);
+	}
+
+	Navigation_ShowService(SERVICE_DOWNLOADS, SHOWMODE_AUTO);
+}
+
+bool DownloadStatus::UpdateStatus(DownloadToken token, size_t downloaded, size_t maxSize)
+{
+	AutoLock lock(statusLock);
+	downloads[token].downloaded = downloaded;
+	downloads[token].maxSize    = maxSize;
+
+	return !!downloads[token].killswitch;
+}
+
+bool DownloadStatus::CurrentlyDownloading()
+{
+	AutoLock lock(statusLock);
+	return !downloads.empty();
+}
+
+void DownloadStatus::GetStatusString( wchar_t *status, size_t len )
+{
+	AutoLock lock( statusLock );
+	Downloads::iterator itr;
+	size_t bytesDownloaded = 0, bytesTotal = 0, numDownloads = 0;
+	bool unknownTotal = false;
+	for ( itr = downloads.begin(); itr != downloads.end(); itr++ )
+	{
+		Status &dlstatus = itr->second;
+		if ( dlstatus.maxSize )
+		{
+			numDownloads++;
+			bytesDownloaded += dlstatus.downloaded;
+			bytesTotal      += dlstatus.maxSize;
+		}
+		else // don't have a max size
+		{
+			if ( dlstatus.downloaded ) // if we've downloaded some then we just don't know the total
+			{
+				unknownTotal = true;
+				numDownloads++;
+				bytesDownloaded += dlstatus.downloaded;
+			}
+		}
+	}
+
+	if ( 0 == numDownloads )
+	{
+		status[ 0 ] = L'\0';
+	}
+	else
+	{
+		if ( unknownTotal || bytesTotal == 0 )
+			StringCchPrintf( status, len, WASABI_API_LNGSTRINGW( IDS_DOWNLOADING_KB_COMPLETE ), numDownloads, bytesDownloaded / 1024);
+		else
+			StringCchPrintf( status, len, WASABI_API_LNGSTRINGW( IDS_DOWNLOADING_KB_PROGRESS ), numDownloads, bytesDownloaded / 1024, bytesTotal / 1024, MulDiv( (int)bytesDownloaded, 100, (int)bytesTotal ) );
+	}
+}