837 lines
22 KiB
C++
837 lines
22 KiB
C++
#include "private.h"
|
|
#include "offline.h"
|
|
#include "updateui.h"
|
|
|
|
//xnotfmgr - most of this file can probably get nuked
|
|
|
|
#define MAX_CAPTION 128
|
|
|
|
#undef TF_THISMODULE
|
|
#define TF_THISMODULE TF_UPDATEAGENT
|
|
|
|
typedef CLSID COOKIE, *PCOOKIE;
|
|
|
|
#define SUBITEM_SIZE 4
|
|
#define SUBITEM_URL 3
|
|
#define SUBITEM_STATUS 2
|
|
#define SUBITEM_IMAGE 1
|
|
|
|
ColInfoType colDlg[] = {
|
|
{0, IDS_NAME_COL, 30, LVCFMT_LEFT},
|
|
{1, 0, 3, LVCFMT_LEFT},
|
|
{2, IDS_STATUS_COL, 10, LVCFMT_LEFT},
|
|
{3, IDS_URL_COL, 40, LVCFMT_LEFT},
|
|
{4, IDS_SIZE_COL, 7, LVCFMT_RIGHT}
|
|
};
|
|
|
|
#define ILI_SUCCEEDED 0
|
|
#define ILI_FAILED 1
|
|
#define ILI_UPDATING 2
|
|
#define ILI_PENDING 3
|
|
#define ILI_SKIPPED 4
|
|
#define ILI_SITE 5
|
|
#define ILI_CHANNEL 6
|
|
#define ILI_DESKTOP 7
|
|
|
|
const int g_aIconResourceID[] = {
|
|
IDI_STAT_SUCCEEDED,
|
|
IDI_STAT_FAILED,
|
|
IDI_STAT_UPDATING,
|
|
IDI_STAT_PENDING,
|
|
IDI_STAT_SKIPPED,
|
|
IDI_WEBDOC,
|
|
IDI_CHANNEL,
|
|
IDI_DESKTOPITEM
|
|
};
|
|
|
|
#define MAX_DLG_COL ARRAYSIZE(colDlg)
|
|
|
|
//struct for saving window state in registry
|
|
typedef struct _PROG_PERSIST_STATE
|
|
{
|
|
short cbSize;
|
|
char bDetails;
|
|
char bAdjustWindowPos;
|
|
RECT rWindow;
|
|
int colOrder [MAX_DLG_COL];
|
|
int colWidth [MAX_DLG_COL];
|
|
} PROG_PERSIST_STATE;
|
|
extern void ResizeDialog(HWND hDlg, BOOL bShowDetail); //in update.cpp
|
|
|
|
const TCHAR c_szProgressWindowSettings[] = TEXT("Progress Preferences");
|
|
|
|
const UINT CookieSeg = 32;
|
|
|
|
CCookieItemMap::CCookieItemMap()
|
|
{
|
|
_map = NULL;
|
|
}
|
|
|
|
CCookieItemMap::~CCookieItemMap()
|
|
{
|
|
SAFELOCALFREE (_map);
|
|
}
|
|
|
|
STDMETHODIMP CCookieItemMap::Init(UINT size)
|
|
{
|
|
// Free old junk.
|
|
SAFELOCALFREE (_map);
|
|
|
|
_lParamNext = 0;
|
|
_count = 0;
|
|
|
|
if (size == 0)
|
|
_capacity = CookieSeg;
|
|
else
|
|
_capacity = size;
|
|
|
|
|
|
_map = (CookieItemMapEntry * )MemAlloc(LPTR, sizeof (CookieItemMapEntry) * _capacity);
|
|
if (!_map) {
|
|
DBG("Failed to allocate memory");
|
|
_capacity = 0;
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CCookieItemMap::ResetMap(void)
|
|
{
|
|
_count = 0;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CCookieItemMap::FindCookie(LPARAM lParam, CLSID * pCookie)
|
|
{
|
|
ASSERT(pCookie);
|
|
|
|
UINT i;
|
|
for (i = 0; i < _count; i ++) {
|
|
if (_map[i]._id == lParam) {
|
|
* pCookie = _map[i]._cookie;
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
*pCookie = CLSID_NULL;
|
|
return E_FAIL;
|
|
}
|
|
|
|
STDMETHODIMP CCookieItemMap::FindLParam(CLSID * pCookie, LPARAM * pLParam)
|
|
{
|
|
ASSERT(pCookie && pLParam);
|
|
|
|
UINT i;
|
|
for (i = 0; i < _count; i ++) {
|
|
if (_map[i]._cookie == *pCookie) {
|
|
* pLParam = _map[i]._id;
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
*pLParam = (LPARAM)-1;
|
|
return E_FAIL;
|
|
}
|
|
|
|
STDMETHODIMP CCookieItemMap::AddCookie(CLSID * pCookie, LPARAM * pLParam)
|
|
{
|
|
HRESULT hr = FindLParam(pCookie, pLParam);
|
|
if (S_OK == hr)
|
|
return S_FALSE;
|
|
|
|
ASSERT(_count <= _capacity);
|
|
ASSERT(CookieSeg);
|
|
|
|
if (_count == _capacity) {
|
|
UINT newSize = CookieSeg + _capacity;
|
|
void * newBuf = MemReAlloc(_map, newSize * sizeof(CookieItemMapEntry), LHND);
|
|
|
|
if (!newBuf) {
|
|
DBG("AddCookie::Failed to reallocate buffer");
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
_map = (CookieItemMapEntry *)newBuf;
|
|
_capacity = newSize;
|
|
}
|
|
|
|
_map[_count]._cookie = *pCookie;
|
|
_map[_count]._id = _lParamNext;
|
|
_count ++;
|
|
|
|
*pLParam = _lParamNext;
|
|
_lParamNext ++;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CCookieItemMap::DelCookie(CLSID * pCookie)
|
|
{
|
|
ASSERT(pCookie);
|
|
|
|
UINT i;
|
|
for (i = 0; i < _count; i ++) {
|
|
if (_map[i]._cookie == *pCookie) {
|
|
if (i == (_count - 1)) {
|
|
_count --;
|
|
return S_OK;
|
|
} else {
|
|
_count --;
|
|
_map[i]._cookie = _map[_count]._cookie;
|
|
_map[i]._id = _map[_count]._id;
|
|
return S_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
return S_FALSE;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Other members
|
|
//
|
|
|
|
int CALLBACK CUpdateDialog::SortUpdatingToTop (LPARAM lParam1,
|
|
LPARAM lParam2,
|
|
LPARAM lParamSort)
|
|
{
|
|
//lparams are cookies; lparamsort is source object
|
|
CUpdateDialog * pUpdater = (CUpdateDialog*)lParamSort;
|
|
if (!pUpdater)
|
|
return 0;
|
|
if (!pUpdater->m_pController)
|
|
return 0;
|
|
|
|
CLSID cookie;
|
|
pUpdater->cookieMap.FindCookie (lParam1, &cookie);
|
|
PReportMap pEntry1 = pUpdater->m_pController->FindReportEntry (&cookie);
|
|
pUpdater->cookieMap.FindCookie (lParam2, &cookie);
|
|
PReportMap pEntry2 = pUpdater->m_pController->FindReportEntry (&cookie);
|
|
|
|
//in progress precedes all else
|
|
if (pEntry1->status == ITEM_STAT_UPDATING)
|
|
return (pEntry2->status == ITEM_STAT_UPDATING ? 0 : -1);
|
|
|
|
if (pEntry2->status == ITEM_STAT_UPDATING)
|
|
return 1;
|
|
|
|
//queued precedes succeeded or skipped
|
|
if (pEntry1->status == ITEM_STAT_QUEUED || pEntry1->status == ITEM_STAT_PENDING)
|
|
return ((pEntry2->status == ITEM_STAT_QUEUED
|
|
|| pEntry2->status == ITEM_STAT_PENDING) ? 0 : -1);
|
|
|
|
if (pEntry2->status == ITEM_STAT_QUEUED || pEntry2->status == ITEM_STAT_PENDING)
|
|
return 1;
|
|
|
|
return 0; //don't care
|
|
}
|
|
|
|
|
|
BOOL CUpdateDialog::SelectFirstUpdatingSubscription()
|
|
{
|
|
LV_ITEM lvi = {0};
|
|
lvi.iSubItem = SUBITEM_IMAGE;
|
|
lvi.mask = LVIF_IMAGE;
|
|
|
|
int cItems = ListView_GetItemCount (m_hLV);
|
|
for (lvi.iItem = 0; lvi.iItem < cItems; lvi.iItem++)
|
|
{
|
|
ListView_GetItem (m_hLV, &lvi);
|
|
if (lvi.iImage == ILI_UPDATING)
|
|
{
|
|
ListView_SetItemState (m_hLV, lvi.iItem, LVIS_SELECTED, LVIS_SELECTED);
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
DWORD CUpdateDialog::SetSiteDownloadSize (PCOOKIE pCookie, DWORD dwNewSize)
|
|
{
|
|
//returns previous size, for bookkeeping purposes
|
|
HRESULT hr;
|
|
|
|
TCHAR szKSuffix[10];
|
|
// Need enough room for DWORD as string + K suffix
|
|
TCHAR szBuf[ARRAYSIZE(szKSuffix) + 11];
|
|
|
|
if (dwNewSize == -1) //shouldn't happen anymore but if it does,
|
|
return -1; //deal gracefully
|
|
|
|
ASSERT(pCookie);
|
|
LPARAM itemParam;
|
|
hr = cookieMap.FindLParam(pCookie, &itemParam);
|
|
if (S_OK != hr)
|
|
{
|
|
return dwNewSize;
|
|
}
|
|
|
|
LV_ITEM lvi = {0};
|
|
LV_FINDINFO lvfi = {0};
|
|
|
|
lvfi.flags = LVFI_PARAM;
|
|
lvfi.lParam = itemParam;
|
|
|
|
lvi.iItem = ListView_FindItem(m_hLV, -1, &lvfi);
|
|
if (lvi.iItem == -1)
|
|
return dwNewSize;
|
|
|
|
lvi.iSubItem = SUBITEM_SIZE;
|
|
lvi.mask = LVIF_TEXT;
|
|
lvi.pszText = szBuf;
|
|
lvi.cchTextMax = sizeof(szBuf);
|
|
|
|
ListView_GetItem(m_hLV, &lvi);
|
|
DWORD dwOldSize = StrToInt (szBuf);
|
|
|
|
MLLoadString (IDS_SIZE_KB, szKSuffix, ARRAYSIZE(szKSuffix));
|
|
wnsprintf (szBuf, ARRAYSIZE(szBuf), "%d%s", dwNewSize, szKSuffix);
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
return dwOldSize;
|
|
}
|
|
|
|
|
|
// This method is actually called from the second thread(only). So far
|
|
// I haven't found any sync problem yet. We only change the internal state
|
|
// of this object after it's creation in this method, so we won't mess
|
|
// it up. About UI, there is a chance when we try to disable 'Skip'
|
|
// button, we may come across another request from the primary thread. Since
|
|
// these 2 requests are both designated to disable the button, it won't
|
|
// matter anyway.
|
|
|
|
STDMETHODIMP CUpdateDialog::RefreshStatus(PCOOKIE pCookie, LPTSTR name, STATUS newStat, LPTSTR extraInfo)
|
|
{
|
|
HRESULT hr;
|
|
TCHAR szBuf[MAX_URL];
|
|
|
|
ASSERT(pCookie);
|
|
LPARAM itemParam;
|
|
hr = cookieMap.FindLParam(pCookie, &itemParam);
|
|
if (S_OK != hr) {
|
|
if (name) {
|
|
hr = AddItem(pCookie, name, newStat);
|
|
if (S_OK != hr) {
|
|
return E_FAIL;
|
|
}
|
|
hr = cookieMap.FindLParam(pCookie, &itemParam);
|
|
if (S_OK != hr) {
|
|
ASSERT(0);
|
|
return E_FAIL;
|
|
}
|
|
} else {
|
|
return hr;
|
|
}
|
|
}
|
|
|
|
LV_ITEM lvi = {0};
|
|
LV_FINDINFO lvfi = {0};
|
|
|
|
lvfi.flags = LVFI_PARAM;
|
|
lvfi.lParam = itemParam;
|
|
|
|
lvi.iItem = ListView_FindItem(m_hLV, -1, &lvfi);
|
|
if (lvi.iItem == -1)
|
|
return E_FAIL;
|
|
|
|
lvi.iSubItem = SUBITEM_STATUS;
|
|
lvi.mask = LVIF_TEXT;
|
|
|
|
ASSERT ((UINT)newStat <= ITEM_STAT_ABORTED);
|
|
if (newStat == ITEM_STAT_UPDATING && extraInfo != NULL) //url available, use it
|
|
{
|
|
TCHAR szFormat[MAX_URL];
|
|
MLLoadString (IDS_ITEM_STAT_UPDATING_URL, szFormat, ARRAYSIZE(szFormat));
|
|
wnsprintf (szBuf, ARRAYSIZE(szBuf), szFormat, extraInfo);
|
|
}
|
|
else
|
|
{
|
|
MLLoadString(IDS_ITEM_STAT_IDLE + newStat, szBuf, ARRAYSIZE(szBuf));
|
|
}
|
|
|
|
lvi.pszText = szBuf;
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
lvi.iSubItem = SUBITEM_IMAGE;
|
|
lvi.mask = LVIF_IMAGE;
|
|
|
|
switch (newStat) {
|
|
case ITEM_STAT_QUEUED:
|
|
case ITEM_STAT_PENDING:
|
|
lvi.iImage = ILI_PENDING;
|
|
break;
|
|
case ITEM_STAT_UPDATING:
|
|
lvi.iImage = ILI_UPDATING;
|
|
//move to top of list -- t-mattgi
|
|
//this happens in sort callback function -- just force resort
|
|
//after we update the LV control
|
|
break;
|
|
case ITEM_STAT_SUCCEEDED:
|
|
lvi.iImage = ILI_SUCCEEDED;
|
|
if (ListView_GetItemState(m_hLV, lvi.iItem, LVIS_SELECTED))
|
|
Button_Enable(GetDlgItem(m_hDlg, IDCMD_SKIP), FALSE);
|
|
break;
|
|
case ITEM_STAT_SKIPPED:
|
|
if (ListView_GetItemState(m_hLV, lvi.iItem, LVIS_SELECTED))
|
|
Button_Enable(GetDlgItem(m_hDlg, IDCMD_SKIP), FALSE);
|
|
lvi.iImage = ILI_SKIPPED;
|
|
break;
|
|
default:
|
|
lvi.iImage = ILI_FAILED;
|
|
break;
|
|
}
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
//force resort since item statuses changed
|
|
ListView_SortItems (m_hLV, SortUpdatingToTop, this);
|
|
|
|
return hr;
|
|
}
|
|
|
|
CUpdateDialog::CUpdateDialog()
|
|
{
|
|
m_bInitialized = FALSE;
|
|
}
|
|
|
|
CUpdateDialog::~CUpdateDialog()
|
|
{
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::CleanUp()
|
|
{
|
|
if (! m_ThreadID || !m_bInitialized)
|
|
return S_OK;
|
|
|
|
if (m_hDlg)
|
|
{
|
|
PersistStateToRegistry (m_hDlg);
|
|
|
|
DestroyWindow(m_hDlg);
|
|
m_hDlg = NULL;
|
|
}
|
|
PostThreadMessage(m_ThreadID, WM_QUIT, 0, 0);
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::Init(HWND hParent, CUpdateController * pController)
|
|
{
|
|
ASSERT(m_ThreadID);
|
|
ASSERT(g_hInst);
|
|
ASSERT(pController);
|
|
if (m_bInitialized) {
|
|
ASSERT(0);
|
|
return S_FALSE;
|
|
}
|
|
|
|
if (FAILED(cookieMap.Init()))
|
|
return E_FAIL;
|
|
|
|
HWND hDlg, hLV;
|
|
|
|
m_pController = pController;
|
|
hDlg = CreateDialogParam(MLGetHinst(), MAKEINTRESOURCE(IDD_PROGRESS), hParent, UpdateDlgProc, (LPARAM)this);
|
|
if (!hDlg)
|
|
return E_FAIL;
|
|
|
|
hLV = GetDlgItem(hDlg, IDL_SUBSCRIPTION);
|
|
if (!hLV) {
|
|
EndDialog(hDlg, FALSE);
|
|
return E_FAIL;
|
|
}
|
|
|
|
HIMAGELIST hImage;
|
|
HICON hIcon;
|
|
|
|
hImage = ImageList_Create(GetSystemMetrics(SM_CXSMICON),
|
|
GetSystemMetrics(SM_CXSMICON),
|
|
ILC_MASK,
|
|
ARRAYSIZE(g_aIconResourceID),
|
|
0);
|
|
|
|
if (hImage == NULL) {
|
|
TraceMsg(TF_ALWAYS, TEXT("CUpdateDialog::Init - Failed to create ImageList"));
|
|
return E_FAIL;
|
|
}
|
|
|
|
for (int i = 0; i < ARRAYSIZE(g_aIconResourceID); i ++) {
|
|
if (g_aIconResourceID[i] == IDI_DESKTOPITEM)
|
|
{
|
|
hinstSrc = MLGetHinst();
|
|
}
|
|
else
|
|
{
|
|
hinstSrc = g_hInst;
|
|
}
|
|
|
|
hIcon = LoadIcon(hinstSrc, MAKEINTRESOURCE(g_aIconResourceID[i]));
|
|
if (hIcon == NULL) {
|
|
ImageList_Destroy(hImage);
|
|
DBG("CUpdateDialog::Init - Failed to load icon");
|
|
return E_FAIL;
|
|
}
|
|
ImageList_AddIcon(hImage, hIcon);
|
|
DestroyIcon(hIcon);
|
|
}
|
|
|
|
ListView_SetImageList(hLV, hImage, LVSIL_SMALL);
|
|
|
|
LV_COLUMN lvc;
|
|
TEXTMETRIC tm;
|
|
HDC hdc;
|
|
|
|
hdc = GetDC(hDlg);
|
|
if (!hdc) {
|
|
EndDialog(hDlg, FALSE);
|
|
return E_FAIL;
|
|
}
|
|
GetTextMetrics(hdc, &tm);
|
|
ReleaseDC(hDlg, hdc);
|
|
|
|
lvc.mask = LVCF_WIDTH | LVCF_TEXT | LVCF_FMT;
|
|
|
|
PROG_PERSIST_STATE state;
|
|
GetPersistentStateFromRegistry(state, tm.tmAveCharWidth);
|
|
|
|
for (UINT ui = 0; ui < MAX_DLG_COL; ui ++)
|
|
{
|
|
int colIndex;
|
|
|
|
TCHAR szCaption[MAX_CAPTION];
|
|
if (colDlg[ui].ids)
|
|
MLLoadString(colDlg[ui].ids, szCaption, MAX_CAPTION);
|
|
else
|
|
szCaption[0] = (TCHAR)0;
|
|
|
|
lvc.pszText = szCaption;
|
|
lvc.cx = state.colWidth[ui];
|
|
lvc.fmt = colDlg[ui].iFmt;
|
|
colIndex = ListView_InsertColumn(hLV, ui, & lvc);
|
|
if ( -1 == colIndex) {
|
|
ASSERT(0);
|
|
EndDialog(hDlg, FALSE);
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
|
|
ListView_SetColumnOrderArray(hLV, MAX_DLG_COL, state.colOrder);
|
|
|
|
SendMessage (hLV, LVM_SETEXTENDEDLISTVIEWSTYLE,
|
|
LVS_EX_HEADERDRAGDROP | LVS_EX_SUBITEMIMAGES,
|
|
LVS_EX_HEADERDRAGDROP | LVS_EX_SUBITEMIMAGES);
|
|
|
|
SendMessage (hDlg, WM_SETICON, ICON_BIG, (LPARAM)LoadIcon (g_hInst, MAKEINTRESOURCE (IDI_DOWNLOAD)));
|
|
SendMessage (hDlg, WM_SETICON, ICON_SMALL, (LPARAM)LoadIcon (g_hInst, MAKEINTRESOURCE (IDI_DOWNLOAD)));
|
|
|
|
if (state.bAdjustWindowPos)
|
|
{
|
|
//adjust size of *details view* to stored state; if we're not in
|
|
//details view, we have to go there temporarily. (The non-details
|
|
//view will pick up the position from the details view when we switch back.)
|
|
m_bDetail = TRUE;
|
|
ResizeDialog (hDlg, m_bDetail);
|
|
|
|
//don't move dialog, just resize it and center it
|
|
//convert right, bottom coordinates to width, height
|
|
state.rWindow.right -= state.rWindow.left;
|
|
state.rWindow.bottom -= state.rWindow.top;
|
|
//calculate left, top to center dialog
|
|
state.rWindow.left = (GetSystemMetrics (SM_CXSCREEN) - state.rWindow.right) / 2;
|
|
state.rWindow.top = (GetSystemMetrics (SM_CYSCREEN) - state.rWindow.bottom) / 2;
|
|
MoveWindow (hDlg, state.rWindow.left, state.rWindow.top,
|
|
state.rWindow.right, state.rWindow.bottom, TRUE);
|
|
|
|
//REVIEW: this centers the details view, then if they don't want details,
|
|
//leaves the small dialog with its upper left where the upper left of the
|
|
//big dialog is when it's centered. I could center it in whatever view
|
|
//it's really in, but if the details view is resized to a fairly large window
|
|
//and we bring it up centered in non-details, then when they click details
|
|
//the position will be the same and the window will potentially extend offscreen
|
|
//to the right and bottom.
|
|
|
|
//set back to non-details view if that was how it was last used
|
|
if (!state.bDetails)
|
|
{
|
|
m_bDetail = state.bDetails;
|
|
ResizeDialog (hDlg, m_bDetail);
|
|
}
|
|
}
|
|
|
|
m_bInitialized = TRUE;
|
|
m_hDlg = hDlg;
|
|
m_hLV = hLV;
|
|
m_hParent = hParent;
|
|
m_cDlKBytes = 0;
|
|
m_cDlDocs = 0;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
BOOL CUpdateDialog::PersistStateToRegistry (HWND hDlg)
|
|
{
|
|
PROG_PERSIST_STATE state;
|
|
|
|
state.cbSize = sizeof(state);
|
|
state.bDetails = m_bDetail;
|
|
state.bAdjustWindowPos = TRUE;
|
|
//save position and size from *details view* -- if we're not there,
|
|
//we'll have to switch temporarily.
|
|
BOOL bTempDetail = m_bDetail;
|
|
if (!bTempDetail)
|
|
{
|
|
ShowWindow (hDlg, SW_HIDE);
|
|
m_bDetail = TRUE;
|
|
ResizeDialog (hDlg, m_bDetail);
|
|
}
|
|
GetWindowRect (hDlg, &state.rWindow);
|
|
if (!bTempDetail)
|
|
{
|
|
m_bDetail = FALSE;
|
|
ResizeDialog (hDlg, m_bDetail);
|
|
ShowWindow (hDlg, SW_SHOW);
|
|
}
|
|
|
|
HWND hLV = GetDlgItem (hDlg, IDL_SUBSCRIPTION);
|
|
ListView_GetColumnOrderArray (hLV, MAX_DLG_COL, state.colOrder);
|
|
for (int i=0; i<MAX_DLG_COL; i++)
|
|
state.colWidth[i] = ListView_GetColumnWidth (hLV, i);
|
|
|
|
HKEY key;
|
|
DWORD dwDisposition;
|
|
if (ERROR_SUCCESS != RegCreateKeyEx (HKEY_CURRENT_USER, c_szRegKey, 0, NULL, REG_OPTION_NON_VOLATILE,
|
|
KEY_WRITE, NULL, &key, &dwDisposition))
|
|
return FALSE;
|
|
|
|
RegSetValueEx (key, c_szProgressWindowSettings, 0, REG_BINARY, (LPBYTE)&state, sizeof(state));
|
|
|
|
RegCloseKey (key);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CUpdateDialog::GetPersistentStateFromRegistry (PROG_PERSIST_STATE& state, int iCharWidth)
|
|
{
|
|
HKEY key;
|
|
DWORD dwType;
|
|
DWORD dwSize = sizeof(state);
|
|
RegOpenKeyEx (HKEY_CURRENT_USER, c_szRegKey, 0, KEY_READ, &key);
|
|
|
|
LONG result = RegQueryValueEx (key, c_szProgressWindowSettings,
|
|
0, &dwType, (LPBYTE)&state, &dwSize);
|
|
|
|
if (ERROR_SUCCESS != result || dwType != REG_BINARY || dwSize != sizeof(state))
|
|
{
|
|
state.cbSize = 0; //flag as error
|
|
}
|
|
|
|
if (state.cbSize != sizeof(state)) //error or incorrect registry format/version
|
|
{ //state not saved in registry; use defaults
|
|
int i;
|
|
|
|
state.bDetails = FALSE;
|
|
state.bAdjustWindowPos = FALSE;
|
|
|
|
state.colOrder[0] = 1;
|
|
state.colOrder[1] = 0;
|
|
for (i=2; i<MAX_DLG_COL; i++)
|
|
state.colOrder[i] = i;
|
|
|
|
for (i=0; i<MAX_DLG_COL; i++)
|
|
state.colWidth[i] = colDlg[i].cchCol * iCharWidth;
|
|
}
|
|
|
|
RegCloseKey (key);
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
STDMETHODIMP CUpdateDialog::Show(BOOL bShow)
|
|
{
|
|
if (!m_bInitialized) {
|
|
ASSERT(0);
|
|
return E_FAIL;
|
|
}
|
|
|
|
ASSERT(m_hDlg);
|
|
|
|
ShowWindow(m_hDlg, (bShow)?SW_SHOW:SW_HIDE);
|
|
ShowWindow(m_hLV, (bShow)?SW_SHOW:SW_HIDE);
|
|
return NOERROR;
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::ResetDialog(void)
|
|
{
|
|
if (!m_bInitialized) {
|
|
return S_OK;
|
|
}
|
|
|
|
ASSERT(m_hLV);
|
|
ListView_DeleteAllItems(m_hLV);
|
|
cookieMap.ResetMap();
|
|
m_bInitialized = FALSE;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::IItem2Cookie(const int iItem, CLSID * pCookie)
|
|
{
|
|
LV_ITEM item = {0};
|
|
HRESULT hr;
|
|
ASSERT(pCookie);
|
|
|
|
item.iItem = iItem;
|
|
item.iSubItem = 0;
|
|
item.mask = LVIF_PARAM;
|
|
|
|
if (!ListView_GetItem(m_hLV, &item)) {
|
|
return E_FAIL;
|
|
}
|
|
|
|
hr = cookieMap.FindCookie(item.lParam, pCookie);
|
|
ASSERT(SUCCEEDED(hr));
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::GetSelectedCookies(CLSID * pCookies, UINT * pCount)
|
|
{
|
|
if (!m_bInitialized)
|
|
return E_INVALIDARG;
|
|
|
|
ASSERT(pCookies && pCount);
|
|
int index = -1;
|
|
|
|
*pCount = 0;
|
|
|
|
index = ListView_GetNextItem(m_hLV, index, LVNI_ALL | LVNI_SELECTED);
|
|
while (-1 != index) {
|
|
if (FAILED(IItem2Cookie(index, pCookies + *pCount))) {
|
|
ASSERT(0);
|
|
return S_FALSE;
|
|
}
|
|
|
|
index = ListView_GetNextItem(m_hLV, index, LVNI_ALL | LVNI_SELECTED);
|
|
*pCount = *pCount + 1;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::GetSelectionCount(UINT * pCount)
|
|
{
|
|
if (!m_bInitialized)
|
|
return E_INVALIDARG;
|
|
|
|
ASSERT(pCount);
|
|
* pCount = ListView_GetSelectedCount(m_hLV);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CUpdateDialog::AddItem(CLSID * pCookie, LPTSTR name, STATUS stat)
|
|
{
|
|
HRESULT hr;
|
|
TCHAR szBuf[MAX_URL];
|
|
|
|
LV_ITEM lvi = {0};
|
|
BOOL bNew;
|
|
|
|
lvi.iSubItem = 0;
|
|
hr = cookieMap.AddCookie(pCookie, &(lvi.lParam));
|
|
if (S_OK == hr) {
|
|
bNew = TRUE;
|
|
} else if (S_FALSE == hr) {
|
|
bNew = FALSE;
|
|
} else {
|
|
return hr;
|
|
}
|
|
|
|
lvi.pszText = name;
|
|
if (bNew) {
|
|
lvi.mask = LVIF_TEXT | LVIF_PARAM | LVIF_IMAGE;
|
|
switch (m_pController->GetSubscriptionType(pCookie))
|
|
{
|
|
case SUBSTYPE_CHANNEL:
|
|
lvi.iImage = ILI_CHANNEL;
|
|
break;
|
|
case SUBSTYPE_DESKTOPURL:
|
|
case SUBSTYPE_DESKTOPCHANNEL:
|
|
lvi.iImage = ILI_DESKTOP;
|
|
break;
|
|
case SUBSTYPE_URL:
|
|
default:
|
|
lvi.iImage = ILI_SITE;
|
|
break;
|
|
}
|
|
lvi.iItem = ListView_InsertItem(m_hLV, &lvi);
|
|
if (lvi.iItem == -1)
|
|
return E_FAIL;
|
|
if (lvi.iItem == 0) {
|
|
ListView_SetItemState(m_hLV, 0, LVIS_SELECTED, LVIS_SELECTED);
|
|
}
|
|
} else {
|
|
LV_FINDINFO lvfi = {0};
|
|
|
|
lvfi.flags = LVFI_PARAM;
|
|
lvfi.lParam = lvi.lParam;
|
|
|
|
lvi.iItem = ListView_FindItem(m_hLV, -1, &lvfi);
|
|
if (lvi.iItem == -1)
|
|
return E_FAIL;
|
|
|
|
lvi.mask = LVIF_TEXT;
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
}
|
|
|
|
//add subitem for status icon
|
|
lvi.mask = LVIF_IMAGE;
|
|
lvi.iSubItem ++; // Icon field.
|
|
switch (stat) {
|
|
case ITEM_STAT_QUEUED:
|
|
case ITEM_STAT_PENDING:
|
|
lvi.iImage = ILI_PENDING;
|
|
break;
|
|
case ITEM_STAT_UPDATING:
|
|
lvi.iImage = ILI_UPDATING;
|
|
break;
|
|
case ITEM_STAT_SUCCEEDED:
|
|
lvi.iImage = ILI_SUCCEEDED;
|
|
if (ListView_GetItemState(m_hLV, lvi.iItem, LVIS_SELECTED))
|
|
Button_Enable(GetDlgItem(m_hDlg, IDCMD_SKIP), FALSE);
|
|
break;
|
|
case ITEM_STAT_SKIPPED:
|
|
lvi.iImage = ILI_SKIPPED;
|
|
if (ListView_GetItemState(m_hLV, lvi.iItem, LVIS_SELECTED))
|
|
Button_Enable(GetDlgItem(m_hDlg, IDCMD_SKIP), FALSE);
|
|
default:
|
|
lvi.iImage = ILI_FAILED;
|
|
break;
|
|
}
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
//add subitem for status text
|
|
lvi.mask = LVIF_TEXT;
|
|
ASSERT ((UINT)stat <= ITEM_STAT_SUCCEEDED);
|
|
MLLoadString(IDS_ITEM_STAT_IDLE + stat, szBuf, ARRAYSIZE(szBuf));
|
|
lvi.pszText = szBuf;
|
|
lvi.iSubItem ++;
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
//add subitem for URL
|
|
PReportMap prm = m_pController->FindReportEntry (pCookie);
|
|
lvi.pszText = prm->url;
|
|
lvi.iSubItem++;
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
//add subitem for size
|
|
lvi.pszText = TEXT("");
|
|
lvi.iSubItem++;
|
|
ListView_SetItem(m_hLV, &lvi);
|
|
|
|
return S_OK;
|
|
}
|