void FindController::countStringMatches(const String& string, FindOptions options, unsigned maxMatchCount)
{
if (maxMatchCount == std::numeric_limits<unsigned>::max())
--maxMatchCount;
PluginView* pluginView = pluginViewForFrame(m_webPage->mainFrame());
unsigned matchCount;
if (pluginView)
matchCount = pluginView->countFindMatches(string, core(options), maxMatchCount + 1);
else {
matchCount = m_webPage->corePage()->countFindMatches(string, core(options), maxMatchCount + 1);
m_webPage->corePage()->unmarkAllTextMatches();
}
if (matchCount > maxMatchCount)
matchCount = static_cast<unsigned>(kWKMoreThanMaximumMatchCount);
m_webPage->send(Messages::WebPageProxy::DidCountStringMatches(string, matchCount));
}
void FindController::updateFindUIAfterPageScroll(bool found, const String& string, FindOptions options, unsigned maxMatchCount)
{
Frame* selectedFrame = frameWithSelection(m_webPage->corePage());
PluginView* pluginView = pluginViewForFrame(m_webPage->mainFrame());
bool shouldShowOverlay = false;
if (!found) {
if (!pluginView)
m_webPage->corePage()->unmarkAllTextMatches();
// Clear the selection.
if (selectedFrame)
selectedFrame->selection()->clear();
hideFindIndicator();
m_webPage->send(Messages::WebPageProxy::DidFailToFindString(string));
} else {
shouldShowOverlay = options & FindOptionsShowOverlay;
bool shouldShowHighlight = options & FindOptionsShowHighlight;
unsigned matchCount = 1;
if (shouldShowOverlay || shouldShowHighlight) {
if (maxMatchCount == numeric_limits<unsigned>::max())
--maxMatchCount;
if (pluginView) {
matchCount = pluginView->countFindMatches(string, core(options), maxMatchCount + 1);
shouldShowOverlay = false;
} else {
m_webPage->corePage()->unmarkAllTextMatches();
matchCount = m_webPage->corePage()->markAllMatchesForText(string, core(options), shouldShowHighlight, maxMatchCount + 1);
}
// Check if we have more matches than allowed.
if (matchCount > maxMatchCount) {
shouldShowOverlay = false;
matchCount = static_cast<unsigned>(kWKMoreThanMaximumMatchCount);
}
}
m_webPage->send(Messages::WebPageProxy::DidFindString(string, matchCount));
if (!(options & FindOptionsShowFindIndicator) || !updateFindIndicator(selectedFrame, shouldShowOverlay)) {
// Either we shouldn't show the find indicator, or we couldn't update it.
hideFindIndicator();
}
}
if (!shouldShowOverlay) {
if (m_findPageOverlay) {
// Get rid of the overlay.
m_webPage->uninstallPageOverlay(m_findPageOverlay, true);
}
} else {
if (!m_findPageOverlay) {
RefPtr<PageOverlay> findPageOverlay = PageOverlay::create(this);
m_findPageOverlay = findPageOverlay.get();
m_webPage->installPageOverlay(findPageOverlay.release(), true);
m_findPageOverlay->setNeedsDisplay();
} else
m_findPageOverlay->setNeedsDisplay();
}
}
void FindController::updateFindUIAfterPageScroll(bool found, const String& string, FindOptions options, unsigned maxMatchCount)
{
Frame* selectedFrame = frameWithSelection(m_webPage->corePage());
PluginView* pluginView = pluginViewForFrame(m_webPage->mainFrame());
bool shouldShowOverlay = false;
if (!found) {
if (!pluginView)
m_webPage->corePage()->unmarkAllTextMatches();
if (selectedFrame)
selectedFrame->selection().clear();
hideFindIndicator();
didFailToFindString();
m_webPage->send(Messages::WebPageProxy::DidFailToFindString(string));
} else {
shouldShowOverlay = options & FindOptionsShowOverlay;
bool shouldShowHighlight = options & FindOptionsShowHighlight;
bool shouldDetermineMatchIndex = options & FindOptionsDetermineMatchIndex;
unsigned matchCount = 1;
if (shouldDetermineMatchIndex) {
if (pluginView)
matchCount = pluginView->countFindMatches(string, core(options), maxMatchCount + 1);
else
matchCount = m_webPage->corePage()->countFindMatches(string, core(options), maxMatchCount + 1);
}
if (shouldShowOverlay || shouldShowHighlight) {
if (maxMatchCount == std::numeric_limits<unsigned>::max())
--maxMatchCount;
if (pluginView) {
if (!shouldDetermineMatchIndex)
matchCount = pluginView->countFindMatches(string, core(options), maxMatchCount + 1);
shouldShowOverlay = false;
} else {
m_webPage->corePage()->unmarkAllTextMatches();
matchCount = m_webPage->corePage()->markAllMatchesForText(string, core(options), shouldShowHighlight, maxMatchCount + 1);
}
// If we have a large number of matches, we don't want to take the time to paint the overlay.
if (matchCount > maxMatchCount) {
shouldShowOverlay = false;
matchCount = static_cast<unsigned>(kWKMoreThanMaximumMatchCount);
}
}
if (matchCount == static_cast<unsigned>(kWKMoreThanMaximumMatchCount))
m_foundStringMatchIndex = -1;
else {
if (m_foundStringMatchIndex < 0)
m_foundStringMatchIndex += matchCount;
if (m_foundStringMatchIndex >= (int) matchCount)
m_foundStringMatchIndex -= matchCount;
}
m_webPage->send(Messages::WebPageProxy::DidFindString(string, matchCount, m_foundStringMatchIndex));
if (!(options & FindOptionsShowFindIndicator) || !selectedFrame || !updateFindIndicator(*selectedFrame, shouldShowOverlay))
hideFindIndicator();
}
if (!shouldShowOverlay) {
if (m_findPageOverlay)
m_webPage->uninstallPageOverlay(m_findPageOverlay, PageOverlay::FadeMode::Fade);
} else {
if (!m_findPageOverlay) {
RefPtr<PageOverlay> findPageOverlay = PageOverlay::create(this);
m_findPageOverlay = findPageOverlay.get();
m_webPage->installPageOverlay(findPageOverlay.release(), PageOverlay::FadeMode::Fade);
}
m_findPageOverlay->setNeedsDisplay();
}
}