3.5

1 files changed, 160 insertions, 0 deletions

diff --git a/library/Taskbar_Seven/taskbar.cpp b/library/Taskbar_Seven/taskbar.cpp
new file mode 100644
index 00000000..4baf2392
--- /dev/null
+++ b/library/Taskbar_Seven/taskbar.cpp
@@ -0,0 +1,160 @@
+// **************************************************************************
+// * This file is part of the FreeFileSync project. It is distributed under *
+// * GNU General Public License: http://www.gnu.org/licenses/gpl.html       *
+// * Copyright (C) 2008-2010 ZenJu (zhnmju123 AT gmx.de)                    *
+// **************************************************************************
+//
+#include "taskbar.h"
+
+#define WIN32_LEAN_AND_MEAN
+#include "windows.h"
+#include <ShObjIdl.h>
+
+#include <map>
+#include <string>
+#include <comdef.h>
+
+
+namespace
+{
+void writeString(const std::wstring& input, wchar_t* output, size_t outputBufferLen)
+{
+    const size_t newSize = min(input.length() + 1, outputBufferLen); //including null-termination
+    memcpy(output, input.c_str(), newSize * sizeof(wchar_t));
+    output[newSize-1] = 0; //if output buffer is too small...
+}
+
+
+std::wstring numberToHexString(const long number)
+{
+    wchar_t result[100];
+    swprintf(result, 100, L"0x%08x", number);
+    return std::wstring(result);
+}
+
+
+std::wstring writeErrorMsg(const wchar_t* input, HRESULT hr)
+{
+    std::wstring output(input);
+    output += L" (";
+    output += numberToHexString(hr);
+    output += L": ";
+    output += _com_error(hr).ErrorMessage();
+    output += L")";
+    return output;
+}
+
+
+using TaskbarSeven::TBHandle;
+typedef std::map<TBHandle, ITaskbarList3*> TaskBarHandleMap;
+
+TaskbarSeven::TBHandle generateHandle()
+{
+    static TBHandle handle = 0;
+    return ++handle; //don't return 0! 0 is reserved for indicating failure
+}
+
+TaskBarHandleMap taskBarHandles;
+
+std::wstring lastErrorMessage;
+}
+//##################################################################################################
+
+
+TaskbarSeven::TBHandle TaskbarSeven::init() //call on app initializaiton; returns handle
+{
+    ITaskbarList3* pto = NULL;
+    HRESULT hr = CoCreateInstance(CLSID_TaskbarList,
+                                  NULL,
+                                  CLSCTX_ALL,
+                                  IID_PPV_ARGS(&pto));
+    if (FAILED(hr))
+    {
+        lastErrorMessage = writeErrorMsg(L"Error calling \"CoCreateInstance\".", hr);
+        return 0;
+    }
+
+    TBHandle newHandle = ::generateHandle();
+    taskBarHandles[newHandle] = pto;
+    return newHandle;
+}
+
+
+void TaskbarSeven::release(TBHandle handle) //release handle on app exit
+{
+    TaskBarHandleMap::const_iterator iter = taskBarHandles.find(handle);
+    if (iter != taskBarHandles.end())
+    {
+        if (iter->second != NULL)
+            iter->second->Release();
+        taskBarHandles.erase(iter);
+    }
+}
+
+
+bool TaskbarSeven::setStatus(TBHandle handle,
+                             void* hwnd, //HWND: window assciated to the taskbar icon
+                             TaskBarStatus status)
+{
+    TBPFLAG flag = TBPF_NORMAL;
+    switch (status)
+    {
+    case STATUS_NOPROGRESS:
+        flag = TBPF_NOPROGRESS;
+        break;
+    case STATUS_INDETERMINATE:
+        flag = TBPF_INDETERMINATE;
+        break;
+    case STATUS_NORMAL:
+        flag = TBPF_NORMAL;
+        break;
+    case STATUS_ERROR:
+        flag = TBPF_ERROR;
+        break;
+    case STATUS_PAUSED:
+        flag = TBPF_PAUSED;
+        break;
+    }
+
+    ITaskbarList3* pto = taskBarHandles[handle];
+    if (pto)
+    {
+        HRESULT hr = pto->SetProgressState(static_cast<HWND>(hwnd),  //[in]  HWND hwnd,
+                                           flag); //[in]  TBPFLAG tbpFlags
+        if (FAILED(hr))
+        {
+            lastErrorMessage = writeErrorMsg(L"Error calling \"SetProgressState\".", hr);
+            return false;
+        }
+    }
+    return true;
+}
+
+
+bool TaskbarSeven::setProgress(TBHandle handle,
+                               void* hwnd, //HWND: window assciated to the taskbar icon
+                               size_t current,
+                               size_t total)
+{
+    ITaskbarList3* pto = taskBarHandles[handle];
+
+    if (pto)
+    {
+        HRESULT hr = pto->SetProgressValue(
+                         static_cast<HWND>(hwnd), //[in]  HWND hwnd,
+                         current, //[in]  ULONGLONG ullCompleted,
+                         total);  //[in]  ULONGLONG ullTotal
+        if (FAILED(hr))
+        {
+            lastErrorMessage = writeErrorMsg(L"Error calling \"SetProgressValue\".", hr);
+            return false;
+        }
+    }
+    return true;
+}
+
+
+void TaskbarSeven::getLastError(wchar_t* errorMessage, size_t errorBufferLen)
+{
+    writeString(lastErrorMessage, errorMessage, errorBufferLen);
+}