chrome/browser/autofill/credit_card_accessory_controller_impl.h

// Copyright 2019 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_AUTOFILL_CREDIT_CARD_ACCESSORY_CONTROLLER_IMPL_H_
#define CHROME_BROWSER_AUTOFILL_CREDIT_CARD_ACCESSORY_CONTROLLER_IMPL_H_

#include "base/memory/raw_ptr.h"
#include "chrome/browser/autofill/credit_card_accessory_controller.h"

#include "base/memory/weak_ptr.h"
#include "components/autofill/core/browser/personal_data_manager.h"
#include "content/public/browser/web_contents_user_data.h"

class ManualFillingController;

namespace autofill {

class AutofillManager;

// Use either CreditCardAccessoryController::GetOrCreate or
// CreditCardAccessoryController::GetIfExisting to obtain instances of this
// class.
class CreditCardAccessoryControllerImpl
    : public CreditCardAccessoryController,
      public content::WebContentsUserData<CreditCardAccessoryControllerImpl> {
 public:
  ~CreditCardAccessoryControllerImpl() override;

  // AccessoryController:
  void RegisterFillingSourceObserver(FillingSourceObserver observer) override;
  absl::optional<AccessorySheetData> GetSheetData() const override;
  void OnFillingTriggered(FieldGlobalId focused_field_id,
                          const AccessorySheetField& selection) override;
  void OnOptionSelected(AccessoryAction selected_action) override;
  void OnToggleChanged(AccessoryAction toggled_action, bool enabled) override;

  // CreditCardAccessoryController:
  void RefreshSuggestions() override;

  // PersonalDataManagerObserver:
  void OnPersonalDataChanged() override;

  // CreditCardAccessManager::Accessor:
  void OnCreditCardFetched(CreditCardFetchResult result,
                           const CreditCard* credit_card,
                           const std::u16string& cvc) override;

  static void CreateForWebContentsForTesting(
      content::WebContents* web_contents,
      base::WeakPtr<ManualFillingController> mf_controller,
      PersonalDataManager* personal_data_manager,
      BrowserAutofillManager* af_manager,
      AutofillDriver* af_driver);

 private:
  friend class content::WebContentsUserData<CreditCardAccessoryControllerImpl>;

  using CardOrVirtualCard =
      absl::variant<const CreditCard*, std::unique_ptr<CreditCard>>;

  // Required for construction via |CreateForWebContents|:
  explicit CreditCardAccessoryControllerImpl(content::WebContents* contents);

  // Used by CreateForWebContentsForTesting:
  CreditCardAccessoryControllerImpl(
      content::WebContents* web_contents,
      base::WeakPtr<ManualFillingController> mf_controller,
      PersonalDataManager* personal_data_manager,
      BrowserAutofillManager* af_manager,
      AutofillDriver* af_driver);

  // Queries the `personal_data_manager_` for regular and virtual credit cards.
  // Virtual cards are (re-)created based on the enrollment status of the cards
  // and only exist temporarily, so in addition to `CreditCard` pointers, the
  // returned array can contain `unique_ptr<CreditCard>`s for virtual cards.
  // Recreation works only because CreditCard::CreateVirtualCard is a constant
  // projection for a card (based only on its GUID and a static suffix).
  std::vector<CardOrVirtualCard> GetAllCreditCards() const;

  // Cards that are already unmasked by the user. These are shown to the user in
  // plaintext and won't require any authentication when filling is triggered.
  std::vector<const CachedServerCardInfo*> GetUnmaskedCreditCards() const;

  // Gets promo code offers from personal data manager.
  std::vector<const AutofillOfferData*> GetPromoCodeOffers() const;

  base::WeakPtr<ManualFillingController> GetManualFillingController();
  AutofillDriver* GetDriver();
  AutofillManager* GetManager() const;

  content::WebContents& GetWebContents() const;

  base::WeakPtr<ManualFillingController> mf_controller_;
  const raw_ptr<PersonalDataManager> personal_data_manager_;
  raw_ptr<BrowserAutofillManager> af_manager_for_testing_ = nullptr;
  raw_ptr<AutofillDriver> af_driver_for_testing_ = nullptr;

  // The observer to notify if available suggestions change.
  FillingSourceObserver source_observer_;

  // OnFillingTriggered() sets this so that OnCreditCardFetched() can assert
  // that the focused frame has not changed and knows the field to be filled.
  FieldGlobalId last_focused_field_id_;

  WEB_CONTENTS_USER_DATA_KEY_DECL();
};

}  // namespace autofill

#endif  // CHROME_BROWSER_AUTOFILL_CREDIT_CARD_ACCESSORY_CONTROLLER_IMPL_H_