#include "ggwave/ggwave.h"

#define DR_WAV_IMPLEMENTATION
#include "dr_wav.h"

#include "ggwave-common.h"

#include <cstdio>
#include <cstring>

int main(int argc, char** argv) {
    printf("Usage: %s \"message\" output.wav [-pN]\n", argv[0]);
    printf("    -pN - select the transmission protocol\n");
    printf("\n");
    printf("    Available protocols:\n");

    const auto & protocols = GGWave::getTxProtocols();
    for (int i = 0; i < (int) protocols.size(); ++i) {
        printf("      %d - %s\n", i, protocols[i].name);
    }
    printf("\n");

    if (argc < 3) {
        return -1;
    }

    std::string message = argv[1];
    std::string fnameOut = argv[2];

    if (message.size() == 0) {
        fprintf(stderr, "Invalid message: size = 0\n");
        return -2;
    }

    if (message.size() > 140) {
        fprintf(stderr, "Invalid message: size > 140\n");
        return -3;
    }

    if (fnameOut.rfind(".wav") != fnameOut.size() - 4) {
        fprintf(stderr, "Invalid output filename\n");
        return -4;
    }

    auto argm = parseCmdArguments(argc, argv);
    int protocolId = argm["p"].empty() ? 1 : std::stoi(argm["p"]);
    int volume = argm["v"].empty() ? 50 : std::stoi(argm["v"]);

    auto sampleRateOut = GGWave::kBaseSampleRate;

    GGWave ggWave(GGWave::kBaseSampleRate, sampleRateOut, 1024, 4, 2);
    ggWave.init(message.size(), message.data(), ggWave.getTxProtocols()[protocolId], volume);

    std::vector<char> bufferPCM;
    GGWave::CBQueueAudio cbQueueAudio = [&](const void * data, uint32_t nBytes) {
        bufferPCM.resize(nBytes);
        std::memcpy(bufferPCM.data(), data, nBytes);
    };

    ggWave.send(cbQueueAudio);

    printf("Output size = %d bytes\n", (int) bufferPCM.size());

    drwav_data_format format;
    format.container = drwav_container_riff;
    format.format = DR_WAVE_FORMAT_PCM;
    format.channels = 1;
    format.sampleRate = sampleRateOut;
    format.bitsPerSample = 16;

    printf("Writing WAV file '%s' ...\n", fnameOut.c_str());

    drwav wav;
    drwav_init_file_write(&wav, fnameOut.c_str(), &format, NULL);
    drwav_uint64 framesWritten = drwav_write_pcm_frames(&wav, bufferPCM.size()/2, bufferPCM.data());

    printf("WAV frames written = %d\n", (int) framesWritten);

    return 0;
}