1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#include "main.h"
#include "./copyfiles.h"
#include "./copyinternal.h"
#include "./resource.h"
#include "../nu/trace.h"
#include <api/service/waServiceFactory.h>
#include <shlwapi.h>
#include <strsafe.h>
typedef struct _FILEMETA
{
LPWSTR pszArtist;
LPWSTR pszAlbum;
LPWSTR pszTitle;
LPWSTR pszGenre;
LPWSTR pszAlbumArtist;
INT nYear;
INT nTrackNum;
INT nTrackCount;
LPWSTR pszDisc;
} FILEMETA;
// sets part and parts to -1 or 0 on fail/missing (e.g. parts will be -1 on "1", but 0 on "1/")
static void ParseIntSlashInt(wchar_t *string, int *part, int *parts)
{
*part = -1;
*parts = -1;
if (string && string[0])
{
*part = _wtoi(string);
while (string && *string && *string != '/')
{
string++;
}
if (*string == '/')
{
string++;
*parts = _wtoi(string);
}
}
}
#define READFILEINFO(__fileName, __tag, __result, __pszBuffer, __cchBuffer)\
(pReader->GetExtendedFileInfo((__fileName), (__tag), (__pszBuffer), (__cchBuffer)) && L'\0' != *(__pszBuffer))
static void CopyFiles_ReadFileMeta(FILEMETA *pMeta, api_metadata *pReader, LPCWSTR pszFileName)
{
WCHAR szBuffer[2048] = {0};
#define GETFILEINFO_STR(__tag, __result, __resId)\
{ szBuffer[0] = L'\0';\
READFILEINFO(pszFileName, __tag, __result, szBuffer, ARRAYSIZE(szBuffer));\
if (TEXT('\0') == *szBuffer)\
{ if (IS_INTRESOURCE(__resId)) WASABI_API_LNGSTRINGW_BUF(((UINT)(UINT_PTR)(__resId)), szBuffer, ARRAYSIZE(szBuffer));\
else StringCchCopyEx(szBuffer, ARRAYSIZE(szBuffer), (__resId), NULL, NULL, STRSAFE_IGNORE_NULLS);\
}\
(__result) = _wcsdup(szBuffer); }
#define GETFILEINFO_INT(__tag, __result) { szBuffer[0] = L'\0';\
if (READFILEINFO(pszFileName, __tag, __result, szBuffer, ARRAYSIZE(szBuffer)))\
{(__result) = _wtoi(szBuffer); }}
#define GETFILEINFO_INTINT(__tag, __result1, __result2) { szBuffer[0] = L'\0';\
if (READFILEINFO(pszFileName, __tag, __result, szBuffer, ARRAYSIZE(szBuffer)))\
{ParseIntSlashInt(szBuffer, (__result1), (__result2)); }}
if (!pMeta) return;
ZeroMemory(pMeta, sizeof(FILEMETA));
#pragma warning(push)
#pragma warning(disable : 4127)
GETFILEINFO_STR(L"artist", pMeta->pszArtist, MAKEINTRESOURCE(IDS_UNKNOWN_ARTIST));
GETFILEINFO_STR(L"album", pMeta->pszAlbum, MAKEINTRESOURCE(IDS_UNKNOWN_ALBUM));
GETFILEINFO_STR(L"title", pMeta->pszTitle, MAKEINTRESOURCE(IDS_UNKNOWN));
GETFILEINFO_STR(L"albumartist", pMeta->pszAlbumArtist, pMeta->pszArtist);
GETFILEINFO_STR(L"genre", pMeta->pszGenre, MAKEINTRESOURCE(IDS_UNKNOWN_GENRE));
GETFILEINFO_INT(L"year", pMeta->nYear);
GETFILEINFO_INTINT(L"track", &pMeta->nTrackNum, &pMeta->nTrackCount);
GETFILEINFO_STR(L"disc", pMeta->pszDisc, MAKEINTRESOURCE(IDS_UNKNOWN));
#pragma warning(pop)
}
static void CopyFiles_ReleaseFileMeta(FILEMETA *pMeta)
{
if (pMeta->pszArtist) free(pMeta->pszArtist);
if (pMeta->pszAlbum) free(pMeta->pszAlbum);
if (pMeta->pszTitle) free(pMeta->pszTitle);
if (pMeta->pszGenre) free(pMeta->pszGenre);
if (pMeta->pszDisc) free(pMeta->pszDisc);
if (pMeta->pszAlbumArtist) free(pMeta->pszAlbumArtist);
ZeroMemory(pMeta, sizeof(FILEMETA));
}
static BOOL CopyFiles_GetFormattedName(LPTSTR pszBuffer, INT cchBufferMax, LPCTSTR pszFileToFormat, LPCTSTR pszOrigFileName, LPCTSTR pszFormat, api_metadata *pMetaReader)
{
HRESULT hr;
FILEMETA meta;
CopyFiles_ReadFileMeta(&meta, pMetaReader, pszFileToFormat);
WCHAR szYear[16] = {0};
StringCchPrintf(szYear, ARRAYSIZE(szYear), TEXT("%d"), meta.nYear);
pszBuffer[0] = TEXT('\0');
hr = FormatFileName(pszBuffer, cchBufferMax, pszFormat,
meta.nTrackNum, meta.pszAlbumArtist,
meta.pszAlbum, meta.pszTitle,
meta.pszGenre, szYear, meta.pszArtist,
pszOrigFileName, meta.pszDisc);
CopyFiles_ReleaseFileMeta(&meta);
if (S_OK != hr)
{
SetLastError(ERROR_OUTOFMEMORY);
return FALSE;
}
return TRUE;
}
BOOL CopyFiles_FormatFileName(LPTSTR pszNewFileName, INT cchBufferMax, LPCTSTR pszFileToRename, LPCTSTR pszOrigFileName, LPCTSTR pszDestination, LPCTSTR pszFormat, api_metadata *pMetaReader)
{
StringCchCopy(pszNewFileName, cchBufferMax, pszDestination);
INT l = lstrlen(pszNewFileName);
if (l) { pszNewFileName[l] = TEXT('\\'); pszNewFileName[l + 1] = TEXT('\0'); l++; }
if (!CopyFiles_GetFormattedName(pszNewFileName + l, cchBufferMax - l, pszFileToRename, pszOrigFileName, pszFormat, pMetaReader))
return FALSE;
LPTSTR p = PathFindFileName(pszNewFileName);
if (p && p > pszNewFileName)
{
*(p - 1) = TEXT('\0');
if (!CopyFiles_CreateDirectory(pszNewFileName)) return FALSE;
*(p - 1) = TEXT('\\');
}
return TRUE;
}
|
