/*
* 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 <CloudGemAWSScriptBehaviors_precompiled.h>

#include "AWSBehaviorHTTP.h"

#include "AWSBehaviorMap.h"

/// To use a specific AWS API request you have to include each of these.
// The AWS Native SDK AWSAllocator triggers a warning due to accessing members of std::allocator directly.
// AWSAllocator.h(70): warning C4996: 'std::allocator<T>::pointer': warning STL4010: Various members of std::allocator are deprecated in C++17.
// Use std::allocator_traits instead of accessing these members directly.
// You can define _SILENCE_CXX17_OLD_ALLOCATOR_MEMBERS_DEPRECATION_WARNING or _SILENCE_ALL_CXX17_DEPRECATION_WARNINGS to acknowledge that you have received this warning.
#include <AzCore/PlatformDef.h>
AZ_PUSH_DISABLE_WARNING(4251 4355 4996, "-Wunknown-warning-option")
#include <aws/core/http/HttpClient.h>
#include <aws/core/http/HttpClientFactory.h>
AZ_POP_DISABLE_WARNING

#include <AzCore/Jobs/JobContext.h>
#include <AzCore/Jobs/JobFunction.h>
#include <AzCore/Jobs/JobManagerBus.h>
#include <AzFramework/AzFramework_Traits_Platform.h>
#include <CloudCanvasCommon/CloudCanvasCommonBus.h>


AZ_PUSH_DISABLE_WARNING(4251 4355 4996, "-Wunknown-warning-option")
#include <aws/lambda/LambdaClient.h>
#include <aws/lambda/model/ListFunctionsRequest.h>
#include <aws/lambda/model/ListFunctionsResult.h>
#include <aws/lambda/model/InvokeRequest.h>
#include <aws/lambda/model/InvokeResult.h>
#include <aws/core/utils/Outcome.h>
#include <aws/core/utils/memory/stl/AWSStringStream.h>
AZ_POP_DISABLE_WARNING

namespace CloudGemAWSScriptBehaviors
{
    AWSBehaviorHTTP::AWSBehaviorHTTP() :
        m_url()
    {

    }

    void AWSBehaviorHTTP::ReflectSerialization(AZ::SerializeContext* serializeContext)
    {
        if (serializeContext)
        {
            serializeContext->Class<AWSBehaviorHTTP>()
                ->Field("URL", &AWSBehaviorHTTP::m_url)
                ->Version(1);
        }
    }

    void AWSBehaviorHTTP::ReflectBehaviors(AZ::BehaviorContext* behaviorContext)
    {
        behaviorContext->Class<AWSBehaviorHTTP>("AWSBehaviorHTTP")
            ->Method("Get", &AWSBehaviorHTTP::Get, nullptr, "HTTP GET operation on AWS")
            ->Property("URL", nullptr, BehaviorValueSetter(&AWSBehaviorHTTP::m_url));

        behaviorContext->EBus<AWSBehaviorHTTPNotificationsBus>("AWSBehaviorHTTPNotificationsBus")
            ->Handler<AWSBehaviorHTTPNotificationsBusHandler>();
    }

    void AWSBehaviorHTTP::ReflectEditParameters(AZ::EditContext* editContext)
    {
        editContext->Class<AWSBehaviorHTTP>("AWSBehaviorHTTP", "Wraps AWS HTTP functionality")
            ->DataElement(AZ::Edit::UIHandlers::Default, &AWSBehaviorHTTP::m_url, "URL", "URL to get");
    }

    void AWSBehaviorHTTP::Get()
    {
        if (m_url.empty())
        {
            EBUS_EVENT(AWSBehaviorHTTPNotificationsBus, OnError, "AWS HTTP Get: No URL provided");
            return;
        }

        AZ::Job* httpGetJob = CreateHttpGetJob(m_url);
        if (httpGetJob)
        {
            httpGetJob->Start();
        }

    }

    AZ::Job* AWSBehaviorHTTP::CreateHttpGetJob(const AZStd::string& url) const
    {
        AZ::JobContext* jobContext{ nullptr };
        EBUS_EVENT_RESULT(jobContext, CloudCanvasCommon::CloudCanvasCommonRequestBus, GetDefaultJobContext);
        AZ::Job* job{ nullptr };
        job = AZ::CreateJobFunction([url]()
        {
            auto config = Aws::Client::ClientConfiguration();
            config.enableTcpKeepAlive = AZ_TRAIT_AZFRAMEWORK_AWS_ENABLE_TCP_KEEP_ALIVE_SUPPORTED;
            AZStd::string caFile;
            CloudCanvas::RequestRootCAFileResult requestResult;
            EBUS_EVENT_RESULT(requestResult, CloudCanvasCommon::CloudCanvasCommonRequestBus, RequestRootCAFile, caFile);
            if (caFile.length())
            {
                AZ_TracePrintf("CloudCanvas", "AWSBehaviorHTTP using caFile %s with request result %d", caFile.c_str(), requestResult);
                config.caFile = caFile.c_str();                
            }

            std::shared_ptr<Aws::Http::HttpClient> httpClient = Aws::Http::CreateHttpClient(config);

            Aws::String requestURL{ url.c_str() };
            auto httpRequest(Aws::Http::CreateHttpRequest(requestURL, Aws::Http::HttpMethod::HTTP_GET, Aws::Utils::Stream::DefaultResponseStreamFactoryMethod));
            auto httpResponse(httpClient->MakeRequest(httpRequest, nullptr, nullptr));

            if (!httpResponse)
            {
                AZStd::function<void()> notifyOnMainThread = []()
                {
                    AWSBehaviorHTTPNotificationsBus::Broadcast(&AWSBehaviorHTTPNotificationsBus::Events::OnError, "No Response Received from request!  (Internal SDK Error)");
                };
                AZ::TickBus::QueueFunction(notifyOnMainThread);
                return;
            }

            int responseCode = static_cast<int>(httpResponse->GetResponseCode());

            auto headerMap = httpResponse->GetHeaders();
            StringMap stringMap;
            for (auto headerContent : headerMap)
            {
                stringMap.SetValue(headerContent.first.c_str(), headerContent.second.c_str());
            }

            AZStd::string contentType = httpResponse->GetContentType().c_str();

            AZStd::string returnString;
            auto& body = httpResponse->GetResponseBody();
            Aws::StringStream readableOut;
            readableOut << body.rdbuf();
            returnString = readableOut.str().c_str();
            
            // respond on the main thread because script context ebus behaviour handlers are not thread safe
            AZStd::function<void()> notifyOnMainThread = [responseCode, stringMap, contentType, returnString]()
            {
                AWSBehaviorHTTPNotificationsBus::Broadcast(&AWSBehaviorHTTPNotificationsBus::Events::OnSuccess, AZStd::string("Success!"));
                AWSBehaviorHTTPNotificationsBus::Broadcast(&AWSBehaviorHTTPNotificationsBus::Events::GetResponse, responseCode, stringMap, contentType, returnString);
            };
            AZ::TickBus::QueueFunction(notifyOnMainThread);
        }, true, jobContext);
        return job;
    }
}