/*
* 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.
*
*/

#include "AssetDetailsPanel.h"

#include "AssetTreeFilterModel.h"
#include "MainWindow.h"
#include "ProductAssetTreeModel.h"
#include "SourceAssetTreeModel.h"

#include <QTabWidget>
#include <QTreeView>

namespace AssetProcessor
{

    AssetDetailsPanel::AssetDetailsPanel(QWidget* parent) : QFrame(parent)
    {
    }

    AssetDetailsPanel::~AssetDetailsPanel()
    {
    }

    void AssetDetailsPanel::RegisterAssociatedWidgets(
        QTreeView* sourceTreeView,
        SourceAssetTreeModel* sourceAssetTreeModel,
        AssetTreeFilterModel* sourceFilterModel,
        QTreeView* productTreeView,
        ProductAssetTreeModel* productAssetTreeModel,
        AssetTreeFilterModel* productFilterModel,
        QTabWidget* assetTab)
    {
        m_sourceTreeView = sourceTreeView;
        m_sourceTreeModel = sourceAssetTreeModel;
        m_sourceFilterModel = sourceFilterModel;
        m_productTreeView = productTreeView;
        m_productTreeModel = productAssetTreeModel;
        m_productFilterModel = productFilterModel;
        m_assetsTab = assetTab;
    }

    void AssetDetailsPanel::GoToSource(const AZStd::string& source)
    {
        if (!m_sourceTreeModel || !m_sourceTreeView || !m_assetsTab || !m_sourceFilterModel)
        {
            return;
        }
        m_assetsTab->setCurrentIndex(static_cast<int>(MainWindow::AssetTabIndex::Source));
        QModelIndex goToIndex = m_sourceTreeModel->GetIndexForSource(source);
        // Make sure this index is visible, even if a search is active.
        m_sourceFilterModel->ForceModelIndexVisible(goToIndex);
        QModelIndex filterIndex = m_sourceFilterModel->mapFromSource(goToIndex);
        // Some tables, like the source dependencies table, may have wildcards or links to files that don't exist.
        if (!filterIndex.isValid())
        {
            return;
        }
        m_sourceTreeView->scrollTo(filterIndex, QAbstractItemView::ScrollHint::EnsureVisible);
        m_sourceTreeView->selectionModel()->setCurrentIndex(filterIndex, AssetTreeModel::GetAssetTreeSelectionFlags());
    }

    void AssetDetailsPanel::GoToProduct(const AZStd::string& product)
    {
        if (!m_productTreeModel || !m_productTreeView || !m_assetsTab || !m_productFilterModel)
        {
            return;
        }
        m_assetsTab->setCurrentIndex(static_cast<int>(MainWindow::AssetTabIndex::Product));
        QModelIndex goToIndex = m_productTreeModel->GetIndexForProduct(product);
        // Make sure this index is visible, even if a search is active.
        m_productFilterModel->ForceModelIndexVisible(goToIndex);
        QModelIndex filterIndex = m_productFilterModel->mapFromSource(goToIndex);
        // Some tables may have wildcards or links to files that don't exist.
        if (!filterIndex.isValid())
        {
            return;
        }
        m_productTreeView->scrollTo(filterIndex, QAbstractItemView::ScrollHint::EnsureVisible);
        m_productTreeView->selectionModel()->setCurrentIndex(filterIndex, AssetTreeModel::GetAssetTreeSelectionFlags());
    }

}