#include <cstdint>
#include <cstring>
#include <iostream>
#include <gtest/gtest.h>

#include "simplesync.hpp"


TEST(simplesync_test, varint) {

    uint8_t buf[1024];
    for (int i = -999999; i < 999999; i += 29) {
        int offset = simplesync::encode_varint(i, buf);
        ASSERT_LE(1, offset);
        for (int o = 0; o < offset - 1; o += 1)
            EXPECT_FALSE(buf[o] & 0x80) << "offset:" << offset << " " << (uint32_t)(buf[o] & 0x7F);
        EXPECT_TRUE(buf[offset - 1] & 0x80)
            << "offset:" << offset << " " << (uint32_t)(buf[offset - 1] & 0x7F);
        int out;
        offset = simplesync::decode_varint(buf, out);
        ASSERT_LE(1, offset);
        ASSERT_EQ(out, i);
    }
}

TEST(simplesync_test, cobs_encode) {
    uint8_t buf_raw[]   = {0x11, 0x22, 0x00, 0x33};
    uint8_t buf_coded[] = {0x03, 0x11, 0x22, 0x02, 0x33, 0x00};

    uint8_t buf[sizeof(buf_coded) + 1];
    unsigned int buf_used = simplesync::cobs_encode(buf_raw, sizeof(buf_raw), buf, sizeof(buf));
    ASSERT_EQ(buf_used, sizeof(buf_coded));
    ASSERT_EQ(0, memcmp(buf_coded, buf, sizeof(buf_coded)));

    unsigned int buf_out_used = sizeof(buf);
    unsigned int buf_in_used  = sizeof(buf_coded);
    int error                 = simplesync::cobs_decode(buf_coded, buf_in_used, buf, buf_out_used);
    ASSERT_EQ(error, 0);
    ASSERT_EQ(buf_used, sizeof(buf_coded));
    ASSERT_EQ(buf_out_used, sizeof(buf_raw));
    ASSERT_EQ(0, memcmp(buf_raw, buf, sizeof(buf_raw)));
}

uint8_t stream_buf[1024];
unsigned int buf_size = sizeof(stream_buf);
unsigned int buf_used = 0;

TEST(simplesync_test, encode_decode) {
    buf_used = 0;
    typedef simplesync::SimpleSync<int, int, 512, true> S;
    S s_send([](uint8_t *buf, unsigned int buf_size) {
        memcpy(stream_buf + buf_used, buf, buf_size);
        buf_used += buf_size;
        return 0;
    });
    S s_recv([](uint8_t *, unsigned int) {
        return 0;
    });

    int n1 = 4;
    int n2 = 2;
    std::string str = "hello";
    S::NumberPtrPeriodic(s_send, "n1",&n1,0);
    S::NumberPtrPeriodic(s_send, "n2",&n2,2);
    S::TextPtrPeriodic(s_send, "s1",&str, 0);

    for (int time = 0; time < 5; time += 1) {
        buf_used = 0;
        s_send.update(time);
        s_send.update(time);
        for(unsigned int i = 0;i < buf_used; i+=1) std::cout << std::hex <<(int) stream_buf[i] << " ";
        std::cout << std::dec << std::endl;
        
        unsigned int parsed = 0;
        for (unsigned int i = 1; i <= buf_used; i += 1)
            parsed += s_recv.parse_stream_buf(stream_buf + parsed, i - parsed);
        ASSERT_EQ(parsed, buf_used);
    }
    str = "hollo2";
    ASSERT_EQ(*s_recv.get_number((char*)"n1"), n1);
    ASSERT_EQ(*s_recv.get_number((char*)"n2"), n2);
    ASSERT_STREQ(s_send.get_string((char*)"s1")->c_str(),"hollo2");
    ASSERT_STREQ(str.c_str(),"hollo2");
    ASSERT_STREQ(s_recv.get_string((char*)"s1")->c_str(),"hello");
    ASSERT_EQ(4, n1);
    ASSERT_EQ(2, n2);
}

int main(int argc, char **argv) {
    testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}