Makes match preview send the dimensions of the omnibox to the page.

chrome/browser/tab_contents/match_preview.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/tab_contents/match_preview.h"
 
 #include <algorithm>
 
 #include "base/command_line.h"
+#include "base/string_number_conversions.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autocomplete/autocomplete.h"
 #include "chrome/browser/favicon_service.h"
 #include "chrome/browser/history/history_marshaling.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
 #include "chrome/browser/renderer_host/render_widget_host.h"
 #include "chrome/browser/renderer_host/render_widget_host_view.h"
 #include "chrome/browser/search_engines/template_url.h"
 #include "chrome/browser/search_engines/template_url_model.h"
@@ skipping 10 matching lines @@
 #include "chrome/common/page_transition_types.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/common/renderer_preferences.h"
 #include "gfx/codec/png_codec.h"
 #include "ipc/ipc_message.h"
 
 namespace {
 
 const char kUserInputScript[] =
     "if (window.chrome.userInput) window.chrome.userInput(\"$1\");";
+const char kUserDoneScript[] =
+    "if (window.chrome.userWantsQuery) window.chrome.userWantsQuery(\"$1\");";
+const char kSetOmniboxBoundsScript[] =
+    "if (window.chrome.setOmniboxDimensions) "
+    "window.chrome.setOmniboxDimensions($1, $2, $3, $4);";
 
 // Sends the user input script to |tab_contents|. |text| is the text the user
 // input into the omnibox.
 void SendUserInputScript(TabContents* tab_contents,
                          const string16& text,
                          bool done) {
-  // TODO: support done.
   string16 escaped_text(text);
   ReplaceSubstringsAfterOffset(&escaped_text, 0L, ASCIIToUTF16("\""),
                                ASCIIToUTF16("\\\""));
-  string16 script = ReplaceStringPlaceholders(ASCIIToUTF16(kUserInputScript),
-                                              escaped_text, NULL);
+  string16 script = ReplaceStringPlaceholders(
+      ASCIIToUTF16(done ? kUserDoneScript : kUserInputScript),
+      escaped_text,
+      NULL);
   tab_contents->render_view_host()->ExecuteJavascriptInWebFrame(
       std::wstring(),
       UTF16ToWide(script));
+}
+
+// Sends the script for setting the bounds of the omnibox to |tab_contents|.
+void SendOmniboxBoundsScript(TabContents* tab_contents,
+                             const gfx::Rect& bounds) {
+  std::vector<string16> bounds_vector;
+  bounds_vector.push_back(base::IntToString16(bounds.x()));
+  bounds_vector.push_back(base::IntToString16(bounds.y()));
+  bounds_vector.push_back(base::IntToString16(bounds.width()));
+  bounds_vector.push_back(base::IntToString16(bounds.height()));
+  string16 script = ReplaceStringPlaceholders(
+      ASCIIToUTF16(kSetOmniboxBoundsScript),
+      bounds_vector,
+      NULL);
+  tab_contents->render_view_host()->ExecuteJavascriptInWebFrame(
+      std::wstring(),
+      UTF16ToWide(script));
 }
 
 }  // namespace
 
 // FrameLoadObserver is responsible for waiting for the TabContents to finish
 // loading and when done sending the necessary script down to the page.
 class MatchPreview::FrameLoadObserver : public NotificationObserver {
  public:
   FrameLoadObserver(MatchPreview* match_preview, const string16& text)
       : match_preview_(match_preview),
@@ skipping 24 matching lines @@
     switch (type.value) {
       case NotificationType::LOAD_COMPLETED_MAIN_FRAME: {
         int page_id = *(Details<int>(details).ptr());
         NavigationEntry* active_entry =
             tab_contents_->controller().GetActiveEntry();
         if (!active_entry || active_entry->page_id() != page_id ||
             active_entry->unique_id() != unique_id_) {
           return;
         }
 
+        if (match_preview_) {
+          gfx::Rect bounds = match_preview_->GetOmniboxBoundsInTermsOfPreview();
+          if (!bounds.IsEmpty())
+            SendOmniboxBoundsScript(tab_contents_, bounds);
+        }
+
         SendUserInputScript(tab_contents_, text_, send_done_);
 
         if (match_preview_)
           match_preview_->PageFinishedLoading();
 
         delete this;
         return;
       }
 
       case NotificationType::TAB_CONTENTS_DESTROYED:
@@ skipping 318 matching lines @@
       match.type == AutocompleteMatch::SEARCH_HISTORY ||
       match.type == AutocompleteMatch::SEARCH_SUGGEST) {
     TemplateURLModel* model = tab_contents->profile()->GetTemplateURLModel();
     template_url = model ? model->GetDefaultSearchProvider() : NULL;
   }
   TemplateURLID template_url_id = template_url ? template_url->id() : 0;
 
   if (template_url && template_url->supports_instant() &&
       TemplateURL::SupportsReplacement(template_url)) {
     if (template_url_id == template_url_id_) {
-      if (frame_load_observer_.get()) {
+      if (is_waiting_for_load()) {
         // The page hasn't loaded yet. We'll send the script down when it does.
         frame_load_observer_->set_text(user_text_);
         return;
       }
       SendUserInputScript(preview_contents_.get(), user_text_, false);
       if (complete_suggested_text_.size() > user_text_.size() &&
           !complete_suggested_text_.compare(0, user_text_.size(), user_text_)) {
         *suggested_text = complete_suggested_text_.substr(user_text_.size());
       }
     } else {
       // TODO: should we use a different url for instant?
       GURL url = GURL(template_url->url()->ReplaceSearchTerms(
           *template_url, std::wstring(),
           TemplateURLRef::NO_SUGGESTIONS_AVAILABLE, std::wstring()));
       // user_text_ is sent once the page finishes loading by FrameLoadObserver.
       preview_contents_->controller().LoadURL(url, GURL(), match.transition);
       frame_load_observer_.reset(new FrameLoadObserver(this, user_text_));
     }
   } else {
+    template_url_id_ = 0;
     frame_load_observer_.reset(NULL);
     preview_contents_->controller().LoadURL(url_, GURL(), match.transition);
   }
 
   template_url_id_ = template_url_id;
 }
 
+void MatchPreview::SetOmniboxBounds(const gfx::Rect& bounds) {
+  if (omnibox_bounds_ == bounds)
+    return;
+
+  omnibox_bounds_ = bounds;
+  if (preview_contents_.get() && is_showing_instant() &&
+      !is_waiting_for_load()) {
+    SendOmniboxBoundsScript(preview_contents_.get(),
+                            GetOmniboxBoundsInTermsOfPreview());
+  }
+}
+
 void MatchPreview::DestroyPreviewContents() {
   delegate_->HideMatchPreview();
   delete ReleasePreviewContents(false);
 }
 
 void MatchPreview::CommitCurrentPreview() {
   DCHECK(preview_contents_.get());
   delegate_->CommitMatchPreview();
 }
 
 TabContents* MatchPreview::ReleasePreviewContents(bool commit_history) {
+  omnibox_bounds_ = gfx::Rect();
   template_url_id_ = 0;
   url_ = GURL();
   user_text_.clear();
   complete_suggested_text_.clear();
   if (frame_load_observer_.get()) {
     frame_load_observer_->DetachFromPreview();
     // FrameLoadObserver will delete itself either when the TabContents is
     // deleted, or when the page finishes loading.
     FrameLoadObserver* unused ALLOW_UNUSED = frame_load_observer_.release();
   }
@@ skipping 34 matching lines @@
 void MatchPreview::PreviewDidPaint() {
   DCHECK(!is_active_);
   is_active_ = true;
   delegate_->ShowMatchPreview();
 }
 
 void MatchPreview::PageFinishedLoading() {
   // FrameLoadObserver deletes itself after this call.
   FrameLoadObserver* unused ALLOW_UNUSED = frame_load_observer_.release();
 }
+
+gfx::Rect MatchPreview::GetOmniboxBoundsInTermsOfPreview() {
+  if (omnibox_bounds_.IsEmpty())
+    return omnibox_bounds_;
+
+  gfx::Rect preview_bounds(delegate_->GetMatchPreviewBounds());
+  return gfx::Rect(omnibox_bounds_.x() - preview_bounds.x(),
+                   omnibox_bounds_.y() - preview_bounds.y(),
+                   omnibox_bounds_.width(),
+                   omnibox_bounds_.height());
+}