/****************************************************************
 * 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.                                           *
 ****************************************************************/

package org.apache.james.mime4j.util;


/**
 * A resizable byte array.
 */
public final class RecycledByteArrayBuffer implements ByteSequence {
    private final BufferRecycler bufferRecycler;
    private byte[] buffer;
    private int len;

    public RecycledByteArrayBuffer(BufferRecycler bufferRecycler, int capacity) {
        super();
        if (capacity < 0) {
            throw new IllegalArgumentException("Buffer capacity may not be negative");
        }
        this.buffer = bufferRecycler.allocByteBuffer(0, capacity);
        this.bufferRecycler = bufferRecycler;
    }

    public RecycledByteArrayBuffer(BufferRecycler bufferRecycler, byte[] bytes, boolean dontCopy) {
        this(bufferRecycler, bytes, bytes.length, dontCopy);
    }

    public RecycledByteArrayBuffer(BufferRecycler bufferRecycler, byte[] bytes, int len, boolean dontCopy) {
        if (bytes == null)
            throw new IllegalArgumentException();
        if (len < 0 || len > bytes.length)
            throw new IllegalArgumentException();

        if (dontCopy) {
            this.buffer = bytes;
        } else {
            this.buffer = bufferRecycler.allocByteBuffer(0, len);
            System.arraycopy(bytes, 0, this.buffer, 0, len);
        }
        this.bufferRecycler = bufferRecycler;
        this.len = len;
    }

    private void expand(int newlen) {
        byte newbuffer[] = bufferRecycler.allocByteBuffer(0, Math.max(this.buffer.length << 1, newlen));
        System.arraycopy(this.buffer, 0, newbuffer, 0, this.len);
        bufferRecycler.releaseByteBuffer(0, buffer);
        this.buffer = newbuffer;
    }

    public void append(final byte[] b, int off, int len) {
        if (b == null) {
            return;
        }
        if ((off < 0) || (off > b.length) || (len < 0) ||
                ((off + len) < 0) || ((off + len) > b.length)) {
            throw new IndexOutOfBoundsException();
        }
        if (len == 0) {
            return;
        }
        int newlen = this.len + len;
        if (newlen > this.buffer.length) {
            expand(newlen);
        }
        System.arraycopy(b, off, this.buffer, this.len, len);
        this.len = newlen;
    }

    public void append(int b) {
        int newlen = this.len + 1;
        if (newlen > this.buffer.length) {
            expand(newlen);
        }
        this.buffer[this.len] = (byte)b;
        this.len = newlen;
    }

    public void clear() {
        this.len = 0;
    }

    public byte[] toByteArray() {
        byte[] b = new byte[this.len];
        if (this.len > 0) {
            System.arraycopy(this.buffer, 0, b, 0, this.len);
        }
        return b;
    }

    public byte byteAt(int i) {
        if (i < 0 || i >= this.len)
            throw new IndexOutOfBoundsException();

        return this.buffer[i];
    }

    public int capacity() {
        return this.buffer.length;
    }

    public int length() {
        return this.len;
    }

    public byte[] buffer() {
        return this.buffer;
    }

    public int indexOf(byte b) {
        return indexOf(b, 0, this.len);
    }

    public int indexOf(byte b, int beginIndex, int endIndex) {
        if (beginIndex < 0) {
            beginIndex = 0;
        }
        if (endIndex > this.len) {
            endIndex = this.len;
        }
        if (beginIndex > endIndex) {
            return -1;
        }
        for (int i = beginIndex; i < endIndex; i++) {
            if (this.buffer[i] == b) {
                return i;
            }
        }
        return -1;
    }

    public void setLength(int len) {
        if (len < 0 || len > this.buffer.length) {
            throw new IndexOutOfBoundsException();
        }
        this.len = len;
    }

    public void remove(int off, int len) {
        if ((off < 0) || (off > this.len) || (len < 0) ||
                ((off + len) < 0) || ((off + len) > this.len)) {
            throw new IndexOutOfBoundsException();
        }
        if (len == 0) {
            return;
        }
        int remaining = this.len - off - len;
        if (remaining > 0) {
            System.arraycopy(this.buffer, off + len, this.buffer, off, remaining);
        }
        this.len -= len;
    }

    public boolean isEmpty() {
        return this.len == 0;
    }

    public boolean isFull() {
        return this.len == this.buffer.length;
    }

    @Override
    public String toString() {
        return new String(toByteArray());
    }

    public void release() {
        bufferRecycler.releaseByteBuffer(0, buffer);
    }

}
