/***
    Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: BZip2Decompressor.hpp
    Date: 2022-2-15
    Author: Reece
***/
#pragma once

#include "bzlib.h"

typedef struct
{
    void *    handle;
    char      buf[BZ_MAX_UNUSED];
    AuInt32   bufN;
    char      writing;
    bz_stream strm;
    AuInt32   lastErr;
    char      initialisedOk;
} bz2_file;

static const char *BShitToString(int error)
{
    int n {};
    bz2_file B;
    B.lastErr = error;
    return BZ2_bzerror((BZFILE *)&B, &n);
}

namespace Aurora::Compression
{
    struct BZIPInflate : BaseStream
    {
        DecompressInfo meta;

        BZIPInflate(const DecompressInfo &meta) : meta(meta), BaseStream(meta.uInternalStreamSize)
        {}

        ~BZIPInflate()
        {
            if (this->bInit_)
            {
                BZ2_bzDecompressEnd(&this->ctx_);
            }
        }

        bool Init(const AuSPtr<IO::IStreamReader> &pReader) override
        {
            this->pReader_ = pReader;

            if (!this->IsValid())
            {
                SysPushErrorMem();
                return false;
            }

            auto ret = BZ2_bzDecompressInit(&this->ctx_, 0, 0);
            if (ret < BZ_OK)
            {
                this->SetLastError(ret, BShitToString(ret));
                return false;
            }

            this->bInit_ = true;

            this->SetArray(this->din_);
            this->SetOutArray(this->dout_);
            return true;
        }

        AuStreamReadWrittenPair_t Ingest_s(AuUInt32 input) override
        {
            AuUInt32 done {}, read {};

            if (!this->pReader_)
            {
                return {};
            }

            while (read < input)
            {
                auto [pMainDOut, uMainDOutLength] = this->GetDOutPair();

                read += IngestForInPointer<char, uInt>(this->pReader_, this->ctx_.next_in, this->ctx_.avail_in, input - read, this);

                if (!this->ctx_.avail_in)
                {
                    return {read, done};
                }

                do
                {
                    this->ctx_.avail_out = uMainDOutLength;
                    this->ctx_.next_out = pMainDOut;

                    auto ret = BZ2_bzDecompress(&this->ctx_);
                    if (ret < BZ_OK)
                    {
                        this->SetLastError(ret, BShitToString(ret));
                        this->pReader_.reset();
                        return AuMakePair(read, 0);
                    }

                    auto have = uMainDOutLength - this->ctx_.avail_out;
                    done += have;

                    if (!Write2(pMainDOut,
                                have))
                    {
                        this->SetLastError(0x69, "OOM");
                        this->pReader_.reset();
                        return AuMakePair(read, 0);
                    }

                }
                while (this->ctx_.avail_out == 0);
            }

            return AuMakePair(read, done);
        }

    private:
        AuSPtr<IO::IStreamReader> pReader_;
        bz_stream ctx_ {};
        bool bInit_ {};
        char dout_[kChunkSize];
        char din_[kChunkSize];
    };
}