/**
 * Orthanc - A Lightweight, RESTful DICOM Store
 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
 * Department, University Hospital of Liege, Belgium
 * Copyright (C) 2017-2021 Osimis S.A., Belgium
 *
 * This program is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public License
 * as published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this program. If not, see
 * <http://www.gnu.org/licenses/>.
 **/


#pragma once

#include "IHttpOutputStream.h"

#include "../ChunkedBuffer.h"
#include "../Compatibility.h"  // For ORTHANC_OVERRIDE

#include <map>


namespace Orthanc
{
  class StringHttpOutput : public IHttpOutputStream
  {
  private:
    HttpStatus    status_;
    ChunkedBuffer body_;
    ChunkedBuffer headers_;
    bool          validBody_;
    bool          validHeaders_;

  public:
    StringHttpOutput();

    virtual void OnHttpStatusReceived(HttpStatus status) ORTHANC_OVERRIDE
    {
      status_ = status;
    }

    virtual void Send(bool isHeader, const void* buffer, size_t length) ORTHANC_OVERRIDE;

    virtual void DisableKeepAlive() ORTHANC_OVERRIDE
    {
    }

    HttpStatus GetStatus() const
    {
      return status_;
    }

    void GetBody(std::string& output);

    void GetHeaders(std::map<std::string, std::string>& target,
                    bool keyToLowerCase);
  };
}