/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#ifndef CRYINCLUDE_EDITOR_UNDODROPDOWN_H
#define CRYINCLUDE_EDITOR_UNDODROPDOWN_H

#pragma once
// UndoDropDown.h : header file
//

#include <QDialog>
#include <QItemSelectionModel>
#include "Undo/IUndoManagerListener.h"

class QAction;
class QListView;
class UndoDropDownListModel;
class QAbstractItemView;
class QContextMenuEvent;

enum class UndoRedoDirection
{
    Undo,
    Redo
};

/// Turns IUndoManagerListener callbacks into signals
class UndoStackStateAdapter
    : public QObject
    , IUndoManagerListener
{
    Q_OBJECT

public:
    explicit UndoStackStateAdapter(QObject* parent = nullptr);

    ~UndoStackStateAdapter();

    void SignalNumUndoRedo(const unsigned int& numUndo, const unsigned int& numRedo)
    {
        emit UndoAvailable(numUndo);
        emit RedoAvailable(numRedo);
    }

signals:
    void UndoAvailable(quint32 count);
    void RedoAvailable(quint32 count);
};

/// Enforces contiguous selections from the top element in the list view to any that is selected below
class UndoStackItemSelectionModel
    : public QItemSelectionModel
{
    Q_OBJECT
public:
    UndoStackItemSelectionModel(QAbstractItemView* view, QAbstractItemModel* model)
        : QItemSelectionModel(model)
        , m_view(view)
    {}

    virtual ~UndoStackItemSelectionModel() {}

public slots:
    void select(const QModelIndex& index, QItemSelectionModel::SelectionFlags command) override;
    void select(const QItemSelection& selection, QItemSelectionModel::SelectionFlags command) override;

private:
    QAbstractItemView* m_view;
};

/////////////////////////////////////////////////////////////////////////////
// CUndoDropDown dialog

class CUndoDropDown
    : public QDialog
{
    Q_OBJECT

    // Construction
public:
    CUndoDropDown(UndoRedoDirection direction, QWidget* pParent = nullptr);   // standard constructor

public slots:
    // Prepare to be shown in the popup/dropdown
    void Prepare();

    // Implementation
protected:
    void OnUndoButton();
    void OnUndoClear();
    void contextMenuEvent(QContextMenuEvent*) override;

    UndoRedoDirection m_direction;
    UndoDropDownListModel* m_model;
    QListView* m_view;
    QPushButton* m_doButton;

private slots:
    void SelectionChanged(const QItemSelection& selected, const QItemSelection& deselected);
};

#endif // CRYINCLUDE_EDITOR_UNDODROPDOWN_H