878 lines
26 KiB
C
878 lines
26 KiB
C
|
//###########################################################################
|
||
|
// Code for the Browse For Starting Folder
|
||
|
//###########################################################################
|
||
|
// Structure to pass information to browse for folder dialog
|
||
|
typedef struct _bfsf
|
||
|
{
|
||
|
HWND hwndOwner;
|
||
|
LPCITEMIDLIST pidlRoot; // Root of search. Typically desktop or my net
|
||
|
LPSTR pszDisplayName;// Return display name of item selected.
|
||
|
int *piImage; // where to return the Image index.
|
||
|
LPCSTR lpszTitle; // resource (or text to go in the banner over the tree.
|
||
|
UINT ulFlags; // Flags that control the return stuff
|
||
|
BFFCALLBACK lpfn;
|
||
|
LPARAM lParam;
|
||
|
HWND hwndDlg; // The window handle to the dialog
|
||
|
HWND hwndTree; // The tree control.
|
||
|
HTREEITEM htiCurParent; // tree item associated with Current shell folder
|
||
|
IShellFolder * psfParent; // Cache of the last IShell folder I needed...
|
||
|
LPITEMIDLIST pidlCurrent; // IDlist of current folder to select
|
||
|
BOOL fShowAllObjects; // Should we Show all ?
|
||
|
} BFSF, *PBFSF;
|
||
|
|
||
|
|
||
|
BOOL CALLBACK _BFSFDlgProc(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam);
|
||
|
LPITEMIDLIST _BFSFUpdateISHCache(PBFSF pbfsf, HTREEITEM hti, LPITEMIDLIST pidlItem);
|
||
|
|
||
|
|
||
|
// _BrowseForStartingFolder - Browse for a folder to start the
|
||
|
// search from.
|
||
|
|
||
|
|
||
|
|
||
|
// BUGBUG, give them a way to turn off the ok button.
|
||
|
|
||
|
LPITEMIDLIST WINAPI SHBrowseForFolder(LPBROWSEINFO lpbi)
|
||
|
{
|
||
|
LPITEMIDLIST lpRet;
|
||
|
BFSF bfsf =
|
||
|
{
|
||
|
lpbi->hwndOwner,
|
||
|
lpbi->pidlRoot,
|
||
|
lpbi->pszDisplayName,
|
||
|
&lpbi->iImage,
|
||
|
lpbi->lpszTitle,
|
||
|
lpbi->ulFlags,
|
||
|
lpbi->lpfn,
|
||
|
lpbi->lParam,
|
||
|
};
|
||
|
HCURSOR hcOld = SetCursor(LoadCursor(NULL,IDC_WAIT));
|
||
|
SHELLSTATE ss;
|
||
|
|
||
|
SHGetSetSettings(&ss, SSF_SHOWALLOBJECTS, FALSE);
|
||
|
bfsf.fShowAllObjects = ss.fShowAllObjects;
|
||
|
|
||
|
// Now Create the dialog that will be doing the browsing.
|
||
|
if (DialogBoxParam(HINST_THISDLL, MAKEINTRESOURCE(DLG_BROWSEFORFOLDER),
|
||
|
lpbi->hwndOwner, _BFSFDlgProc, (LPARAM)&bfsf))
|
||
|
lpRet = bfsf.pidlCurrent;
|
||
|
else
|
||
|
lpRet = NULL;
|
||
|
|
||
|
if (hcOld)
|
||
|
SetCursor(hcOld);
|
||
|
|
||
|
return lpRet;
|
||
|
}
|
||
|
|
||
|
void BFSFCallback(PBFSF pbfsf, UINT uMsg, LPARAM lParam)
|
||
|
{
|
||
|
if (pbfsf->lpfn) {
|
||
|
pbfsf->lpfn(pbfsf->hwndDlg, uMsg, lParam, pbfsf->lParam);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
// Some helper functions for processing the dialog
|
||
|
|
||
|
HTREEITEM _AddItemToTree(HWND hwndTree, HTREEITEM htiParent, LPITEMIDLIST pidl, int cChildren)
|
||
|
{
|
||
|
TV_INSERTSTRUCT tii;
|
||
|
|
||
|
// Initialize item to add with callback for everything
|
||
|
tii.item.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE |
|
||
|
TVIF_PARAM | TVIF_CHILDREN;
|
||
|
tii.hParent = htiParent;
|
||
|
tii.hInsertAfter = TVI_FIRST;
|
||
|
tii.item.iImage = I_IMAGECALLBACK;
|
||
|
tii.item.iSelectedImage = I_IMAGECALLBACK;
|
||
|
tii.item.pszText = LPSTR_TEXTCALLBACK; //
|
||
|
tii.item.cChildren = cChildren; // Assume it has children
|
||
|
tii.item.lParam = (LPARAM)pidl;
|
||
|
|
||
|
return TreeView_InsertItem(hwndTree, &tii);
|
||
|
}
|
||
|
|
||
|
|
||
|
LPITEMIDLIST _GetIDListFromTreeItem(HWND hwndTree, HTREEITEM hti)
|
||
|
{
|
||
|
LPITEMIDLIST pidl;
|
||
|
LPITEMIDLIST pidlT;
|
||
|
TV_ITEM tvi;
|
||
|
|
||
|
// If no hti passed in, get the selected on.
|
||
|
if (hti == NULL)
|
||
|
{
|
||
|
hti = TreeView_GetSelection(hwndTree);
|
||
|
if (hti == NULL)
|
||
|
return(NULL);
|
||
|
}
|
||
|
|
||
|
// now lets get the information about the item
|
||
|
tvi.mask = TVIF_PARAM | TVIF_HANDLE;
|
||
|
tvi.hItem = hti;
|
||
|
if (!TreeView_GetItem(hwndTree, &tvi))
|
||
|
return NULL; // Failed again
|
||
|
|
||
|
pidl = ILClone((LPITEMIDLIST)tvi.lParam);
|
||
|
|
||
|
// Now walk up parents.
|
||
|
while ((tvi.hItem = TreeView_GetParent(hwndTree, tvi.hItem)) && pidl)
|
||
|
{
|
||
|
if (!TreeView_GetItem(hwndTree, &tvi))
|
||
|
return(pidl); // will assume I messed up...
|
||
|
pidlT = ILCombine((LPITEMIDLIST)tvi.lParam, pidl);
|
||
|
|
||
|
ILFree(pidl);
|
||
|
|
||
|
pidl = pidlT;
|
||
|
|
||
|
}
|
||
|
return(pidl);
|
||
|
}
|
||
|
|
||
|
|
||
|
int CALLBACK _BFSFTreeCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
|
||
|
{
|
||
|
IShellFolder *psfParent = (IShellFolder *)lParamSort;
|
||
|
HRESULT hres = psfParent->lpVtbl->CompareIDs(psfParent, 0, (LPITEMIDLIST)lParam1, (LPITEMIDLIST)lParam2);
|
||
|
|
||
|
Assert(SUCCEEDED(hres));
|
||
|
|
||
|
return (short)SCODE_CODE(GetScode(hres));
|
||
|
}
|
||
|
|
||
|
void _BFSFSort(PBFSF pbfsf, HTREEITEM hti, IShellFolder * psf)
|
||
|
{
|
||
|
TV_SORTCB sSortCB;
|
||
|
|
||
|
sSortCB.hParent = hti;
|
||
|
sSortCB.lpfnCompare = _BFSFTreeCompare;
|
||
|
|
||
|
psf->lpVtbl->AddRef(psf);
|
||
|
sSortCB.lParam = (LPARAM)psf;
|
||
|
TreeView_SortChildrenCB(pbfsf->hwndTree, &sSortCB, FALSE);
|
||
|
psf->lpVtbl->Release(psf);
|
||
|
}
|
||
|
|
||
|
|
||
|
BOOL _BFSFHandleItemExpanding(PBFSF pbfsf, LPNM_TREEVIEW pnmtv)
|
||
|
{
|
||
|
LPITEMIDLIST pidlToExpand;
|
||
|
LPITEMIDLIST pidl;
|
||
|
IShellFolder * psf;
|
||
|
IShellFolder * psfDesktop = Desktop_GetShellFolder(TRUE);
|
||
|
BYTE bType;
|
||
|
DWORD grfFlags;
|
||
|
BOOL fPrinterTest = FALSE;
|
||
|
int cAdded = 0;
|
||
|
TV_ITEM tvi;
|
||
|
|
||
|
IEnumIDList * penum; // Enumerator in use.
|
||
|
|
||
|
if (pnmtv->action != TVE_EXPAND)
|
||
|
return FALSE;
|
||
|
|
||
|
if ((pnmtv->itemNew.state & TVIS_EXPANDEDONCE))
|
||
|
return FALSE;
|
||
|
|
||
|
// set this bit now because we might be reentered via the wnet apis
|
||
|
tvi.mask = TVIF_STATE;
|
||
|
tvi.hItem = pnmtv->itemNew.hItem;
|
||
|
tvi.state = TVIS_EXPANDEDONCE;
|
||
|
tvi.stateMask = TVIS_EXPANDEDONCE;
|
||
|
TreeView_SetItem(pbfsf->hwndTree, &tvi);
|
||
|
|
||
|
|
||
|
if (pnmtv->itemNew.hItem == NULL)
|
||
|
{
|
||
|
pnmtv->itemNew.hItem = TreeView_GetSelection(pbfsf->hwndTree);
|
||
|
if (pnmtv->itemNew.hItem == NULL)
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
pidlToExpand = _GetIDListFromTreeItem(pbfsf->hwndTree, pnmtv->itemNew.hItem);
|
||
|
|
||
|
if (pidlToExpand == NULL)
|
||
|
return FALSE;
|
||
|
|
||
|
// Now lets get the IShellFolder and iterator for this object
|
||
|
// special case to handle if the Pidl is the desktop
|
||
|
// This is rather gross, but the desktop appears to be simply a pidl
|
||
|
// of length 0 and ILIsEqual will not work...
|
||
|
if (pidlToExpand->mkid.cb == 0)
|
||
|
{
|
||
|
psf = psfDesktop;
|
||
|
psfDesktop->lpVtbl->AddRef(psf);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (FAILED(psfDesktop->lpVtbl->BindToObject(psfDesktop,
|
||
|
pidlToExpand, NULL, &IID_IShellFolder, &psf)))
|
||
|
{
|
||
|
ILFree(pidlToExpand);
|
||
|
return FALSE; // Could not get IShellFolder.
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Need to do a couple of special cases here to allow us to
|
||
|
// browse for a network printer. In this case if we are at server
|
||
|
// level we then need to change what we search for non folders when
|
||
|
// we are the level of a server.
|
||
|
if (pbfsf->ulFlags & BIF_BROWSEFORPRINTER)
|
||
|
{
|
||
|
grfFlags = SHCONTF_FOLDERS | SHCONTF_NETPRINTERSRCH;
|
||
|
pidl = ILFindLastID(pidlToExpand);
|
||
|
bType = SIL_GetType(pidl);
|
||
|
fPrinterTest = ((bType & (SHID_NET|SHID_INGROUPMASK))==SHID_NET_SERVER);
|
||
|
if (fPrinterTest)
|
||
|
grfFlags |= SHCONTF_NONFOLDERS;
|
||
|
}
|
||
|
else
|
||
|
grfFlags = SHCONTF_FOLDERS;
|
||
|
|
||
|
if (pbfsf->fShowAllObjects)
|
||
|
grfFlags |= SHCONTF_INCLUDEHIDDEN;
|
||
|
|
||
|
|
||
|
|
||
|
if (FAILED(psf->lpVtbl->EnumObjects(psf, pbfsf->hwndDlg, grfFlags, &penum)))
|
||
|
{
|
||
|
psf->lpVtbl->Release(psf);
|
||
|
ILFree(pidlToExpand);
|
||
|
return FALSE;
|
||
|
}
|
||
|
// psf->lpVtbl->AddRef(psf);
|
||
|
|
||
|
while (pidl = _NextIDL(psf, penum))
|
||
|
{
|
||
|
int cChildren = I_CHILDRENCALLBACK; // Do call back for children
|
||
|
//
|
||
|
// We need to special case here in the netcase where we onlyu
|
||
|
// browse down to workgroups...
|
||
|
//
|
||
|
//
|
||
|
// Here is where I also need to special case to not go below
|
||
|
// workgroups when the appropriate option is set.
|
||
|
//
|
||
|
bType = SIL_GetType(pidl);
|
||
|
if ((pbfsf->ulFlags & BIF_DONTGOBELOWDOMAIN) && (bType & SHID_NET))
|
||
|
{
|
||
|
switch (bType & (SHID_NET | SHID_INGROUPMASK))
|
||
|
{
|
||
|
case SHID_NET_SERVER:
|
||
|
ILFree(pidl); // Dont want to add this one
|
||
|
continue; // Try the next one
|
||
|
case SHID_NET_DOMAIN:
|
||
|
cChildren = 0; // Force to not have children;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
else if ((pbfsf->ulFlags & BIF_BROWSEFORCOMPUTER) && (bType & SHID_NET))
|
||
|
{
|
||
|
if ((bType & (SHID_NET | SHID_INGROUPMASK)) == SHID_NET_SERVER)
|
||
|
cChildren = 0; // Don't expand below it...
|
||
|
}
|
||
|
else if (fPrinterTest)
|
||
|
{
|
||
|
// Special case when we are only allowing printers.
|
||
|
// for now I will simply key on the fact that it is non-FS.
|
||
|
ULONG ulAttr = SFGAO_FILESYSTEM;
|
||
|
|
||
|
psf->lpVtbl->GetAttributesOf(psf, 1, &pidl, &ulAttr);
|
||
|
|
||
|
if ((ulAttr & SFGAO_FILESYSTEM)== 0)
|
||
|
{
|
||
|
cChildren = 0; // Force to not have children;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
ILFree(pidl); // Dont want to add this one
|
||
|
continue; // Try the next one
|
||
|
}
|
||
|
}
|
||
|
|
||
|
_AddItemToTree(pbfsf->hwndTree, pnmtv->itemNew.hItem,
|
||
|
pidl, cChildren);
|
||
|
cAdded++;
|
||
|
}
|
||
|
|
||
|
// Now Cleanup after ourself
|
||
|
penum->lpVtbl->Release(penum);
|
||
|
|
||
|
_BFSFSort(pbfsf, pnmtv->itemNew.hItem, psf);
|
||
|
psf->lpVtbl->Release(psf);
|
||
|
ILFree(pidlToExpand);
|
||
|
|
||
|
// If we did not add anything we should update this item to let
|
||
|
// the user know something happened.
|
||
|
//
|
||
|
if (cAdded == 0)
|
||
|
{
|
||
|
TV_ITEM tvi;
|
||
|
tvi.mask = TVIF_CHILDREN | TVIF_HANDLE; // only change the number of children
|
||
|
tvi.hItem = pnmtv->itemNew.hItem;
|
||
|
tvi.cChildren = 0;
|
||
|
|
||
|
TreeView_SetItem(pbfsf->hwndTree, &tvi);
|
||
|
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
void _BFSFHandleDeleteItem(PBFSF pbfsf, LPNM_TREEVIEW pnmtv)
|
||
|
{
|
||
|
// We need to free the IDLists that we allocated previously
|
||
|
if (pnmtv->itemOld.lParam != 0)
|
||
|
ILFree((LPITEMIDLIST)pnmtv->itemOld.lParam);
|
||
|
}
|
||
|
|
||
|
|
||
|
LPITEMIDLIST _BFSFUpdateISHCache(PBFSF pbfsf, HTREEITEM hti,
|
||
|
LPITEMIDLIST pidlItem)
|
||
|
{
|
||
|
HTREEITEM htiParent;
|
||
|
IShellFolder * psfDesktop = Desktop_GetShellFolder(TRUE);
|
||
|
|
||
|
if (pidlItem == NULL)
|
||
|
return(NULL);
|
||
|
|
||
|
// Need to handle the root case here!
|
||
|
htiParent = TreeView_GetParent(pbfsf->hwndTree, hti);
|
||
|
if ((htiParent != pbfsf->htiCurParent) || (pbfsf->psfParent == NULL))
|
||
|
{
|
||
|
LPITEMIDLIST pidl;
|
||
|
|
||
|
if (pbfsf->psfParent)
|
||
|
{
|
||
|
|
||
|
if (pbfsf->psfParent != psfDesktop)
|
||
|
pbfsf->psfParent->lpVtbl->Release(pbfsf->psfParent);
|
||
|
pbfsf->psfParent = NULL;
|
||
|
}
|
||
|
|
||
|
if (htiParent)
|
||
|
{
|
||
|
pidl = _GetIDListFromTreeItem(pbfsf->hwndTree, htiParent);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// If No Parent then the item here is one of our roots which
|
||
|
// should be fully qualified. So try to get the parent by
|
||
|
// decomposing the ID.
|
||
|
//
|
||
|
LPITEMIDLIST pidlT = (LPITEMIDLIST)ILFindLastID(pidlItem);
|
||
|
if (pidlT != pidlItem)
|
||
|
{
|
||
|
pidl = ILClone(pidlItem);
|
||
|
ILRemoveLastID(pidl);
|
||
|
pidlItem = pidlT;
|
||
|
}
|
||
|
else
|
||
|
pidl = NULL;
|
||
|
}
|
||
|
|
||
|
pbfsf->htiCurParent = htiParent;
|
||
|
|
||
|
// If still NULL then we use root of evil...
|
||
|
if (pidl == NULL || (pidl->mkid.cb == 0))
|
||
|
{
|
||
|
// Still one m
|
||
|
pbfsf->psfParent = psfDesktop;
|
||
|
|
||
|
if (pidl)
|
||
|
ILFree(pidl);
|
||
|
}
|
||
|
|
||
|
else
|
||
|
{
|
||
|
psfDesktop->lpVtbl->BindToObject(psfDesktop,
|
||
|
pidl, NULL, &IID_IShellFolder, &pbfsf->psfParent);
|
||
|
ILFree(pidl);
|
||
|
|
||
|
if (pbfsf->psfParent == NULL)
|
||
|
return NULL;
|
||
|
}
|
||
|
}
|
||
|
return(ILFindLastID(pidlItem));
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
void _BFSFGetDisplayInfo(PBFSF pbfsf, TV_DISPINFO *pnm)
|
||
|
{
|
||
|
TV_ITEM ti;
|
||
|
LPITEMIDLIST pidlItem = (LPITEMIDLIST)pnm->item.lParam;
|
||
|
|
||
|
if ((pnm->item.mask & (TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_TEXT | TVIF_CHILDREN)) == 0)
|
||
|
return; // nothing for us to do here.
|
||
|
|
||
|
pidlItem = _BFSFUpdateISHCache(pbfsf, pnm->item.hItem, pidlItem);
|
||
|
|
||
|
ti.mask = 0;
|
||
|
ti.hItem = (HTREEITEM)pnm->item.hItem;
|
||
|
|
||
|
// They are asking for IconIndex. See if we can find it now.
|
||
|
// Once found update their list, such that they wont call us back for
|
||
|
// it again.
|
||
|
if (pnm->item.mask & (TVIF_IMAGE | TVIF_SELECTEDIMAGE))
|
||
|
{
|
||
|
// We now need to map the item into the right image index.
|
||
|
ti.iImage = pnm->item.iImage = SHMapPIDLToSystemImageListIndex(
|
||
|
pbfsf->psfParent, pidlItem, &ti.iSelectedImage);
|
||
|
// we should save it back away to
|
||
|
pnm->item.iSelectedImage = ti.iSelectedImage;
|
||
|
ti.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE;
|
||
|
}
|
||
|
// Also see if this guy has any child folders
|
||
|
if (pnm->item.mask & TVIF_CHILDREN)
|
||
|
{
|
||
|
ULONG ulAttrs;
|
||
|
|
||
|
ulAttrs = SFGAO_HASSUBFOLDER;
|
||
|
pbfsf->psfParent->lpVtbl->GetAttributesOf(pbfsf->psfParent,
|
||
|
1, &pidlItem, &ulAttrs);
|
||
|
|
||
|
ti.cChildren = pnm->item.cChildren =
|
||
|
(ulAttrs & SFGAO_HASSUBFOLDER)? 1 : 0;
|
||
|
|
||
|
ti.mask |= TVIF_CHILDREN;
|
||
|
|
||
|
}
|
||
|
|
||
|
if (pnm->item.mask & TVIF_TEXT)
|
||
|
{
|
||
|
STRRET str;
|
||
|
pbfsf->psfParent->lpVtbl->GetDisplayNameOf(pbfsf->psfParent,
|
||
|
pidlItem, SHGDN_INFOLDER, &str);
|
||
|
|
||
|
StrRetToStrN(pnm->item.pszText, pnm->item.cchTextMax, &str, pidlItem);
|
||
|
ti.mask |= TVIF_TEXT;
|
||
|
ti.pszText = pnm->item.pszText;
|
||
|
}
|
||
|
|
||
|
// Update the item now
|
||
|
TreeView_SetItem(pbfsf->hwndTree, &ti);
|
||
|
}
|
||
|
|
||
|
|
||
|
void _BFSFHandleSelChanged(PBFSF pbfsf, LPNM_TREEVIEW pnmtv)
|
||
|
{
|
||
|
LPITEMIDLIST pidl;
|
||
|
ULONG ulAttrs = SFGAO_FILESYSTEM;
|
||
|
BYTE bType;
|
||
|
|
||
|
// We only need to do anything if we only want to return File system
|
||
|
// level objects.
|
||
|
if ((pbfsf->ulFlags & (BIF_RETURNONLYFSDIRS|BIF_RETURNFSANCESTORS|BIF_BROWSEFORPRINTER|BIF_BROWSEFORCOMPUTER)) == 0)
|
||
|
goto NotifySelChange;
|
||
|
|
||
|
// We need to get the attributes of this object...
|
||
|
pidl = _BFSFUpdateISHCache(pbfsf, pnmtv->itemNew.hItem,
|
||
|
(LPITEMIDLIST)pnmtv->itemNew.lParam);
|
||
|
|
||
|
if (pidl)
|
||
|
{
|
||
|
BOOL fEnable;
|
||
|
|
||
|
bType = SIL_GetType(pidl);
|
||
|
if ((pbfsf->ulFlags & (BIF_RETURNFSANCESTORS|BIF_RETURNONLYFSDIRS)) != 0)
|
||
|
{
|
||
|
int i;
|
||
|
// if this is the root pidl, then do a get attribs on 0
|
||
|
// so that we'll get the attributes on the root, rather than
|
||
|
// random returned values returned by FSFolder
|
||
|
if (ILIsEmpty(pidl)) {
|
||
|
i = 0;
|
||
|
} else
|
||
|
i = 1;
|
||
|
|
||
|
pbfsf->psfParent->lpVtbl->GetAttributesOf(pbfsf->psfParent,
|
||
|
i, &pidl, &ulAttrs);
|
||
|
|
||
|
fEnable = (((ulAttrs & SFGAO_FILESYSTEM) && (pbfsf->ulFlags & BIF_RETURNONLYFSDIRS)) ||
|
||
|
((ulAttrs & SFGAO_FILESYSANCESTOR) && (pbfsf->ulFlags & BIF_RETURNFSANCESTORS))) ||
|
||
|
((bType & (SHID_NET | SHID_INGROUPMASK)) == SHID_NET_SERVER);
|
||
|
}
|
||
|
else if ((pbfsf->ulFlags & BIF_BROWSEFORCOMPUTER) != 0)
|
||
|
fEnable = ((bType & (SHID_NET | SHID_INGROUPMASK)) == SHID_NET_SERVER);
|
||
|
else if ((pbfsf->ulFlags & BIF_BROWSEFORPRINTER) != 0)
|
||
|
{
|
||
|
// Printers are of type Share and usage Print...
|
||
|
fEnable = ((bType & (SHID_NET | SHID_INGROUPMASK)) == SHID_NET_SHARE);
|
||
|
}
|
||
|
|
||
|
EnableWindow(GetDlgItem(pbfsf->hwndDlg, IDOK),fEnable);
|
||
|
|
||
|
}
|
||
|
|
||
|
NotifySelChange:
|
||
|
if (pbfsf->lpfn) {
|
||
|
pidl = _GetIDListFromTreeItem(pbfsf->hwndTree, pnmtv->itemNew.hItem);
|
||
|
BFSFCallback(pbfsf, BFFM_SELCHANGED, (LPARAM)pidl);
|
||
|
ILFree(pidl);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
BOOL BrowseSelectPidl(PBFSF pbfsf, LPCITEMIDLIST pidl)
|
||
|
{
|
||
|
HTREEITEM htiParent;
|
||
|
LPITEMIDLIST pidlTemp;
|
||
|
LPITEMIDLIST pidlNext = NULL;
|
||
|
LPITEMIDLIST pidlParent = NULL;
|
||
|
BOOL fRet = FALSE;
|
||
|
|
||
|
htiParent = TreeView_GetChild(pbfsf->hwndTree, NULL);
|
||
|
if (htiParent) {
|
||
|
|
||
|
// step through each item of the pidl
|
||
|
for (;;) {
|
||
|
TreeView_Expand(pbfsf->hwndTree, htiParent, TVE_EXPAND);
|
||
|
pidlParent = _GetIDListFromTreeItem(pbfsf->hwndTree, htiParent);
|
||
|
if (!pidlParent)
|
||
|
break;
|
||
|
|
||
|
pidlNext = ILClone(pidl);
|
||
|
if (!pidlNext)
|
||
|
break;
|
||
|
|
||
|
pidlTemp = ILFindChild(pidlParent, pidlNext);
|
||
|
if (!pidlTemp)
|
||
|
break;
|
||
|
|
||
|
if (ILIsEmpty(pidlTemp)) {
|
||
|
// found it!
|
||
|
//
|
||
|
TreeView_SelectItem(pbfsf->hwndTree, htiParent);
|
||
|
fRet = TRUE;
|
||
|
break;
|
||
|
} else {
|
||
|
// loop to find the next item
|
||
|
HTREEITEM htiChild;
|
||
|
|
||
|
pidlTemp = ILGetNext(pidlTemp);
|
||
|
if (!pidlTemp)
|
||
|
break;
|
||
|
else
|
||
|
pidlTemp->mkid.cb = 0;
|
||
|
|
||
|
|
||
|
htiChild = TreeView_GetChild(pbfsf->hwndTree, htiParent);
|
||
|
while (htiChild) {
|
||
|
BOOL fEqual;
|
||
|
pidlTemp = _GetIDListFromTreeItem(pbfsf->hwndTree, htiChild);
|
||
|
if (!pidlTemp) {
|
||
|
htiChild = NULL;
|
||
|
break;
|
||
|
}
|
||
|
fEqual = ILIsEqual(pidlTemp, pidlNext);
|
||
|
|
||
|
ILFree(pidlTemp);
|
||
|
if (fEqual) {
|
||
|
break;
|
||
|
} else {
|
||
|
htiChild = TreeView_GetNextSibling(pbfsf->hwndTree, htiChild);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!htiChild) {
|
||
|
// we didn't find the next one... bail
|
||
|
break;
|
||
|
} else {
|
||
|
// the found child becomes the next parent
|
||
|
htiParent = htiChild;
|
||
|
ILFree(pidlParent);
|
||
|
ILFree(pidlNext);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (pidlParent) ILFree(pidlParent);
|
||
|
if (pidlNext) ILFree(pidlNext);
|
||
|
return fRet;
|
||
|
}
|
||
|
|
||
|
|
||
|
// _BFSFOnInitDlg - Process the init dialog
|
||
|
|
||
|
BOOL _BFSFOnInitDlg(HWND hwnd, HWND hwndFocus, LPARAM lParam)
|
||
|
{
|
||
|
HTREEITEM hti;
|
||
|
PBFSF pbfsf = (PBFSF)lParam;
|
||
|
HIMAGELIST himl;
|
||
|
LPSTR lpsz;
|
||
|
char szTitle[80]; // no title should be bigger than this!
|
||
|
HWND hwndTree;
|
||
|
|
||
|
lpsz = ResourceCStrToStr(HINST_THISDLL, pbfsf->lpszTitle);
|
||
|
SetDlgItemText(hwnd, IDD_BROWSETITLE, lpsz);
|
||
|
if (lpsz != pbfsf->lpszTitle)
|
||
|
{
|
||
|
LocalFree(lpsz);
|
||
|
lpsz = NULL;
|
||
|
}
|
||
|
|
||
|
|
||
|
SetWindowLong(hwnd, DWL_USER, (LONG)lParam);
|
||
|
pbfsf->hwndDlg = hwnd;
|
||
|
hwndTree = pbfsf->hwndTree = GetDlgItem(hwnd, IDD_FOLDERLIST);
|
||
|
|
||
|
if (hwndTree)
|
||
|
{
|
||
|
UINT swpFlags = SWP_FRAMECHANGED | SWP_NOMOVE | SWP_NOSIZE | SWP_NOZORDER
|
||
|
| SWP_NOACTIVATE;
|
||
|
RECT rc;
|
||
|
POINT pt = {0,0};
|
||
|
|
||
|
if (!(pbfsf->ulFlags & BIF_STATUSTEXT)) {
|
||
|
HWND hwndStatus = GetDlgItem(hwnd, IDD_BROWSESTATUS);
|
||
|
// nuke the status window
|
||
|
ShowWindow(hwndStatus, SW_HIDE);
|
||
|
MapWindowPoints(hwndStatus, hwnd, &pt, 1);
|
||
|
GetClientRect(hwndTree, &rc);
|
||
|
MapWindowPoints(hwndTree, hwnd, (POINT*)&rc, 2);
|
||
|
rc.top = pt.y;
|
||
|
swpFlags = SWP_FRAMECHANGED | SWP_NOZORDER | SWP_NOACTIVATE;
|
||
|
}
|
||
|
|
||
|
Shell_GetImageLists(NULL, &himl);
|
||
|
TreeView_SetImageList(hwndTree, himl, TVSIL_NORMAL);
|
||
|
|
||
|
SetWindowLong(hwndTree, GWL_EXSTYLE,
|
||
|
GetWindowLong(hwndTree, GWL_EXSTYLE) | WS_EX_CLIENTEDGE);
|
||
|
|
||
|
// Now try to get this window to know to recalc
|
||
|
SetWindowPos(hwndTree, NULL, rc.left, rc.top,
|
||
|
rc.right - rc.left, rc.bottom - rc.top, swpFlags);
|
||
|
|
||
|
}
|
||
|
|
||
|
// If they passed in a root, add it, else add the contents of the
|
||
|
// Root of evil... to the list as ROOT objects.
|
||
|
if (pbfsf->pidlRoot)
|
||
|
{
|
||
|
LPITEMIDLIST pidl;
|
||
|
if (!HIWORD(pbfsf->pidlRoot)) {
|
||
|
pidl = SHCloneSpecialIDList(NULL, (UINT)pbfsf->pidlRoot, TRUE);
|
||
|
} else {
|
||
|
pidl = ILClone(pbfsf->pidlRoot);
|
||
|
}
|
||
|
// Now lets insert the Root object
|
||
|
hti = _AddItemToTree(hwndTree, TVI_ROOT, pidl, 1);
|
||
|
// Still need to expand below this point. to the starting location
|
||
|
// That was passed in. But for now expand the first level.
|
||
|
TreeView_Expand(hwndTree, hti, TVE_EXPAND);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
LPCITEMIDLIST pidlDrives = GetSpecialFolderIDList(NULL, CSIDL_DRIVES, FALSE);
|
||
|
LPITEMIDLIST pidlDesktop = SHCloneSpecialIDList(NULL, CSIDL_DESKTOP, FALSE);
|
||
|
HTREEITEM htiRoot = _AddItemToTree(hwndTree, TVI_ROOT, pidlDesktop, 1);
|
||
|
|
||
|
// Expand the first level under the desktop
|
||
|
TreeView_Expand(hwndTree, htiRoot, TVE_EXPAND);
|
||
|
|
||
|
// Lets Preexpand the Drives portion....
|
||
|
hti = TreeView_GetChild(hwndTree, htiRoot);
|
||
|
while (hti)
|
||
|
{
|
||
|
LPITEMIDLIST pidl = _GetIDListFromTreeItem(hwndTree, hti);
|
||
|
if (ILIsEqual(pidl, pidlDrives))
|
||
|
{
|
||
|
|
||
|
TreeView_Expand(hwndTree, hti, TVE_EXPAND);
|
||
|
|
||
|
TreeView_SelectItem(hwndTree, hti);
|
||
|
ILFree(pidl);
|
||
|
break;
|
||
|
}
|
||
|
ILFree(pidl);
|
||
|
hti = TreeView_GetNextSibling(hwndTree, hti);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// go to our internal selection changed code to do any window enabling needed
|
||
|
{
|
||
|
NM_TREEVIEW nmtv;
|
||
|
hti = TreeView_GetSelection(hwndTree);
|
||
|
if (hti) {
|
||
|
TV_ITEM ti;
|
||
|
ti.mask = TVIF_PARAM;
|
||
|
ti.hItem = hti;
|
||
|
TreeView_GetItem(hwndTree, &ti);
|
||
|
nmtv.itemNew.hItem = hti;
|
||
|
nmtv.itemNew.lParam = ti.lParam;
|
||
|
|
||
|
_BFSFHandleSelChanged(pbfsf, &nmtv);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ((pbfsf->ulFlags & BIF_BROWSEFORCOMPUTER) != 0)
|
||
|
{
|
||
|
LoadString(HINST_THISDLL, IDS_FINDSEARCH_COMPUTER, szTitle, sizeof(szTitle));
|
||
|
SetWindowText(hwnd, szTitle);
|
||
|
}
|
||
|
else if ((pbfsf->ulFlags & BIF_BROWSEFORPRINTER) != 0)
|
||
|
{
|
||
|
LoadString(HINST_THISDLL, IDS_FINDSEARCH_PRINTER, szTitle, sizeof(szTitle));
|
||
|
SetWindowText(hwnd, szTitle);
|
||
|
}
|
||
|
|
||
|
BFSFCallback(pbfsf, BFFM_INITIALIZED, 0);
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
void _BFSFSetStatusText(PBFSF pbfsf, LPCSTR lpszText)
|
||
|
{
|
||
|
char szText[100];
|
||
|
if (!HIWORD(lpszText)) {
|
||
|
LoadString(HINST_THISDLL, LOWORD(lpszText), szText, sizeof(szText));
|
||
|
lpszText = szText;
|
||
|
}
|
||
|
SetDlgItemText(pbfsf->hwndDlg, IDD_BROWSESTATUS, lpszText);
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
// _BFSFOnCommand - Process the WM_COMMAND message
|
||
|
|
||
|
void _BFSFOnCommand(PBFSF pbfsf, int id, HWND hwndCtl, UINT codeNotify)
|
||
|
{
|
||
|
HTREEITEM hti;
|
||
|
switch (id)
|
||
|
{
|
||
|
case IDOK:
|
||
|
// We can now update the structure with the idlist of the item selected
|
||
|
hti = TreeView_GetSelection(pbfsf->hwndTree);
|
||
|
pbfsf->pidlCurrent = _GetIDListFromTreeItem(pbfsf->hwndTree,
|
||
|
hti);
|
||
|
if (pbfsf->pszDisplayName || pbfsf->piImage)
|
||
|
{
|
||
|
TV_ITEM tvi;
|
||
|
tvi.mask = (pbfsf->pszDisplayName)? (TVIF_TEXT | TVIF_IMAGE) :
|
||
|
TVIF_IMAGE;
|
||
|
tvi.hItem = hti;
|
||
|
tvi.pszText = pbfsf->pszDisplayName;
|
||
|
tvi.cchTextMax = MAX_PATH;
|
||
|
TreeView_GetItem(pbfsf->hwndTree, &tvi);
|
||
|
|
||
|
if (pbfsf->piImage)
|
||
|
*pbfsf->piImage = tvi.iImage;
|
||
|
}
|
||
|
EndDialog(pbfsf->hwndDlg, 1); // To return TRUE.
|
||
|
break;
|
||
|
case IDCANCEL:
|
||
|
EndDialog(pbfsf->hwndDlg, 0); // to return FALSE from this.
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
// _BSFSDlgProc - The dialog procedure for processing the browse
|
||
|
// for starting folder dialog.
|
||
|
|
||
|
#pragma data_seg(".text", "CODE")
|
||
|
const static DWORD aBrowseHelpIDs[] = { // Context Help IDs
|
||
|
IDD_BROWSETITLE, NO_HELP,
|
||
|
IDD_BROWSESTATUS, NO_HELP,
|
||
|
IDD_FOLDERLIST, IDH_BROWSELIST,
|
||
|
|
||
|
0, 0
|
||
|
};
|
||
|
#pragma data_seg()
|
||
|
|
||
|
BOOL CALLBACK _BFSFDlgProc(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
|
||
|
{
|
||
|
PBFSF pbfsf = (PBFSF)GetWindowLong(hwndDlg, DWL_USER);
|
||
|
|
||
|
switch (msg) {
|
||
|
HANDLE_MSG(pbfsf, WM_COMMAND, _BFSFOnCommand);
|
||
|
HANDLE_MSG(hwndDlg, WM_INITDIALOG, _BFSFOnInitDlg);
|
||
|
|
||
|
case WM_DESTROY:
|
||
|
if (pbfsf->psfParent && (pbfsf->psfParent != Desktop_GetShellFolder(TRUE)))
|
||
|
{
|
||
|
pbfsf->psfParent->lpVtbl->Release(pbfsf->psfParent);
|
||
|
pbfsf->psfParent = NULL;
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case BFFM_SETSTATUSTEXT:
|
||
|
_BFSFSetStatusText(pbfsf, (LPCSTR)lParam);
|
||
|
break;
|
||
|
|
||
|
case BFFM_SETSELECTION:
|
||
|
{
|
||
|
BOOL fRet;
|
||
|
|
||
|
// wParam TRUE means path, not pidl
|
||
|
if (wParam) {
|
||
|
lParam = (LPARAM)SHSimpleIDListFromPath((LPSTR)lParam);
|
||
|
if (!lParam)
|
||
|
return FALSE;
|
||
|
}
|
||
|
fRet = BrowseSelectPidl(pbfsf, (LPITEMIDLIST)lParam);
|
||
|
|
||
|
if (wParam)
|
||
|
ILFree((LPITEMIDLIST)lParam);
|
||
|
return fRet;
|
||
|
}
|
||
|
|
||
|
case BFFM_ENABLEOK:
|
||
|
EnableWindow(GetDlgItem(hwndDlg, IDOK), lParam);
|
||
|
break;
|
||
|
|
||
|
|
||
|
case WM_NOTIFY:
|
||
|
switch (((NMHDR *)lParam)->code)
|
||
|
{
|
||
|
case TVN_GETDISPINFO:
|
||
|
_BFSFGetDisplayInfo(pbfsf, (TV_DISPINFO *)lParam);
|
||
|
break;
|
||
|
|
||
|
case TVN_ITEMEXPANDING:
|
||
|
SetCursor(LoadCursor(NULL, IDC_WAIT));
|
||
|
_BFSFHandleItemExpanding(pbfsf, (LPNM_TREEVIEW)lParam);
|
||
|
break;
|
||
|
|
||
|
case TVN_ITEMEXPANDED:
|
||
|
SetCursor(LoadCursor(NULL, IDC_ARROW));
|
||
|
break;
|
||
|
|
||
|
case TVN_DELETEITEM:
|
||
|
_BFSFHandleDeleteItem(pbfsf, (LPNM_TREEVIEW)lParam);
|
||
|
break;
|
||
|
|
||
|
case TVN_SELCHANGED:
|
||
|
_BFSFHandleSelChanged(pbfsf, (LPNM_TREEVIEW)lParam);
|
||
|
break;
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case WM_HELP:
|
||
|
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, NULL,
|
||
|
HELP_WM_HELP, (DWORD)(LPSTR) aBrowseHelpIDs);
|
||
|
break;
|
||
|
|
||
|
case WM_CONTEXTMENU:
|
||
|
WinHelp((HWND) wParam, NULL, HELP_CONTEXTMENU,
|
||
|
(DWORD)(LPVOID) aBrowseHelpIDs);
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|