winamp/Src/Plugins/Library/ml_disc/formatfilename.cpp
2024-09-24 14:54:57 +02:00

269 lines
No EOL
8.2 KiB
C++

#include "main.h"
#include "../nu/ns_wc.h"
#include "../winamp/wa_ipc.h"
#include <shlwapi.h>
#include <strsafe.h>
#define TID_UNKNOWN 0
#define TID_ARTIST 1
#define TID_ALBUM 2
#define TID_TITLE 3
#define TID_GENRE 4
#define TID_YEAR 5
#define TID_TRACKARTIST 6
#define TID_FILENAME 7
#define TID_EXTENSION 8
#define TID_DISC 9
#define TID_DISCS 10
#define TOKEN_ARTIST TEXT("<artist>")
#define TOKEN_ALBUM TEXT("<album>")
#define TOKEN_TITLE TEXT("<title>")
#define TOKEN_GENRE TEXT("<genre>")
#define TOKEN_YEAR TEXT("<year>")
#define TOKEN_TRACKARTIST TEXT("<trackartist>")
#define TOKEN_FILENAME TEXT("<filename>")
#define TOKEN_EXTENSION TEXT("<extension>")
#define TOKEN_DISC TEXT("<disc>")
#define TOKEN_DISCS TEXT("<discs>")
#define TOKEN_LEN_ARTIST 8
#define TOKEN_LEN_ALBUM 7
#define TOKEN_LEN_TITLE 7
#define TOKEN_LEN_GENRE 7
#define TOKEN_LEN_YEAR 6
#define TOKEN_LEN_TRACKARTIST 13
#define TOKEN_LEN_FILENAME 10
#define TOKEN_LEN_EXTENSION 11
#define TOKEN_LEN_DISC 6
#define TOKEN_LEN_DISCS 7
#define STRCOMP_INVARIANT MAKELCID(MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), SORT_DEFAULT)
#define CHECK_TOKEN(token, format) (CSTR_EQUAL == CompareString(STRCOMP_INVARIANT, NORM_IGNORECASE, TOKEN_##token##, TOKEN_LEN_##token##, format, TOKEN_LEN_##token##))
static TCHAR PathValidateChar(TCHAR cVal, BOOL bAllowBackSlash)
{
switch(cVal)
{
case TEXT('\\'): if (!bAllowBackSlash) return TEXT('-');
break;
case TEXT('/'): if (!bAllowBackSlash) return TEXT('-');
return TEXT('\\');
case TEXT(':'): return TEXT('-');
case TEXT('*'): return TEXT('_');
case TEXT('?'): return TEXT('_');
case TEXT('\"'): return TEXT('\'');
case TEXT('<'): return TEXT('(');
case TEXT('>'): return TEXT(')');
case TEXT('|'): return TEXT('_');
}
return cVal;
}
void CleanupDirectoryString(LPTSTR pszDirectory)
{
PathUnquoteSpaces(pszDirectory);
PathRemoveBlanks(pszDirectory);
LPTSTR pc = pszDirectory;
while (TEXT('\0') != *pc) { if (TEXT('/') == *pc) *pc = TEXT('\\'); pc++; }
if (pc > pszDirectory)
{
pc--;
while (pszDirectory != pc &&
(TEXT('.') == *pc || TEXT(' ') == *pc || TEXT('\\') == *pc))
{
*pc = TEXT('\0');
pc--;
}
}
}
LPWSTR GetExtensionString(LPWSTR pszBuffer, INT cchBufferMax, DWORD fourcc)
{
char configExt[10] = { '\0', };
pszBuffer[0] = L'\0';
convertConfigItem cfi;
cfi.configfile = 0;
cfi.data = configExt;
cfi.format = fourcc;
cfi.item = "extension";
cfi.len = 10;
SENDWAIPC(plugin.hwndWinampParent, IPC_CONVERT_CONFIG_GET_ITEM, (WPARAM)&cfi);
if ('\0' != *configExt)
{
if (!MultiByteToWideCharSZ(CP_ACP, 0, configExt, -1, pszBuffer, cchBufferMax))
return NULL;
}
else
{
if (cchBufferMax < 5) return NULL;
pszBuffer[0] = (TCHAR)((fourcc) & 0xff);
pszBuffer[1] = (TCHAR)((fourcc >> 8) & 0xff);
pszBuffer[2] = (TCHAR)((fourcc >> 16) & 0xff);
pszBuffer[3] = TEXT('\0');
for (LPTSTR p = &pszBuffer[2]; p >= pszBuffer && TEXT(' ') == *p; p--) *p = TEXT('\0');
}
return pszBuffer;
}
// if trackno is 0xdeadbeef, or title is 0, they are both ignored
// TODO: use ATF instead
HRESULT FormatFileName(LPTSTR pszTextOut, INT cchTextMax, LPCTSTR pszFormat,
INT nTrackNo, LPCTSTR pszArtist,
LPCTSTR pszAlbum, LPCTSTR pszTitle,
LPCTSTR pszGenre, LPCTSTR pszYear,
LPCTSTR pszTrackArtist,
LPCTSTR pszFileName, LPCTSTR pszDisc)
{
HRESULT hr = S_OK;
TCHAR szBuffer[MAX_PATH] = {0};
LPTSTR pszStart = pszTextOut + lstrlen(pszTextOut);
while (pszFormat && TEXT('\0') != *pszFormat)
{
int whichstr = TID_UNKNOWN;
if (*pszFormat == TEXT('#') && nTrackNo != 0xdeadbeef)
{
int cnt = 0;
while (pszFormat && *pszFormat == TEXT('#')) { pszFormat++; cnt++; }
if (cnt > 8) cnt = 8;
TCHAR szFormat[32] = {0};
hr = StringCchPrintf(szFormat, ARRAYSIZE(szFormat), TEXT("%%%02dd"), cnt);
if (S_OK == hr) StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), szFormat, nTrackNo);
if (S_OK == hr) StringCchCat(pszTextOut, cchTextMax, szBuffer);
if (S_OK != hr) return hr;
}
else if (pszArtist && CHECK_TOKEN(ARTIST, pszFormat)) whichstr = TID_ARTIST;
else if (pszAlbum && CHECK_TOKEN(ALBUM, pszFormat)) whichstr = TID_ALBUM;
else if (pszTitle && CHECK_TOKEN(TITLE, pszFormat)) whichstr = TID_TITLE;
else if (pszGenre && CHECK_TOKEN(GENRE, pszFormat)) whichstr = TID_GENRE;
else if (pszYear && CHECK_TOKEN(YEAR, pszFormat)) whichstr = TID_YEAR;
else if (pszTrackArtist && CHECK_TOKEN(TRACKARTIST, pszFormat)) whichstr = TID_TRACKARTIST;
else if (pszFileName && CHECK_TOKEN(FILENAME, pszFormat)) whichstr = TID_FILENAME;
else if (pszFileName && CHECK_TOKEN(EXTENSION, pszFormat)) whichstr = TID_EXTENSION;
else if (pszDisc && CHECK_TOKEN(DISC, pszFormat)) whichstr = TID_DISC;
else if (pszDisc && CHECK_TOKEN(DISCS, pszFormat)) whichstr = TID_DISCS;
else
{
INT l = lstrlen(pszTextOut);
pszTextOut += l;
cchTextMax -= l;
pszTextOut[0] = *pszFormat++;
if (cchTextMax < 2) return STRSAFE_E_INSUFFICIENT_BUFFER;
pszTextOut[0] = PathValidateChar(pszTextOut[0], TRUE);
if (TEXT('\\') == *pszTextOut)
{
// remove end spaces and dots
while (pszTextOut > pszStart &&
(TEXT('\\') == *(pszTextOut - 1) || TEXT(' ') == *(pszTextOut - 1) || TEXT('.') == *(pszTextOut - 1)))
{
pszTextOut--;
cchTextMax--;
}
if (pszTextOut == pszStart)
{
pszTextOut--;
cchTextMax--;
}
else *pszTextOut = TEXT('\\');
}
pszTextOut++;
cchTextMax--;
*pszTextOut = TEXT('\0');
if (S_OK != hr) return hr;
}
if (whichstr != TID_UNKNOWN)
{
LPCTSTR pszSrc = NULL;
int islow = IsCharLower(pszFormat[1]) && IsCharLower(pszFormat[2]);
int ishi = IsCharUpper(pszFormat[1]) && IsCharUpper(pszFormat[2]);
switch(whichstr)
{
case TID_ARTIST: pszSrc = pszArtist; pszFormat += TOKEN_LEN_ARTIST; break;
case TID_ALBUM: pszSrc = pszAlbum; pszFormat += TOKEN_LEN_ALBUM; break;
case TID_TITLE: pszSrc = pszTitle; pszFormat += TOKEN_LEN_TITLE; break;
case TID_GENRE: pszSrc = pszGenre; pszFormat += TOKEN_LEN_GENRE; break;
case TID_YEAR: pszSrc = pszYear; pszFormat += TOKEN_LEN_YEAR; break;
case TID_TRACKARTIST: pszSrc = pszTrackArtist; pszFormat += TOKEN_LEN_TRACKARTIST; break;
case TID_DISC:
if (pszDisc && *pszDisc && TEXT('\0') == *pszDisc) pszSrc = L"1";
else
{
// default to 1 when we've not got a proper value passed to us
int disc = _wtoi(pszDisc);
if(disc <= 0) disc = 1;
StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), L"%d", disc);
pszSrc = szBuffer;
}
pszFormat += TOKEN_LEN_DISC;
break;
case TID_DISCS:
if (pszDisc && *pszDisc && TEXT('\0') == *pszDisc) pszSrc = L"1";
else
{
LPTSTR pszTemp = wcschr((LPTSTR)pszDisc, L'/');
if(pszTemp == NULL) pszTemp = wcschr((LPTSTR)pszDisc, L'\\');
if(pszTemp == NULL)
{
pszTemp = (LPTSTR)pszDisc;
}
else
{
pszTemp = CharNext(pszTemp);
}
int disc = _wtoi(pszTemp);
if(disc <= 0) disc = 1;
StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), L"%d", disc);
pszSrc = szBuffer;
}
pszFormat += TOKEN_LEN_DISCS;
break;
case TID_FILENAME:
pszSrc = PathFindExtension(pszFileName);
if (TEXT('\0') == *pszSrc) pszSrc = pszFileName;
else
{
StringCchCopy(szBuffer, ARRAYSIZE(szBuffer), pszFileName);
PathRemoveExtension(szBuffer);
pszSrc = szBuffer;
}
pszFormat += TOKEN_LEN_FILENAME;
break;
case TID_EXTENSION:
pszSrc = PathFindExtension(pszFileName); pszFormat += TOKEN_LEN_EXTENSION;
while (pszTextOut > pszStart && TEXT('.') == *(pszTextOut - 1))
{
pszTextOut--;
*pszTextOut = TEXT('\0');
cchTextMax++;
}
break;
}
INT l = lstrlen(pszTextOut);
pszTextOut += l;
cchTextMax -= l;
while (pszSrc && TEXT('\0') != *pszSrc && cchTextMax > 0)
{
pszTextOut[0] = pszSrc[0];
if (ishi) CharUpperBuffW(pszTextOut, 1);
else if (islow) CharLowerBuffW(pszTextOut, 1);
pszTextOut[0] = PathValidateChar(pszTextOut[0], FALSE);
cchTextMax--;
if (cchTextMax) pszTextOut++;
pszSrc++;
}
pszTextOut[0] = TEXT('\0');
if (0 == cchTextMax) return STRSAFE_E_INSUFFICIENT_BUFFER;
}
}
return hr;
}