X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Fserial%2FImporter.java;h=a017fabfd362ea3ad27f72e0a6305856db2e41a1;hb=564bbbdbc349805ff7fc1142facbedae87e5dc0e;hp=e7285c42ca329c7a6e84c73fba0189c228493781;hpb=f4053377fa15da2f11e82955bfab86e673fa371c;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/utils/serial/Importer.java b/src/be/nikiroo/utils/serial/Importer.java
index e7285c4..a017fab 100644
--- a/src/be/nikiroo/utils/serial/Importer.java
+++ b/src/be/nikiroo/utils/serial/Importer.java
@@ -1,11 +1,14 @@
 package be.nikiroo.utils.serial;
 
 import java.io.IOException;
+import java.io.InputStream;
 import java.lang.reflect.Field;
 import java.util.HashMap;
 import java.util.Map;
-import java.util.Scanner;
 
+import be.nikiroo.utils.IOUtils;
+import be.nikiroo.utils.NextableInputStream;
+import be.nikiroo.utils.NextableInputStreamStep;
 import be.nikiroo.utils.StringUtils;
 
 /**
@@ -58,31 +61,42 @@ public class Importer {
 	 *             if a class described in the serialised data cannot be found
 	 * @throws IOException
 	 *             if the content cannot be read (for instance, corrupt data)
+	 * @throws NullPointerException
+	 *             if the stream is empty
 	 */
-	public Importer read(String data) throws NoSuchFieldException,
-			NoSuchMethodException, ClassNotFoundException, IOException {
+	public Importer read(InputStream in) throws NoSuchFieldException,
+			NoSuchMethodException, ClassNotFoundException, IOException,
+			NullPointerException {
+
+		NextableInputStream stream = new NextableInputStream(in,
+				new NextableInputStreamStep('\n'));
 
-		Scanner scan = new Scanner(data);
 		try {
-			scan.useDelimiter("\n");
-			while (scan.hasNext()) {
-				String line = scan.next();
+			if (in == null || stream.eof()) {
+				if (in == null) {
+					throw new NullPointerException("InputStream is null");
+				}
+				throw new NullPointerException("InputStream is empty");
+			}
 
-				if (line.startsWith("ZIP:")) {
+			while (stream.next()) {
+				boolean zip = stream.startsWiths("ZIP:");
+				boolean b64 = stream.startsWiths("B64:");
+
+				if (zip || b64) {
+					InputStream decoded = StringUtils.unbase64(stream.open(),
+							zip);
 					try {
-						line = StringUtils.unzip64(line.substring("ZIP:"
-								.length()));
-					} catch (IOException e) {
-						throw new IOException(
-								"Internal error when decoding ZIP content: input may be corrupt");
+						read(decoded);
+					} finally {
+						decoded.close();
 					}
-					read(line);
 				} else {
-					processLine(line);
+					processLine(stream);
 				}
 			}
 		} finally {
-			scan.close();
+			stream.close(false);
 		}
 
 		return this;
@@ -108,11 +122,12 @@ public class Importer {
 	 * @throws IOException
 	 *             if the content cannot be read (for instance, corrupt data)
 	 */
-	private boolean processLine(String line) throws NoSuchFieldException,
+	private boolean processLine(InputStream in) throws NoSuchFieldException,
 			NoSuchMethodException, ClassNotFoundException, IOException {
+
 		// Defer to latest child if any
 		if (child != null) {
-			if (child.processLine(line)) {
+			if (child.processLine(in)) {
 				if (currentFieldName != null) {
 					setField(currentFieldName, child.getValue());
 					currentFieldName = null;
@@ -123,6 +138,9 @@ public class Importer {
 			return false;
 		}
 
+		// TODO use the stream, Luke
+		String line = IOUtils.readSmallStream(in);
+
 		if (line.equals("{")) { // START: new child if needed
 			if (link != null) {
 				child = new Importer(map);