#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <string>

#include "../../public/common/tracy_lz4hc.hpp"

static void Usage()
{
    fprintf( stderr, "Usage: embed [-t] <objectName> <source> <destination>\n" );
    fprintf( stderr, "  destination should be without extension, will create cpp, hpp pair\n" );
    fprintf( stderr, "  -t: treat source as text, convert line endings to unix\n" );
}

int main( int argc, char** argv )
{
    bool text = false;
    if( argc >= 2 && strcmp( argv[1], "-t" ) == 0 )
    {
        text = true;
        argc--;
        argv++;
    }

    if( argc < 4 )
    {
        Usage();
        return 1;
    }

    const char* objectName = argv[1];
    const char* source = argv[2];
    const char* destination = argv[3];

    FILE* src = fopen( source, "rb" );
    if( !src )
    {
        fprintf( stderr, "Failed to open source file %s\n", source );
        return 1;
    }

    size_t sz;
    fseek( src, 0, SEEK_END );
    sz = ftell( src );
    fseek( src, 0, SEEK_SET );

    auto data = new uint8_t[sz];
    fread( data, 1, sz, src );
    fclose( src );

    if( text )
    {
        size_t pos = 0;
        for( size_t i=0; i<sz; i++ )
        {
            if( data[i] != '\r' ) data[pos++] = data[i];
        }
        sz = pos;
    }

    const auto lz4szMax = tracy::LZ4_compressBound( sz );
    auto lz4data = new uint8_t[lz4szMax];
    const auto lz4sz = tracy::LZ4_compress_HC( (const char*)data, (char*)lz4data, sz, lz4szMax, 6 );
    delete[] data;

    FILE* hdr = fopen( ( std::string(destination) + ".hpp" ).c_str(), "wb" );
    fprintf( hdr, "// This file is generated by embed tool, do not modify\n" );
    fprintf( hdr, "// Source: %s\n\n", source );
    fprintf( hdr, "#pragma once\n\n" );
    fprintf( hdr, "#include <stddef.h>\n" );
    fprintf( hdr, "#include <stdint.h>\n\n" );
    fprintf( hdr, "namespace Embed\n{\n" );
    fprintf( hdr, "constexpr size_t %sSize = %zu;\n", objectName, sz );
    fprintf( hdr, "constexpr size_t %sLz4Size = %i;\n", objectName, lz4sz );
    fprintf( hdr, "extern const uint8_t %sData[];\n", objectName );
    fprintf( hdr, "}\n" );
    fclose( hdr );

    FILE* cpp = fopen( ( std::string(destination) + ".cpp" ).c_str(), "wb" );
    fprintf( cpp, "// This file is generated by embed tool, do not modify\n" );
    fprintf( cpp, "// Source: %s\n\n", source );
    fprintf( cpp, "#include \"%s.hpp\"\n\n", destination );
    fprintf( cpp, "namespace Embed\n{\n" );
    fprintf( cpp, "const uint8_t %sData[] =\n", objectName );
    fprintf( cpp, "{\n" );
    for( size_t i=0; i<lz4sz; i++ )
    {
        fprintf( cpp, "%d,", lz4data[i] );
    }
    fprintf( cpp, "};\n" );
    fprintf( cpp, "}\n" );

    fclose( cpp );
    delete[] lz4data;
    return 0;
}
