780 lines
23 KiB
C++
780 lines
23 KiB
C++
|
#include "pch.h"
|
||
|
|
#include "thisdll.h"
|
||
|
|
#include "cowsite.h"
|
||
|
|
#include <shlobj.h>
|
||
|
|
#include "ids.h"
|
||
|
|
|
||
|
|
// Context menu offset IDs
|
||
|
|
enum {
|
||
|
|
CMD_ORGANIZE = 0,
|
||
|
|
CMD_ORGANIZE_DEEP = 1,
|
||
|
|
CMD_ORGANIZE_FLAT = 2,
|
||
|
|
CMD_MAX = 3,
|
||
|
|
};
|
||
|
|
|
||
|
|
class COrganizeFiles;
|
||
|
|
|
||
|
|
class COrganizeFiles : public IContextMenu, IShellExtInit, INamespaceWalkCB, CObjectWithSite
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
COrganizeFiles();
|
||
|
|
|
||
|
|
// IUnknown
|
||
|
|
STDMETHODIMP QueryInterface(REFIID riid, void **ppv);
|
||
|
|
STDMETHODIMP_(ULONG) AddRef();
|
||
|
|
STDMETHODIMP_(ULONG) Release();
|
||
|
|
|
||
|
|
// IShellExtInit
|
||
|
|
STDMETHODIMP Initialize(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hKeyID);
|
||
|
|
|
||
|
|
// IContextMenu
|
||
|
|
STDMETHODIMP QueryContextMenu(HMENU hMenu, UINT uIndex, UINT uIDFirst, UINT uIDLast, UINT uFlags);
|
||
|
|
STDMETHODIMP InvokeCommand(LPCMINVOKECOMMANDINFO pcmici);
|
||
|
|
STDMETHODIMP GetCommandString(UINT_PTR uID, UINT uFlags, UINT *res, LPSTR pName, UINT ccMax);
|
||
|
|
|
||
|
|
// INamespaceWalkCB
|
||
|
|
STDMETHODIMP FoundItem(IShellFolder *psf, LPCITEMIDLIST pidl);
|
||
|
|
STDMETHODIMP EnterFolder(IShellFolder *psf, LPCITEMIDLIST pidl);
|
||
|
|
STDMETHODIMP LeaveFolder(IShellFolder *psf, LPCITEMIDLIST pidl);
|
||
|
|
STDMETHOD(InitializeProgressDialog)(LPWSTR *ppszTitle, LPWSTR *ppszCancel)
|
||
|
|
{ *ppszTitle = NULL; *ppszCancel = NULL; return E_NOTIMPL; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
~COrganizeFiles();
|
||
|
|
|
||
|
|
class CMD_THREAD_DATA
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
CMD_THREAD_DATA(COrganizeFiles *pof, UINT idCmd) : _pof(pof), _idCmd(idCmd)
|
||
|
|
{
|
||
|
|
_pof->AddRef();
|
||
|
|
_pstmDataObj = NULL;
|
||
|
|
}
|
||
|
|
~CMD_THREAD_DATA()
|
||
|
|
{
|
||
|
|
_pof->Release();
|
||
|
|
ATOMICRELEASE(_pstmDataObj);
|
||
|
|
}
|
||
|
|
|
||
|
|
COrganizeFiles *_pof; // back ptr to object
|
||
|
|
UINT _idCmd;
|
||
|
|
IStream *_pstmDataObj; // the IDataObject marshalled to the background thread
|
||
|
|
};
|
||
|
|
|
||
|
|
static DWORD CALLBACK COrganizeFiles::_CmdThreadProc(void *pv);
|
||
|
|
void _CreateCmdThread(UINT idCmd);
|
||
|
|
void _DoCmd(UINT idCmd, IDataObject *pdtobj);
|
||
|
|
void _DoOrganizeMusic(UINT idCmd, IDataObject *pdtobj);
|
||
|
|
HRESULT _GetPropertyUI(IPropertyUI **pppui);
|
||
|
|
HRESULT _NameFromPropertiesString(LPCTSTR pszProps, IPropertySetStorage *ppss, LPTSTR pszRoot, LPTSTR pszName, UINT cchName);
|
||
|
|
|
||
|
|
LONG _cRef;
|
||
|
|
IDataObject *_pdtobj;
|
||
|
|
|
||
|
|
IPropertyUI *_ppui;
|
||
|
|
BOOL _bCountFiles; // call back is in "count files" mode
|
||
|
|
UINT _cFilesTotal; // total computed in the count
|
||
|
|
UINT _cFileCur; // current, for progress UI
|
||
|
|
IProgressDialog *_ppd;
|
||
|
|
TCHAR _szRootFolder[MAX_PATH];
|
||
|
|
LPCTSTR _pszProps;
|
||
|
|
LPTSTR _pszTemplateFlat;
|
||
|
|
LPTSTR _pszTemplate;
|
||
|
|
};
|
||
|
|
|
||
|
|
COrganizeFiles::COrganizeFiles() : _cRef(1)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
COrganizeFiles::~COrganizeFiles()
|
||
|
|
{
|
||
|
|
CoTaskMemFree(_pszTemplateFlat);
|
||
|
|
CoTaskMemFree(_pszTemplate);
|
||
|
|
|
||
|
|
IUnknown_Set(&_punkSite, NULL);
|
||
|
|
IUnknown_Set((IUnknown**)&_pdtobj, NULL);
|
||
|
|
IUnknown_Set((IUnknown**)&_ppui, NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDAPI COrganizeFiles_CreateInstance(IUnknown *pUnkOuter, IUnknown **ppunk, LPCOBJECTINFO poi)
|
||
|
|
{
|
||
|
|
COrganizeFiles *psid = new COrganizeFiles();
|
||
|
|
if (!psid)
|
||
|
|
{
|
||
|
|
*ppunk = NULL; // incase of failure
|
||
|
|
return E_OUTOFMEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT hr = psid->QueryInterface(IID_PPV_ARG(IUnknown, ppunk));
|
||
|
|
psid->Release();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP COrganizeFiles::QueryInterface(REFIID riid, void **ppv)
|
||
|
|
{
|
||
|
|
static const QITAB qit[] =
|
||
|
|
{
|
||
|
|
QITABENT(COrganizeFiles, IShellExtInit),
|
||
|
|
QITABENT(COrganizeFiles, IContextMenu),
|
||
|
|
QITABENT(COrganizeFiles, IObjectWithSite),
|
||
|
|
QITABENT(COrganizeFiles, INamespaceWalkCB),
|
||
|
|
{ 0 },
|
||
|
|
};
|
||
|
|
return QISearch(this, qit, riid, ppv);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG) COrganizeFiles::AddRef()
|
||
|
|
{
|
||
|
|
return InterlockedIncrement(&_cRef);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG) COrganizeFiles::Release()
|
||
|
|
{
|
||
|
|
if (InterlockedDecrement(&_cRef))
|
||
|
|
return _cRef;
|
||
|
|
|
||
|
|
delete this;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
// IShellExtInit
|
||
|
|
STDMETHODIMP COrganizeFiles::Initialize(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hKeyID)
|
||
|
|
{
|
||
|
|
IUnknown_Set((IUnknown**)&_pdtobj, pdtobj);
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
// IContextMenu
|
||
|
|
STDMETHODIMP COrganizeFiles::QueryContextMenu(HMENU hMenu, UINT uIndex, UINT uIDFirst, UINT uIDLast, UINT uFlags)
|
||
|
|
{
|
||
|
|
if (NULL == _pszTemplate)
|
||
|
|
{
|
||
|
|
IAssociationArray *paa;
|
||
|
|
if (SUCCEEDED(IUnknown_QueryService(_punkSite, SID_CtxQueryAssociations, IID_PPV_ARG(IAssociationArray, &paa))))
|
||
|
|
{
|
||
|
|
paa->QueryString(ASSOCELEM_MASK_QUERYNORMAL, AQN_NAMED_VALUE, L"OrganizeTemplate", &_pszTemplate);
|
||
|
|
paa->QueryString(ASSOCELEM_MASK_QUERYNORMAL, AQN_NAMED_VALUE, L"OrganizeTemplateFlat", &_pszTemplateFlat);
|
||
|
|
paa->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!(uFlags & CMF_DEFAULTONLY))
|
||
|
|
{
|
||
|
|
InsertMenu(hMenu, uIndex++, MF_BYPOSITION | MF_SEPARATOR, 0, NULL);
|
||
|
|
|
||
|
|
TCHAR szBuffer[128];
|
||
|
|
LoadString(m_hInst, IDS_ORGANIZE_MUSIC, szBuffer, ARRAYSIZE(szBuffer));
|
||
|
|
InsertMenu(hMenu, uIndex++, MF_BYPOSITION | MF_STRING, uIDFirst + CMD_ORGANIZE, szBuffer);
|
||
|
|
|
||
|
|
// if (uFlags & CMF_EXTENDEDVERBS)
|
||
|
|
{
|
||
|
|
LoadString(m_hInst, IDS_ORGANIZE_MUSIC_FLAT, szBuffer, ARRAYSIZE(szBuffer));
|
||
|
|
InsertMenu(hMenu, uIndex++, MF_BYPOSITION | MF_STRING, uIDFirst + CMD_ORGANIZE_FLAT, szBuffer);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return MAKE_HRESULT(SEVERITY_SUCCESS, 0, CMD_MAX);
|
||
|
|
}
|
||
|
|
|
||
|
|
typedef struct
|
||
|
|
{
|
||
|
|
LPCTSTR psz;
|
||
|
|
UINT id;
|
||
|
|
} CMD_MAP;
|
||
|
|
|
||
|
|
const CMD_MAP c_szVerbs[] =
|
||
|
|
{
|
||
|
|
{TEXT("OrganizeMusic"), CMD_ORGANIZE},
|
||
|
|
{TEXT("OrganizeMusicFlat"), CMD_ORGANIZE_FLAT},
|
||
|
|
{TEXT("OrganizeMusicDeep"), CMD_ORGANIZE_DEEP},
|
||
|
|
};
|
||
|
|
|
||
|
|
UINT GetCommandId(LPCMINVOKECOMMANDINFO pcmici, const CMD_MAP rgMap[], UINT cMap)
|
||
|
|
{
|
||
|
|
UINT id = -1;
|
||
|
|
|
||
|
|
if (IS_INTRESOURCE(pcmici->lpVerb))
|
||
|
|
{
|
||
|
|
id = LOWORD((UINT_PTR)pcmici->lpVerb);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
TCHAR szCmd[64];
|
||
|
|
szCmd[0] = 0;
|
||
|
|
|
||
|
|
// Check first whether the caller passed a EX structure by checking
|
||
|
|
// the size...
|
||
|
|
if (pcmici->cbSize == sizeof(CMINVOKECOMMANDINFOEX))
|
||
|
|
{
|
||
|
|
LPCMINVOKECOMMANDINFOEX pcmiciEX = (LPCMINVOKECOMMANDINFOEX)pcmici;
|
||
|
|
if (pcmici->fMask & CMIC_MASK_UNICODE)
|
||
|
|
{
|
||
|
|
SHUnicodeToTChar(pcmiciEX->lpVerbW, szCmd, ARRAYSIZE(szCmd));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// If we don't yet have a command verb, it must have been passed
|
||
|
|
// ANSI or in a regular CMINVOKECOMMANDINFO structure (which means
|
||
|
|
// it is by default ANSI). In either case, we need to convert it to
|
||
|
|
// UNICODE before proceeding...
|
||
|
|
|
||
|
|
if (szCmd[0] == 0)
|
||
|
|
{
|
||
|
|
SHAnsiToTChar(pcmici->lpVerb, szCmd, ARRAYSIZE(szCmd));
|
||
|
|
}
|
||
|
|
|
||
|
|
for (UINT i = 0; i < cMap; i++)
|
||
|
|
{
|
||
|
|
if (!StrCmpIC(szCmd, rgMap[i].psz))
|
||
|
|
{
|
||
|
|
id = rgMap[i].id;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return id;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP COrganizeFiles::InvokeCommand(LPCMINVOKECOMMANDINFO pcmici)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
UINT idCmd = GetCommandId(pcmici, c_szVerbs, ARRAYSIZE(c_szVerbs));
|
||
|
|
switch (idCmd)
|
||
|
|
{
|
||
|
|
case CMD_ORGANIZE:
|
||
|
|
case CMD_ORGANIZE_DEEP:
|
||
|
|
case CMD_ORGANIZE_FLAT:
|
||
|
|
_CreateCmdThread(idCmd);
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP COrganizeFiles::GetCommandString(UINT_PTR uID, UINT uFlags, UINT *res, LPSTR pName, UINT cchMax)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
UINT idSel = (UINT)uID;
|
||
|
|
|
||
|
|
switch (uFlags)
|
||
|
|
{
|
||
|
|
case GCS_VERBW:
|
||
|
|
case GCS_VERBA:
|
||
|
|
if (idSel < ARRAYSIZE(c_szVerbs))
|
||
|
|
{
|
||
|
|
if (uFlags == GCS_VERBW)
|
||
|
|
{
|
||
|
|
SHTCharToUnicode(c_szVerbs[idSel].psz, (LPWSTR)pName, cchMax);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
SHTCharToAnsi(c_szVerbs[idSel].psz, pName, cchMax);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case GCS_HELPTEXTA:
|
||
|
|
case GCS_HELPTEXTW:
|
||
|
|
case GCS_VALIDATEA:
|
||
|
|
case GCS_VALIDATEW:
|
||
|
|
hr = E_NOTIMPL;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CALLBACK COrganizeFiles::_CmdThreadProc(void *pv)
|
||
|
|
{
|
||
|
|
CMD_THREAD_DATA *potd = (CMD_THREAD_DATA *)pv;
|
||
|
|
IDataObject *pdtobj;
|
||
|
|
HRESULT hr = CoGetInterfaceAndReleaseStream(potd->_pstmDataObj, IID_PPV_ARG(IDataObject, &pdtobj));
|
||
|
|
potd->_pstmDataObj = NULL;
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
potd->_pof->_DoCmd(potd->_idCmd, pdtobj);
|
||
|
|
pdtobj->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
delete potd;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void COrganizeFiles::_CreateCmdThread(UINT idCmd)
|
||
|
|
{
|
||
|
|
CMD_THREAD_DATA *potd = new CMD_THREAD_DATA(this, idCmd);
|
||
|
|
if (potd)
|
||
|
|
{
|
||
|
|
if (FAILED(CoMarshalInterThreadInterfaceInStream(IID_IDataObject, _pdtobj, &potd->_pstmDataObj)) ||
|
||
|
|
!SHCreateThread(_CmdThreadProc, potd, CTF_COINIT, NULL))
|
||
|
|
{
|
||
|
|
delete potd;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void COrganizeFiles::_DoCmd(UINT idCmd, IDataObject *pdtobj)
|
||
|
|
{
|
||
|
|
switch (idCmd)
|
||
|
|
{
|
||
|
|
case CMD_ORGANIZE:
|
||
|
|
case CMD_ORGANIZE_DEEP:
|
||
|
|
case CMD_ORGANIZE_FLAT:
|
||
|
|
_DoOrganizeMusic(idCmd, pdtobj);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT ReadPropertyAsString(IPropertyUI *ppui, IPropertySetStorage *ppss, LPCSHCOLUMNID pscid, LPTSTR psz, UINT cch)
|
||
|
|
{
|
||
|
|
*psz = 0;
|
||
|
|
|
||
|
|
IPropertyStorage *pps;
|
||
|
|
HRESULT hr = ppss->Open(pscid->fmtid, STGM_READ | STGM_SHARE_EXCLUSIVE, &pps);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
PROPSPEC ps = {PRSPEC_PROPID, pscid->pid};
|
||
|
|
PROPVARIANT v = {0};
|
||
|
|
hr = pps->ReadMultiple(1, &ps, &v);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
hr = ppui->FormatForDisplay(pscid->fmtid, pscid->pid, &v, PUIFFDF_DEFAULT, psz, cch);
|
||
|
|
if (SUCCEEDED(hr) && (0 == *psz))
|
||
|
|
{
|
||
|
|
hr = E_FAIL;
|
||
|
|
}
|
||
|
|
PropVariantClear(&v);
|
||
|
|
}
|
||
|
|
pps->Release();
|
||
|
|
}
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
void FixBadFilenameChars(LPTSTR pszName)
|
||
|
|
{
|
||
|
|
while (*pszName)
|
||
|
|
{
|
||
|
|
switch (*pszName)
|
||
|
|
{
|
||
|
|
case '?':
|
||
|
|
case '*':
|
||
|
|
case '/':
|
||
|
|
case '\\':
|
||
|
|
case '!':
|
||
|
|
*pszName = '_';
|
||
|
|
break;
|
||
|
|
|
||
|
|
case '\"':
|
||
|
|
*pszName = '\'';
|
||
|
|
break;
|
||
|
|
case ':':
|
||
|
|
*pszName = '-';
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
pszName++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
LPTSTR PathCombineRemoveDups(LPTSTR pszNewPath, LPCTSTR pszRoot, LPCTSTR pszTail)
|
||
|
|
{
|
||
|
|
for (LPCTSTR psz = pszTail; (psz && *psz); psz = PathFindNextComponent(psz))
|
||
|
|
{
|
||
|
|
LPCTSTR pszNext = PathFindNextComponent(psz);
|
||
|
|
if (pszNext && *pszNext)
|
||
|
|
{
|
||
|
|
TCHAR szPart[MAX_PATH];
|
||
|
|
UINT_PTR len = pszNext - psz;
|
||
|
|
StrCpyN(szPart, psz, (int)min(ARRAYSIZE(szPart), len));
|
||
|
|
|
||
|
|
LPCTSTR pszFound = StrStr(pszRoot, szPart);
|
||
|
|
if ((pszFound > pszRoot) &&
|
||
|
|
(*(pszFound - 1) == TEXT('\\')) &&
|
||
|
|
((*(pszFound + len - 1) == TEXT('\\')) || (*(pszFound + len - 1) == 0)))
|
||
|
|
{
|
||
|
|
pszTail = pszNext;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return PathCombine(pszNewPath, pszRoot, pszTail);
|
||
|
|
}
|
||
|
|
|
||
|
|
LPTSTR PathRemovePart(LPTSTR pszPath, LPCTSTR pszToRemove)
|
||
|
|
{
|
||
|
|
LPTSTR pszFound = StrStr(pszPath, pszToRemove);
|
||
|
|
if (pszFound && (pszFound > pszPath) &&
|
||
|
|
(*(pszFound - 1) == TEXT('\\')))
|
||
|
|
{
|
||
|
|
LPTSTR pszTail = pszFound + lstrlen(pszToRemove);
|
||
|
|
if ((*pszTail == TEXT('\\')) || (*pszTail == 0))
|
||
|
|
{
|
||
|
|
*pszFound = 0;
|
||
|
|
if (*pszTail)
|
||
|
|
{
|
||
|
|
PathAppend(pszPath, pszTail + 1);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
PathRemoveBackslash(pszPath); // trim extra stuff
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return pszPath;
|
||
|
|
}
|
||
|
|
|
||
|
|
// pszProps == "%Artist% - %Album% - %Track% - %DocTitle%"
|
||
|
|
|
||
|
|
HRESULT COrganizeFiles::_NameFromPropertiesString(LPCTSTR pszProps, IPropertySetStorage *ppss, LPTSTR pszRoot, LPTSTR pszName, UINT cchName)
|
||
|
|
{
|
||
|
|
*pszName = 0; // start empty, we build this up
|
||
|
|
|
||
|
|
BOOL bSomeProps = FALSE;
|
||
|
|
|
||
|
|
IPropertyUI *ppui;
|
||
|
|
if (SUCCEEDED(_GetPropertyUI(&ppui)))
|
||
|
|
{
|
||
|
|
LPCTSTR psz = StrChr(pszProps, TEXT('%'));
|
||
|
|
while (psz)
|
||
|
|
{
|
||
|
|
psz++; // skip first %
|
||
|
|
|
||
|
|
LPCTSTR pszTail = StrChr(psz, TEXT('%'));
|
||
|
|
if (pszTail)
|
||
|
|
{
|
||
|
|
TCHAR szPropName[128];
|
||
|
|
StrCpyN(szPropName, psz, (int)min(ARRAYSIZE(szPropName), pszTail - psz + 1));
|
||
|
|
|
||
|
|
psz = pszTail + 1; // just past second %, make sure we advance every time through the loop
|
||
|
|
|
||
|
|
if (szPropName[0])
|
||
|
|
{
|
||
|
|
SHCOLUMNID scid;
|
||
|
|
ULONG chEaten = 0; // gets incremented by ParsePropertyName
|
||
|
|
if (SUCCEEDED(ppui->ParsePropertyName(szPropName, &scid.fmtid, &scid.pid, &chEaten)))
|
||
|
|
{
|
||
|
|
TCHAR szValue[128];
|
||
|
|
szValue[0] = 0;
|
||
|
|
|
||
|
|
if (SUCCEEDED(ReadPropertyAsString(ppui, ppss, &scid, szValue, ARRAYSIZE(szValue))))
|
||
|
|
{
|
||
|
|
bSomeProps = TRUE;
|
||
|
|
FixBadFilenameChars(szValue);
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
TCHAR szUnknown[64], szPropDisplayName[128];
|
||
|
|
LoadString(m_hInst, IDS_UNKNOWN, szUnknown, ARRAYSIZE(szUnknown));
|
||
|
|
|
||
|
|
if (SUCCEEDED(ppui->GetDisplayName(scid.fmtid, scid.pid, PUIFNF_DEFAULT, szPropDisplayName, ARRAYSIZE(szPropDisplayName))))
|
||
|
|
{
|
||
|
|
wnsprintf(szValue, ARRAYSIZE(szValue), TEXT("%s %s"), szUnknown, szPropDisplayName);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
StrCpyN(szValue, szUnknown, ARRAYSIZE(szValue));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// clean stuff out of the root path that we find
|
||
|
|
PathRemovePart(pszRoot, szValue);
|
||
|
|
|
||
|
|
if (szValue[0])
|
||
|
|
{
|
||
|
|
StrCatBuff(pszName, szValue, cchName); // the value
|
||
|
|
|
||
|
|
// add extra formatting stuff next
|
||
|
|
LPCTSTR pszNext = StrChr(psz, TEXT('%'));
|
||
|
|
if (NULL == pszNext)
|
||
|
|
pszNext = psz + lstrlen(psz); // end of string case
|
||
|
|
|
||
|
|
// get extra stuff
|
||
|
|
TCHAR szExtra[64];
|
||
|
|
StrCpyN(szExtra, psz, (int)min((UINT)(pszNext - psz + 1), ARRAYSIZE(szExtra)));
|
||
|
|
StrCatBuff(pszName, szExtra, cchName);
|
||
|
|
|
||
|
|
psz = *pszNext ? pszNext : NULL; // maybe end, maybe more
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
psz = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ppui->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
return bSomeProps ? S_OK : S_FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
// do both files binary compare
|
||
|
|
BOOL IsSameFile(LPCTSTR pszFile1, LPCTSTR pszFile2)
|
||
|
|
{
|
||
|
|
BOOL bSame = FALSE;
|
||
|
|
IStream *pstm1;
|
||
|
|
HRESULT hr = SHCreateStreamOnFileEx(pszFile1, STGM_READ | STGM_SHARE_DENY_NONE, 0, FALSE, NULL, &pstm1);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
IStream *pstm2;
|
||
|
|
hr = SHCreateStreamOnFileEx(pszFile2, STGM_READ | STGM_SHARE_DENY_NONE, 0, FALSE, NULL, &pstm2);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
ULONG cb1;
|
||
|
|
do
|
||
|
|
{
|
||
|
|
char buf1[4096];
|
||
|
|
hr = pstm1->Read(buf1, sizeof(buf1), &cb1);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
char buf2[4096];
|
||
|
|
ULONG cb2;
|
||
|
|
hr = pstm2->Read(buf2, sizeof(buf2), &cb2);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
if (cb1 == cb2)
|
||
|
|
{
|
||
|
|
if (0 == memcmp(buf1, buf2, cb1))
|
||
|
|
bSame = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
bSame = FALSE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
while (bSame && SUCCEEDED(hr) && cb1);
|
||
|
|
|
||
|
|
pstm2->Release();
|
||
|
|
}
|
||
|
|
pstm1->Release();
|
||
|
|
}
|
||
|
|
return bSame;
|
||
|
|
}
|
||
|
|
|
||
|
|
// returns win32 error code
|
||
|
|
// 0 == success
|
||
|
|
|
||
|
|
int MoveFileAndCreateDirectory(LPCTSTR pszOldPath, LPCTSTR pszNewPath)
|
||
|
|
{
|
||
|
|
int err = ERROR_SUCCESS;
|
||
|
|
if (!MoveFile(pszOldPath, pszNewPath))
|
||
|
|
{
|
||
|
|
err = GetLastError();
|
||
|
|
if (ERROR_PATH_NOT_FOUND == err)
|
||
|
|
{
|
||
|
|
// maybe the target folder does not exist, lets
|
||
|
|
// create it and try again
|
||
|
|
|
||
|
|
TCHAR szNewPath[MAX_PATH];
|
||
|
|
StrCpyN(szNewPath, pszNewPath, ARRAYSIZE(szNewPath));
|
||
|
|
PathRemoveFileSpec(szNewPath);
|
||
|
|
|
||
|
|
if (ERROR_SUCCESS == SHCreateDirectoryEx(NULL, szNewPath, NULL))
|
||
|
|
{
|
||
|
|
if (MoveFile(pszOldPath, pszNewPath))
|
||
|
|
{
|
||
|
|
// failure now success
|
||
|
|
err = ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
|
||
|
|
LPCWSTR LoadStr(UINT id, LPWSTR psz, UINT cch)
|
||
|
|
{
|
||
|
|
LoadString(m_hInst, id, psz, cch);
|
||
|
|
return psz;
|
||
|
|
}
|
||
|
|
|
||
|
|
// INamespaceWalkCB
|
||
|
|
STDMETHODIMP COrganizeFiles::FoundItem(IShellFolder *psf, LPCITEMIDLIST pidl)
|
||
|
|
{
|
||
|
|
if (_bCountFiles)
|
||
|
|
{
|
||
|
|
_cFilesTotal++;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
_cFileCur++;
|
||
|
|
|
||
|
|
// if we were invoked on just a file we never got an ::EnterFolder()
|
||
|
|
if (0 == _szRootFolder[0])
|
||
|
|
{
|
||
|
|
DisplayNameOf(psf, pidl, SHGDN_FORPARSING, _szRootFolder, ARRAYSIZE(_szRootFolder));
|
||
|
|
PathRemoveFileSpec(_szRootFolder);
|
||
|
|
}
|
||
|
|
|
||
|
|
TCHAR szOldPath[MAX_PATH];
|
||
|
|
DisplayNameOf(psf, pidl, SHGDN_FORPARSING, szOldPath, ARRAYSIZE(szOldPath));
|
||
|
|
|
||
|
|
_ppd->SetLine(2, szOldPath, TRUE, NULL);
|
||
|
|
_ppd->SetProgress64(_cFileCur, _cFilesTotal);
|
||
|
|
|
||
|
|
IPropertySetStorage *ppss;
|
||
|
|
if (SUCCEEDED(psf->BindToObject(pidl, NULL, IID_PPV_ARG(IPropertySetStorage, &ppss))))
|
||
|
|
{
|
||
|
|
TCHAR szRoot[MAX_PATH];
|
||
|
|
TCHAR szNewName[MAX_PATH];
|
||
|
|
|
||
|
|
StrCpyN(szRoot, _szRootFolder, ARRAYSIZE(szRoot));
|
||
|
|
|
||
|
|
if (S_OK == _NameFromPropertiesString(_pszProps, ppss, szRoot, szNewName, ARRAYSIZE(szNewName)))
|
||
|
|
{
|
||
|
|
PathRenameExtension(szNewName, PathFindExtension(szOldPath));
|
||
|
|
|
||
|
|
TCHAR szNewPath[MAX_PATH];
|
||
|
|
// PathCombineRemoveDups(szNewPath, szRoot, szNewName);
|
||
|
|
PathCombine(szNewPath, szRoot, szNewName);
|
||
|
|
|
||
|
|
ASSERT(0 != *PathFindExtension(szNewPath));
|
||
|
|
|
||
|
|
if (StrCmpC(szNewPath, szOldPath))
|
||
|
|
{
|
||
|
|
int err = MoveFileAndCreateDirectory(szOldPath, szNewPath);
|
||
|
|
|
||
|
|
if (ERROR_ALREADY_EXISTS == err)
|
||
|
|
{
|
||
|
|
// maybe the source and target are the same file
|
||
|
|
// if so lets get rid of one of them
|
||
|
|
WCHAR szBuf[64];
|
||
|
|
|
||
|
|
_ppd->SetLine(2, LoadStr(IDS_COMPARING_FILES, szBuf, ARRAYSIZE(szBuf)), FALSE, NULL);
|
||
|
|
|
||
|
|
if (IsSameFile(szOldPath, szNewPath))
|
||
|
|
{
|
||
|
|
DeleteFile(szOldPath);
|
||
|
|
}
|
||
|
|
|
||
|
|
_ppd->SetLine(2, L"", FALSE, NULL);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ppss->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP COrganizeFiles::EnterFolder(IShellFolder *psf, LPCITEMIDLIST pidl)
|
||
|
|
{
|
||
|
|
if (0 == _szRootFolder[0])
|
||
|
|
{
|
||
|
|
DisplayNameOf(psf, pidl, SHGDN_FORPARSING, _szRootFolder, ARRAYSIZE(_szRootFolder));
|
||
|
|
}
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL IsEmptyFolder(IShellFolder *psf)
|
||
|
|
{
|
||
|
|
// don't look for SHCONTF_INCLUDEHIDDEN items, as we want to still delete the folder
|
||
|
|
// if there are some of these
|
||
|
|
IEnumIDList *penum;
|
||
|
|
HRESULT hr = psf->EnumObjects(NULL, SHCONTF_NONFOLDERS | SHCONTF_FOLDERS, &penum);
|
||
|
|
if (S_OK == hr)
|
||
|
|
{
|
||
|
|
LPITEMIDLIST pidl;
|
||
|
|
hr = penum->Next(1, &pidl, NULL);
|
||
|
|
if (S_OK == hr)
|
||
|
|
{
|
||
|
|
ILFree(pidl);
|
||
|
|
}
|
||
|
|
penum->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
return S_FALSE == hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
void RemoveEmptyFolder(IShellFolder *psf, LPCITEMIDLIST pidl)
|
||
|
|
{
|
||
|
|
IShellFolder *psfItem;
|
||
|
|
if (SUCCEEDED(psf->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder, &psfItem))))
|
||
|
|
{
|
||
|
|
if (IsEmptyFolder(psfItem))
|
||
|
|
{
|
||
|
|
TCHAR szPath[MAX_PATH] = {0}; // zero init to dbl null terminate
|
||
|
|
DisplayNameOf(psf, pidl, SHGDN_FORPARSING, szPath, ARRAYSIZE(szPath));
|
||
|
|
|
||
|
|
SHFILEOPSTRUCT fo = { NULL, FO_DELETE, szPath, NULL,
|
||
|
|
FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOERRORUI };
|
||
|
|
SHFileOperation(&fo);
|
||
|
|
}
|
||
|
|
psfItem->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP COrganizeFiles::LeaveFolder(IShellFolder *psf, LPCITEMIDLIST pidl)
|
||
|
|
{
|
||
|
|
if (!_bCountFiles)
|
||
|
|
{
|
||
|
|
// if the folder is not empty we try to remove it knowing that this call will
|
||
|
|
// fail if there are files in the folder
|
||
|
|
RemoveEmptyFolder(psf, pidl);
|
||
|
|
}
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT COrganizeFiles::_GetPropertyUI(IPropertyUI **pppui)
|
||
|
|
{
|
||
|
|
if (!_ppui)
|
||
|
|
SHCoCreateInstance(NULL, &CLSID_PropertiesUI, NULL, IID_PPV_ARG(IPropertyUI, &_ppui));
|
||
|
|
|
||
|
|
return _ppui ? _ppui->QueryInterface(IID_PPV_ARG(IPropertyUI, pppui)) : E_NOTIMPL;
|
||
|
|
}
|
||
|
|
|
||
|
|
void COrganizeFiles::_DoOrganizeMusic(UINT idCmd, IDataObject *pdtobj)
|
||
|
|
{
|
||
|
|
_pszProps = (idCmd == CMD_ORGANIZE_FLAT) ?
|
||
|
|
(_pszTemplateFlat ? _pszTemplateFlat : TEXT("%Artist% - %Album% - %Track% - %DocTitle%")) :
|
||
|
|
(_pszTemplate ? _pszTemplate : TEXT("%Artist%\\%Album%\\%Track% - %DocTitle%"));
|
||
|
|
|
||
|
|
INamespaceWalk *pnsw;
|
||
|
|
HRESULT hr = CoCreateInstance(CLSID_NamespaceWalker, NULL, CLSCTX_INPROC, IID_PPV_ARG(INamespaceWalk, &pnsw));
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
IProgressDialog *ppd;
|
||
|
|
hr = CoCreateInstance(CLSID_ProgressDialog, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IProgressDialog, &ppd));
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
ppd->StartProgressDialog(NULL, NULL, PROGDLG_AUTOTIME, NULL);
|
||
|
|
|
||
|
|
WCHAR szBuf[64];
|
||
|
|
ppd->SetTitle(LoadStr(IDS_ORGANIZE_MUSIC, szBuf, ARRAYSIZE(szBuf)));
|
||
|
|
ppd->SetLine(1, LoadStr(IDS_FINDING_FILES, szBuf, ARRAYSIZE(szBuf)), FALSE, NULL);
|
||
|
|
|
||
|
|
_ppd = ppd;
|
||
|
|
|
||
|
|
_bCountFiles = TRUE;
|
||
|
|
_cFileCur = _cFilesTotal = 0;
|
||
|
|
// everything happens in our callback interface
|
||
|
|
hr = pnsw->Walk(pdtobj, NSWF_DONT_TRAVERSE_LINKS | NSWF_DONT_ACCUMULATE_RESULT, 8, SAFECAST(this, INamespaceWalkCB *));
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
ppd->SetLine(1, LoadStr(IDS_ORGANIZING_FILES, szBuf, ARRAYSIZE(szBuf)), FALSE, NULL);
|
||
|
|
_bCountFiles = FALSE;
|
||
|
|
hr = pnsw->Walk(pdtobj, NSWF_DONT_TRAVERSE_LINKS | NSWF_DONT_ACCUMULATE_RESULT, 8, SAFECAST(this, INamespaceWalkCB *));
|
||
|
|
}
|
||
|
|
ppd->StopProgressDialog();
|
||
|
|
ppd->Release();
|
||
|
|
}
|
||
|
|
pnsw->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (_szRootFolder[0])
|
||
|
|
{
|
||
|
|
SHChangeNotify(SHCNE_UPDATEDIR, SHCNF_PATH | SHCNF_FLUSHNOWAIT, _szRootFolder, NULL);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|