Request.h

Go to the documentation of this file.

/*
 * SNode.C - a slim toolkit for network communication
 * Copyright (C) Volker Christian <me@vchrist.at>
 *               2020, 2021, 2022, 2023, 2024, 2025
 *
 * 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/>.
 */
 
#ifndef WEB_HTTP_CLIENT_REQUEST_H
#define WEB_HTTP_CLIENT_REQUEST_H
 
#include "core/pipe/Sink.h"
#include "web/http/ConnectionState.h"
#include "web/http/TransferEncoding.h"
 
namespace web::http::client {
    class RequestCommand;
    class Response;
    class SocketContext;
    namespace commands {
        class SendFileCommand;
        class SendFragmentCommand;
        class SendHeaderCommand;
        class UpgradeCommand;
        class EndCommand;
    } // namespace commands
} // namespace web::http::client
 
#ifndef DOXYGEN_SHOULD_SKIP_THIS
 
#include "web/http/CiStringMap.h" // IWYU pragma: export
 
#include <cstddef>
#include <functional>
#include <list>
#include <memory>
#include <string>
 
#endif /* DOXYGEN_SHOULD_SKIP_THIS */
 
namespace web::http::client {
 
    class Request : public core::pipe::Sink {
    public:
        explicit Request(web::http::client::SocketContext* socketContext, const std::string& host);
 
        explicit Request(Request&) = delete;
        explicit Request(Request&&) noexcept;
 
        Request& operator=(Request&) = delete;
        Request& operator=(Request&&) noexcept = delete;
 
        ~Request() override;
 
        void setMasterRequest(const std::shared_ptr<Request>& masterRequest);
        virtual void init();
 
        Request& host(const std::string& hostFieldValue);
        Request& append(const std::string& field, const std::string& value);
        Request& set(const std::string& field, const std::string& value, bool overwrite = true);
        Request& set(const std::map<std::string, std::string>& headers, bool overwrite = true);
        Request& type(const std::string& type);
        Request& cookie(const std::string& name, const std::string& value);
        Request& cookie(const std::map<std::string, std::string>& cookies);
        Request& query(const std::string& key, const std::string& value);
        Request& setTrailer(const std::string& field, const std::string& value, bool overwrite = true);
 
        static void responseParseError(const std::shared_ptr<Request>& request, const std::string& message);
 
        bool
        send(const char* chunk,
             std::size_t chunkLen,
             const std::function<void(const std::shared_ptr<Request>&, const std::shared_ptr<Response>&)>& onResponseReceived,
             const std::function<void(const std::shared_ptr<Request>&, const std::string&)>& onResponseParseError = responseParseError);
        bool
        send(const std::string& chunk,
             const std::function<void(const std::shared_ptr<Request>&, const std::shared_ptr<Response>&)>& onResponseReceived,
             const std::function<void(const std::shared_ptr<Request>&, const std::string&)>& onResponseParseError = responseParseError);
        bool
        upgrade(const std::string& url,
                const std::string& protocols,
                const std::function<void(const std::shared_ptr<Request>&, const std::shared_ptr<Response>&)>& onResponseReceived,
                const std::function<void(const std::shared_ptr<Request>&, const std::string&)>& onResponseParseError = responseParseError);
        void upgrade(const std::shared_ptr<Response>& response, const std::function<void(const std::string&)>& status);
        bool
        sendFile(const std::string& file,
                 const std::function<void(int errnum)>& onStatus,
                 const std::function<void(const std::shared_ptr<Request>&, const std::shared_ptr<Response>&)>& onResponseReceived,
                 const std::function<void(const std::shared_ptr<Request>&, const std::string&)>& onResponseParseError = responseParseError);
        Request& sendHeader();
        Request& sendFragment(const char* chunk, std::size_t chunkLen);
        Request& sendFragment(const std::string& data);
        bool end(const std::function<void(const std::shared_ptr<Request>&, const std::shared_ptr<Response>&)>& onResponseReceived,
                 const std::function<void(const std::shared_ptr<Request>&, const std::string&)>& onResponseParseError = responseParseError);
 
    private:
        bool initiate();
 
        friend class commands::SendFileCommand;
        friend class commands::SendFragmentCommand;
        friend class commands::SendHeaderCommand;
        friend class commands::UpgradeCommand;
        friend class commands::EndCommand;
 
        bool executeSendFile(const std::string& file, const std::function<void(int)>& onStatus);
        bool executeUpgrade(const std::string& url, const std::string& protocols);
        bool executeEnd();
        bool executeSendHeader();
        bool executeSendFragment(const char* chunk, std::size_t chunkLen);
 
        void requestPrepared();
        void requestDelivered();
 
        void deliverResponse(const std::shared_ptr<Request>& request, const std::shared_ptr<Response>& response);
        void deliverResponseParseError(const std::shared_ptr<Request>& request, const std::string& message);
 
        void onSourceConnect(core::pipe::Source* source) override;
        void onSourceData(const char* chunk, std::size_t chunkLen) override;
        void onSourceEof() override;
        void onSourceError(int errnum) override;
 
    public:
        std::string header(const std::string& field);
 
        const CiStringMap<std::string>& getQueries() const;
        const CiStringMap<std::string>& getHeaders() const;
        const CiStringMap<std::string>& getCookies() const;
 
        web::http::client::SocketContext* getSocketContext() const;
 
        std::string hostFieldValue;
        std::string method = "GET";
        std::string url = "/";
        int httpMajor = 1;
        int httpMinor = 1;
 
    protected:
        CiStringMap<std::string> queries;
        CiStringMap<std::string> headers;
        CiStringMap<std::string> cookies;
        CiStringMap<std::string> trailer;
 
    private:
        std::list<RequestCommand*> requestCommands;
 
        TransferEncoding transferEncoding = TransferEncoding::HTTP10;
 
        std::size_t contentLength = 0;
        std::size_t contentLengthSent = 0;
 
        ConnectionState connectionState = ConnectionState::Default;
 
        std::function<void(const std::shared_ptr<Request>&, const std::shared_ptr<Response>&)> onResponseReceived;
        std::function<void(const std::shared_ptr<Request>& request, const std::string& message)> onResponseParseError;
 
        std::weak_ptr<Request> masterRequest;
 
        web::http::client::SocketContext* socketContext;
 
        friend class SocketContext;
    };
 
} // namespace web::http::client
 
#endif // WEB_HTTP_CLIENT_REQUEST_H