main.cpp

#include <iostream>
#include "include/Huffman.h"

using namespace std;

int main(int argc, char *argv[])
{
    // Usage: ./byteshrink -c input.txt compressed.bin
    if (argc < 4)
    {
        cout << "=== ByteShrink CLI v1.0 ===" << endl;
        cout << "Usage: " << argv[0] << " [flag] <input_file> <output_file>" << endl;
        cout << "Flags:" << endl;
        cout << "  -c : Compress a text file to binary" << endl;
        cout << "  -d : Decompress a binary file to text" << endl;
        return 1;
    }

    string flag = argv[1];
    string inputFile = argv[2];
    string outputFile = argv[3];

    if (flag == "-c")
    {
        cout << "Compressing: " << inputFile << " -> " << outputFile << "..." << endl;
        Compress(inputFile, outputFile);
        cout << "Compression successful!" << endl;
    }
    else if (flag == "-d")
    {
        cout << "Decompressing: " << inputFile << " -> " << outputFile << "..." << endl;
        Decompresss(inputFile, outputFile);
        cout << "Decompression successful!" << endl;
    }
    else
    {
        cout << "Error: Invalid flag '" << flag << "'. Use -c or -d." << endl;
        return 1;
    }

    return 0;
}
/*
                COMMANDS
    to Compile    -  g++ main.cpp src/Huffman.cpp src/RLE.cpp src/FileM.cpp -Iinclude -o byteshrink
    to Compress   -  ./byteshrink -c input.txt compressed.bin
    to Decompress -  ./byteshrink -d compressed.bin restored.txt
    to Check      -  fc.exe input.txt restored.txt

*/