#ifndef MUDLET_TTREEWIDGET_H
#define MUDLET_TTREEWIDGET_H

/***************************************************************************
 *   Copyright (C) 2008-2009 by Heiko Koehn - KoehnHeiko@googlemail.com    *
 *   Copyright (C) 2014 by Ahmed Charles - acharles@outlook.com            *
 *   Copyright (C) 2022 by Stephen Lyons - slysven@virginmedia.com         *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/


#include <QPointer>
#include <QTreeWidget>

class Host;

enum class TreeType {
    None,
    Trigger,
    Alias,
    Timer,
    Script,
    Key,
    Action,
    Var
};

class TTreeWidget : public QTreeWidget
{
    Q_OBJECT

public:
    Q_DISABLE_COPY(TTreeWidget)
    explicit TTreeWidget(QWidget* pW);
    Qt::DropActions supportedDropActions() const override;
    void dragEnterEvent(QDragEnterEvent* event) override;
    void dragLeaveEvent(QDragLeaveEvent* event) override;
    void dragMoveEvent(QDragMoveEvent* event) override;
    void dropEvent(QDropEvent* event) override;
    void startDrag(Qt::DropActions supportedActions) override;
    bool dropMimeData(QTreeWidgetItem* parent, int index, const QMimeData* data, Qt::DropAction action) override;
    void rowsAboutToBeRemoved(const QModelIndex& parent, int start, int end) override;
    void rowsInserted(const QModelIndex& parent, int start, int end) override;
    void mousePressEvent(QMouseEvent* event) override;
    void mouseReleaseEvent(QMouseEvent* event) override;
    void setHost(Host* pH);
    void setTreeType(TreeType type);
    void beginInsertRows(const QModelIndex& parent, int first, int last);
    void getAllChildren(QTreeWidgetItem*, QList<QTreeWidgetItem*>&);

    void updateTriggerIcon(QTreeWidgetItem* pItem, int triggerID);
    void updateTriggerIconsRecursively(QTreeWidgetItem* pItem);
    QTreeWidgetItem* findItemByTriggerID(QTreeWidgetItem* pParent, int triggerID);

signals:
    void itemMoved(int itemID, int oldParentID, int newParentID, int oldPosition, int newPosition);
    void batchMoveStarted();
    void batchMoveEnded();

private:
    // Structure to hold information about items being moved
    struct MoveInfo {
        int childID;
        int oldParentID;
        int oldPosition;
    };

    bool mIsDropAction;
    QPointer<Host> mpHost;
    QList<MoveInfo> mPendingMoves;  // Stores info for all items being moved
    int mOldParentID;  // Deprecated: kept for compatibility, will be removed
    int mOldPosition;  // Deprecated: kept for compatibility, will be removed
    int mChildID;      // Deprecated: kept for compatibility, will be removed
    TreeType mTreeType = TreeType::None;
    // CHECK: Should this actually be a: QPersistentModelIndex ?
    QModelIndex mClickedItem;
};

#endif // MUDLET_TTREEWIDGET_H
