#include "irc_message.h"

namespace roro {

IrcMessage IrcMessage::parse(const std::string& raw) {
    IrcMessage m;
    std::string line = raw;
    while (!line.empty() && (line.back() == '\r' || line.back() == '\n')) line.pop_back();

    size_t pos = 0;
    auto word = [&]() {
        size_t end = line.find(' ', pos);
        std::string w = line.substr(pos, end == std::string::npos ? std::string::npos : end - pos);
        pos = end == std::string::npos ? line.size() : end + 1;
        while (pos < line.size() && line[pos] == ' ') pos++;
        return w;
    };
    if (pos < line.size() && line[pos] == '@') word();
    if (pos < line.size() && line[pos] == ':') m.prefix = word().substr(1);
    m.command = word();
    while (pos < line.size()) {
        if (line[pos] == ':') {
            m.params.push_back(line.substr(pos + 1));
            break;
        }
        m.params.push_back(word());
    }
    return m;
}

std::string IrcMessage::serialize(const std::string& command, std::initializer_list<std::string> params) {
    std::string out = command;
    size_t i = 0;
    for (auto& p : params) {
        bool last = ++i == params.size();
        out += ' ';
        if (last && (p.empty() || p[0] == ':' || p.find(' ') != std::string::npos)) out += ':';
        out += p;
    }
    return out;
}

}  // namespace roro