// Copyright 2026 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_UI_WEBUI_SKILLS_SKILLS_DIALOG_HANDLER_H_
#define CHROME_BROWSER_UI_WEBUI_SKILLS_SKILLS_DIALOG_HANDLER_H_

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "chrome/browser/optimization_guide/optimization_guide_keyed_service.h"
#include "chrome/browser/ui/webui/skills/skills.mojom.h"
#include "components/optimization_guide/core/model_execution/optimization_guide_model_execution_error.h"
#include "components/skills/public/skill.h"
#include "components/skills/public/skill.mojom-forward.h"
#include "components/skills/public/skills_metrics.h"
#include "mojo/public/cpp/bindings/receiver.h"

namespace content {
class WebContents;
}  // namespace content

class Profile;

namespace skills {

class SkillsDialogDelegate;

class SkillsDialogHandler : public skills::mojom::DialogHandler {
 public:
  SkillsDialogHandler(
      mojo::PendingReceiver<skills::mojom::DialogHandler> receiver,
      content::WebContents* web_contents,
      OptimizationGuideKeyedService* optimization_guide_keyed_service,
      skills::Skill initial_skill,
      SkillsDialogEntryPoint entrypoint,
      mojom::SkillsDialogType dialog_type,
      base::WeakPtr<SkillsDialogDelegate> delegate);

  SkillsDialogHandler(const SkillsDialogHandler&) = delete;
  SkillsDialogHandler& operator=(const SkillsDialogHandler&) = delete;

  ~SkillsDialogHandler() override;

  // skills::mojom::DialogHandler:
  void SubmitSkill(
      const skills::Skill& skill,
      skills::mojom::SkillsPromptRefinementOutcome refinement_outcome,
      skills::mojom::DialogHandler::SubmitSkillCallback callback) override;
  void DeleteSkill(const std::string& skill_id) override;
  void CloseDialog() override;
  void ShowEmojiPicker() override;
  void GetInitialState(GetInitialStateCallback callback) override;
  void RefineSkill(
      const skills::Skill& skill,
      skills::mojom::DialogHandler::RefineSkillCallback callback) override;
  void GenerateNameAndEmoji(
      const skills::Skill& skill,
      skills::mojom::DialogHandler::GenerateNameAndEmojiCallback callback)
      override;
  void GetSignedInEmail(GetSignedInEmailCallback callback) override;

 protected:
  virtual const skills::Skill* SaveOrUpdateSkill(const skills::Skill& skill);

 private:
  // Callback for the model execution result for `RefineSkill`.
  void OnRefineSkillResponse(
      skills::mojom::DialogHandler::RefineSkillCallback callback,
      base::TimeTicks start_time,
      optimization_guide::OptimizationGuideModelExecutionResult result,
      std::unique_ptr<optimization_guide::ModelQualityLogEntry> log_entry);

  // Callback for the model execution result for `GenerateNameAndEmoji`.
  void OnGenerateNameAndEmojiResponse(
      skills::mojom::DialogHandler::GenerateNameAndEmojiCallback callback,
      optimization_guide::OptimizationGuideModelExecutionResult result,
      std::unique_ptr<optimization_guide::ModelQualityLogEntry> log_entry);

  mojo::Receiver<skills::mojom::DialogHandler> receiver_;
  const raw_ref<content::WebContents> web_contents_;
  raw_ptr<OptimizationGuideKeyedService> optimization_guide_keyed_service_ =
      nullptr;
  // The skill data used to pre-populate the dialog's input fields.
  Skill initial_skill_;
  // The entry point from which this dialog instance was initiated (i.e. web
  // client, management page). This is set at creation time and used for metrics
  // logging.
  SkillsDialogEntryPoint entrypoint_;
  // The type of dialog to open.
  mojom::SkillsDialogType dialog_type_;
  base::WeakPtr<SkillsDialogDelegate> delegate_;

  // Initialized with the browser_context passed in the constructor.
  const raw_ref<Profile> profile_;
  base::WeakPtrFactory<SkillsDialogHandler> weak_ptr_factory_{this};
};

}  // namespace skills

#endif  // CHROME_BROWSER_UI_WEBUI_SKILLS_SKILLS_DIALOG_HANDLER_H_