Move the logic to retrieve the WebPluginContainer to LocalFrame and Node.

third_party/WebKit/Source/web/WebLocalFrameImpl.cpp

 /*
  * Copyright (C) 2009 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 228 matching lines @@
 static int g_frame_count = 0;
 
 static HeapVector<ScriptSourceCode> CreateSourcesVector(
     const WebScriptSource* sources_in,
     unsigned num_sources) {
   HeapVector<ScriptSourceCode> sources;
   sources.Append(sources_in, num_sources);
   return sources;
 }
 
-WebPluginContainerBase* WebLocalFrameImpl::PluginContainerFromFrame(
-    LocalFrame* frame) {
-  if (!frame)
-    return 0;
-  if (!frame->GetDocument() || !frame->GetDocument()->IsPluginDocument())
-    return 0;
-  PluginDocument* plugin_document = ToPluginDocument(frame->GetDocument());
-  return ToWebPluginContainerBase(plugin_document->GetPluginView());
-}
-
-WebPluginContainerBase* WebLocalFrameImpl::CurrentPluginContainer(
-    LocalFrame* frame,
-    Node* node) {
-  WebPluginContainerBase* plugin_container = PluginContainerFromFrame(frame);
-  if (plugin_container)
-    return plugin_container;
-
-  if (!node) {
-    DCHECK(frame->GetDocument());
-    node = frame->GetDocument()->FocusedElement();
-  }
-  return ToWebPluginContainerBase(WebNode::PluginContainerFromNode(node));
-}
-
 // Simple class to override some of PrintContext behavior. Some of the methods
 // made virtual so that they can be overridden by ChromePluginPrintContext.
 class ChromePrintContext : public PrintContext {
   WTF_MAKE_NONCOPYABLE(ChromePrintContext);
 
  public:
   explicit ChromePrintContext(LocalFrame* frame)
       : PrintContext(frame), printed_page_width_(0) {}
 
   ~ChromePrintContext() override {}
@@ skipping 774 matching lines @@
   // Make sure the first letter is upper case.
   command.replace(0, 1, command.Substring(0, 1).UpperASCII());
 
   // Remove the trailing ':' if existing.
   if (command[command.length() - 1] == UChar(':'))
     command = command.Substring(0, command.length() - 1);
 
   Node* plugin_lookup_context_node =
       context_menu_node_ && name == "Copy" ? context_menu_node_ : nullptr;
   WebPluginContainerBase* plugin_container =
-      CurrentPluginContainer(GetFrame(), plugin_lookup_context_node);
+      GetFrame()->GetWebPluginContainerBase(plugin_lookup_context_node);
   if (plugin_container && plugin_container->ExecuteEditCommand(name))
     return true;
 
   return GetFrame()->GetEditor().ExecuteCommand(command);
 }
 
 bool WebLocalFrameImpl::ExecuteCommand(const WebString& name,
                                        const WebString& value) {
   DCHECK(GetFrame());
 
-  WebPluginContainerBase* plugin_container = CurrentPluginContainer(GetFrame());
+  WebPluginContainerBase* plugin_container =
+      GetFrame()->GetWebPluginContainerBase();
   if (plugin_container && plugin_container->ExecuteEditCommand(name, value))
     return true;
 
   return GetFrame()->GetEditor().ExecuteCommand(name, value);
 }
 
 bool WebLocalFrameImpl::IsCommandEnabled(const WebString& name) const {
   DCHECK(GetFrame());
   return GetFrame()->GetEditor().CreateCommand(name).IsEnabled();
 }
 
 void WebLocalFrameImpl::EnableSpellChecking(bool enable) {
   if (enable == IsSpellCheckingEnabled())
     return;
   GetFrame()->GetSpellChecker().ToggleSpellCheckingEnabled();
 }
 
 bool WebLocalFrameImpl::IsSpellCheckingEnabled() const {
   return GetFrame()->GetSpellChecker().IsSpellCheckingEnabled();
 }
 
 void WebLocalFrameImpl::ReplaceMisspelledRange(const WebString& text) {
   // If this caret selection has two or more markers, this function replace the
   // range covered by the first marker with the specified word as Microsoft Word
   // does.
-  if (PluginContainerFromFrame(GetFrame()))
+  if (GetFrame()->GetWebPluginContainerBase())
     return;
 
   // TODO(editing-dev): The use of updateStyleAndLayoutIgnorePendingStylesheets
   // needs to be audited.  see http://crbug.com/590369 for more details.
   GetFrame()->GetDocument()->UpdateStyleAndLayoutIgnorePendingStylesheets();
 
   GetFrame()->GetSpellChecker().ReplaceMisspelledRange(text);
 }
 
 void WebLocalFrameImpl::RemoveSpellingMarkers() {
   GetFrame()->GetSpellChecker().RemoveSpellingMarkers();
 }
 
 void WebLocalFrameImpl::RemoveSpellingMarkersUnderWords(
     const WebVector<WebString>& words) {
   Vector<String> converted_words;
   converted_words.Append(words.Data(), words.size());
   GetFrame()->RemoveSpellingMarkersUnderWords(converted_words);
 }
 
 bool WebLocalFrameImpl::HasSelection() const {
+  DCHECK(GetFrame());
   WebPluginContainerBase* plugin_container =
-      PluginContainerFromFrame(GetFrame());
+      GetFrame()->GetWebPluginContainerBase();
   if (plugin_container)
     return plugin_container->Plugin()->HasSelection();
 
   // frame()->selection()->isNone() never returns true.
   return GetFrame()
              ->Selection()
              .ComputeVisibleSelectionInDOMTreeDeprecated()
              .Start() != GetFrame()
                              ->Selection()
                              .ComputeVisibleSelectionInDOMTreeDeprecated()
                              .end();
 }
 
 WebRange WebLocalFrameImpl::SelectionRange() const {
   // TODO(editing-dev): The use of updateStyleAndLayoutIgnorePendingStylesheets
   // needs to be audited.  See http://crbug.com/590369 for more details.
   GetFrame()->GetDocument()->UpdateStyleAndLayoutIgnorePendingStylesheets();
 
   return GetFrame()
       ->Selection()
       .ComputeVisibleSelectionInDOMTreeDeprecated()
       .ToNormalizedEphemeralRange();
 }
 
 WebString WebLocalFrameImpl::SelectionAsText() const {
+  DCHECK(GetFrame());
   WebPluginContainerBase* plugin_container =
-      PluginContainerFromFrame(GetFrame());
+      GetFrame()->GetWebPluginContainerBase();
   if (plugin_container)
     return plugin_container->Plugin()->SelectionAsText();
 
   // TODO(editing-dev): The use of updateStyleAndLayoutIgnorePendingStylesheets
   // needs to be audited.  See http://crbug.com/590369 for more details.
   GetFrame()->GetDocument()->UpdateStyleAndLayoutIgnorePendingStylesheets();
 
   String text = GetFrame()->Selection().SelectedText(
       TextIteratorBehavior::EmitsObjectReplacementCharacterBehavior());
 #if OS(WIN)
   ReplaceNewlinesWithWindowsStyleNewlines(text);
 #endif
   ReplaceNBSPWithSpace(text);
   return text;
 }
 
 WebString WebLocalFrameImpl::SelectionAsMarkup() const {
   WebPluginContainerBase* plugin_container =
-      PluginContainerFromFrame(GetFrame());
+      GetFrame()->GetWebPluginContainerBase();
   if (plugin_container)
     return plugin_container->Plugin()->SelectionAsMarkup();
 
   // TODO(editing-dev): The use of updateStyleAndLayoutIgnorePendingStylesheets
   // needs to be audited.  See http://crbug.com/590369 for more details.
   // Selection normalization and markup generation require clean layout.
   GetFrame()->GetDocument()->UpdateStyleAndLayoutIgnorePendingStylesheets();
 
   return GetFrame()->Selection().SelectedHTMLForClipboard();
 }
@@ skipping 191 matching lines @@
   GetFrame()->Selection().SetCaretVisible(visible);
 }
 
 VisiblePosition WebLocalFrameImpl::VisiblePositionForViewportPoint(
     const WebPoint& point_in_viewport) {
   return VisiblePositionForContentsPoint(
       GetFrame()->View()->ViewportToContents(point_in_viewport), GetFrame());
 }
 
 WebPlugin* WebLocalFrameImpl::FocusedPluginIfInputMethodSupported() {
-  WebPluginContainerBase* container =
-      WebLocalFrameImpl::CurrentPluginContainer(GetFrame());
+  WebPluginContainerBase* container = GetFrame()->GetWebPluginContainerBase();
   if (container && container->SupportsInputMethod())
     return container->Plugin();
   return 0;
 }
 
 int WebLocalFrameImpl::PrintBegin(const WebPrintParams& print_params,
                                   const WebNode& constrain_to_node) {
   DCHECK(!GetFrame()->GetDocument()->IsFrameSet());
   WebPluginContainerBase* plugin_container = nullptr;
   if (constrain_to_node.IsNull()) {
     // If this is a plugin document, check if the plugin supports its own
     // printing. If it does, we will delegate all printing to that.
-    plugin_container = PluginContainerFromFrame(GetFrame());
+    plugin_container = GetFrame()->GetWebPluginContainerBase();
   } else {
     // We only support printing plugin nodes for now.
     plugin_container =
         ToWebPluginContainerBase(constrain_to_node.PluginContainer());
   }
 
   if (plugin_container && plugin_container->SupportsPaginatedPrint()) {
     print_context_ = new ChromePluginPrintContext(GetFrame(), plugin_container,
                                                   print_params);
   } else {
@@ skipping 23 matching lines @@
   return print_context_->SpoolSinglePage(canvas, page);
 }
 
 void WebLocalFrameImpl::PrintEnd() {
   DCHECK(print_context_);
   print_context_->EndPrintMode();
   print_context_.Clear();
 }
 
 bool WebLocalFrameImpl::IsPrintScalingDisabledForPlugin(const WebNode& node) {
+  DCHECK(GetFrame());
   WebPluginContainerBase* plugin_container =
-      node.IsNull() ? PluginContainerFromFrame(GetFrame())
+      node.IsNull() ? GetFrame()->GetWebPluginContainerBase()
                     : ToWebPluginContainerBase(node.PluginContainer());
 
   if (!plugin_container || !plugin_container->SupportsPaginatedPrint())
     return false;
 
   return plugin_container->IsPrintScalingDisabled();
 }
 
 bool WebLocalFrameImpl::GetPrintPresetOptionsForPlugin(
     const WebNode& node,
     WebPrintPresetOptions* preset_options) {
   WebPluginContainerBase* plugin_container =
-      node.IsNull() ? PluginContainerFromFrame(GetFrame())
+      node.IsNull() ? GetFrame()->GetWebPluginContainerBase()
                     : ToWebPluginContainerBase(node.PluginContainer());
 
   if (!plugin_container || !plugin_container->SupportsPaginatedPrint())
     return false;
 
   return plugin_container->GetPrintPresetOptionsFromDocument(preset_options);
 }
 
 bool WebLocalFrameImpl::HasCustomPageSizeStyle(int page_index) {
   return GetFrame()
@@ skipping 441 matching lines @@
 
 void WebLocalFrameImpl::DidFail(const ResourceError& error,
                                 bool was_provisional,
                                 HistoryCommitType commit_type) {
   if (!Client())
     return;
   WebURLError web_error = error;
   WebHistoryCommitType web_commit_type =
       static_cast<WebHistoryCommitType>(commit_type);
 
-  if (WebPluginContainerBase* plugin = PluginContainerFromFrame(GetFrame()))
+  if (WebPluginContainerBase* plugin = GetFrame()->GetWebPluginContainerBase())
     plugin->DidFailLoading(error);
 
   if (was_provisional)
     Client()->DidFailProvisionalLoad(web_error, web_commit_type);
   else
     Client()->DidFailLoad(web_error, web_commit_type);
 }
 
 void WebLocalFrameImpl::DidFinish() {
   if (!Client())
     return;
 
-  if (WebPluginContainerBase* plugin = PluginContainerFromFrame(GetFrame()))
+  if (WebPluginContainerBase* plugin = GetFrame()->GetWebPluginContainerBase())
     plugin->DidFinishLoading();
 
   Client()->DidFinishLoad();
 }
 
 void WebLocalFrameImpl::SetCanHaveScrollbars(bool can_have_scrollbars) {
   if (FrameView* view = GetFrameView())
     view->SetCanHaveScrollbars(can_have_scrollbars);
 }
 
@@ skipping 660 matching lines @@
 TextCheckerClient& WebLocalFrameImpl::GetTextCheckerClient() const {
   return *text_checker_client_;
 }
 
 void WebLocalFrameImpl::SetTextCheckClient(
     WebTextCheckClient* text_check_client) {
   text_check_client_ = text_check_client;
 }
 
 }  // namespace blink