// Copyright 2024 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_IOS_BROWSER_AUTOFILL_CLIENT_IOS_H_
#define COMPONENTS_AUTOFILL_IOS_BROWSER_AUTOFILL_CLIENT_IOS_H_

#import "base/memory/weak_ptr.h"
#import "components/autofill/core/browser/foundations/autofill_client.h"
#import "components/autofill/ios/browser/autofill_driver_ios_factory.h"
#import "ios/web/public/web_state.h"

namespace autofill {

// Common base class for all AutofillClients on iOS.
//
// There must be at most one instance per web::WebState. Unfortunately, the
// production-code destruction order is complicated and embedder-specific.
//
// Therefore, the contract of AutofillClientIOS is as follows:
// - web_state() must be non-null throughout the lifetime of AutofillClientIOS
//   except perhaps during destruction.
// - AutofillDriverIOSFactory::WebStateDestroyed() must be called at least once
//   and only while the AutofillClientIOS is fully functional (i.e., before any
//   of its or a derived class's members have been destroyed and before
//   web_state() has become null).
class AutofillClientIOS : public AutofillClient {
 public:
  // Returns the `AutofillClientIOS` that is associated with `web_state`, if any
  // exists.
  static AutofillClientIOS* FromWebState(web::WebState* web_state);

  // At most one `AutofillClientIOS` may exist per `web_state` at any point in
  // time. This is enforced by this constructor.
  AutofillClientIOS(web::WebState* web_state,
                    id<AutofillDriverIOSBridge> bridge);
  AutofillClientIOS(const AutofillClientIOS&) = delete;
  AutofillClientIOS& operator=(const AutofillClientIOS&) = delete;
  ~AutofillClientIOS() override;

  // Non-null throughout the lifetime of the AutofillClientIOS except perhaps
  // its destructor.
  web::WebState* web_state() const { return web_state_.get(); }

  // Intentionally final to allow it to be called during construction (in
  // particular, transitively by members of subclasses).
  AutofillDriverIOSFactory& GetAutofillDriverFactory() final;

  // AutofillClient:
  std::u16string_view GetPageTitle() const override;
  AutofillManager* GetAutofillManagerForPrimaryMainFrame() override;

  base::WeakPtr<AutofillClientIOS> AsWeakPtr();

 private:
  base::WeakPtr<web::WebState> web_state_;

  AutofillDriverIOSFactory autofill_driver_factory_;

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

}  // namespace autofill

#endif  // COMPONENTS_AUTOFILL_IOS_BROWSER_AUTOFILL_CLIENT_IOS_H_