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

// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com

#ifndef XFA_FWL_CFWL_WIDGET_H_
#define XFA_FWL_CFWL_WIDGET_H_

#include <stdint.h>

#include "core/fxcrt/fx_coordinates.h"
#include "core/fxcrt/mask.h"
#include "core/fxcrt/unowned_ptr.h"
#include "core/fxcrt/widestring.h"
#include "core/fxge/dib/fx_dib.h"
#include "fxjs/gc/heap.h"
#include "v8/include/cppgc/garbage-collected.h"
#include "v8/include/cppgc/macros.h"
#include "v8/include/cppgc/member.h"
#include "v8/include/cppgc/prefinalizer.h"
#include "xfa/fde/cfde_data.h"
#include "xfa/fwl/cfwl_themepart.h"
#include "xfa/fwl/cfwl_widgetmgr.h"
#include "xfa/fwl/fwl_widgethit.h"
#include "xfa/fwl/ifwl_widgetdelegate.h"

namespace pdfium {

class CFWL_App;
class CFWL_Event;
class CFWL_Widget;
class IFWL_ThemeProvider;

enum class WidgetStyle : uint32_t {
  kOverLapper = 0,
  kPopup = 1L << 0,
  kChild = 2L << 0,
  kBorder = 1L << 2,
  kVScroll = 1L << 11,
  kGroup = 1L << 22,
  kNoBackground = 1L << 28,
};

inline constexpr Mask<WidgetStyle> kWidgetStyleWindowTypeMask = {
    WidgetStyle::kPopup, WidgetStyle::kChild};

// Widget state flags. Per-widget state values (kHovered/kPressed and the
// checkbox-specific Check* bits, which are also shared by some buttons) are
// folded into this enum because they all live in CFWL_Widget::Properties
// `states_`.
enum class WidgetState : uint32_t {
  // Used by CFWL_Caret.
  kCaretHighlight = 1L << 0,
  kDisabled = 1L << 2,
  kFocused = 1L << 4,
  kInvisible = 1L << 5,
  kHovered = 1L << 6,
  kPressed = 1L << 7,
  kCheckboxChecked = 1L << 8,
  kCheckboxNeutral = 2L << 8,
  kCheckboxCheckMask = 3L << 8,
  // Aliases the same bit as kCheckboxChecked; widgets only use one role.
  kPushbuttonDefault = 1L << 8,
};

enum class FWL_Type {
  Unknown = 0,

  Barcode,
  Caret,
  CheckBox,
  ComboBox,
  DateTimePicker,
  Edit,
  Form,
  FormProxy,
  ListBox,
  MonthCalendar,
  PictureBox,
  PushButton,
  ScrollBar,
  SpinButton,
  ToolTip
};

// NOTE: CFWL_Widget serves as its own delegate until replaced at runtime.
class CFWL_Widget : public cppgc::GarbageCollected<CFWL_Widget>,
                    public IFWL_WidgetDelegate {
  CPPGC_USING_PRE_FINALIZER(CFWL_Widget, PreFinalize);

 public:
  class AdapterIface : public cppgc::GarbageCollectedMixin {
   public:
    virtual ~AdapterIface() = default;
    virtual CFX_Matrix GetRotateMatrix() = 0;
    virtual void DisplayCaret(bool bVisible, const CFX_RectF* pRtAnchor) = 0;
    virtual void GetBorderColorAndThickness(FX_ARGB* cr, float* fWidth) = 0;
  };

  class Properties {
   public:
    Mask<WidgetStyle> styles_ = WidgetStyle::kChild;
    uint32_t style_exts_ = 0;  // Mask of FWL_STYLEEXT_*_*.
    Mask<WidgetState> states_;
  };

  class ScopedUpdateLock {
    CPPGC_STACK_ALLOCATED();  // Allow raw/unowned pointers.

   public:
    explicit ScopedUpdateLock(CFWL_Widget* widget);
    ~ScopedUpdateLock();

   private:
    UnownedPtr<CFWL_Widget> const widget_;  // Ok, stack-only.
  };

  CONSTRUCT_VIA_MAKE_GARBAGE_COLLECTED;
  ~CFWL_Widget() override;

  virtual void PreFinalize();
  void Trace(cppgc::Visitor* visitor) const override;

  virtual FWL_Type GetClassID() const = 0;
  virtual bool IsForm() const;
  virtual CFX_RectF GetAutosizedWidgetRect();
  virtual CFX_RectF GetWidgetRect();
  virtual CFX_RectF GetClientRect();
  virtual void ModifyStyleExts(uint32_t dwStyleExtsAdded,
                               uint32_t dwStyleExtsRemoved);
  virtual void SetStates(Mask<WidgetState> states);
  virtual void ClearStates(Mask<WidgetState> states);
  virtual void Update() = 0;
  virtual FWL_WidgetHit HitTest(const CFX_PointF& point);
  virtual void DrawWidget(CFGAS_GEGraphics* pGraphics,
                          const CFX_Matrix& matrix) = 0;

  // IFWL_WidgetDelegate:
  void OnProcessMessage(CFWL_Message* pMessage) override;
  void OnProcessEvent(CFWL_Event* pEvent) override;

  void InflateWidgetRect(CFX_RectF& rect);
  void SetWidgetRect(const CFX_RectF& rect);

  bool IsVisible() const;
  bool IsOverLapper() const;
  bool IsPopup() const;
  bool IsChild() const;

  CFWL_WidgetMgr* GetWidgetMgr() const { return widget_mgr_; }
  CFWL_Widget* GetOuter() const { return outer_; }
  CFWL_Widget* GetOutmost() const;

  void SetStyles(Mask<WidgetStyle> styles);
  uint32_t GetStyleExts() const { return properties_.style_exts_; }
  Mask<WidgetState> GetStates() const { return properties_.states_; }

  CFX_PointF TransformTo(CFWL_Widget* pWidget, const CFX_PointF& point);
  CFX_Matrix GetMatrix() const;
  IFWL_ThemeProvider* GetThemeProvider() const;
  void SetDelegate(IFWL_WidgetDelegate* delegate) { delegate_ = delegate; }
  IFWL_WidgetDelegate* GetDelegate() {
    return delegate_ ? delegate_.Get() : this;
  }
  const IFWL_WidgetDelegate* GetDelegate() const {
    return delegate_ ? delegate_.Get() : this;
  }

  CFWL_App* GetFWLApp() const { return fwlapp_; }
  uint64_t GetEventKey() const { return event_key_; }
  void SetEventKey(uint64_t key) { event_key_ = key; }

  AdapterIface* GetAdapterIface() const { return adapter_iface_; }
  void SetAdapterIface(AdapterIface* pItem) { adapter_iface_ = pItem; }
  void RepaintRect(const CFX_RectF& pRect);

 protected:
  CFWL_Widget(CFWL_App* app, const Properties& properties, CFWL_Widget* pOuter);

  bool IsEnabled() const;
  bool IsLocked() const { return lock_ > 0; }
  bool HasBorder() const;
  CFX_RectF GetEdgeRect() const;
  float GetCXBorderSize() const;
  float GetCYBorderSize() const;
  CFX_RectF GetRelativeRect() const;
  CFX_SizeF CalcTextSize(const WideString& wsText, bool bMultiLine);
  void CalcTextRect(const WideString& wsText,
                    const FDE_TextStyle& dwTTOStyles,
                    FDE_TextAlignment iTTOAlign,
                    CFX_RectF* pRect);
  void SetGrab(bool bSet);
  void UnregisterEventTarget();
  void DispatchEvent(CFWL_Event* pEvent);
  void DrawBorder(CFGAS_GEGraphics* pGraphics,
                  CFWL_ThemePart::Part iPartBorder,
                  const CFX_Matrix& pMatrix);

  Properties properties_;
  CFX_RectF widget_rect_;

 private:
  void LockUpdate() { lock_++; }
  void UnlockUpdate() {
    if (IsLocked()) {
      lock_--;
    }
  }

  CFWL_Widget* GetParent() const { return widget_mgr_->GetParentWidget(this); }
  CFX_SizeF GetOffsetFromParent(CFWL_Widget* pParent);
  void DrawBackground(CFGAS_GEGraphics* pGraphics,
                      CFWL_ThemePart::Part iPartBk,
                      const CFX_Matrix& mtMatrix);
  void NotifyDriver();
  bool IsParent(CFWL_Widget* pParent);

  int32_t lock_ = 0;
  uint64_t event_key_ = 0;
  cppgc::Member<AdapterIface> adapter_iface_;
  cppgc::Member<CFWL_App> const fwlapp_;
  cppgc::Member<CFWL_WidgetMgr> const widget_mgr_;
  cppgc::Member<IFWL_WidgetDelegate> delegate_;
  cppgc::Member<CFWL_Widget> const outer_;
};

}  // namespace pdfium

// TODO(crbug.com/42271761): Remove.
using pdfium::CFWL_Widget;
using pdfium::FWL_Type;

#endif  // XFA_FWL_CFWL_WIDGET_H_