/*
 * 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 "Utils.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
    char userTempFolder[] = "/tmp/";
#endif

    for (int i = 0; i < MaxAttempts; ++i)
    {
        auto randomFolder = AZ::Uuid::CreateRandom().ToString<AZStd::string>(false, false);
        AZStd::string testPath;
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER 
        auto path = userTempFolder / ("UnitTest-" + randomFolder).c_str();
        testPath = path.string().c_str();
#else
        AZ::StringFunc::Path::Join(userTempFolder, AZStd::string::format("UnitTest-%s",randomFolder).c_str(), testPath);
#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::LocalFileIO::GetInstance()->CreatePath(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::LocalFileIO::GetInstance()->DestroyPath(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