/*
* 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.
*
*/
#pragma once

#include "AssetTreeModel.h"


namespace AssetProcessor
{
    class ProductAssetTreeItemData;

    class ProductAssetTreeModel : public AssetTreeModel
    {
    public:
        ProductAssetTreeModel(AZStd::shared_ptr<AzToolsFramework::AssetDatabase::AssetDatabaseConnection> sharedDbConnection, QObject *parent = nullptr);
        virtual ~ProductAssetTreeModel();

        // AssetDatabaseNotificationBus::Handler
        void OnProductFileChanged(const AzToolsFramework::AssetDatabase::ProductDatabaseEntry& entry) override;
        void OnProductFileRemoved(AZ::s64 productId) override;
        void OnProductFilesRemoved(const AzToolsFramework::AssetDatabase::ProductDatabaseEntryContainer& products) override;

        QModelIndex GetIndexForProduct(const AZStd::string& product);

    protected:
        void ResetModel() override;

        void AddOrUpdateEntry(
            const AzToolsFramework::AssetDatabase::ProductDatabaseEntry& product,
            bool modelIsResetting);

        void RemoveAsset(AZ::s64 productId);

        void RemoveAssetTreeItem(AssetTreeItem* assetToRemove);
        void RemoveFoldersIfEmpty(AssetTreeItem* folderToCheck);

        void CheckForUnresolvedIssues(AZStd::shared_ptr<ProductAssetTreeItemData> productItemData);

        AZStd::unordered_map<AZStd::string, AssetTreeItem*> m_productToTreeItem;
        // Mapping product Ids to asset tree items makes cleanup easier when files are deleted.
        AZStd::unordered_map<AZ::s64, AssetTreeItem*> m_productIdToTreeItem;
    };
}