diff options
Diffstat (limited to 'Src/Plugins/Library/ml_local/TitleInfo.cpp')
| -rw-r--r-- | Src/Plugins/Library/ml_local/TitleInfo.cpp | 321 |
1 files changed, 321 insertions, 0 deletions
diff --git a/Src/Plugins/Library/ml_local/TitleInfo.cpp b/Src/Plugins/Library/ml_local/TitleInfo.cpp new file mode 100644 index 00000000..3d52a4c1 --- /dev/null +++ b/Src/Plugins/Library/ml_local/TitleInfo.cpp @@ -0,0 +1,321 @@ +#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; +}
\ No newline at end of file |