MultipartStreamParser.hpp

Go to the documentation of this file.

#pragma once
#include <string>
#include <string_view>
#include <functional>
#include <fstream>
#include <filesystem>
 
namespace http {
 
class MultipartStreamParser {
public:
    using OnFieldCallback = std::function<void(const std::string& name, const std::string& value)>;
    using OnFileCallback = std::function<void(const std::string& name, const std::string& filename, const std::string& content_type, const std::string& tmp_filepath)>;
 
    MultipartStreamParser(const std::string& boundary, OnFieldCallback on_field, OnFileCallback on_file);
    ~MultipartStreamParser();
 
    void feed(std::string_view chunk);
 
    void end();
 
private:
    std::string boundary_;
    std::string dash_boundary_;
    OnFieldCallback on_field_;
    OnFileCallback on_file_;
 
    enum class State {
        FINDING_BOUNDARY,
        READING_HEADERS,
        READING_BODY
    };
 
    State state_ = State::FINDING_BOUNDARY;
    std::string buffer_;
    
    std::string current_name_;
    std::string current_filename_;
    std::string current_content_type_;
    std::string current_field_value_;
    std::ofstream current_file_;
    std::string current_tmp_filepath_;
    
    void process_buffer();
    void parse_headers(std::string_view headers);
    void open_temp_file();
    void close_current_part();
};
 
} // namespace http