winamp/Src/Plugins/General/gen_ml/fileview_filesystem.cpp
2024-09-24 14:54:57 +02:00

290 lines
8.4 KiB
C++

#include "main.h"
#include "./fileview.h"
#include "./fileview_internal.h"
#include <vector>
#include "../playlist/svc_playlisthandler.h"
#include "../playlist/api_playlistmanager.h"
#include <api/service/waservicefactorybase.h>
#include <api/service/services.h>
#include <shlwapi.h>
#include <strsafe.h>
#include <algorithm>
#define FILEREC_ALLOCATION_STEP 1000
#define FILETYPE_REREAD ((UINT)(0 - 1))
typedef struct _FILETYPEREC
{
WCHAR szExtension[32];
WCHAR szFamily[128];
UINT Type;
} FILETYPEREC;
static std::vector<FILETYPEREC> supportedFilesList;
//__inline static int __cdecl QSort_StrCmpI(const void *elem1, const void *elem2)
//{
// return (CompareStringW(CSTR_INVARIANT, NORM_IGNORECASE, (LPCWSTR)elem1, -1, (LPCWSTR)elem2, -1) - 2);
//}
static BOOL FileView_GetPLExtensionName(LPCWSTR pszExt, LPWSTR pszDest, INT cchDest)
{
BOOL result(FALSE);
DWORD lcid;
int n(0);
waServiceFactory *sf = 0;
LPCWSTR ext;
lcid = MAKELCID(MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), SORT_DEFAULT);
while (NULL != (sf = WASABI_API_SVC->service_enumService(WaSvc::PLAYLISTHANDLER, n++)))
{
svc_playlisthandler * handler = static_cast<svc_playlisthandler *>(sf->getInterface());
if (handler)
{
int k(0);
while (NULL != (ext = handler->EnumerateExtensions(k++)))
{
if (CSTR_EQUAL == CompareStringW(lcid, NORM_IGNORECASE, pszExt, -1, ext, -1))
{
result = (S_OK == StringCchCopyW(pszDest, cchDest, handler->GetName()));
if (result && CSTR_EQUAL == CompareStringW(lcid, NORM_IGNORECASE, pszExt, -1, L"M3U8", -1)) // ugly...
result = (S_OK == StringCchCatW(pszDest, cchDest, L" (Unicode)"));
break;
}
}
sf->releaseInterface(handler);
}
}
return result;
}
static void FileView_ReadSupportedTypesInfo()
{
LPWSTR pszTypes, p;
pszTypes = (LPWSTR)SendMessageW(plugin.hwndParent, WM_WA_IPC, 0, IPC_GET_EXTLISTW);
if (pszTypes)
{
INT i;
for(i = 0, p = pszTypes; *p != 0 && *(p+1) != 0; p += lstrlenW(p) + 1, i++)
{
FILETYPEREC ftr;
if (S_OK == StringCchCopyW(ftr.szExtension, sizeof(ftr.szExtension) / sizeof(ftr.szExtension[0]), p))
{
ftr.Type = FILETYPE_REREAD;
supportedFilesList.push_back(ftr);
}
}
GlobalFree(pszTypes);
}
if (WASABI_API_SVC)
{
api_playlistmanager *plMngr = 0;
waServiceFactory *factory = WASABI_API_SVC->service_getServiceByGuid(api_playlistmanagerGUID);
if (factory) plMngr = (api_playlistmanager*) factory->getInterface();
if (plMngr)
{
FILETYPEREC ftr;
ftr.Type = FVFT_PLAYLIST;
size_t playlistEnum=0;
const wchar_t *playlistExt=0;
while (NULL != (playlistExt = plMngr->EnumExtension(playlistEnum++)))
{
if (S_OK ==StringCchCopyW(ftr.szExtension, sizeof(ftr.szExtension) / sizeof(ftr.szExtension[0]), playlistExt))
{
FileView_GetPLExtensionName(ftr.szExtension, ftr.szFamily, sizeof(ftr.szFamily)/sizeof(ftr.szFamily[0]));
supportedFilesList.push_back(ftr);
}
}
factory->releaseInterface(plMngr);
}
}
}
static UINT FileView_GetFileType(LPCWSTR pszExtension)
{
FILETYPEREC *pftr = nullptr;
if (!pszExtension) return FVFT_UNKNOWN;
if ( 0 == supportedFilesList.size())
{
FileView_ReadSupportedTypesInfo();
if (supportedFilesList.size()) {
//qsort(supportedFilesList.begin(), supportedFilesList.size(), sizeof(FILETYPEREC), QSort_StrCmpI);
std::sort(supportedFilesList.begin(), supportedFilesList.end(),
[&](const FILETYPEREC &lhs, const FILETYPEREC &rhs) -> bool
{
return CSTR_LESS_THAN == CompareStringW(CSTR_INVARIANT, NORM_IGNORECASE, (LPCWSTR)(&lhs), -1, (LPCWSTR)(&rhs), -1);
}
);
}
}
//pftr = (FILETYPEREC*)bsearch(pszExtension, supportedFilesList.begin(), supportedFilesList.size(), sizeof(FILETYPEREC), QSort_StrCmpI);
auto it = std::find_if(supportedFilesList.begin(), supportedFilesList.end(),
[&](const FILETYPEREC& rec) -> bool
{
return CSTR_EQUAL == CompareStringW(CSTR_INVARIANT, NORM_IGNORECASE, (LPCWSTR)(&rec), -1, (LPCWSTR)(pszExtension), -1);
}
);
if (it != supportedFilesList.end())
{
pftr = &(*it);
}
if (pftr)
{
if (FILETYPE_REREAD == pftr->Type)
{
wchar_t szTest[MAX_PATH] = {0};
pftr->Type = FVFT_UNKNOWN;
if (S_OK == StringCchPrintfW(szTest, sizeof(szTest)/sizeof(wchar_t), L"test.%s", pftr->szExtension))
{
wchar_t szResult[MAX_PATH] = {0};
extendedFileInfoStructW efis = { szTest, L"type", szResult, sizeof(szResult)/sizeof(szResult[0]), };
if (SendMessageW(plugin.hwndParent, WM_WA_IPC, (WPARAM)&efis, IPC_GET_EXTENDED_FILE_INFOW))
{
switch(szResult[0])
{
case L'0': pftr->Type = FVFT_AUDIO; break;
case L'1': pftr->Type = FVFT_VIDEO; break;
}
pftr->szFamily[0] = L'\0';
switch(pftr->Type)
{
case FVFT_AUDIO:
case FVFT_VIDEO:
extendedFileInfoStructW efis = { szTest, L"family", pftr->szFamily, sizeof(pftr->szFamily)/sizeof(pftr->szFamily[0]), };
SendMessageW(plugin.hwndParent, WM_WA_IPC, (WPARAM)&efis, IPC_GET_EXTENDED_FILE_INFOW);
break;
}
}
}
}
return pftr->Type;
}
return FVFT_UNKNOWN;
}
size_t FileView_ReadFileData(FILEDATA *pfd, LPCWSTR pszPath, UINT fStyle, FILESYSTEMINFO *pfsi)
{
HANDLE hFile;
wchar_t szSearch[2 * MAX_PATH + 4] = {0};
if (!pfd) return ((size_t)-1);
size_t c = 0;
pfd->count = 0;
pfd->folderSize = 0;
if (S_OK != StringCchPrintfW(szSearch, sizeof(szSearch)/sizeof(szSearch[0]), L"%s\\*", pszPath)) return ((size_t)-1);
if (0 == pfd->allocated)
{
pfd->pRec = (FILERECORD*)calloc(FILEREC_ALLOCATION_STEP, sizeof(FILERECORD));
if (!pfd->pRec)
{
return ((size_t)-1);
}
pfd->pSort = (size_t*)calloc(FILEREC_ALLOCATION_STEP, sizeof(size_t));
if (!pfd->pSort)
{
if (pfd->pRec) free(pfd->pRec);
ZeroMemory(&pfd, sizeof(FILEDATA));
return ((size_t)-1);
}
pfd->allocated = FILEREC_ALLOCATION_STEP;
}
hFile = pfsi->fnFindFirstFile(szSearch, &pfd->pRec[c].Info);
if (INVALID_HANDLE_VALUE != hFile)
{
do
{
if (0 == (FILE_ATTRIBUTE_DIRECTORY & pfd->pRec[c].Info.dwFileAttributes) &&
(0 == (FVS_IGNOREHIDDEN & fStyle) || 0 == (FILE_ATTRIBUTE_HIDDEN & pfd->pRec[c].Info.dwFileAttributes)))
{
LPCWSTR pszExt = PathFindExtensionW(pfd->pRec[c].Info.cFileName);
if (L'.' == *pszExt) pszExt++;
else pszExt = NULL;
pfd->pRec[c].extOffset = (pszExt && *pszExt) ? (pszExt - pfd->pRec[c].Info.cFileName) : 0;
pfd->pRec[c].fileType = FileView_GetFileType(pszExt);
pfd->pRec[c].pMeta = NULL;
if ((FVFT_UNKNOWN == pfd->pRec[c].fileType && (FVS_SHOWUNKNOWN & fStyle)) ||
(FVFT_AUDIO == pfd->pRec[c].fileType && (FVS_SHOWAUDIO & fStyle)) ||
(FVFT_VIDEO == pfd->pRec[c].fileType && (FVS_SHOWVIDEO & fStyle)) ||
(FVFT_PLAYLIST == pfd->pRec[c].fileType && (FVS_SHOWPLAYLIST & fStyle)))
{
if (NULL != pfd->pRec[c].extOffset && FVFT_UNKNOWN != pfd->pRec[c].fileType && (FVS_HIDEEXTENSION & fStyle))
pfd->pRec[c].Info.cFileName[pfd->pRec[c].extOffset -1] = L'\0';
pfd->pSort[c] = c;
pfd->folderSize += (ULONGLONG)(((__int64)pfd->pRec[c].Info.nFileSizeHigh << 32) | pfd->pRec[c].Info.nFileSizeLow);
c++;
if (c == pfd->allocated)
{
void *pData;
pData = realloc(pfd->pRec, sizeof(FILERECORD) * (pfd->allocated + FILEREC_ALLOCATION_STEP));
if (!pData) { c = ((size_t)-1); break; }
pfd->pRec = (FILERECORD*)pData;
pData = realloc(pfd->pSort, sizeof(size_t) * (pfd->allocated + FILEREC_ALLOCATION_STEP));
if (!pData) { c = ((size_t)-1); break; }
pfd->pSort= (size_t*)pData;
pfd->allocated += FILEREC_ALLOCATION_STEP;
}
}
}
} while (pfsi->fnFindNextFile(hFile, &pfd->pRec[c].Info));
pfsi->fnFindClose(hFile);
}
else
{
DWORD e = GetLastError();
if (0 != e) c = ((size_t)-1);
}
if (((size_t)-1) != c) pfd->count = c;
return c;
}
LPCWSTR FileView_GetTypeFamily(LPCWSTR pszExtension)
{
if (!pszExtension || 0 == supportedFilesList.size())
return NULL;
//FILETYPEREC *pftr = (FILETYPEREC*)bsearch(pszExtension, supportedFilesList.begin(), supportedFilesList.size(), sizeof(FILETYPEREC), QSort_StrCmpI);
FILETYPEREC* pftr = nullptr;
auto it = std::find_if(supportedFilesList.begin(), supportedFilesList.end(),
[&](const FILETYPEREC& rec) -> bool
{
return CSTR_EQUAL == CompareStringW(CSTR_INVARIANT, NORM_IGNORECASE, (LPCWSTR)(&rec), -1, (LPCWSTR)(pszExtension), -1);
}
);
if (it != supportedFilesList.end())
{
pftr = &(*it);
}
return (pftr) ? pftr->szFamily : NULL;
}