Experimental AutocompleteProvider for zerosuggest.

chrome/browser/autocomplete/zerosuggest_provider.h

+// Copyright (c) 2012 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.
+//
+// This file contains the zerosuggest autocomplete provider. This experimental
+// provider is invoked when the user focuses in the omnibox prior to editing,
+// and generates search query suggestions based on the current URL. To enable
+// this provider, point --zerosuggest-url-prefix at an appropriate suggestion
+// service.
+
+#ifndef CHROME_BROWSER_AUTOCOMPLETE_ZEROSUGGEST_PROVIDER_H_
+#define CHROME_BROWSER_AUTOCOMPLETE_ZEROSUGGEST_PROVIDER_H_
+
+#include <string>
+#include <vector>
+
+#include "base/basictypes.h"
+#include "base/compiler_specific.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/string16.h"
+#include "base/timer.h"
+#include "chrome/browser/autocomplete/autocomplete_provider.h"
+#include "net/url_request/url_fetcher_delegate.h"
+
+class AutocompleteInput;
+class GURL;
+class TemplateURLService;
+
+namespace base {
+class Value;
+}
+
+namespace net {
+class URLFetcher;
+}
+
+// Autocomplete provider for searches based on the current URL.
+//
+// The controller will call StartZerosuggest when the user focuses in the
+// omnibox. After construction, the autocomplete controller repeatedly calls
+// Start() with some user input, each time expecting to receive an updated
+// set of matches.
+class ZerosuggestProvider : public AutocompleteProvider,
+                            public net::URLFetcherDelegate {
+ public:
+  ZerosuggestProvider(AutocompleteProviderListener* listener,
+                      Profile* profile,
+                      const std::string& url_prefix);
+
+  // AutocompleteProvider:
+  virtual void Start(const AutocompleteInput& input,
+                     bool /*minimal_changes*/) OVERRIDE;
+  virtual void Stop(bool clear_cached_results) OVERRIDE;
+
+  // net::URLFetcherDelegate
+  virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE;
+
+  // Sets a timer to initiate a new fetch for the given |url|, limiting
+  // suggestions to those matching |user_text|. |user_text| may be non-empty
+  // if the user previously interacted with zerosuggest suggestions and then
+  // unfocused the omnibox.
+  void StartZerosuggest(const GURL& url, const string16& user_text);
+
+ private:
+  virtual ~ZerosuggestProvider();
+
+  // Update matches given the user has typed |user_text|.
+  void UpdateMatches(const string16& user_text);
+
+  // Fetches zerosuggest suggestions for |current_query_|.
+  void Run();
+
+  // Parses results from the zerosuggest server and updates results.
+  // Returns true if results were updated.
+  bool ParseSuggestResults(base::Value* root_val);
+
+  // Converts the parsed results to a set of AutocompleteMatches and adds them
+  // to |matches_|.
+  void ConvertResultsToAutocompleteMatches();
+
+  // Adds a URL suggestion for the current URL. This should be in the top
+  // position so that pressing enter has the effect of reloading the page.
+  void AddMatchForCurrentURL();
+
+  // Adds a query suggestion from response position |result_index| with text
+  // |result| to |matches_|. Uses |search_provider| to build a search URL for
+  // this match.
+  void AddMatchForResult(const TemplateURL* search_provider,
+                         size_t result_index,
+                         const string16& result);
+
+  // Timer which must elapse before a fetch begins.
+  base::OneShotTimer<ZerosuggestProvider> timer_;
+
+  // Prefix of the URL from which to fetch zerosuggest suggestions.
+  const std::string url_prefix_;
+
+  // Used to build default search engine URLs for suggested queries.
+  TemplateURLService* template_url_service_;
+
+  // The URL for which a suggestion fetch is pending.
+  std::string current_query_;
+
+  // Current query as UTF-16.
+  string16 current_query_text_;
+
+  // What the user has typed.
+  string16 user_text_;
+
+  // Fetcher used to retrieve results.
+  scoped_ptr<net::URLFetcher> fetcher_;
+
+  // Suggestions for the most recent query.
+  std::vector<string16> results_;
+
+  DISALLOW_COPY_AND_ASSIGN(ZerosuggestProvider);
+};
+
+#endif  // CHROME_BROWSER_AUTOCOMPLETE_ZEROSUGGEST_PROVIDER_H_