HttpReader.h

/*
╔══════════════════════════════════════════════════════════════════════════════╗
║                       SIMPLY POWERFUL TOOLKIT (SPTK)                         ║
╟──────────────────────────────────────────────────────────────────────────────╢
║  copyright            © 1999-2022 Alexey Parshin. All rights reserved.       ║
║  email                alexeyp@gmail.com                                      ║
╚══════════════════════════════════════════════════════════════════════════════╝
┌──────────────────────────────────────────────────────────────────────────────┐
│   This library is free software; you can redistribute it and/or modify it    │
│   under the terms of the GNU Library General Public License as published by  │
│   the Free Software Foundation; either version 2 of the License, or (at your │
│   option) any later version.                                                 │
│                                                                              │
│   This library 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 Library   │
│   General Public License for more details.                                   │
│                                                                              │
│   You should have received a copy of the GNU Library General Public License  │
│   along with this library; if not, write to the Free Software Foundation,    │
│   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.               │
│                                                                              │
│   Please report all bugs and problems to alexeyp@gmail.com.                  │
└──────────────────────────────────────────────────────────────────────────────┘
*/
 
#pragma once
 
#include <sptk5/Buffer.h>
#include <sptk5/CaseInsensitiveCompare.h>
#include <sptk5/RegularExpression.h>
#include <sptk5/net/TCPSocket.h>
 
#include <mutex>
 
namespace sptk {
 
using HttpHeaders = std::map<String, String, CaseInsensitiveCompare>;
 
class SP_EXPORT HttpReader
{
public:
    enum class State : unsigned
    {
        READY = 0,           
        READING_HEADERS = 1, 
        READING_DATA = 2,    
        COMPLETED = 4,       
        READ_ERROR = 8       
    };
 
    enum class ReadMode
    {
        REQUEST,
        RESPONSE
    };
 
    State getReaderState() const;
 
    HttpHeaders& getHttpHeaders();
 
    String httpHeader(const String& headerName) const;
 
    HttpReader(TCPSocket& socket, Buffer& output, ReadMode readMode);
 
    TCPSocket& socket();
 
    Buffer& output();
 
    void read();
 
    bool readHttpRequest();
 
    void readHttpHeaders();
 
    int readAll(std::chrono::milliseconds timeout);
 
    int getStatusCode() const;
 
    const String& getStatusText() const;
 
    String getRequestType() const;
    String getRequestURL() const;
 
    void close();
 
private:
    TCPSocket& m_socket;                                                                    
    ReadMode m_readMode;                                                                    
    State m_readerState {State::READY};                                                     
    mutable std::mutex m_mutex;                                                             
    String m_statusText;                                                                    
    int m_statusCode {0};                                                                   
    size_t m_contentLength {0};                                                             
    size_t m_contentReceivedLength {0};                                                     
    bool m_contentIsChunked {false};                                                        
    HttpHeaders m_httpHeaders;                                                              
    RegularExpression m_matchProtocolAndResponseCode {"^(HTTP\\S+)\\s+(\\d+)\\s+(.*)?\r?"}; 
    Buffer& m_output;                                                                       
    Buffer m_read_buffer;                                                                   
    String m_requestType;                                                                   
    String m_requestURL;                                                                    
 
    void reset();
 
    bool readStatus();
 
    bool readData();
 
    void readDataChunk(bool& done);
};
 
} // namespace sptk