/*
 *   Copyright (C) 2021-2022 by Geoffrey Merck F4FXL / KC3FRA
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 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 General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the Free Software
 *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <boost/algorithm/string.hpp>

#include "RSMS1AMessageBuilder.h"
#include "StringUtils.h"

std::vector<signed char> CRSMS1AMessageBuilder::m_charsToEscape = {-17, 0, 17, 19, -2, -25, 26, -3, -1, 36, 13, 44};

void CRSMS1AMessageBuilder::buildMessage(std::string& message, const std::string& sender, const std::string& recipient, const std::string body)
{
    auto bodyCrc = calculateBodyCRC(body);
    std::string bodyTmp;
    escapeBody(bodyTmp, body + (char)bodyCrc);
    
    std::string header = CStringUtils::string_format("%s,%s,0011", sender.c_str(), recipient.c_str());
    signed char c1, c2;
    calcMsgIcomCRC(header, c1, c2);
    header.push_back(c1);
    header.push_back(c2);
    message = "$$Msg," + header + bodyTmp + '\r';
}   

signed char CRSMS1AMessageBuilder::calculateBodyCRC(const std::string& body)
{
    if(body.length() == 1)
        return body[0];

    signed int num = 0;
    for(signed char c : body) {
        num += c;
    }

    signed int res = (num & 255);
    if(res >= 128)
        res -= 128;
    
    return (signed char)res;
}

void CRSMS1AMessageBuilder::escapeBody(std::string& output, const std::string& body)
{
    output.clear();
    for(char c : body) {
        if(std::find(m_charsToEscape.begin(), m_charsToEscape.end(), c) != m_charsToEscape.end()) {
            output.push_back('o');
        }
        output.push_back(c);
    }
}

void CRSMS1AMessageBuilder::calcMsgIcomCRC(const std::string& msg, signed char& c1, signed char& c2)
{
	int num = 0;
	for(unsigned int i = 0U; i < msg.length(); i++) {
		num += msg[i];
	}

	c1 = doWhatever((signed char)((num >> 4) & 15));
	c2 = doWhatever((signed char)(num & 15));
}

signed char CRSMS1AMessageBuilder::doWhatever(signed char b2) {
    int i;
    int i2 = b2 & 255;
    if (i2 >= 0 && i2 <= 9) {
        i = b2 + 48;
    } else if (10 > i2 || i2 > 15) {
        return 0;
    } else {
        i = b2 + 55;
    }
    return (signed char) i;
}

RSMS1A_PARSE_STATUS CRSMS1AMessageBuilder::parseMessage(std::string& sender, std::string& recipient, std::string& body, const std::string& message)
{
    sender.clear();
    recipient.clear();
    body.clear();

    if(!boost::starts_with(message, "$$Msg,"))
        return RSMS_FAIL;

    auto firstCommaPos = message.find_first_of(',');
    if(firstCommaPos == std::string::npos || (firstCommaPos + 1) >= message.length())
        return RSMS_FAIL;
    
    auto secondCommaPos = message.find_first_of(',', firstCommaPos + 1);
    if(secondCommaPos == std::string::npos || (secondCommaPos + 1) >= message.length())
        return RSMS_FAIL;

    auto thirdCommaPos = message.find_first_of(',', secondCommaPos + 1);
    if(thirdCommaPos == std::string::npos || (thirdCommaPos + 1 + 6) >= message.length())
        return RSMS_FAIL;

    sender.assign(message.substr(firstCommaPos + 1, secondCommaPos - firstCommaPos - 1));
    recipient.assign(message.substr(secondCommaPos + 1, thirdCommaPos - secondCommaPos - 1));
    body.assign(message.substr(thirdCommaPos + 1 + 6));
    unescapeBody(body, std::string(body));
    if(body.length() >= 2U) body.resize(body.length() - 2U);

    // build a message out of what we received in order to check if all checksums matches
    // use only header of message to match checksum
    std::string messagecheck;
    buildMessage(messagecheck, sender, recipient, " ");
    messagecheck.resize(messagecheck.length() - 3U);// get rid of body, body check byte and trailing \r

    if(messagecheck != message.substr(0, messagecheck.length())) {
        sender.clear();
        recipient.clear();
        body.clear();
        return RSMS_FAIL; // we do not allow any errors in message header
    }

    //rebuild complete messsage with body
    buildMessage(messagecheck, sender, recipient, body);
    if(messagecheck != message)
        return RSMS_ERRONEOUS_MSG; // we allow erros to occur in the message body

    return RSMS_OK;
}

void CRSMS1AMessageBuilder::unescapeBody(std::string& output, const std::string& body)
{
    output.clear();
    if(body.empty())
        return;

    for(unsigned int i = 0U; i < body.length(); i++) {
        auto c = body[i];
        auto next = body[i + 1U];
        if(c == 'o' && std::find(m_charsToEscape.begin(), m_charsToEscape.end(), next) != m_charsToEscape.end()) {
            output.push_back(next);
            i++;
            continue;
        }
        
        output.push_back(c);
    }
}