/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#include <AssetSeedUtil.h>

#include "AssetValidationTestShared.h"
#include <AzFramework/Platform/PlatformDefaults.h>
#include <AzFramework/Asset/AssetSeedList.h>
#include <AzFramework/Gem/GemInfo.h>

// Needs SPEC-2324
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER 
bool AssetValidationTest::CreateDummyFile(const char* path, const char* seedFileName, AZStd::string_view contents, AZStd::string& subfolderPath) const
{
    subfolderPath = (m_tempDir / path / seedFileName).concat(".").concat(AssetValidation::AssetSeed::SeedFileExtension).string().c_str();

    AZ::IO::FileIOStream fileStream(subfolderPath.c_str(), AZ::IO::OpenMode::ModeAppend | AZ::IO::OpenMode::ModeUpdate | AZ::IO::OpenMode::ModeCreatePath);
    
    return fileStream.Write(contents.size(), contents.data()) != 0;
}

TEST_F(AssetValidationTest, DefaultSeedList_ReturnsExpectedSeedLists)
{
    AZStd::vector<AzFramework::GemInfo> gemInfo;

    AZ::IO::Path gemSeedList, engineSeedList, projectSeedList;

    ASSERT_TRUE(CreateDummyFile((AZ::IO::Path("mockGem") / AzFramework::GemInfo::GetGemAssetFolder()).c_str(), "seedList", "Mock Gem Seed List", gemSeedList.Native()));
    ASSERT_TRUE(CreateDummyFile((AZ::IO::Path("Assets") / "Engine").c_str(), "SeedAssetList", "Engine Seed List", engineSeedList.Native()));

    AZ::SettingsRegistryInterface::FixedValueString projectName = AZ::Utils::GetProjectName();
    ASSERT_FALSE(projectName.empty());
    ASSERT_TRUE(CreateDummyFile(projectName.c_str(), "SeedAssetList", "Project Seed List", projectSeedList.Native()));

    AzFramework::GemInfo mockGem("MockGem");
    mockGem.m_absoluteSourcePaths.push_back((m_tempDir / "mockGem").string().c_str());
    gemInfo.push_back(mockGem);

    AZStd::vector<AZStd::string> defaultSeedStringList = AssetValidation::AssetSeed::GetDefaultSeedListFiles(gemInfo, AzFramework::PlatformFlags::Platform_PC);
    AZStd::vector<AZ::IO::Path> defaultSeedLists{ AZStd::make_move_iterator(defaultSeedStringList.begin()), AZStd::make_move_iterator(defaultSeedStringList.end()) };

    ASSERT_THAT(defaultSeedLists, ::testing::UnorderedElementsAre(gemSeedList, engineSeedList, projectSeedList));
}
#endif // AZ_TRAIT_USE_POSIX_TEMP_FOLDER

TEST_F(AssetValidationTest, SeedListDependencyTest_AddAndRemoveList_Success)
{
    MockValidationComponent testComponent;

    AzFramework::AssetSeedList seedList;
    const char DummyAssetFile[] = "Dummy";
    seedList.push_back({ testComponent.m_assetIds[6], AzFramework::PlatformFlags::Platform_PC, DummyAssetFile });

    testComponent.TestAddSeedsFor(seedList, 2);
    testComponent.SeedMode();

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);

    testComponent.TestRemoveSeedsFor(seedList, 0);

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);

    testComponent.TestRemoveSeedsFor(seedList, 2);

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);
}


TEST_F(AssetValidationTest, SeedAssetDependencyTest_AddSingleAsset_DependenciesFound)
{
    MockValidationComponent testComponent;
    testComponent.AddSeedAssetId(testComponent.m_assetIds[8], 0);
    testComponent.SeedMode();

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath7"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);
}

TEST_F(AssetValidationTest, SeedListDependencyTest_AddAndRemoveListPaths_Success)
{
    MockValidationComponent testComponent;

    const char seedListPathName[] = "ValidPath";
    const char invalidSeedListPathName[] = "InvalidPath";

    AzFramework::AssetSeedList seedList;
    const char DummyAssetFile[] = "Dummy";
    seedList.push_back({ testComponent.m_assetIds[6], AzFramework::PlatformFlags::Platform_PC, DummyAssetFile });

    testComponent.m_validSeedList = seedList;
    testComponent.m_validSeedPath = seedListPathName;

    testComponent.TestAddSeedList(invalidSeedListPathName);
    testComponent.SeedMode();

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);

    testComponent.TestAddSeedList(seedListPathName);

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);

    testComponent.TestRemoveSeedList(invalidSeedListPathName);

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), true);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);

    testComponent.TestRemoveSeedList(seedListPathName);

    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath5"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath6"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath8"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath9"), false);
    EXPECT_EQ(testComponent.IsKnownAsset("AssetPath10"), false);
}


AZ_UNIT_TEST_HOOK(DEFAULT_UNIT_TEST_ENV);