/*
 * Copyright (c) Contributors to the Open 3D Engine Project
 * 
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#include "Utils.h"
#include <AzCore/IO/Path/Path.h>
#include <AzCore/StringFunc/StringFunc.h>

UnitTest::ScopedTemporaryDirectory::ScopedTemporaryDirectory()
{
    constexpr int MaxAttempts = 255;
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER     
    const auto userTempFolder = std::filesystem::temp_directory_path();
#else
    AZ::IO::Path userTempFolder("/tmp");
#endif

    for (int i = 0; i < MaxAttempts; ++i)
    {
        auto randomFolder = AZ::Uuid::CreateRandom().ToString<AZStd::fixed_string<512>>(false, false);
        AZ::IO::FixedMaxPath testPath;
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER 
        auto path = userTempFolder / ("UnitTest-" + randomFolder).c_str();
        testPath = path.string().c_str();
#else
        userTempFolder /= ("UnitTest-" + randomFolder).c_str(); 
        testPath = userTempFolder.c_str();
#endif
        if (!AZ::IO::SystemFile::Exists(testPath.c_str()))
        {
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER   
            m_path = path;
            m_tempDirectory = m_path.string().c_str();
#else
            m_tempDirectory = testPath;
#endif
            m_directoryExists = AZ::IO::SystemFile::CreateDir(m_tempDirectory.c_str());
            break;
        }
    }

    AZ_Error("ScopedTemporaryDirectory", !m_tempDirectory.empty(), "Failed to create unique temporary directory after attempting %d random folder names", MaxAttempts);
}

UnitTest::ScopedTemporaryDirectory::~ScopedTemporaryDirectory()
{
    if (m_directoryExists)
    {
        AZ::IO::SystemFile::DeleteDir(m_tempDirectory.c_str());
    }
}

bool UnitTest::ScopedTemporaryDirectory::IsValid() const
{
    return m_directoryExists;
}

const char* UnitTest::ScopedTemporaryDirectory::GetDirectory() const
{
    return m_tempDirectory.c_str();
}


#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER     
const std::filesystem::path& UnitTest::ScopedTemporaryDirectory::GetPath() const
{
    return m_path;
}

std::filesystem::path UnitTest::ScopedTemporaryDirectory::operator/(const std::filesystem::path& rhs) const
{
    return m_path / rhs;
}

#endif // !AZ_TRAIT_USE_POSIX_TEMP_FOLDER