windows-nt/Source/XPSP1/NT/admin/activec/samples/displ2/enumtask.cpp
2020-09-26 16:20:57 +08:00

674 lines
21 KiB
C++

//+-------------------------------------------------------------------------
//
// Microsoft Windows
//
// Copyright (C) Microsoft Corporation, 1999 - 1999
//
// File: enumtask.cpp
//
//--------------------------------------------------------------------------
#include "stdafx.h"
#include "displ2.h"
#include "DsplMgr2.h"
extern HINSTANCE g_hinst; // in displ2.cpp
#define NEW_WAY
LPOLESTR CoTaskDupString (LPOLESTR szString)
{
#ifdef NEW_WAY
OLECHAR * lpString = (OLECHAR *)CoTaskMemAlloc (sizeof(OLECHAR)*(wcslen(szString)+1));
if (lpString)
wcscpy (lpString, szString);
return lpString;
#else
return(LPOLESTR)SysAllocString (szString);
#endif
}
void CoTaskFreeString (LPOLESTR szString)
{
#ifdef NEW_WAY
CoTaskMemFree (szString);
#else
SysFreeString (szString);
#endif
}
CEnumTasks::CEnumTasks()
{
m_refs = 0;
m_index = 0;
m_type = 0; // default group/category
// filenames for wallpaper mode
m_bmps = NULL;
}
CEnumTasks::~CEnumTasks()
{
if (m_bmps)
delete m_bmps;
}
HRESULT CEnumTasks::QueryInterface (REFIID riid, LPVOID FAR* ppv)
{
if ( (riid == IID_IUnknown) ||
(riid == IID_IEnumTASK) )
{
*ppv = this;
((LPUNKNOWN)(*ppv))->AddRef();
return NOERROR;
}
*ppv = NULL;
return E_NOINTERFACE;
}
ULONG CEnumTasks::AddRef ()
{
return ++m_refs;
}
ULONG CEnumTasks::Release ()
{
if (--m_refs == 0)
{
delete this;
return 0;
}
return m_refs;
}
#define NUMBER_OF_TASKS 4
LPTSTR g_bitmaps[NUMBER_OF_TASKS] = { _T("/img\\ntauto.gif"),
_T("/img\\mariners.gif"),
_T("/img\\ntstart.gif"),
_T("/img\\ntmonitor.gif")};
LPTSTR g_text [NUMBER_OF_TASKS] = { _T("Set Wallpaper"),
_T("Wallpaper Node"),
_T("Wallpaper Options"),
_T("Alert Script")};
LPTSTR g_help [NUMBER_OF_TASKS] = { _T("Pick Bitmap Image for Wallpaper"),
_T("Jump to Wallpaper Node"),
_T("Select Stretch, Tile, or Center"),
_T("Script demo")};
LPTSTR g_action [NUMBER_OF_TASKS] = { _T("/wallpapr.htm"),
_T("1"),
_T("/default.htm#wallpaper_options"),
_T("JSCRIPT:alert('my location is: ' + location);")};
// _T("vbscript:MsgBox 'hi' ")};
HRESULT OneOfEach(ULONG index, MMC_TASK *rgelt, ULONG *pceltFetched)
{ // NOTE: not bothering with error checking!!!
if (index >= 20 /*NUMBER_OF_TASKS*/)
{
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // all done
}
USES_CONVERSION;
// setup path for reuse
TCHAR szPath[MAX_PATH*2]; // that should be enough
lstrcpy (szPath, _T("res://"));
::GetModuleFileName (g_hinst, szPath + lstrlen(szPath), MAX_PATH);
TCHAR * szBuffer = szPath + lstrlen(szPath);
MMC_TASK * task = rgelt;
MMC_TASK_DISPLAY_OBJECT* pdo = &task->sDisplayObject;
MMC_TASK_DISPLAY_BITMAP* pdb = &pdo->uBitmap;
MMC_TASK_DISPLAY_SYMBOL* pds = &pdo->uSymbol;
switch (index)
{
default:
{
TCHAR szNumber[10];
if (index < 200)
_itot (index, szNumber, 10);
else
_itot (index-200, szNumber, 10);
task->szText = CoTaskDupString (T2OLE(szNumber));
}
task->szHelpString = CoTaskDupString (T2OLE(g_help[0]));
task->eActionType = MMC_ACTION_LINK;
_tcscpy (szBuffer, _T("/wallpapr.htm"));
task->szActionURL = CoTaskDupString (T2OLE(szPath));
lstrcpy (szPath, _T("res://"));
::GetModuleFileName (NULL, szPath + lstrlen(szPath), MAX_PATH);
szBuffer = szPath + lstrlen(szPath);
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_SYMBOL;
if (index < 200)
{
pds->szFontFamilyName = CoTaskDupString (L"Glyph 100"); // name of font family
_tcscpy (szBuffer, _T("/GLYPH100.eot"));
}
else
{
pds->szFontFamilyName = CoTaskDupString (L"Glyph 110"); // name of font family
_tcscpy (szBuffer, _T("/GLYPH110.eot"));
}
pds->szURLtoEOT = CoTaskDupString (T2OLE(szPath)); // "res://"-type URL to EOT file
{
OLECHAR szChar[2] = {0,0};
szChar[0] = (WORD) (index%20 + 32); // cycle the same 20 symbols starting at 32
pds->szSymbolString= CoTaskDupString (szChar); // 1 or more symbol characters
}
break;
case 0:
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_VANILLA_GIF;
_tcscpy (szBuffer, _T("/img\\vanilla.gif"));
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(szPath));
pdb->szMouseOverBitmap = NULL; // skipping mouse over bitmap
task->szText = CoTaskDupString (T2OLE(g_text[index]));
task->szHelpString = CoTaskDupString (T2OLE(g_help[index]));
task->eActionType = MMC_ACTION_LINK;
_tcscpy (szBuffer, _T("/wallpapr.htm"));
task->szActionURL = CoTaskDupString (T2OLE(szPath));
break;
case 1:
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_CHOCOLATE_GIF;
_tcscpy (szBuffer, _T("/img\\chocolate.gif"));
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(szPath));
pdb->szMouseOverBitmap = NULL; // skipping mouse off bitmap
task->szText = CoTaskDupString (T2OLE(g_text[index]));
task->szHelpString = CoTaskDupString (T2OLE(g_help[index]));
task->eActionType = MMC_ACTION_ID;
task->nCommandID = 1;
break;
case 2:
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_BITMAP;
_tcscpy (szBuffer, _T("/img\\ntstart.gif"));
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(szPath));
_tcscpy (szBuffer, _T("/img\\dax.bmp"));
pdb->szMouseOverBitmap = CoTaskDupString (T2OLE(szPath));
task->szText = CoTaskDupString (T2OLE(g_text[index]));
task->szHelpString = CoTaskDupString (T2OLE(g_help[index]));
task->eActionType = MMC_ACTION_LINK;
_tcscpy (szBuffer, _T("/default.htm#wallpaper_options"));
task->szActionURL = CoTaskDupString (T2OLE(szPath));
break;
case 3:
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_SYMBOL;
pds->szFontFamilyName = CoTaskDupString (L"Kingston"); // name of font family
_tcscpy (szBuffer, _T("/KINGSTON.eot"));
pds->szURLtoEOT = CoTaskDupString (T2OLE(szPath)); // "res://"-type URL to EOT file
pds->szSymbolString = CoTaskDupString (T2OLE(_T("A"))); // 1 or more symbol characters
task->szText = CoTaskDupString (T2OLE(g_text[index]));
task->szHelpString = CoTaskDupString (T2OLE(g_help[index]));
task->eActionType = MMC_ACTION_SCRIPT;
task->szScript = CoTaskDupString (T2OLE(g_action[index]));
break;
}
return S_OK;
}
HRESULT CEnumTasks::Next (ULONG celt, MMC_TASK *rgelt, ULONG *pceltFetched)
{ // will be called with celt == 1
// caller alloc's array of MMC_TASKs
// callee fills MMC_TASK elements (via CoTaskDupString)
_ASSERT (!IsBadWritePtr (rgelt, celt*sizeof(MMC_TASK)));
if (m_type == 2)
return EnumBitmaps (celt, rgelt, pceltFetched);
if (m_type == 3)
return EnumOptions (celt, rgelt, pceltFetched);
// new stuff
return OneOfEach (m_index++, rgelt, pceltFetched);
// new stuff
if (m_index >= NUMBER_OF_TASKS)
{
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}
USES_CONVERSION;
// setup path for reuse
TCHAR szBuffer[MAX_PATH*2]; // that should be enough
lstrcpy (szBuffer, _T("res://"));
::GetModuleFileName (g_hinst, szBuffer + lstrlen(szBuffer), MAX_PATH);
TCHAR * temp = szBuffer + lstrlen(szBuffer);
MMC_TASK * task = rgelt;
MMC_TASK_DISPLAY_OBJECT* pdo = &task->sDisplayObject;
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_BITMAP;
MMC_TASK_DISPLAY_BITMAP *pdb = &pdo->uBitmap;
// fill out bitmap URL
lstrcpy (temp, g_bitmaps[m_index]);
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(szBuffer));
if (pdb->szMouseOffBitmap)
{
lstrcpy (temp, _T("/img\\dax.bmp"));
pdb->szMouseOverBitmap = CoTaskDupString (T2OLE(szBuffer));
if (pdb->szMouseOverBitmap)
{
// add button text
task->szText = CoTaskDupString (T2OLE(g_text[m_index]));
if (task->szText)
{
// add help string
task->szHelpString = CoTaskDupString (T2OLE(g_help[m_index]));
if (task->szHelpString)
{
// add action URL (link or script)
switch (m_index)
{
default:
case 0:
task->eActionType = MMC_ACTION_LINK;
break;
case 1:
task->eActionType = MMC_ACTION_ID;
break;
case 2:
task->eActionType = MMC_ACTION_LINK;
break;
case 3:
task->eActionType = MMC_ACTION_SCRIPT;
break;
}
// the stuff below works, because of the nameless union.
if (m_index == 1)
{
task->nCommandID = _ttol (g_action[m_index]);
m_index++;
// if we get here all is well
if (pceltFetched)
*pceltFetched = 1;
return S_OK;
}
else
{
if (m_index == 0)
{
lstrcpy (temp, g_action[m_index]);
task->szActionURL = CoTaskDupString (T2OLE(szBuffer));
}
else
task->szActionURL = CoTaskDupString (T2OLE(g_action[m_index]));
if (task->szActionURL)
{
m_index++;
// if we get here all is well
if (pceltFetched)
*pceltFetched = 1;
return S_OK;
}
}
CoTaskFreeString (task->szHelpString);
}
CoTaskFreeString (task->szText);
}
CoTaskFreeString (pdb->szMouseOverBitmap);
}
CoTaskFreeString (pdb->szMouseOffBitmap);
}
// if we get here, we have some kinda failure
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}
HRESULT CEnumTasks::Skip (ULONG celt)
{ // won't be called
m_index += celt;
return S_OK;
}
HRESULT CEnumTasks::Reset()
{
m_index = 0;
return S_OK;
}
HRESULT CEnumTasks::Clone(IEnumTASK **ppenum)
{ // won't be called
// clone maintaining state info
CEnumTasks * pet = new CEnumTasks();
if (pet)
{
pet->m_index = m_index;
return pet->QueryInterface (IID_IEnumTASK, (void **)ppenum); // can't fail
}
return E_OUTOFMEMORY;
}
HRESULT CEnumTasks::Init (IDataObject * pdo, LPOLESTR szTaskGroup)
{ // return ok, if we can handle data object and group
if (!wcscmp (szTaskGroup, L""))
m_type = 1; // default tasks
else
if (!wcscmp (szTaskGroup, L"wallpaper"))
m_type = 2; // enum wallpaper tasks
else
if (!wcscmp (szTaskGroup, L"wallpaper_options"))
m_type = 3; // enum option-tasks (tile/center/stretch)
// else
// if (!wcscmp (szTaskGroup, L"ListPad"))
// m_type = 4; // default tasks
return S_OK;
}
HRESULT CEnumTasks::EnumBitmaps (ULONG celt, MMC_TASK *rgelt, ULONG *pceltFetched)
{ // will be called with celt == 1
// enum wallpaper tasks
// may only be called when m_type == 2
_ASSERT (m_type == 2);
USES_CONVERSION;
TCHAR temp2[MAX_PATH*2];
// setup path for reuse
TCHAR szBuffer[MAX_PATH*2]; // that should be enough
lstrcpy (szBuffer, _T("file://"));
TCHAR * path = szBuffer + lstrlen(szBuffer);
::GetWindowsDirectory (path, MAX_PATH);
lstrcat (path, _T("\\"));
path = szBuffer + lstrlen(szBuffer);
// if we haven't already, get all .bmp files in the windows directory
if (!m_bmps)
GetBitmaps ();
if (!m_bmps)
{
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}
TCHAR * temp = m_bmps;
// skip past names of stuff we've already enum'ed
for (ULONG j=0; j<m_index; j++)
temp += lstrlen (temp) + 1;
if (*temp == 0)
{
// all done!
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}
MMC_TASK * task = rgelt;
MMC_TASK_DISPLAY_OBJECT* pdo = &task->sDisplayObject;
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_BITMAP;
MMC_TASK_DISPLAY_BITMAP *pdb = &pdo->uBitmap;
// fill out bitmap URL
lstrcpy (path, temp);
if (!lstrcmp (temp, _T("(none)")))
{
// special case for none
lstrcpy (temp2, _T("res://"));
::GetModuleFileName (g_hinst, temp2 + lstrlen (temp2), MAX_PATH);
lstrcat (temp2, _T("/img\\none.gif"));
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(temp2));
}
else
{
lstrcpy (temp2, _T("\""));
lstrcat (temp2, szBuffer);
lstrcat (temp2, _T("\""));
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(temp2));
}
if (pdb->szMouseOffBitmap)
{
// am using same bitmap for both!!!
pdb->szMouseOverBitmap = CoTaskDupString ((LPOLESTR)pdb->szMouseOffBitmap);
if (pdb->szMouseOverBitmap)
{
// add button text
task->szText = CoTaskDupString (T2OLE(temp));
if (task->szText)
{
// add help string
OLECHAR help[] = L"Add this Bitmap as Wallpaper";
task->szHelpString = CoTaskDupString (help);
if (task->szHelpString)
{
// add action URL (link or script)
task->eActionType = MMC_ACTION_LINK; // always link to scriptlet
TCHAR wallpaper[] = _T("#wallpaper");
#ifndef TRY_THIS
TCHAR action[] = _T("/button.htm#");
lstrcpy (temp2, _T("res://"));
::GetModuleFileName (g_hinst, temp2 + lstrlen (temp2), MAX_PATH);
lstrcat (temp2, action);
#else
TCHAR action[] = _T("button.htm#");
lstrcpy (temp2, action);
#endif
TCHAR * sztemp = temp2 + lstrlen (temp2);
lstrcat (temp2, temp);
// replace any spaces with '*' char
// script can't handle hashes with ' ' in 'em
// and there can't be any filenames with '*' char,
// so this works ok.
TCHAR * space;
while (space = _tcschr (sztemp, ' '))
*space = '*';
lstrcat (temp2, wallpaper);
task->szActionURL = CoTaskDupString (T2OLE(temp2));
if (task->szActionURL)
{
m_index++;
// if we get here all is well
if (pceltFetched)
*pceltFetched = 1;
return S_OK;
}
CoTaskFreeString (task->szHelpString);
}
CoTaskFreeString (task->szText);
}
CoTaskFreeString (pdb->szMouseOverBitmap);
}
CoTaskFreeString (pdb->szMouseOffBitmap);
}
// if we get here, we failed above
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}
void CEnumTasks::GetBitmaps (void)
{
if (m_bmps)
return; // knuckle-head
TCHAR path[MAX_PATH];
GetWindowsDirectory (path, MAX_PATH);
lstrcat (path, _T("\\*.bmp"));
// count up *.bmp files in windows directory (also add an entry for "(none)" )
int numBMPs = 0;
int length = 0; // get total length of all filenames
WIN32_FIND_DATA fd;
ZeroMemory(&fd, sizeof(fd));
HANDLE hFind = FindFirstFile (path, &fd);
if (hFind != INVALID_HANDLE_VALUE)
{
do
{
if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
(fd.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) ||
(fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) )
continue; // files only
numBMPs++;
length += lstrlen (fd.cFileName) + 1;
} while (FindNextFile (hFind, &fd) == TRUE);
FindClose(hFind);
}
numBMPs++; // one for "(none)"
length += lstrlen (_T("(none)")) + 1;
length++; // add trailing double NULL
// alloc space to hold filenames (plus extra NULL entry)
m_bmps = new TCHAR[length];
if (!m_bmps)
return; // fail, but no return mechanism
// add none first
TCHAR * temp = m_bmps;
lstrcpy (temp, _T("(none)"));
temp += lstrlen (temp) + 1;
// add all bmp files
ZeroMemory(&fd, sizeof(fd));
hFind = FindFirstFile (path, &fd);
if (hFind != INVALID_HANDLE_VALUE)
{
do
{
if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
(fd.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) ||
(fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) )
continue; // files only
lstrcpy (temp, fd.cFileName);
temp += lstrlen (temp) + 1;
} while (FindNextFile (hFind, &fd) == TRUE);
FindClose(hFind);
}
*temp = 0; // double null terminator
}
#define NUMBER_OF_O_TASKS 3
LPTSTR g_o_bitmaps[NUMBER_OF_O_TASKS] = {_T("/img\\ntauto.gif"),
_T("/img\\mariners.gif"),
_T("/img\\ntstart.gif")};
LPTSTR g_o_text [NUMBER_OF_O_TASKS] = {_T("Center"),
_T("Tile"),
_T("Stretch")};
LPTSTR g_o_help [NUMBER_OF_O_TASKS] = {_T("Center Wallpaper"),
_T("Tile Wallpaper"),
_T("Stretch Wallpaper")};
LPTSTR g_o_action [NUMBER_OF_O_TASKS] = {_T("2"), // command ids
_T("3"),
_T("4")};
HRESULT CEnumTasks::EnumOptions (ULONG celt, MMC_TASK *rgelt, ULONG *pceltFetched)
{ // will be called with celt == 1
// enum option tasks
// may only be called when m_type == 3
_ASSERT (m_type == 3);
_ASSERT (celt == 1);
if (m_index >= NUMBER_OF_O_TASKS)
{
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}
USES_CONVERSION;
// setup path for reuse
TCHAR szBuffer[MAX_PATH*2]; // that should be enough
_tcscpy (szBuffer, _T("res://"));
::GetModuleFileName (g_hinst, szBuffer + _tcslen(szBuffer), MAX_PATH);
TCHAR * path = szBuffer + _tcslen(szBuffer);
MMC_TASK * task = rgelt;
MMC_TASK_DISPLAY_OBJECT* pdo = &task->sDisplayObject;
#ifdef BITMAP_CASE
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_BITMAP;
MMC_TASK_DISPLAY_BITMAP *pdb = &pdo->uBitmap;
// fill out bitmap URL
lstrcpy (path, g_o_bitmaps[m_index]);
pdb->szMouseOffBitmap = CoTaskDupString (T2OLE(szBuffer));
if (pdb->szMouseOffBitmap)
{
// am using same bitmap for both!!!
pdb->szMouseOverBitmap = CoTaskDupString (T2OLE(szBuffer));
if (pdb->szMouseOverBitmap)
{
#else
// symbol case
pdo->eDisplayType = MMC_TASK_DISPLAY_TYPE_SYMBOL;
MMC_TASK_DISPLAY_SYMBOL *pds = &pdo->uSymbol;
// fill out symbol stuff
pds->szFontFamilyName = CoTaskDupString (L"Kingston"); // name of font family
if (pds->szFontFamilyName)
{
_tcscpy (path, _T("/KINGSTON.eot"));
pds->szURLtoEOT = CoTaskDupString (T2OLE(szBuffer)); // "res://"-type URL to EOT file
if (pds->szURLtoEOT)
{
TCHAR szSymbols[2];
szSymbols[0] = (TCHAR)(m_index + 'A');
szSymbols[1] = 0;
pds->szSymbolString = CoTaskDupString (T2OLE(szSymbols)); // 1 or more symbol characters
if (pds->szSymbolString)
{
#endif
// add button text
task->szText = CoTaskDupString (T2OLE(g_o_text[m_index]));
if (task->szText)
{
// add help string
task->szHelpString = CoTaskDupString (T2OLE(g_o_help[m_index]));
if (task->szHelpString)
{
// add action
task->eActionType = MMC_ACTION_ID;
task->nCommandID = _ttol(g_o_action[m_index]);
m_index++;
return S_OK; // all is well
}
CoTaskFreeString (task->szText);
}
#ifdef BITMAP_CASE
CoTaskFreeString (pdb->szMouseOverBitmap);
}
CoTaskFreeString (pdb->szMouseOffBitmap);
#else
CoTaskFreeString (pds->szSymbolString);
}
CoTaskFreeString (pds->szURLtoEOT);
}
CoTaskFreeString (pds->szFontFamilyName);
#endif
}
// we get here on failure
if (pceltFetched)
*pceltFetched = 0;
return S_FALSE; // failure
}