X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Fstreams%2FBufferedInputStream.java;h=336dba4000c54a26b25e33d0e84a2d8f46b2a01d;hb=dce410fec00030c0affde7b19bcde6de847fae13;hp=397f6fcf9b22f414631cddb899ab38a3b38f3fa1;hpb=f04d5e49e91832e122617fbbaa5cdb053459a7e7;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/utils/streams/BufferedInputStream.java b/src/be/nikiroo/utils/streams/BufferedInputStream.java
index 397f6fc..336dba4 100644
--- a/src/be/nikiroo/utils/streams/BufferedInputStream.java
+++ b/src/be/nikiroo/utils/streams/BufferedInputStream.java
@@ -4,6 +4,8 @@ import java.io.IOException;
 import java.io.InputStream;
 import java.util.Arrays;
 
+import be.nikiroo.utils.StringUtils;
+
 /**
  * A simple {@link InputStream} that is buffered with a bytes array.
  * <p>
@@ -14,13 +16,22 @@ import java.util.Arrays;
  * @author niki
  */
 public class BufferedInputStream extends InputStream {
+	/**
+	 * The size of the internal buffer (can be different if you pass your own
+	 * buffer, of course).
+	 * <p>
+	 * A second buffer of twice the size can sometimes be created as needed for
+	 * the {@link BufferedInputStream#startsWith(byte[])} search operation.
+	 */
+	static private final int BUFFER_SIZE = 4096;
+
 	/** The current position in the buffer. */
 	protected int start;
 	/** The index of the last usable position of the buffer. */
 	protected int stop;
 	/** The buffer itself. */
 	protected byte[] buffer;
-	/** An End-Of-File (or buffer, here) marker. */
+	/** An End-Of-File (or {@link InputStream}, here) marker. */
 	protected boolean eof;
 
 	private boolean closed;
@@ -45,7 +56,7 @@ public class BufferedInputStream extends InputStream {
 	public BufferedInputStream(InputStream in) {
 		this.in = in;
 
-		this.buffer = new byte[4096];
+		this.buffer = new byte[BUFFER_SIZE];
 		this.originalBuffer = this.buffer;
 		this.start = 0;
 		this.stop = 0;
@@ -94,6 +105,15 @@ public class BufferedInputStream extends InputStream {
 		this.stop = length;
 	}
 
+	/**
+	 * The internal buffer size (can be useful to know for search methods).
+	 * 
+	 * @return the size of the internal buffer, in bytes.
+	 */
+	public int getInternalBufferSize() {
+		return originalBuffer.length;
+	}
+
 	/**
 	 * Return this very same {@link BufferedInputStream}, but keep a counter of
 	 * how many streams were open this way. When calling
@@ -116,6 +136,50 @@ public class BufferedInputStream extends InputStream {
 		return this;
 	}
 
+	/**
+	 * Check if the current content (until eof) is equal to the given search
+	 * term.
+	 * <p>
+	 * Note: the search term size <b>must</b> be smaller or equal the internal
+	 * buffer size.
+	 * 
+	 * @param search
+	 *            the term to search for
+	 * 
+	 * @return TRUE if the content that will be read starts with it
+	 * 
+	 * @throws IOException
+	 *             in case of I/O error or if the size of the search term is
+	 *             greater than the internal buffer
+	 */
+	public boolean is(String search) throws IOException {
+		return is(StringUtils.getBytes(search));
+	}
+
+	/**
+	 * Check if the current content (until eof) is equal to the given search
+	 * term.
+	 * <p>
+	 * Note: the search term size <b>must</b> be smaller or equal the internal
+	 * buffer size.
+	 * 
+	 * @param search
+	 *            the term to search for
+	 * 
+	 * @return TRUE if the content that will be read starts with it
+	 * 
+	 * @throws IOException
+	 *             in case of I/O error or if the size of the search term is
+	 *             greater than the internal buffer
+	 */
+	public boolean is(byte[] search) throws IOException {
+		if (startsWith(search)) {
+			return stop == search.length;
+		}
+
+		return false;
+	}
+
 	/**
 	 * Check if the current content (what will be read next) starts with the
 	 * given search term.
@@ -133,7 +197,7 @@ public class BufferedInputStream extends InputStream {
 	 *             greater than the internal buffer
 	 */
 	public boolean startsWiths(String search) throws IOException {
-		return startsWith(search.getBytes("UTF-8"));
+		return startsWith(StringUtils.getBytes(search));
 	}
 
 	/**