void CReBar::OnCommand(WORD id, WORD notifyCode, HWND hWndControl)
{
switch (id)
{
case ID_TOOLS:
{
// If the button is clicked again while menu is shown, we shouldn't show it again
static bool bIgnoreNext = false;
if (hWndControl == (HWND)-1)
{
bIgnoreNext = true;
return;
}
if (bIgnoreNext)
{
bIgnoreNext = false;
return;
}
// Install a hook to capture mouse events when the menu is shown
HHOOK hHook = SetWindowsHookEx(WH_MOUSE_LL, LLMouseProc, GetModuleHandle(NULL), 0);
ASSERT(hHook);
LLMouseProc(-1, -1, (LPARAM)hHook);
LLMouseProc(-2, -2, (LPARAM)this);
// Show menu, make sure commands get sent to the main window
RECT rc;
GetWindowRect(m_btnTools, &rc);
VERIFY(TrackPopupMenu(m_hToolsMenu, TPM_RIGHTALIGN|TPM_TOPALIGN, rc.right, rc.bottom,
0, GetTopLevelParent(m_hWnd), NULL));
UnhookWindowsHookEx(hHook);
// Restore button state
POINT pt;
GetCursorPos(&pt);
ScreenToClient(m_btnTools, &pt);
PostMessage(m_btnTools, WM_LBUTTONUP, 0, MAKELPARAM(pt.x, pt.y));
}
break;
case ID_TOOLS_STATUSBAR:
CheckMenuItem(m_hToolsMenu, ID_TOOLS_STATUSBAR,
(GETPREFBOOL(_T("MainWnd"), _T("ShowStatusBar")) ? MF_CHECKED : MF_UNCHECKED));
break;
case ID_TOOLS_LOG:
CheckMenuItem(m_hToolsMenu, ID_TOOLS_LOG,
(GETPREFBOOL(_T("MainWnd"), _T("ShowLog")) ? MF_CHECKED : MF_UNCHECKED));
break;
case ID_SEEN:
m_btnSeen.SetCheck(!m_btnSeen.GetCheck());
SETPREFBOOL(_T("ShowSeenMovies"), m_btnSeen.GetCheck());
SendMessage(*GetDB(), WM_PREFCHANGED);
GetDB()->Filter();
break;
case ID_VIEW:
m_btnView.SetCheck(!m_btnView.GetCheck());
SETPREFBOOL(_T("ViewType"), m_btnView.GetCheck());
SendMessage(GetMainWnd(), WM_SWITCHVIEW);
break;
case ID_TOGGLEVIEWSTATUS:
m_btnView.SetCheck(!m_btnView.GetCheck());
break;
case ID_SORT:
{
// If the button is clicked again while menu is shown, we shouldn't show it again
static bool bIgnoreNext = false;
if (hWndControl == (HWND)-1)
{
bIgnoreNext = true;
return;
}
if (bIgnoreNext)
{
bIgnoreNext = false;
return;
}
// Install a hook to capture mouse events when the menu is shown
HHOOK hHook = SetWindowsHookEx(WH_MOUSE_LL, LLMouseProc, GetModuleHandle(NULL), 0);
ASSERT(hHook);
LLMouseProc(-1, -1, (LPARAM)hHook);
LLMouseProc(-2, -2, (LPARAM)this);
// Show menu, make sure commands get sent to THIS window
RECT rc;
GetWindowRect(m_btnSort, &rc);
VERIFY(TrackPopupMenu(m_hSortMenu, TPM_RIGHTALIGN|TPM_TOPALIGN, rc.right, rc.bottom,
0, m_hWnd, NULL));
UnhookWindowsHookEx(hHook);
// Restore button state
POINT pt;
GetCursorPos(&pt);
ScreenToClient(m_btnSort, &pt);
PostMessage(m_btnSort, WM_LBUTTONUP, 0, MAKELPARAM(pt.x, pt.y));
}
break;
case ID_SEARCH:
if (notifyCode == EN_RETURN || (m_bInstantSearch && notifyCode == EN_CHANGE))
{
RString strText = m_eSearch.GetText();
RObArray<RString> keywords;
if (m_bLiteralSearch)
{
if (!strText.IsEmpty())
keywords.Add(strText);
}
else
{
RRegEx re(_T("(?:\"([^\"]+)\")|([^ ]+)"));
if (!strText.IsEmpty())
keywords.Add(strText);
while (re.Search(strText))
{
RString strMatch;
for (INT_PTR i = 1; i < re.GetMatchCount() && strMatch.IsEmpty(); ++i)
strMatch = re.GetMatch(i);
strMatch.Replace(_T("\""), _T(""));
keywords.Add(strMatch);
strText.Replace(re.GetMatch(0), _T(""));
}
}
GetDB()->FilterByKeywords(keywords);
}
break;
case ID_SORT_TITLEASC:
case ID_SORT_TITLEDES:
case ID_SORT_YEARASC:
case ID_SORT_YEARDES:
case ID_SORT_RATINGASC:
case ID_SORT_RATINGDES:
case ID_SORT_FILETIMEASC:
case ID_SORT_FILETIMEDES:
SETPREFINT(_T("SortBy"), id - ID_SORT_TITLEASC);
CheckMenuRadioItem(m_hSortMenu, ID_SORT_TITLEASC, ID_SORT_FILETIMEDES, id, MF_CHECKED);
PostMessage(*GetDB(), WM_PREFCHANGED);
PostMessage(m_hWnd, WM_COMMAND, MAKEWPARAM(ID_SEARCH, EN_RETURN), (LPARAM)(HWND)m_eSearch);
break;
default:
ASSERT(false);
}
}
void ParseFileName(RString_ strFileName, RString &strTitle, RString &strYear)
{
INT_PTR m, n;
RString strTemp;
strTitle = strFileName;
strYear.Empty();
// only keep file name
n = strTitle.ReverseFind(_T('\\'));
if (n != -1)
strTitle = strTitle.Mid(n + 1);
//strTitle.MakeLower();
// strip file extension (max 4 chars)
n = strTitle.ReverseFind(_T('.'));
if (n != -1 && strTitle.GetLength() - n < 6)
strTitle = strTitle.Left(n);
// replace []{} by ()
strTitle.Replace(_T('['), _T('('));
strTitle.Replace(_T(']'), _T(')'));
strTitle.Replace(_T('{'), _T('('));
strTitle.Replace(_T('}'), _T(')'));
/*
// replace anything not ,()0-9a-zA-Z by a space
for (int i = 0; i < strTitle.GetLength(); i++)
if (!(strTitle[i] == _T(',')) &&
!(strTitle[i] == _T('\'')) &&
!(strTitle[i] == _T('(')) &&
!(strTitle[i] == _T(')')) &&
!(strTitle[i] >= _T('0') && strTitle[i] <= _T('9')) &&
!(strTitle[i] >= _T('A') && strTitle[i] <= _T('Z')) &&
!(strTitle[i] >= _T('a') && strTitle[i] <= _T('z')))
*((LPTSTR)(LPCTSTR)strTitle + i) = _T(' ');
*/
// replace ._ by a space
strTitle.Replace(_T('.'), _T(' '));
strTitle.Replace(_T('_'), _T(' '));
// remove redundant space
while (strTitle.Replace(_T(" "), _T(" ")));
strTitle.Trim();
// find year between (), strip anything following it
for (n = 0; (n = strTitle.Find(_T('('), n)) != -1; n++)
{
if (n+5 < strTitle.GetLength() && strTitle[n+5] == _T(')'))
{
strTemp = strTitle.Mid(n+1, 4);
if (StringToNumber(strTemp) > 1900)
{
strYear = strTemp;
strTitle = strTitle.Left(n);
break;
}
}
}
// remove anything between ()
for (m = 0, n = 0; (m = strTitle.Find(_T('('), n)) != -1 &&
(n = strTitle.Find(_T(')'), m)) != -1; m = 0, n = 0)
strTitle = strTitle.Left(m) + strTitle.Mid(n+1);
// find year not in (), but not as first word, strip anything following it
strTitle.Replace(_T('('), _T(' '));
strTitle.Replace(_T(')'), _T(' '));
while (strTitle.Replace(_T(" "), _T(" ")));
strTitle.Trim();
if (strYear.IsEmpty())
{
m = 0;
while (true)
{
if (m >= strTitle.GetLength())
break;
n = strTitle.Find(_T(' '), m);
if (n == -1)
n = strTitle.GetLength();
if (m == n)
{m = n + 1; continue;}
if (n - m == 4)
{
strTemp = strTitle.Mid(m, n - m);
if (StringToNumber(strTemp) > 1900 && m > 0)
{
strYear = strTemp;
strTitle = strTitle.Left(m);
break;
}
}
m = n + 1;
}
}
while (strTitle.Replace(_T(" "), _T(" ")));
strTitle.Trim();
// place some literals in front
if (strTitle.GetLength() > 5 && strTitle.Right(5) == _T(", the"))
strTitle = _T("the ") + strTitle.Left(strTitle.GetLength() - 5);
if (strTitle.GetLength() > 5 && strTitle.Right(5) == _T(", The"))
strTitle = _T("The ") + strTitle.Left(strTitle.GetLength() - 5);
if (strTitle.GetLength() > 3 && strTitle.Right(3) == _T(", a"))
strTitle = _T("a ") + strTitle.Left(strTitle.GetLength() - 3);
if (strTitle.GetLength() > 3 && strTitle.Right(3) == _T(", A"))
strTitle = _T("A ") + strTitle.Left(strTitle.GetLength() - 3);
strTitle.Replace(_T(','), _T(' '));
while (strTitle.Replace(_T(" "), _T(" ")));
strTitle.Trim();
//TRACE0(_T("ParseFileName\n strFileName = ") + strFileName + _T("\n strTitle = ") +
// strTitle + _T("\n strYear = ") + strYear + _T("\n"));
}
void ParseFileName(RString_ strFileName, RString &strTitle, RString &strYear, INT_PTR &nSeason, INT_PTR &nEpisode, RString &strAirDate, BYTE &bType)
{
INT_PTR m, n;
RString strTemp, strSeason, strEpisode;
RString strYearTmp, strMonthTmp, strDayTmp;
strTitle = strFileName;
strYear.Empty();
// only keep file name
n = strTitle.ReverseFind(_T('\\'));
if (n != -1)
strTitle = strTitle.Mid(n + 1);
//strTitle.MakeLower();
// strip file extension (max 4 chars)
n = strTitle.ReverseFind(_T('.'));
if (n != -1 && strTitle.GetLength() - n < 6)
strTitle = strTitle.Left(n);
// replace []{} by ()
strTitle.Replace(_T('['), _T('('));
strTitle.Replace(_T(']'), _T(')'));
strTitle.Replace(_T('{'), _T('('));
strTitle.Replace(_T('}'), _T(')'));
/*
// replace anything not ,()0-9a-zA-Z by a space
for (int i = 0; i < strTitle.GetLength(); i++)
if (!(strTitle[i] == _T(',')) &&
!(strTitle[i] == _T('\'')) &&
!(strTitle[i] == _T('(')) &&
!(strTitle[i] == _T(')')) &&
!(strTitle[i] >= _T('0') && strTitle[i] <= _T('9')) &&
!(strTitle[i] >= _T('A') && strTitle[i] <= _T('Z')) &&
!(strTitle[i] >= _T('a') && strTitle[i] <= _T('z')))
*((LPTSTR)(LPCTSTR)strTitle + i) = _T(' ');
*/
// remove urls
RString strUrl;
if (GetFirstMatch(strTitle, _T("([wW][wW][wW]\\.[^\\.]*?\\.[cC][oO][mM])"), &strUrl, NULL))
{
m = strTitle.Find(strUrl, 0);
if (m >= 0)
strTitle = strTitle.Left(m) + strTitle.Right(strTitle.GetLength() - (m + strUrl.GetLength()));
}
// replace ._ -by a space
strTitle.Replace(_T('.'), _T(' '));
strTitle.Replace(_T('_'), _T(' '));
strTitle.Replace(_T('-'), _T(' '));
// remove redundant space
while (strTitle.Replace(_T(" "), _T(" ")));
strTitle.Trim();
// find year between (), strip anything following it
for (n = 0; (n = strTitle.Find(_T('('), n)) != -1; n++)
{
if (n + 5 < strTitle.GetLength() && strTitle[n + 5] == _T(')'))
{
strTemp = strTitle.Mid(n + 1, 4);
if (StringToNumber(strTemp) > 1900)
{
strYear = strTemp;
strTitle = strTitle.Left(n);
break;
}
}
}
// remove anything between ()
for (m = 0, n = 0; (m = strTitle.Find(_T('('), n)) != -1 &&
(n = strTitle.Find(_T(')'), m)) != -1; m = 0, n = 0)
strTitle = strTitle.Left(m) + strTitle.Mid(n + 1);
// for TV shows. Find the season and episode and remove everything following it
if (GetFirstMatch(strTitle, _T("([Ss]\\d?\\d[Ee]\\d?\\d)"), &strTemp, NULL))
{
m = strTitle.Find(strTemp, 0);
if (GetFirstMatch(strTemp, _T("[Ss](\\d?\\d)[Ee](\\d?\\d)"), &strSeason, &strEpisode, NULL))
{
nSeason = StringToNumber(strSeason);
nEpisode = StringToNumber(strEpisode);
bType = DB_TYPE_TV;
}
if (m >= 0)
strTitle = strTitle.Left(m);
}
// for TV shows by date and parse date aired
if (GetFirstMatch(strTitle, _T("(\\d\\d\\d\\d) (\\d?\\d) (\\d?\\d)"), &strYearTmp, &strMonthTmp, &strDayTmp, NULL))
{
const RString Month[12] = { _T("Jan"), _T("Feb"), _T("Mar"), _T("Apr"),
_T("May"), _T("Jun"), _T("Jul"), _T("Aug"), _T("Sep"), _T("Oct"), _T("Nov"), _T("Dec") };
strAirDate = strDayTmp + _T(" ") + Month[StringToNumber(strMonthTmp) - 1] + _T(". ") + strYearTmp;
bType = DB_TYPE_TV;
}
// find year not in (), but not as first word, strip anything following it
strTitle.Replace(_T('('), _T(' '));
strTitle.Replace(_T(')'), _T(' '));
while (strTitle.Replace(_T(" "), _T(" ")));
strTitle.Trim();
if (strYear.IsEmpty())
{
m = 0;
while (m < strTitle.GetLength())
{
n = strTitle.Find(_T(' '), m);
if (n == -1)
n = strTitle.GetLength();
if (m == n)
{
m = n + 1; continue;
}
if (n - m == 4)
{
strTemp = strTitle.Mid(m, n - m);
if (StringToNumber(strTemp) > 1900 && m > 0)
{
strYear = strTemp;
strTitle = strTitle.Left(m);
break;
}
}
m = n + 1;
}
}
// strip 'season[s] \\d' and anything following it
RString strSeasons;
if (GetFirstMatch(strTitle, _T("([Ss]eason[s]? ?\\d?\\d(?: ?- ?\\d?\\d)?)"), &strSeasons, NULL))
{
m = strTitle.Find(strSeasons, 0);
if (m >= 0)
strTitle = strTitle.Left(m);
bType = DB_TYPE_TV;
}
//Remove episode numbers of the form XXofYY - TODO: process to return episode number
RString strOf;
if (GetFirstMatch(strTitle, _T("(\\d?\\d[oO][fF]\\d?\\d)"), &strOf, NULL))
{
m = strTitle.Find(strOf, 0);
if (m > 0)
strTitle = strTitle.Left(m);
bType = DB_TYPE_TV;
}
// strip common movie descriptors and everything after
static const RString strDescriptors[] = { _T("webrip"), _T("dvdrip"), _T("dvdscr"), _T("xvid"), _T("bdrip"),
_T("brrip"), _T("hdtv"), _T("pdtv"), _T("box set"), _T("box-set"), _T("x264") };
foreach(strDescriptors, strD)
{
m = strTitle.FindNoCase(strD, 0);
if (m > 0)
strTitle = strTitle.Left(m);
}
