@use itertools::Itertools;
@use crate::data::RPC;
@use crate::generators::cpp_s::*;

@(rpc: &RPC)
#pragma once
#ifndef MRPC_GEN_H
#define MRPC_GEN_H

#include <unordered_map>
#include <memory>
#include <mutex>
#include <iosfwd>
#include <string>
#include <cstdint>
#include <crow.h>
#define RAPIDJSON_HAS_STDSTRING 1
#include <rapidjson/stringbuffer.h>
#include <rapidjson/writer.h>
#include <rapidjson/document.h>

namespace mrpc @{
    using MRPCJWriter = rapidjson::Writer<rapidjson::StringBuffer>;
@for e in &rpc.enums {
enum struct @e.name : std::uint64_t @{
    @e.values.iter().map(|(k,v)| format!("{k} = {v}")).join(",\n    ")
@};
}
@for s in &rpc.structs {
struct @s.name;
}
@for s in &rpc.structs {
struct @s.name @{
@for f in &s.fields {    @ty_to_str(&f.ty) @f.name;
}
    MRPCJWriter& operator >>(MRPCJWriter&) const;
    @(s.name)& operator <<(const rapidjson::Value&);
@};
}

struct MRPCStreamImpl @{
    virtual void close() noexcept final;
    virtual void abort() noexcept final;
    virtual bool is_open() noexcept final;
protected:
    MRPCStreamImpl(crow::websocket::connection *conn, uint64_t id) : conn(conn), id(id) @{@}
    crow::websocket::connection* conn;
    std::uint64_t id;
@};

template<class T>
struct MRPCStream final : MRPCStreamImpl @{
    MRPCStream(crow::websocket::connection *conn, uint64_t id) : MRPCStreamImpl(conn, id) @{@}
    bool send(const T &v) noexcept @{
        if (!conn) return false;
        try @{
            rapidjson::StringBuffer s;
            mrpc::MRPCJWriter writer@{s@};
            writer.StartObject();
            writer.Key("id");
            writer.Uint64(id);
            writer.Key("data");
            v >> writer;
            writer.EndObject();
            conn->send_text(s.GetString());
        @} catch (const std::exception &_) @{
            abort();
            return false;
        @}
        return true;
    @}
@};

struct MRPCServer @{
    virtual void install(crow::SimpleApp &app, std::string &&route) final;
private:
@for s in &rpc.services {@for m in &s.methods {    virtual @method_ret(m) @(s.name)_@(m.name)(@method_args(m)) = 0;
}}
    virtual void msg_handler(crow::websocket::connection&, std::string, bool) final;

    std::mutex __streams_mutex;
    std::unordered_multimap<crow::websocket::connection*, std::shared_ptr<MRPCStreamImpl>> __streams;
@};
@}

#endif // MRPC_GEN_H