// Copyright 2013 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_ACCESSIBILITY_ACCESSIBILITY_UI_H_
#define CHROME_BROWSER_UI_WEBUI_ACCESSIBILITY_ACCESSIBILITY_UI_H_

#include <memory>
#include <string>
#include <vector>

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/timer/timer.h"
#include "base/values.h"
#include "content/public/browser/web_contents_observer.h"
#include "content/public/browser/web_ui_controller.h"
#include "content/public/browser/web_ui_data_source.h"
#include "content/public/browser/web_ui_message_handler.h"
#include "content/public/browser/webui_config.h"
#include "ui/accessibility/platform/inspect/ax_api_type.h"

namespace ui {
struct AXUpdatesAndEvents;
}

namespace content {
class WebContents;
}  // namespace content

namespace user_prefs {
class PrefRegistrySyncable;
}  // namespace user_prefs

class AccessibilityUI;

class AccessibilityUIConfig
    : public content::DefaultWebUIConfig<AccessibilityUI> {
 public:
  AccessibilityUIConfig();
  ~AccessibilityUIConfig() override;
};

// Controls the accessibility web UI page.
class AccessibilityUI : public content::WebUIController {
 public:
  explicit AccessibilityUI(content::WebUI* web_ui);
  ~AccessibilityUI() override;
};

// Observes accessibility events from web contents.
class AccessibilityUIObserver : public content::WebContentsObserver {
 public:
  AccessibilityUIObserver(content::WebContents* web_contents,
                          std::vector<std::string>* event_logs);
  ~AccessibilityUIObserver() override;

  void AccessibilityEventReceived(
      const ui::AXUpdatesAndEvents& details) override;

 private:
  raw_ptr<std::vector<std::string>> event_logs_;
};

// Manages messages sent from accessibility.js via json.
class AccessibilityUIMessageHandler : public content::WebUIMessageHandler,
                                      public content::WebContentsObserver {
 public:
  AccessibilityUIMessageHandler();

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

  ~AccessibilityUIMessageHandler() override;

  void OnJavascriptAllowed() override;
  void OnJavascriptDisallowed() override;
  void RegisterMessages() override;

  static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);

 private:
  void HandleInitialize(const base::ListValue& args);
  void HandleToggleAccessibilityForWebContents(const base::ListValue& args);
  void HandleSetGlobalFlag(const base::ListValue& args);
  void HandleSetGlobalString(const base::ListValue& args);

  void GetRequestTypeAndFilters(const base::DictValue& data,
                                std::string& request_type,
                                std::string& allow,
                                std::string& allow_empty,
                                std::string& deny);
  void HandleRequestWebContentsTree(const base::ListValue& args);
  void HandleRequestNativeUITree(const base::ListValue& args);
#if defined(USE_AURA) && !BUILDFLAG(IS_CHROMEOS)
  void HandleRequestWidgetsTree(const base::ListValue& args);
#endif
  void HandleRequestAccessibilityEvents(const base::ListValue& args);
  void Callback(const std::string&);
  void StopRecording(content::WebContents* web_contents);

  // Returns the user-set API type. or the platform's default recording type if
  // the user-set type is not supported.
  ui::AXApiType::Type GetRecordingApiType();

  // content::WebContentsObserver:
  void OnVisibilityChanged(content::Visibility visibility) override;

  // Updates the UI with new data. Called periodically to keep the UI up-to-date
  // while it is visible.
  void OnUpdateDisplayTimer();

  std::vector<std::string> event_logs_;
  std::unique_ptr<AccessibilityUIObserver> observer_;

  // The last data for display sent to the UI by OnUpdateDisplayTimer.
  base::DictValue last_data_;

  // A timer that runs while the UI is visible to call OnUpdateDisplayTimer.
  base::RepeatingTimer update_display_timer_;

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

#endif  // CHROME_BROWSER_UI_WEBUI_ACCESSIBILITY_ACCESSIBILITY_UI_H_