package aQute.lib.io;

import java.io.*;

public class LimitedInputStream extends InputStream {

	final InputStream	in;
	final int			size;
	int					left;

	public LimitedInputStream(InputStream in, int size) {
		this.in = in;
		this.left = size;
		this.size = size;
	}

	@Override public int read() throws IOException {
		if (left <= 0) {
			eof();
			return -1;
		}

		left--;
		return in.read();
	}

	@Override public int available() throws IOException {		
		return Math.min(left, in.available());
	}

	@Override public void close() throws IOException {
		eof();
		in.close();
	}

	protected void eof() {
	}

	@Override public synchronized void mark(int readlimit) {
		throw new UnsupportedOperationException();
	}

	@Override public boolean markSupported() {
		return false;
	}

	@Override public int read(byte[] b, int off, int len) throws IOException {
		int min = Math.min(len, left);
		if (min == 0)
			return 0;

		int read = in.read(b, off, min);
		if (read > 0)
			left -= read;
		return read;
	}

	@Override public int read(byte[] b) throws IOException {
		return read(b,0,b.length);
	}

	@Override public synchronized void reset() throws IOException {
		throw new UnsupportedOperationException();
	}

	@Override public long skip(long n) throws IOException {
		long count = 0;
		byte buffer[] = new byte[1024];
		while ( n > 0 && read() >= 0) {
			int size = read(buffer);
			if ( size <= 0)
				return count;
			count+=size;
			n-=size;
		}
		return count;
	}
}