BOOL install_util::DeleteFolder(LPCTSTR pszFolder)
{
if(IsFolderEmpty(pszFolder))
return RemoveDirectory(pszFolder);
//下面的实现据说有隐患。应改为递归删除所有子文件及文件夹
_TCHAR szPath[MAX_PATH + 1] = {0};
_sntprintf_s(szPath, _countof(szPath), sizeof(szPath), _TEXT("%s%c"), pszFolder, 0);
SHFILEOPSTRUCT fos ;
ZeroMemory(&fos, sizeof( fos)) ;
fos.hwnd = HWND_DESKTOP;
fos.wFunc = FO_DELETE ;
fos.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
fos.pFrom = szPath;
// 删除文件夹及其内容
if (0 == SHFileOperation(&fos))
return TRUE;
wstring tmpFile = szPath;
tmpFile += L"_yytmp";
if (MoveFileEx(szPath, tmpFile.c_str(), MOVEFILE_REPLACE_EXISTING))
return MoveFileEx(tmpFile.c_str(),NULL,MOVEFILE_DELAY_UNTIL_REBOOT);
return FALSE;
}
/* #FN#
Fills a branch of the TreeView control.
Given the shell folder, enumerate the subitems of this folder, and
add the appropriate items to the tree. This function enumerates the
items in the folder identifed by lpsf. Note that since we are filling
the left hand pane, we will only add items that are folders and/or
have sub-folders. We *could* put all items in here if we wanted, but
that's not the intent. */
void
/* #AS#
Nothing */
CShellTree::
FillTreeView(
LPSHELLFOLDER lpsf, /* #IN# Pointer to shell folder that we want to enumerate items */
LPITEMIDLIST lpifq, /* #IN# Fully qualified item id list to the item that we are enumerating items for; in other words, this is the PIDL to the item identified by the lpsf parameter */
HTREEITEM hParent /* #IN# Parent node */
)
{
TV_ITEM tvi; /* TreeView Item */
TV_INSERTSTRUCT tvins; /* TreeView Insert Struct */
HTREEITEM hPrev = NULL; /* Previous Item Added */
LPSHELLFOLDER lpsf2 = NULL;
LPENUMIDLIST lpe = NULL;
LPITEMIDLIST lpi = NULL;
LPITEMIDLIST lpifqThisItem = NULL;
LPTVITEMDATA lptvid = NULL;
LPMALLOC lpMalloc = NULL;
ULONG ulFetched = 0;
HWND hwnd = ::GetParent( m_hWnd );
char szBuff[ MAX_PATH + 1 ];
char szPath[ MAX_PATH + 1 ];
HRESULT hr;
/* Allocate a shell memory object */
hr = ::SHGetMalloc( &lpMalloc );
if( FAILED(hr) )
return;
if( SUCCEEDED(hr) )
{
/* Get the IEnumIDList object for the given folder */
hr = lpsf->EnumObjects( hwnd, SHCONTF_FOLDERS | SHCONTF_NONFOLDERS | SHCONTF_INCLUDEHIDDEN, &lpe );
if( S_OK == hr && lpe )
{
/* Enumerate throught the list of folder and non-folder objects */
while( S_OK == lpe->Next( 1, &lpi, &ulFetched ) )
{
/* Create a fully qualified path to the current item. The SH* shell API's
take a fully qualified path pidl, (see GetIcon above where I call
SHGetFileInfo) whereas the interface methods take a relative path pidl */
ULONG ulAttrs = SFGAO_FOLDER | SFGAO_HASSUBFOLDER;
if( !m_bFolderMode )
ulAttrs |= SFGAO_FILESYSTEM | SFGAO_LINK;
/* Determine what type of object we have */
lpsf->GetAttributesOf( 1, (const struct _ITEMIDLIST **)&lpi, &ulAttrs );
if( m_bFolderMode && ulAttrs & (SFGAO_HASSUBFOLDER | SFGAO_FOLDER) ||
(!m_bFolderMode && ulAttrs & (SFGAO_HASSUBFOLDER | SFGAO_FOLDER | SFGAO_FILESYSTEM) && !(ulAttrs & SFGAO_LINK)) )
{
tvi.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
/* We need this next if statement so that we don't add things like
the MSN to our tree. MSN is not a folder, but according to the
shell it has subfolders */
/* OK, let's get some memory for our ITEMDATA struct */
lptvid = (LPTVITEMDATA)lpMalloc->Alloc( sizeof(TVITEMDATA) );
if( !lptvid )
goto Done; /* Error - could not allocate memory */
/* Now get the friendly name that we'll put in the treeview */
if( !GetName( lpsf, lpi, SHGDN_NORMAL, szBuff ) )
goto Done; /* Error - could not get friendly name */
tvi.pszText = szBuff;
tvi.cchTextMax = MAX_PATH;
/* Allocate/create the fully qualified PIDL, consisting
of the parents full PIDL and our relative PIDL */
lpifqThisItem = ConcatPidls( lpifq, lpi );
if( ulAttrs & SFGAO_FOLDER && (ulAttrs & SFGAO_HASSUBFOLDER ||
/* There are not any subfolders but what about files? */
(!m_bFolderMode && SHGetPathFromIDList( lpifqThisItem, szPath ) && !IsFolderEmpty( szPath ))) )
{
/* This item has sub-folders, so let's put the + in the TreeView.
The first time the user clicks on the item, we'll populate the
sub-folders */
tvi.cChildren = 1;
tvi.mask |= TVIF_CHILDREN;
}
/* Now, make a copy of the ITEMIDLIST (non-qualified) */
lptvid->lpi = CopyITEMID( lpMalloc, lpi );
tvi.iImage =
GetItemIcon( lpifqThisItem,
SHGFI_PIDL |
SHGFI_SYSICONINDEX |
SHGFI_SMALLICON );
tvi.iSelectedImage =
GetItemIcon( lpifqThisItem,
SHGFI_PIDL |
SHGFI_SYSICONINDEX |
SHGFI_SMALLICON |
SHGFI_OPENICON );
lptvid->lpsfParent = lpsf; /* Store the parent folders SF */
lpsf->AddRef(); /* Increment our saved reference */
/* Now create another PIDL from our Full parents PIDL and
the releative one that we'll save */
lptvid->lpifq = ConcatPidls( lpifq, lpi );
/* Populate the TreeView Insert Struct. The item is the one
filled above. Insert it after the last item inserted at
this level. And indicate this is a root entry */
tvi.lParam = (LPARAM)lptvid;
tvins.item = tvi;
tvins.hInsertAfter = hPrev;
tvins.hParent = hParent;
/* Add the item to the tree */
hPrev = InsertItem( &tvins );
/* Free this item with task allocator */
lpMalloc->Free( lpifqThisItem );
lpifqThisItem = 0;
}
lpMalloc->Free( lpi ); /* Free the pidl that the shell gave us */
lpi = 0;
}
}
}
else
return;
Done:
if( lpe )
lpe->Release();
/* The following 2 if statements will only be TRUE if we got here
on an error condition from the "goto" statement. Otherwise, we
free this memory at the end of the while loop above */
if( lpi && lpMalloc )
lpMalloc->Free( lpi );
if( lpifqThisItem && lpMalloc )
lpMalloc->Free( lpifqThisItem );
if( lpMalloc )
lpMalloc->Release();
} /* #OF# CShellTree::FillTreeView */
BOOL CPropertiesCtrl::OnExpanding( HTREEITEM hItem, UINT nCode )
{
int i, nChild = 0;
CFTCItemData * pItemData = (CFTCItemData*)GetItemData( hItem );
switch( nCode )
{
case TVE_EXPAND:
{
CWaitCursor cur;
nChild = GetChildCount( hItem );
for( i = 0; i < nChild; i++ )
{
DelItemRecursive( GetNextItem( hItem, TVGN_CHILD ) );
Invalidate();
}
CComPtr<IEnumIDList> piIDList;
ULONG nRes = 0;
ITEMIDLIST * pItem = NULL;
HRESULT hr = pItemData->m_piFolder->EnumObjects( NULL, SHCONTF_FOLDERS, &piIDList );
_STRRET strret;
CSCADString sPath, sName;
IExtractIcon * piEI = NULL;
IShellFolder * piChildFolder = NULL;
if( hr != NOERROR )
{
Invalidate();
return FALSE;
}
for( hr = piIDList->Next( 1, &pItem, &nRes ); hr == NOERROR && nRes > 0; hr = piIDList->Next( 1, &pItem, &nRes ) )
{
hr = pItemData->m_piFolder->GetDisplayNameOf( pItem, SHGDN_INFOLDER, &strret );
sName = StrRetToStr(pItem, strret, true, m_piMalloc );
hr = pItemData->m_piFolder->GetDisplayNameOf( pItem, SHGDN_NORMAL | SHGDN_FORPARSING, &strret );
sPath = StrRetToStr(pItem, strret, true, m_piMalloc );
hr = pItemData->m_piFolder->GetUIObjectOf( NULL, 1, (LPCITEMIDLIST*)&pItem, IID_IExtractIcon, NULL, (void**)&piEI );
hr = pItemData->m_piFolder->BindToObject( pItem, NULL, IID_IShellFolder, (void**)&piChildFolder );
if( piChildFolder )
{
if( !piEI )
{
//hr = piChildFolder->CreateViewObject( NULL, IID_IExtractIcon, (void**)&piEI );
}
//InsertFolder( pItem, piChildFolder, piEI, sName, sPath, hItem );
piChildFolder->Release();
piChildFolder = NULL;
}
if( piEI )
{
piEI->Release();
piEI = NULL;
}
}
}
return TRUE;
default:
nChild = GetChildCount( hItem );
for( i = 0; i < nChild; i++ )
DelItemRecursive( GetNextItem( hItem, TVGN_CHILD ) );
if( !IsFolderEmpty( pItemData->m_piFolder ) )
InsertItem( _T(""), hItem, TVI_LAST );
break;
}
return TRUE;
}
