/*
  * Copyright 2010-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License").
  * You may not use this file except in compliance with the License.
  * A copy of the License is located at
  *
  *  http://aws.amazon.com/apache2.0
  *
  * or in the "license" file accompanying this file. This file is distributed
  * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
  * express or implied. See the License for the specific language governing
  * permissions and limitations under the License.
  */

#pragma once

#include <aws/core/Core_EXPORTS.h>

#include <aws/core/http/HttpResponse.h>
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/stream/ResponseStream.h>
#include <aws/core/utils/memory/stl/AWSString.h>

namespace Aws
{
    namespace Http
    {
        namespace Standard
        {
            /**
             * Simple STL representation an Http Response, implements HttpResponse.
             */
            class AWS_CORE_API StandardHttpResponse : public HttpResponse
            {
            public:
                /**
                 * Initializes an http response with the originalRequest and the response code.
                 */
                StandardHttpResponse(const HttpRequest& originatingRequest) :
                    HttpResponse(originatingRequest),
                    bodyStream(originatingRequest.GetResponseStreamFactory())
                {}

                /**
                 * Initializes an http response with the originalRequest and the response code.
                 */
                StandardHttpResponse(const std::shared_ptr<const HttpRequest>& originatingRequest) :
                    HttpResponse(originatingRequest),
                    bodyStream(originatingRequest->GetResponseStreamFactory())
                {}

                ~StandardHttpResponse() = default;

                /**
                 * Get the headers from this response
                 */
                HeaderValueCollection GetHeaders() const;
                /**
                 * Returns true if the response contains a header by headerName
                 */
                bool HasHeader(const char* headerName) const;
                /**
                 * Returns the value for a header at headerName if it exists.
                 */
                const Aws::String& GetHeader(const Aws::String&) const;
                /**
                 * Gets the response body of the response.
                 */
                inline Aws::IOStream& GetResponseBody() const { return bodyStream.GetUnderlyingStream(); }
                /**
                 * Gives full control of the memory of the ResponseBody over to the caller. At this point, it is the caller's
                 * responsibility to clean up this object.
                 */
                inline Utils::Stream::ResponseStream&& SwapResponseStreamOwnership() { return std::move(bodyStream); }
                /**
                 * Adds a header to the http response object.
                 */
                void AddHeader(const Aws::String&, const Aws::String&);

            private:
                StandardHttpResponse(const StandardHttpResponse&);                

                Aws::Map<Aws::String, Aws::String> headerMap;
                Utils::Stream::ResponseStream bodyStream;
            };

        } // namespace Standard
    } // namespace Http
} // namespace Aws


