Abstract notion of off-the-record in OmniboxMetricsProvider

chrome/browser/metrics/omnibox_metrics_provider.cc

 // Copyright 2014 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/metrics/omnibox_metrics_provider.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
-#include "chrome/browser/ui/browser_otr_state.h"
 #include "components/metrics/metrics_log.h"
 #include "components/metrics/proto/omnibox_event.pb.h"
 #include "components/metrics/proto/omnibox_input_type.pb.h"
 #include "components/omnibox/browser/autocomplete_match.h"
 #include "components/omnibox/browser/autocomplete_provider.h"
 #include "components/omnibox/browser/autocomplete_result.h"
 #include "components/omnibox/browser/omnibox_log.h"
 
 using metrics::OmniboxEventProto;
 
@@ skipping 41 matching lines @@
     case AutocompleteMatchType::CONTACT_DEPRECATED:
     case AutocompleteMatchType::NUM_TYPES:
       break;
   }
   NOTREACHED();
   return OmniboxEventProto::Suggestion::UNKNOWN_RESULT_TYPE;
 }
 
 }  // namespace
 
-OmniboxMetricsProvider::OmniboxMetricsProvider() {
-}
+OmniboxMetricsProvider::OmniboxMetricsProvider(
+    const base::Callback<bool(void)>& is_off_the_record_callback)
+    : is_off_the_record_callback_(is_off_the_record_callback) {}
 
 OmniboxMetricsProvider::~OmniboxMetricsProvider() {
 }
 
 void OmniboxMetricsProvider::OnRecordingEnabled() {
   subscription_ = OmniboxEventGlobalTracker::GetInstance()->RegisterCallback(
       base::Bind(&OmniboxMetricsProvider::OnURLOpenedFromOmnibox,
                  base::Unretained(this)));
 }
 
 void OmniboxMetricsProvider::OnRecordingDisabled() {
   subscription_.reset();
 }
 
 void OmniboxMetricsProvider::ProvideGeneralMetrics(
     metrics::ChromeUserMetricsExtension* uma_proto) {
   uma_proto->mutable_omnibox_event()->Swap(
       omnibox_events_cache.mutable_omnibox_event());
 }
 
 void OmniboxMetricsProvider::OnURLOpenedFromOmnibox(OmniboxLog* log) {
-  // We simply don't log events to UMA if there is a single incognito
-  // session visible. In the future, it may be worth revisiting this to
-  // still log events from non-incognito sessions.
-  if (!chrome::IsOffTheRecordSessionActive())
+  // Do not log events to UMA if the embedder reports that the user is in an
+  // off-the-record context.
+  if (is_off_the_record_callback_.Run())
     RecordOmniboxOpenedURL(*log);
 }
 
 void OmniboxMetricsProvider::RecordOmniboxOpenedURL(const OmniboxLog& log) {
   std::vector<base::StringPiece16> terms = base::SplitStringPiece(
       log.text, base::kWhitespaceUTF16,
       base::KEEP_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
 
   OmniboxEventProto* omnibox_event = omnibox_events_cache.add_omnibox_event();
   omnibox_event->set_time(metrics::MetricsLog::GetCurrentTime());
@@ skipping 39 matching lines @@
     if (i->typed_count != -1)
       suggestion->set_typed_count(i->typed_count);
   }
   for (ProvidersInfo::const_iterator i(log.providers_info.begin());
        i != log.providers_info.end(); ++i) {
     OmniboxEventProto::ProviderInfo* provider_info =
         omnibox_event->add_provider_info();
     provider_info->CopyFrom(*i);
   }
 }