TheaninovOS/desktops/hyprland/ags/modules/lib/actioncenter.js

// Not yet used. For cool drag and drop stuff. Thanks DevAlien

const Toggles = {}
Toggles.Wifi = NetworkToggle
Toggles.Bluetooth = BluetoothToggle
Toggles.DND = DNDToggle
Toggles.ThemeToggle = ThemeToggle
Toggles.ProfileToggle = ProfileToggle
// Toggles.Record = RecordToggle;
// Toggles.Airplane = AirplaneToggle;
// Toggles.DoNotDisturb = DoNotDisturbToggle;
const TARGET = [Gtk.TargetEntry.new("text/plain", Gtk.TargetFlags.SAME_APP, 0)]

export class ActionCenter extends Gtk.Box {
  static {
    GObject.registerClass(
      {
        GTypeName: "ActionCenter",
        Properties: {},
      },
      this,
    )
  }

  constructor({className = "ActionCenter", toggles, ...rest}) {
    super(rest)
    this.toggles = Toggles
    this.currentToggles = Settings.getSetting("toggles", [])
    this.mainFlowBox = this._setupFlowBox(className + QSView.editing && className + "Editing")
    this.mainFlowBox.connect("drag_motion", this._dragMotionMain)
    this.mainFlowBox.connect("drag_drop", this._dragDropMain)
    this._dragged = {}
    this._draggedExtra = {}

    this._dragged
    this._currentPosition = 0
    this._orderedState
    this._draggedName

    this.updateList(toggles, this.mainFlowBox)

    this.set_orientation(Gtk.Orientation.VERTICAL)
    this.add(this.mainFlowBox)
    this.mainFlowBox.set_size_request(1, 30)
    if (QSView.editing) {
      this.extraFlowBox = this._setupFlowBox(className)
      this.extraFlowBox.connect("drag_motion", this._dragMotionExtra)
      this.extraFlowBox.connect("drag_drop", this._dragDropExtra)
      this.updateList(this._getExtraToggles(), this.extraFlowBox)
      this.add(
        Box({
          vertical: true,
          children: [
            Label("Extra widgets"),
            Label("Drop here to remove or drag from here to add"),
            this.extraFlowBox,
          ],
        }),
      )
    }
  }

  _getExtraToggles() {
    let toggles = {...this.toggles}
    this.currentToggles.map(t => {
      if (toggles[t]) {
        delete toggles[t]
      }
    })
    return Object.keys(toggles)
  }

  _setupFlowBox(className) {
    const flowBox = new Gtk.FlowBox()
    flowBox.set_valign(Gtk.Align.FILL)
    flowBox.set_min_children_per_line(2)
    flowBox.set_max_children_per_line(2)
    flowBox.set_selection_mode(Gtk.SelectionMode.NONE)
    flowBox.get_style_context().add_class(className)
    flowBox.set_homogeneous(true)
    flowBox.drag_dest_set(Gtk.DestDefaults.ALL, TARGET, Gdk.DragAction.COPY)

    return flowBox
  }

  createWidget = (name, index, type) => {
    const editSetup = widget => {
      widget.drag_source_set(Gdk.ModifierType.BUTTON1_MASK, TARGET, Gdk.DragAction.COPY)

      widget.connect("drag-begin", (w, context) => {
        const widgetContainer = widget.get_parent()

        Gtk.drag_set_icon_surface(context, createSurfaceFromWidget(widgetContainer))
        this._dragged = {
          widget: widgetContainer.get_parent().get_parent(),
          container: widgetContainer,
          name: name,
          currentPosition: type === "Main" ? index : null,
          currentPositionExtra: type === "Extra" ? index : null,
          from: type,
        }
        widgetContainer.get_style_context().add_class("hidden")
        if (type !== "Main") {
          this.extraFlowBox.remove(this._dragged.widget)
        }

        return true
      })
      widget.connect("drag-failed", () => {
        this.updateList(Settings.getSetting("toggles"), this.mainFlowBox)
        this.updateList(this._getExtraToggles(), this.extraFlowBox)
      })
    }

    let row = new Gtk.FlowBoxChild({visible: true})
    row.add(Toggles[name]({setup: QSView.editing && editSetup, QSView: QSView}))
    row._index = index
    row._name = name
    return row
  }

  updateList(toggles, flowBox) {
    let type = flowBox === this.mainFlowBox ? "Main" : "Extra"
    var childrenBox = flowBox.get_children()
    childrenBox.forEach(element => {
      flowBox.remove(element)
      element.destroy()
    })

    if (!toggles) return

    toggles.forEach((name, i) => {
      if (Toggles[name]) flowBox.add(this.createWidget(name, i, type))
    })
    flowBox.show_all()
  }

  _dragMotionMain = (widget, context, x, y, time) => {
    if (this._dragged.currentPositionExtra !== null) {
      this._dragged.currentPositionExtra = null
      if (this._isChild(this.extraFlowBox, this._dragged.widget)) {
        this.extraFlowBox.remove(this._dragged.widget)
      }
    }
    const children = this.mainFlowBox.get_children()
    const sampleItem = children[0]
    const sampleWidth = sampleItem.get_allocation().width
    const sampleHeight = sampleItem.get_allocated_height()
    const perLine = Math.floor(this.mainFlowBox.get_allocation().width / sampleWidth)
    const pos = Math.floor(y / sampleHeight) * perLine + Math.floor(x / sampleWidth)
    if (pos >= children.length && pos !== 0) return false

    if (this._dragged.currentPosition === null) {
      this.mainFlowBox.insert(this._dragged.widget, pos)

      this._dragged.currentPosition = pos
    } else if (this._dragged.currentPosition !== pos) {
      if (this._isChild(this.mainFlowBox, this._dragged.widget)) {
        this.mainFlowBox.remove(this._dragged.widget)
      }

      this.mainFlowBox.insert(this._dragged.widget, pos)

      this._dragged.currentPosition = pos
    }

    return true
  }

  _dragDropMain = () => {
    if (this._dragged.from !== "Main") {
      this.currentToggles.splice(this._dragged.currentPosition, 0, this._dragged.name)
    } else {
      const indexCurrentToggle = this.currentToggles.indexOf(this._dragged.name)
      this.currentToggles.splice(indexCurrentToggle, 1)
      this.currentToggles.splice(this._dragged.currentPosition, 0, this._dragged.name)
    }

    Settings.setSetting("toggles", this.currentToggles)
    this._dragged.container.get_style_context().remove_class("hidden")
    return true
  }

  _dragDropExtra = () => {
    if (this._dragged.from === "Main") {
      const indexCurrentToggle = this.currentToggles.indexOf(this._dragged.name)
      this.currentToggles.splice(indexCurrentToggle, 1)
    }

    Settings.setSetting("toggles", this.currentToggles)
    this._dragged.container.get_style_context().remove_class("hidden")
    return true
  }

  _dragMotionExtra = (widget, context, x, y, time) => {
    if (this._dragged.currentPosition !== null) {
      this._dragged.currentPosition = null
      if (this._isChild(this.mainFlowBox, this._dragged.widget)) {
        this.mainFlowBox.remove(this._dragged.widget)
      }
    }

    const children = this.extraFlowBox.get_children()
    const sampleItem = children[0]
    let pos = 0
    if (sampleItem) {
      const sampleWidth = sampleItem.get_allocation().width
      const sampleHeight = sampleItem.get_allocated_height()
      const perLine = Math.floor(this.extraFlowBox.get_allocation().width / sampleWidth)
      pos = Math.floor(y / sampleHeight) * perLine + Math.floor(x / sampleWidth)
    }

    if (pos >= children.length && pos !== 0) return false

    if (this._dragged.currentPositionExtra === null) {
      this.extraFlowBox.insert(this._dragged.widget, pos)

      this._dragged.currentPositionExtra = pos
    }

    if (this._dragged.currentPositionExtra !== pos) {
      if (this._isChild(this.extraFlowBox, this._dragged.widget)) {
        this.extraFlowBox.remove(this._dragged.widget)
      }

      this.extraFlowBox.insert(this._dragged.widget, pos)

      this._dragged.currentPositionExtra = pos
    }

    return true
  }

  _isChild(container, widget) {
    let found = false
    container.get_children().forEach(c => {
      if (c === widget) found = true
    })
    return found
  }
}