Adding previews information to PLM UKM

chrome/browser/previews/previews_infobar_tab_helper.h

 // Copyright 2016 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.
 
 #ifndef CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_
 #define CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_
 
 #include "base/macros.h"
 #include "base/optional.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_pingback_client.h"
+#include "components/ukm/ukm_source.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/browser/web_contents_user_data.h"
 
 // Tracks whether a previews infobar has been shown for a page. Handles showing
 // the infobar when the main frame response indicates a Lite Page.
 class PreviewsInfoBarTabHelper
     : public content::WebContentsObserver,
       public content::WebContentsUserData<PreviewsInfoBarTabHelper> {
  public:
   ~PreviewsInfoBarTabHelper() override;
 
   // Indicates whether the InfoBar for a preview has been shown for the page.
   bool displayed_preview_infobar() const {
     return displayed_preview_infobar_;
   }
 
   // Sets whether the InfoBar for a preview has been shown for the page.
   // |displayed_preview_infobar_| is reset to false on
   // DidStartProvisionalLoadForFrame for the main frame.
   void set_displayed_preview_infobar(bool displayed) {
     displayed_preview_infobar_ = displayed;
   }
 
   // The data saver page identifier of the current page load.
   const base::Optional<data_reduction_proxy::NavigationID>&
   committed_data_saver_navigation_id() {
     return committed_data_saver_navigation_id_;
   }
 
+  // Source ID for the last main frame navigation.
+  base::Optional<ukm::SourceId> source_id() const { return source_id_; }
+
  private:
   friend class content::WebContentsUserData<PreviewsInfoBarTabHelper>;
   friend class PreviewsInfoBarTabHelperUnitTest;
 
   explicit PreviewsInfoBarTabHelper(content::WebContents* web_contents);
 
   // Clears the last navigation from this tab in the pingback client.
   void ClearLastNavigationAsync() const;
 
   // Overridden from content::WebContentsObserver:
   void DidFinishNavigation(
       content::NavigationHandle* navigation_handle) override;
 
   // Stored for use in the destructor.
   content::BrowserContext* browser_context_;
 
   // True if the InfoBar for a preview has been shown for the page.
   bool displayed_preview_infobar_;
 
   // The data saver page identifier of the current page load.
   base::Optional<data_reduction_proxy::NavigationID>
       committed_data_saver_navigation_id_;
 
+  // Source ID for the last main frame navigation.
+  base::Optional<ukm::SourceId> source_id_;
+
   DISALLOW_COPY_AND_ASSIGN(PreviewsInfoBarTabHelper);
 };
 
 #endif  // CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_