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

#ifndef CONTENT_BROWSER_ANDROID_SELECTION_SELECTION_POPUP_CONTROLLER_H_
#define CONTENT_BROWSER_ANDROID_SELECTION_SELECTION_POPUP_CONTROLLER_H_

#include <jni.h>

#include <memory>

#include "base/android/jni_weak_ref.h"
#include "base/memory/raw_ptr.h"
#include "content/browser/android/render_widget_host_connector.h"
#include "content/public/browser/android/selection_popup_delegate.h"
#include "third_party/blink/public/mojom/input/input_handler.mojom-forward.h"
#include "ui/gfx/geometry/rect_f.h"
#include "ui/touch_selection/selection_event_type.h"

namespace gfx {
class PointF;
}

namespace ui {
class MenuModel;
class MenuModelBridge;
}

namespace content {

class RenderWidgetHostViewAndroid;
struct ContextMenuParams;
class SelectionPopupDelegate;

class SelectionPopupController : public RenderWidgetHostConnector {
 public:
  static SelectionPopupController* FromWebContents(WebContents& web_contents);

  SelectionPopupController(JNIEnv* env,
                           const base::android::JavaParamRef<jobject>& obj,
                           WebContents* web_contents);

  void SetTextHandlesHiddenForDropdownMenu(JNIEnv* env, jboolean hidden);

  void SetTextHandlesTemporarilyHidden(JNIEnv* env, jboolean hidden);

  base::android::ScopedJavaLocalRef<jobjectArray> GetTouchHandleRects(
      JNIEnv* env);

  void SetDelegate(
      std::unique_ptr<SelectionPopupDelegate> selection_popup_delegate) {
    selection_popup_delegate_ = std::move(selection_popup_delegate);
  }

  // RendetWidgetHostConnector implementation.
  void UpdateRenderProcessConnection(
      RenderWidgetHostViewAndroid* old_rwhva,
      RenderWidgetHostViewAndroid* new_rhwva) override;

  // Called from native -> java
  void OnSelectionEvent(ui::SelectionEventType event,
                        const gfx::RectF& selection_rect);
  void OnDragUpdate(const ui::TouchSelectionDraggable::Type type,
                    const gfx::PointF& position);
  void OnSelectionChanged(const std::string& text);
  bool ShowSelectionMenu(RenderFrameHost* render_frame_host,
                         const ContextMenuParams& params,
                         int handle_height);
  void OnSelectAroundCaretAck(int startOffset,
                              int endOffset,
                              int surroundingTextLength,
                              blink::mojom::SelectAroundCaretResultPtr result);
  void HidePopupsAndPreserveSelection();
  void RestoreSelectionPopupsIfNecessary();
  void ChildLocalSurfaceIdChanged();
  std::unique_ptr<ui::TouchHandleDrawable> CreateTouchHandleDrawable(
      gfx::NativeView parent_native_view,
      cc::slim::Layer* parent_layer);
  void MoveRangeSelectionExtent(const gfx::PointF& extent);

  void SelectBetweenCoordinates(const gfx::PointF& base,
                                const gfx::PointF& extent);

 private:
  ~SelectionPopupController() override;
  base::android::ScopedJavaLocalRef<jobject> GetContext() const;
  raw_ptr<RenderWidgetHostViewAndroid> rwhva_ = nullptr;
  std::unique_ptr<SelectionPopupDelegate> selection_popup_delegate_;
  // Retained to keep the model in scope until the menu is dismissed.
  std::unique_ptr<ui::MenuModelBridge> menu_model_bridge_;
  std::unique_ptr<ui::MenuModel> extra_items_menu_model_;

  JavaObjectWeakGlobalRef java_obj_;
};

}  // namespace content

#endif  // CONTENT_BROWSER_ANDROID_SELECTION_SELECTION_POPUP_CONTROLLER_H_