#ifndef MYSQL_FFTO_HPP
#define MYSQL_FFTO_HPP

#include "TrafficObserver.hpp"
#include <vector>
#include <string>
#include <unordered_map>
#include <stdint.h>

class MySQL_Session;

class MySQLFFTO : public TrafficObserver {
public:
    explicit MySQLFFTO(MySQL_Session* session);
    virtual ~MySQLFFTO();

    void on_client_data(const char* buf, size_t len) override;
    void on_server_data(const char* buf, size_t len) override;
    void on_close() override;

private:
    enum State {
        IDLE,
        AWAITING_PREPARE_OK,
        AWAITING_RESPONSE,
        READING_COLUMNS,
        READING_ROWS
    };

    MySQL_Session* m_session;
    State m_state;
    std::vector<char> m_client_buffer;
    std::vector<char> m_server_buffer;
    
    std::string m_current_query;
    std::string m_pending_prepare_query;
    unsigned long long m_query_start_time;
    uint64_t m_affected_rows;
    uint64_t m_rows_sent;

    std::unordered_map<uint32_t, std::string> m_statements;

    void process_client_packet(const unsigned char* data, size_t len);
    void process_server_packet(const unsigned char* data, size_t len);
    void report_query_stats(const std::string& query, unsigned long long duration_us, uint64_t affected_rows = 0, uint64_t rows_sent = 0);
};

#endif // MYSQL_FFTO_HPP