#include <ctype.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <inttypes.h>
#include <string.h>
#include <getopt.h>
#include "libae.h"

#define CHUNK 1024

int main(int argc, char *argv[])
{
    ae_stream strm;
    uint8_t *in;
    uint8_t *out;
    int chunk, c, total_out, status;
    int input_avail, output_avail;
    char *outfn, *infn, *ext;
    FILE *infp, *outfp;
    int cflag = 0;

    chunk = CHUNK;
    strm.bit_per_sample = 8;
    strm.block_size = 8;
    strm.rsi = 2;
    strm.flags = AE_DATA_MSB | AE_DATA_PREPROCESS;
    opterr = 0;

    while ((c = getopt (argc, argv, "cb:B:J:R:")) != -1)
        switch (c)
        {
        case 'b':
            chunk = 2 * atoi(optarg);
            break;
        case 'B':
            strm.bit_per_sample = atoi(optarg);
            break;
        case 'J':
            strm.block_size = atoi(optarg);
            break;
        case 'R':
            strm.rsi = atoi(optarg);
            break;
        case 'c':
            cflag = 1;
            break;
        case '?':
            if (optopt == 'b')
                fprintf (stderr, "Option -%c requires an argument.\n", optopt);
            else if (isprint (optopt))
                fprintf (stderr, "Unknown option `-%c'.\n", optopt);
            else
                fprintf (stderr,
                         "Unknown option character `\\x%x'.\n",
                         optopt);
            return 1;
        default:
            abort ();
        }

    if (optind < argc)
    {
        infn = argv[optind];
    }
    else
    {
        fprintf(stderr, "Usage: %s [ -c ] [ -b chunksize ] name\n", argv[0]);
        exit(-1);
    }

    in = (uint8_t *)malloc(chunk);
    out = (uint8_t *)malloc(chunk * sizeof(uint8_t));
    if (in == NULL || out == NULL)
        return 1;

    if (cflag)
    {
        outfp = stdout;
    }
    else
    {
        outfn = malloc(strlen(infn));
        if (outfn == NULL)
            exit(-1);

        if ((ext = strstr(infn, ".aee")) == NULL)
        {
            fprintf(stderr, "Error: input file needs to end with .aee\n");
            exit(-1);
        }
        strncpy(outfn, infn, ext - infn);

        if ((outfp = fopen(outfn, "w")) == NULL)
            exit(-1);
    }
    if ((infp = fopen(infn, "r")) == NULL)
        exit(-1);

    if (ae_decode_init(&strm) != AE_OK)
        return 1;

    total_out = 0;
    strm.avail_in = 0;
    strm.avail_out = chunk;
    strm.next_out = (uint8_t *)out;

    input_avail = 1;
    output_avail = 1;

    while(input_avail || output_avail)
    {
        if (strm.avail_in == 0)
        {
            strm.avail_in = fread(in, 1, chunk, infp);
            if (strm.avail_in != chunk)
                input_avail = 0;

            strm.next_in = in;
        }

        if ((status = ae_decode(&strm, AE_NO_FLUSH)) != AE_OK)
        {
            fprintf(stderr, "error is %i\n", status);
            return 1;
        }

        if (strm.total_out - total_out > 0)
        {
            fwrite(out, strm.total_out - total_out, 1, outfp);
            total_out = strm.total_out;
            output_avail = 1;
            strm.next_out = (uint8_t *)out;
            strm.avail_out = chunk;
        }
        else
        {
            output_avail = 0;
        }

    }

    ae_decode_end(&strm);
    fclose(infp);
    fclose(outfp);
    free(in);
    free(out);
    if (!cflag)
    {
        unlink(infn);
        free(outfn);
    }
    return 0;
}