/*
* 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 <AzCore/std/smart_ptr/shared_ptr.h>
#include <AzCore/std/containers/vector.h>
#include <AzCore/Asset/AssetCommon.h>
#include <AzToolsFramework/AssetBrowser/Entries/RootAssetBrowserEntry.h>

namespace AzToolsFramework
{
    namespace AssetDatabase
    {
        class AssetDatabaseConnection;
        class FileDatabaseEntry;
        class ScanFolderDatabaseEntry;
        class SourceDatabaseEntry;
        class CombinedDatabaseEntry;
        class ProductDatabaseEntry;
    }

    namespace AssetBrowser
    {
        class AssetEntryChange;

        class AssetEntryChangeset
        {
        public:
            AssetEntryChangeset(
                AZStd::shared_ptr<AssetDatabase::AssetDatabaseConnection> databaseConnection,
                AZStd::shared_ptr<RootAssetBrowserEntry> rootEntry);
            ~AssetEntryChangeset();

            void PopulateEntries();
            void Update();
            void Synchronize();

            void AddFile(const AZ::s64& fileId);
            void RemoveFile(const AZ::s64& fileId);
            void AddSource(const AZ::Uuid& sourceUuid);
            void RemoveSource(const AZ::Uuid& sourceUuid);
            void AddProduct(const AZ::Data::AssetId& assetId);
            void RemoveProduct(const AZ::Data::AssetId& assetId);

        private:
            AZStd::shared_ptr<AssetDatabase::AssetDatabaseConnection> m_databaseConnection;

            AZStd::shared_ptr<RootAssetBrowserEntry> m_rootEntry;

            //! protects read/write to m_entries
            AZStd::mutex m_mutex;

            AZStd::atomic_bool m_fullUpdate;
            bool m_updated;

            AZStd::vector<AssetEntryChange*> m_changes;

            AZStd::vector<AZ::s64> m_fileIdsToAdd;
            AZStd::vector<AZ::Uuid> m_sourceUuidsToAdd;
            AZStd::vector<AZ::Data::AssetId> m_productAssetIdsToAdd;

            AZStd::string m_relativePath;

            void QueryEntireDatabase();
            void QueryChangeset();
            
        };
    } // namespace AssetBrowser
} // namespace AzToolsFramework