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

#ifndef COMPONENTS_AUTOFILL_CONTENT_RENDERER_AUTOFILL_AGENT_TEST_API_H_
#define COMPONENTS_AUTOFILL_CONTENT_RENDERER_AUTOFILL_AGENT_TEST_API_H_

#include "base/memory/raw_ref.h"
#include "base/types/optional_ref.h"
#include "components/autofill/content/renderer/autofill_agent.h"
#include "components/autofill/content/renderer/javascript_autofill_tracker.h"
#include "components/autofill/content/renderer/password_autofill_agent.h"

namespace autofill {

class AutofillAgentTestApi {
 public:
  explicit AutofillAgentTestApi(AutofillAgent* agent) : agent_(*agent) {}

  bool is_dom_content_loaded() const { return agent_->is_dom_content_loaded_; }

  FormSubmissionTracker& form_tracker() { return *agent_->form_tracker_; }
  void set_form_tracker(std::unique_ptr<FormSubmissionTracker> form_tracker) {
    agent_->form_tracker_ = std::move(form_tracker);
  }

  void FocusedElementChanged(blink::WebElement new_focused_element) {
    agent_->FocusedElementChanged(new_focused_element);
  }

  void ShowSuggestions(
      const blink::WebFormControlElement& element,
      AutofillSuggestionTriggerSource trigger_source,
      const SynchronousFormCache& form_cache,
      std::optional<PasswordSuggestionRequest> password_request) {
    agent_->ShowSuggestions(element, trigger_source, form_cache,
                            password_request);
  }

  void ShowSuggestionsForContentEditable(
      const blink::WebElement& element,
      AutofillSuggestionTriggerSource trigger_source) {
    agent_->ShowSuggestionsForContentEditable(element, trigger_source);
  }

  void ContentEditableDidChange(const blink::WebElement& element) {
    agent_->ContentEditableDidChange(element);
  }

  void TextFieldValueChanged(const blink::WebFormControlElement& element) {
    agent_->TextFieldValueChanged(element);
  }

  void SelectFieldOptionsChanged(const blink::WebFormControlElement& element) {
    agent_->SelectFieldOptionsChanged(element);
  }

  void DidChangeScrollOffset() { agent_->DidChangeScrollOffset(); }

  bool ShouldThrottleAskForValuesToFill(
      FieldRendererId field,
      AutofillSuggestionTriggerSource trigger_source) {
    return agent_->ShouldThrottleAskForValuesToFill(field, trigger_source);
  }

  void set_focus_requires_scroll(bool focus_requires_scroll) {
    const_cast<AutofillAgent::Config&>(agent_->config_).focus_requires_scroll =
        AutofillAgent::FocusRequiresScroll(focus_requires_scroll);
  }

  const FormCache& form_cache() { return agent_->form_cache_; }

  PasswordAutofillAgent* password_autofill_agent() {
    return agent_->password_autofill_agent_.get();
  }

  const base::OneShotTimer& process_forms_after_dynamic_change_timer() {
    return agent_->process_forms_after_dynamic_change_timer_;
  }

  EmailVerificationHandler& email_verification_handler() {
    return agent_->email_verification_handler_;
  }

  JavaScriptAutofillTracker& javascript_autofill_tracker() {
    return agent_->javascript_autofill_tracker_;
  }

  base::WeakPtr<AutofillAgent> GetWeakPtr() {
    return agent_->weak_ptr_factory_.GetWeakPtr();
  }

 private:
  const raw_ref<AutofillAgent> agent_;
};

inline AutofillAgentTestApi test_api(AutofillAgent& agent) {
  return AutofillAgentTestApi(&agent);
}

}  // namespace autofill

#endif  // COMPONENTS_AUTOFILL_CONTENT_RENDERER_AUTOFILL_AGENT_TEST_API_H_