// Copyright 2023 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_SUPERVISED_USER_ANDROID_SUPERVISED_USER_WEB_CONTENT_HANDLER_IMPL_H_
#define CHROME_BROWSER_SUPERVISED_USER_ANDROID_SUPERVISED_USER_WEB_CONTENT_HANDLER_IMPL_H_

#include <string>

#include "base/gtest_prod_util.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/supervised_user/chrome_supervised_user_web_content_handler_base.h"
#include "components/supervised_user/core/browser/web_content_handler.h"
#include "content/public/browser/frame_tree_node_id.h"

enum class AndroidLocalWebApprovalFlowOutcome;

// Android specific implementation of web content handler.
class SupervisedUserWebContentHandlerImpl
    : public ChromeSupervisedUserWebContentHandlerBase {
 public:
  SupervisedUserWebContentHandlerImpl(content::WebContents* web_contents,
                                      content::FrameTreeNodeId frame_id,
                                      int64_t interstitial_navigation_id);

  SupervisedUserWebContentHandlerImpl(
      const SupervisedUserWebContentHandlerImpl&) = delete;
  SupervisedUserWebContentHandlerImpl& operator=(
      const SupervisedUserWebContentHandlerImpl&) = delete;
  ~SupervisedUserWebContentHandlerImpl() override;

  // ChromeSupervisedUserWebContentHandlerBase implementation:
  void RequestLocalApproval(
      const GURL& target_url,
      supervised_user::WebFilteringResult filtering_result,
      const std::u16string& child_display_name,
      ApprovalRequestInitiatedCallback callback) override;

 private:
  void OnLocalApprovalRequestCompleted(
      supervised_user::FamilyLinkSettingsService& settings_service,
      const GURL& url,
      base::TimeTicks start_time,
      AndroidLocalWebApprovalFlowOutcome request_outcome);

  // Helpers for private method testing.
  FRIEND_TEST_ALL_PREFIXES(SupervisedUserWebContentHandlerImplTest,
                           LocalWebApprovalDurationHistogramRejectionTest);
  FRIEND_TEST_ALL_PREFIXES(SupervisedUserWebContentHandlerImplTest,
                           LocalWebApprovalDurationHistogramApprovalTest);
  FRIEND_TEST_ALL_PREFIXES(SupervisedUserWebContentHandlerImplTest,
                           LocalWebApprovalDurationHistogramCancellationTest);

  base::WeakPtrFactory<SupervisedUserWebContentHandlerImpl> weak_ptr_factory_{
      this};
};

#endif  // CHROME_BROWSER_SUPERVISED_USER_ANDROID_SUPERVISED_USER_WEB_CONTENT_HANDLER_IMPL_H_