/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "DeflaterOutputStream.h"

using namespace decaf;
using namespace decaf::io;
using namespace decaf::lang;
using namespace decaf::lang::exceptions;
using namespace decaf::util;
using namespace decaf::util::zip;

////////////////////////////////////////////////////////////////////////////////
const std::size_t DeflaterOutputStream::DEFAULT_BUFFER_SIZE = 512;

////////////////////////////////////////////////////////////////////////////////
DeflaterOutputStream::DeflaterOutputStream(OutputStream* outputStream, bool own) :
    FilterOutputStream(outputStream, own), deflater(new Deflater()), buf(), ownDeflater(true), isDone(false) {

    this->buf.resize(DEFAULT_BUFFER_SIZE);
}

////////////////////////////////////////////////////////////////////////////////
DeflaterOutputStream::DeflaterOutputStream(OutputStream* outputStream, Deflater* deflater, bool own, bool ownDeflater) :
    FilterOutputStream(outputStream, own), deflater(deflater), buf(), ownDeflater(ownDeflater), isDone(false) {

    if (deflater == NULL) {
        throw NullPointerException(
            __FILE__, __LINE__, "Deflater passed was NULL.");
    }

    this->buf.resize(DEFAULT_BUFFER_SIZE);
}

////////////////////////////////////////////////////////////////////////////////
DeflaterOutputStream::DeflaterOutputStream(OutputStream* outputStream, Deflater* deflater, int bufferSize, bool own, bool ownDeflater) :
    FilterOutputStream(outputStream, own), deflater(deflater), buf(), ownDeflater(ownDeflater), isDone(false) {

    if (deflater == NULL) {
        throw NullPointerException(
            __FILE__, __LINE__, "Deflater passed was NULL.");
    }

    if (bufferSize == 0) {
        throw IllegalArgumentException(
            __FILE__, __LINE__, "Cannot create a zero sized buffer.");
    }

    this->buf.resize(bufferSize);
}

////////////////////////////////////////////////////////////////////////////////
DeflaterOutputStream::~DeflaterOutputStream() {
    try {

        this->close();

        if (ownDeflater) {
            delete this->deflater;
        }
    }
    DECAF_CATCH_NOTHROW(Exception)
    DECAF_CATCHALL_NOTHROW()}

////////////////////////////////////////////////////////////////////////////////
void DeflaterOutputStream::finish() {

    try {

        if (isDone) {
            return;
        }

        int result;
        this->deflater->finish();

        while (!this->deflater->finished()) {

            if (this->deflater->needsInput()) {
                this->deflater->setInput(buf, 0, 0);
            }
            result = this->deflater->deflate(&buf[0], (int) buf.size(), 0, (int) buf.size());
            this->outputStream->write(&buf[0], (int) buf.size(), 0, result);
        }

        this->isDone = true;
    }
    DECAF_CATCH_RETHROW(IOException)
    DECAF_CATCHALL_THROW(IOException)
}

////////////////////////////////////////////////////////////////////////////////
void DeflaterOutputStream::close() {

    try {

        if (!this->deflater->finished()) {
            this->finish();
        }
        this->deflater->end();
        FilterOutputStream::close();
    }
    DECAF_CATCH_RETHROW(IOException)
    DECAF_CATCHALL_THROW(IOException)
}

////////////////////////////////////////////////////////////////////////////////
void DeflaterOutputStream::doWriteByte(unsigned char value) {

    try {
        this->doWriteArrayBounded(&value, 1, 0, 1);
    }
    DECAF_CATCH_RETHROW(IOException)
    DECAF_CATCHALL_THROW(IOException)
}

////////////////////////////////////////////////////////////////////////////////
void DeflaterOutputStream::doWriteArrayBounded(const unsigned char* buffer, int size, int offset, int length) {

    try {

        if (isDone) {
            throw IOException(
                __FILE__, __LINE__, "Finish was already called on this DeflaterOutputStream.");
        }

        if (buffer == NULL) {
            throw NullPointerException(
                __FILE__, __LINE__, "Buffer passed was NULL.");
        }

        if (size < 0) {
            throw IndexOutOfBoundsException(
                __FILE__, __LINE__, "size parameter out of Bounds: %d.", size);
        }

        if (offset > size || offset < 0) {
            throw IndexOutOfBoundsException(
                __FILE__, __LINE__, "offset parameter out of Bounds: %d.", offset);
        }

        if (length < 0 || length > size - offset) {
            throw IndexOutOfBoundsException(
                __FILE__, __LINE__, "length parameter out of Bounds: %d.", length);
        }

        if (length == 0) {
            return;
        }

        if (isClosed()) {
            throw IOException(
                __FILE__, __LINE__, "The stream is already closed.");
        }

        if (!this->deflater->needsInput()) {
            throw IOException(
                __FILE__, __LINE__, "The Deflater is in an Invalid State.");
        }

        this->deflater->setInput(buffer, size, offset, length);

        this->deflate();
    }
    DECAF_CATCH_RETHROW(IOException)
    DECAF_CATCH_RETHROW(NullPointerException)
    DECAF_CATCH_RETHROW(IndexOutOfBoundsException)
    DECAF_CATCHALL_THROW(IOException)
}

////////////////////////////////////////////////////////////////////////////////
void DeflaterOutputStream::deflate() {

    try {

        int result;
        do {
            result = this->deflater->deflate(&buf[0], (int) buf.size(), 0, (int) buf.size());
            this->outputStream->write(&buf[0], (int) buf.size(), 0, result);
        } while (!this->deflater->needsInput());
    }
    DECAF_CATCH_RETHROW(IOException)
    DECAF_CATCHALL_THROW(IOException)
}