// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_ENTERPRISE_CONNECTORS_ANALYSIS_CONTENT_ANALYSIS_INFO_H_
#define CHROME_BROWSER_ENTERPRISE_CONNECTORS_ANALYSIS_CONTENT_ANALYSIS_INFO_H_

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/safe_browsing/cloud_content_scanning/binary_upload_service.h"
#include "components/enterprise/connectors/core/content_analysis_info_base.h"
#include "components/safe_browsing/core/common/proto/csd.pb.h"
#include "content/public/browser/clipboard_types.h"

namespace download {
class DownloadItem;
}  // namespace download

namespace enterprise_connectors {

// Implementation of `ContentAnalysisInfoBase` for chrome/ platforms.
class ContentAnalysisInfo : public ContentAnalysisInfoBase {
 public:
  // Returns the web contents associated to the user action. This can return
  // nullptr, for example in cases where the tab is closed after the content
  // analysis triggering action has happened.
  virtual content::WebContents* web_contents() const = 0;

  // Adds shared fields to `request` before sending it to the binary upload
  // service. Connector-specific fields need to be added to the request
  // separately.
  void InitializeRequest(safe_browsing::BinaryUploadService::Request* request,
                         bool include_enterprise_only_fields = true);

  // Returns email of the active Gaia user based on the values provided by
  // `tab_url()` and `identity_manager()`. Only returns a value for Workspace
  // sites.
  std::string GetContentAreaAccountEmail() const;
};

// Simple implementation of `ContentAnalysisInfo` meant to obtain the value
// returned by `GetContentAreaAccountEmail`. This class shouldn't expose
// non-static methods unless every other override required to support it are
// implemented.
class ContentAreaUserProvider : public ContentAnalysisInfo {
 public:
  static std::string GetUser(Profile* profile,
                             content::WebContents* web_contents,
                             const GURL& tab_url);
  static std::string GetUser(const content::ClipboardEndpoint& endpoint);

  ~ContentAreaUserProvider();

 private:
  const AnalysisSettings& settings() const override;
  signin::IdentityManager* identity_manager() const override;
  int user_action_requests_count() const override;
  std::string tab_title() const override;
  std::string user_action_id() const override;
  std::string email() const override;
  const GURL& url() const override;
  const GURL& tab_url() const override;
  ContentAnalysisRequest::Reason reason() const override;
  google::protobuf::RepeatedPtrField<::safe_browsing::ReferrerChainEntry>
  referrer_chain() const override;
  google::protobuf::RepeatedPtrField<std::string> frame_url_chain()
      const override;
  content::WebContents* web_contents() const override;

  explicit ContentAreaUserProvider(
      signin::IdentityManager* im,
      content::WebContents* web_contents,
      const GURL& tab_url);

  raw_ptr<signin::IdentityManager> im_;
  base::WeakPtr<content::WebContents> web_contents_;
  raw_ref<const GURL> tab_url_;
  google::protobuf::RepeatedPtrField<::safe_browsing::ReferrerChainEntry>
      referrer_chain_;
};

// Download-specific implementation of `ContentAnalysisInfo`. This is meant to
// be used only for reporting, so only public fields should be called.
class DownloadContentAreaUserProvider : public ContentAnalysisInfo {
 public:
  explicit DownloadContentAreaUserProvider(
      download::DownloadItem& download_item);
  ~DownloadContentAreaUserProvider();

  // ContentAnalysisInfo:
  const GURL& url() const override;
  const GURL& tab_url() const override;
  signin::IdentityManager* identity_manager() const override;
  content::WebContents* web_contents() const override;
  google::protobuf::RepeatedPtrField<::safe_browsing::ReferrerChainEntry>
  referrer_chain() const override;
  google::protobuf::RepeatedPtrField<std::string> frame_url_chain()
      const override;

 private:
  // ContentAnalysisInfo:
  const AnalysisSettings& settings() const override;
  int user_action_requests_count() const override;
  std::string tab_title() const override;
  std::string user_action_id() const override;
  std::string email() const override;
  ContentAnalysisRequest::Reason reason() const override;

  GURL url_;
  GURL tab_url_;
  raw_ptr<signin::IdentityManager> im_;
  google::protobuf::RepeatedPtrField<::safe_browsing::ReferrerChainEntry>
      referrer_chain_;
  google::protobuf::RepeatedPtrField<std::string> frame_url_chain_;
  base::WeakPtr<content::WebContents> web_contents_;
};

}  // namespace enterprise_connectors

#endif  // CHROME_BROWSER_ENTERPRISE_CONNECTORS_ANALYSIS_CONTENT_ANALYSIS_INFO_H_