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

321 lines
8.7 KiB
C++

#include "main.h"
#include "../replicant/nu/AutoChar.h"
#include "../replicant/nu/ns_wc.h"
#include <shlwapi.h>
#include <malloc.h> // for alloca
static wchar_t m_lastfn[2048];
static wchar_t m_lastartist[256], m_lasttitle[256], m_lastalbum[256], m_gracenotefileid[128];
static int m_lasttrack;
static int m_playcount;
static int m_ispodcast;
static int m_rating;
static int m_db_found;
void ClearTitleHookCache()
{
memset(m_lastfn, 0, sizeof(m_lastfn));
}
BOOL IPC_HookExtInfoW(INT_PTR param)
{
if (skipTitleInfo) // we're reading metadata and being asked to skip title hooking (so we can get a valid title for guessing, if need be)
return false;
extendedFileInfoStructW *p = (extendedFileInfoStructW *)param;
// fill in our own titles from db? not for now, just let it default to hitting the tags?
if (!g_config->ReadInt(L"newtitle", 1))
return FALSE;
if (NULL == p->filename || L'\0' == *p->filename ||
NULL == p->metadata || L'\0' == *p->metadata)
{
return FALSE;
}
int which = 0;
if (!_wcsicmp(p->metadata, DB_FIELDNAME_artist )) which = 1;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_title )) which = 2;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_album )) which = 3;
//else if (!_wcsicmp(p->metadata, L"tuid")) which = 4;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_track )) which = 5;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_rating )) which = 6;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_playcount )) which = 7;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_GracenoteFileID )) which=8;
else if (!_wcsicmp(p->metadata, DB_FIELDNAME_ispodcast )) which=9;
if (which)
{
if (_wcsicmp(m_lastfn, p->filename))
{
if (!g_table) openDb();
if (!g_table) return 0;
m_lastartist[0] = 0;
m_lasttitle[0] = 0;
m_lastalbum[0] = 0;
m_gracenotefileid[0]=0;
m_lasttrack = 0;
m_rating=-1;
m_playcount=-1;
m_ispodcast=-1;
lstrcpynW(m_lastfn, p->filename, sizeof(m_lastfn)/sizeof(wchar_t));
wchar_t filename2[MAX_PATH] = {0}; // full lfn path if set
EnterCriticalSection(&g_db_cs);
nde_scanner_t s = NDE_Table_CreateScanner(g_table);
int found = FindFileInDatabase(s, MAINTABLE_ID_FILENAME, m_lastfn, filename2);
if (found == 2)
lstrcpynW(m_lastfn, filename2, sizeof(m_lastfn)/sizeof(m_lastfn));
if (found)
{
nde_field_t f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_ARTIST);
if (f) lstrcpynW(m_lastartist, NDE_StringField_GetString(f), sizeof(m_lastartist)/sizeof(wchar_t));
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_ALBUM);
if (f) lstrcpynW(m_lastalbum, NDE_StringField_GetString(f), sizeof(m_lastalbum)/sizeof(wchar_t));
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_TITLE);
if (f) lstrcpynW(m_lasttitle, NDE_StringField_GetString(f), sizeof(m_lasttitle)/sizeof(wchar_t));
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_TRACKNB);
if (f) m_lasttrack = NDE_IntegerField_GetValue(f);
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_PLAYCOUNT);
if (f) m_playcount = NDE_IntegerField_GetValue(f);
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_RATING);
if (f) m_rating = NDE_IntegerField_GetValue(f);
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_ISPODCAST);
if (f) m_ispodcast = NDE_IntegerField_GetValue(f);
f = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_GRACENOTEFILEID);
if (f) lstrcpynW(m_gracenotefileid, NDE_StringField_GetString(f), sizeof(m_gracenotefileid)/sizeof(wchar_t));
m_db_found = 1;
}
else m_db_found = 0;
NDE_Table_DestroyScanner(g_table, s);
LeaveCriticalSection(&g_db_cs);
}
if (m_db_found == 1)
{
switch(which)
{
case 1:
if (m_lastartist[0])
{
lstrcpynW(p->ret, m_lastartist, p->retlen);
return 1;
}
break;
case 2:
if (m_lasttitle[0])
{
lstrcpynW(p->ret, m_lasttitle, p->retlen);
return 1;
}
break;
case 3:
if (m_lastalbum[0])
{
lstrcpynW(p->ret, m_lastalbum, p->retlen);
return 1;
}
break;
case 5:
if (m_lasttrack && m_lasttrack != -1)
{
_snwprintf(p->ret, p->retlen, L"%d", m_lasttrack);
return 1;
}
break;
case 6:
if (m_rating != -1)
{
_snwprintf(p->ret, p->retlen, L"%d", m_rating);
return 1;
}
break;
case 7:
if (m_playcount != -1)
{
_snwprintf(p->ret, p->retlen, L"%d", m_playcount);
return 1;
}
break;
case 8:
if (m_gracenotefileid[0])
{
lstrcpynW(p->ret, m_gracenotefileid, p->retlen);
return 1;
}
break;
case 9:
if (m_ispodcast != -1)
{
_snwprintf(p->ret, p->retlen, L"%d", m_ispodcast);
return 1;
}
break;
}
}
}
return FALSE;
}
BOOL IPC_HookExtInfo(INT_PTR param)
{
extendedFileInfoStruct *p = (extendedFileInfoStruct*)param;
// fill in our own titles from db? not for now, just let it default to hitting the tags?
if (!g_config->ReadInt(L"newtitle", 1))
return FALSE;
if (NULL == p->filename || '\0' == *p->filename ||
NULL == p->metadata || '\0' == *p->metadata)
{
return FALSE;
}
extendedFileInfoStructW pW = {0};
AutoWide wideFn(p->filename), wideMetadata(p->metadata);
pW.filename = wideFn;
pW.metadata = wideMetadata;
pW.retlen = p->retlen;
pW.ret = (wchar_t *)alloca(pW.retlen * sizeof(wchar_t));
if (IPC_HookExtInfoW((INT_PTR)&pW))
{
WideCharToMultiByteSZ(CP_ACP, 0, pW.ret, -1, p->ret, p->retlen, 0, 0);
return 1;
}
return 0;
}
BOOL IPC_HookTitleInfo(INT_PTR param)
{
if (skipTitleInfo) // we're reading metadata and being asked to skip title hooking (so we can get a valid title for guessing, if need be)
return false;
waHookTitleStructW *hts = (waHookTitleStructW*)param;
if (NULL != hts->filename &&
!StrStrW(hts->filename, L"://") &&
g_config->ReadInt(L"newtitle", 1))
{
if (!g_table) openDb();
if (!g_table) return 0;
wchar_t filename2[MAX_PATH] = {0}; // full lfn path if set
EnterCriticalSection(&g_db_cs);
nde_scanner_t s = NDE_Table_CreateScanner(g_table);
int found=FindFileInDatabase(s, MAINTABLE_ID_FILENAME, hts->filename, filename2);
if (found)
{
nde_field_t length = NDE_Scanner_GetFieldByID(s, MAINTABLE_ID_LENGTH);
int l = -1;
if (length)
l = NDE_IntegerField_GetValue(length);
if (l > 0)
hts->length = l;
else hts->length = -1;
if (hts->title)
{
TAG_FMT_EXT(hts->filename, fieldTagFunc, ndeTagFuncFree, (void*)s, hts->title, 2048, 1);
}
}
NDE_Table_DestroyScanner(g_table, s);
LeaveCriticalSection(&g_db_cs);
if (found) return 1;
} // not http://
return FALSE;
}
DWORD doGuessProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
{
// call the old wndproc, and if it produces empty results or no results, try to guess
extendedFileInfoStruct *p = (extendedFileInfoStruct*)wParam;
LRESULT ret = CallWindowProc(wa_oldWndProc, hwndDlg, uMsg, wParam, lParam);
if (NULL != p &&
(!ret || !p->ret[0]) &&
g_config->ReadInt(L"guessmode", 0) != 2)
{
int which = 0;
if (NULL != p->metadata)
{
if (!_stricmp(p->metadata, "artist")) which = 1;
else if (!_stricmp(p->metadata, "title")) which = 2;
else if (!_stricmp(p->metadata, "album")) which = 3;
else if (!_stricmp(p->metadata, "track")) which = 5;
else which = 0;
}
else
which = 0;
if (which)
{
static char m_lastfn[2048];
if (NULL != p->filename && _strnicmp(m_lastfn, p->filename, sizeof(m_lastfn)))
{
m_db_found = 2;
m_lastartist[0] = 0;
m_lasttitle[0] = 0;
m_lastalbum[0] = 0;
m_lasttrack = 0;
StringCbCopyA(m_lastfn, sizeof(m_lastfn), p->filename);
int tn = 0;
wchar_t *artist = 0, *album = 0, *title = 0;
wchar_t *guessbuf = guessTitles(AutoWide(m_lastfn), &tn, &artist, &album, &title);
if (guessbuf)
{
if (artist) StringCbCopyW(m_lastartist, sizeof(m_lastartist), artist);
if (album) StringCbCopyW(m_lastalbum, sizeof(m_lastalbum), album);
if (title) StringCbCopyW(m_lasttitle, sizeof(m_lasttitle), title);
m_lasttrack = tn;
free(guessbuf);
}
}
if (m_db_found == 2)
{
if (which == 1 && m_lastartist[0])
{
WideCharToMultiByteSZ(CP_ACP, 0, m_lastartist, -1, p->ret, p->retlen, 0, 0);
return 1;
}
if (which == 2 && m_lasttitle[0])
{
WideCharToMultiByteSZ(CP_ACP, 0, m_lasttitle, -1, p->ret, p->retlen, 0, 0);
return 1;
}
if (which == 3 && m_lastalbum[0])
{
WideCharToMultiByteSZ(CP_ACP, 0, m_lastalbum, -1, p->ret, p->retlen, 0, 0);
return 1;
}
if (which == 5 && m_lasttrack)
{
_snprintf(p->ret, p->retlen, "%d", m_lasttrack);
return 1;
}
if (which == 6 && m_rating != -1)
{
_snprintf(p->ret, p->retlen, "%d", m_rating);
return 1;
}
if (which == 7 && m_playcount != -1)
{
_snprintf(p->ret, p->retlen, "%d", m_playcount);
return 1;
}
}
}
}
return (int)ret;
}