package org.jruby.runtime.load;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;

Author:
nicksieger/**
 *
 * @author nicksieger
 */
public class LoadServiceResourceInputStream extends ByteArrayInputStream {

    private static final byte[] NULL_BYTE_ARRAY = new byte[0];

    
Construct a LoadServiceResourceInputStream from the given bytes.
Params:
bytes – the bytes to wrap in this stream/**
     * Construct a LoadServiceResourceInputStream from the given bytes.
     *
     * @param bytes the bytes to wrap in this stream
     */
    public LoadServiceResourceInputStream(byte[] bytes) {
        super(bytes);
    }

    
Construct a new LoadServiceInputStream by reading all bytes from the
specified stream.
You are responsible for the lifecycle of the given stream after this
constructor has been called (i.e. it will not be closed for you).
Params:
stream – the stream from which to read bytes
Throws:
IOException – if the reading causes an IOException/**
     * Construct a new LoadServiceInputStream by reading all bytes from the
     * specified stream.
     *
     * You are responsible for the lifecycle of the given stream after this
     * constructor has been called (i.e. it will not be closed for you).
     *
     * @param stream the stream from which to read bytes
     * @throws IOException if the reading causes an IOException
     */
    public LoadServiceResourceInputStream(InputStream stream) throws IOException {
        super(NULL_BYTE_ARRAY);
        bufferEntireStream(stream);
    }

    public byte[] getBytes() {
        if (buf.length != count) {
            byte[] b = new byte[count];
            System.arraycopy(buf, 0, b, 0, count);
            return b;
        }
        return buf;
    }

    private static final int READ_CHUNK_SIZE = 16384;

    private void bufferEntireStream(InputStream stream) throws IOException {
        byte[] chunk = new byte[READ_CHUNK_SIZE];
        int bytesRead;
        while ((bytesRead = stream.read(chunk)) != -1) {
            byte[] newbuf = new byte[buf.length + bytesRead];
            System.arraycopy(buf, 0, newbuf, 0, buf.length);
            System.arraycopy(chunk, 0, newbuf, buf.length, bytesRead);
            buf = newbuf;
            count = buf.length;
        }
    }
}