fix/perf base64/serial
[nikiroo-utils.git] / src / be / nikiroo / utils / StringUtils.java
index 79bccec76f26ba9148c3b7658e219e863ec7ac65..ef3d84b921aea488f5ce17e5b8b52334cf24a375 100644 (file)
@@ -1,7 +1,8 @@
 package be.nikiroo.utils;
 
-import java.io.ByteArrayInputStream;
 import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
 import java.io.UnsupportedEncodingException;
 import java.security.MessageDigest;
 import java.security.NoSuchAlgorithmException;
@@ -14,7 +15,6 @@ import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
 import java.util.Map.Entry;
-import java.util.Scanner;
 import java.util.regex.Pattern;
 
 import org.unbescape.html.HtmlEscape;
@@ -523,7 +523,7 @@ public class StringUtils {
        @Deprecated
        public static String zip64(String data) {
                try {
-                       return Base64.encodeBytes(data.getBytes(), Base64.GZIP);
+                       return Base64.encodeBytes(data.getBytes("UTF-8"), Base64.GZIP);
                } catch (IOException e) {
                        e.printStackTrace();
                        return null;
@@ -546,16 +546,26 @@ public class StringUtils {
         */
        @Deprecated
        public static String unzip64(String data) throws IOException {
-               ByteArrayInputStream in = new ByteArrayInputStream(Base64.decode(data,
-                               Base64.GZIP));
+               return new String(Base64.decode(data, Base64.GZIP), "UTF-8");
+       }
 
-               Scanner scan = new Scanner(in);
-               scan.useDelimiter("\\A");
-               try {
-                       return scan.next();
-               } finally {
-                       scan.close();
-               }
+       /**
+        * Convert the given data to Base64 format.
+        * 
+        * @param data
+        *            the data to convert
+        * @param zip
+        *            TRUE to also compress the data in GZIP format; remember that
+        *            compressed and not-compressed content are different; you need
+        *            to know which is which when decoding
+        * 
+        * @return the Base64 {@link String} representation of the data
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static String base64(String data, boolean zip) throws IOException {
+               return base64(data.getBytes("UTF-8"), zip);
        }
 
        /**
@@ -578,14 +588,73 @@ public class StringUtils {
        }
 
        /**
-        * Unonvert the given data from Base64 format back to a raw array of bytes.
+        * Convert the given data to Base64 format.
         * 
         * @param data
-        *            the data to unconvert
+        *            the data to convert
         * @param zip
         *            TRUE to also uncompress the data from a GZIP format; take care
         *            about this flag, as it could easily cause errors in the
         *            returned content or an {@link IOException}
+        * @param breakLines
+        *            TRUE to break lines on every 76th character
+        * 
+        * @return the Base64 {@link String} representation of the data
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static OutputStream base64(OutputStream data, boolean zip,
+                       boolean breakLines) throws IOException {
+               OutputStream out = new Base64.OutputStream(data,
+                               breakLines ? Base64.DO_BREAK_LINES & Base64.ENCODE
+                                               : Base64.ENCODE);
+
+               if (zip) {
+                       out = new java.util.zip.GZIPOutputStream(out);
+               }
+
+               return out;
+       }
+
+       /**
+        * Convert the given data to Base64 format.
+        * 
+        * @param data
+        *            the data to convert
+        * @param zip
+        *            TRUE to also uncompress the data from a GZIP format; take care
+        *            about this flag, as it could easily cause errors in the
+        *            returned content or an {@link IOException}
+        * @param breakLines
+        *            TRUE to break lines on every 76th character
+        * 
+        * @return the Base64 {@link String} representation of the data
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static InputStream base64(InputStream data, boolean zip,
+                       boolean breakLines) throws IOException {
+               if (zip) {
+                       data = new java.util.zip.GZIPInputStream(data);
+               }
+
+               return new Base64.InputStream(data, breakLines ? Base64.DO_BREAK_LINES
+                               & Base64.ENCODE : Base64.ENCODE);
+       }
+
+       /**
+        * Unconvert the given data from Base64 format back to a raw array of bytes.
+        * <p>
+        * Will automatically detect zipped data and also uncompress it before
+        * returning, unless ZIP is false.
+        * 
+        * @param data
+        *            the data to unconvert
+        * @param zip
+        *            TRUE to also uncompress the data from a GZIP format
+        *            automatically; if set to FALSE, zipped data can be returned
         * 
         * @return the raw data represented by the given Base64 {@link String},
         *         optionally compressed with GZIP
@@ -594,11 +663,38 @@ public class StringUtils {
         *             in case of I/O errors
         */
        public static byte[] unbase64(String data, boolean zip) throws IOException {
-               return Base64.decode(data, zip ? Base64.GZIP : Base64.NO_OPTIONS);
+               return Base64
+                               .decode(data, zip ? Base64.NO_OPTIONS : Base64.DONT_GUNZIP);
        }
 
        /**
-        * Unonvert the given data from Base64 format back to a {@link String}.
+        * Unconvert the given data from Base64 format back to a raw array of bytes.
+        * 
+        * @param data
+        *            the data to unconvert
+        * @param zip
+        *            TRUE to also uncompress the data from a GZIP format; take care
+        *            about this flag, as it could easily cause errors in the
+        *            returned content or an {@link IOException}
+        * 
+        * @return the raw data represented by the given Base64 {@link String}
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static OutputStream unbase64(OutputStream data, boolean zip)
+                       throws IOException {
+               OutputStream out = new Base64.OutputStream(data, Base64.DECODE);
+
+               if (zip) {
+                       out = new java.util.zip.GZIPOutputStream(out);
+               }
+
+               return out;
+       }
+
+       /**
+        * Unconvert the given data from Base64 format back to a raw array of bytes.
         * 
         * @param data
         *            the data to unconvert
@@ -607,6 +703,61 @@ public class StringUtils {
         *            about this flag, as it could easily cause errors in the
         *            returned content or an {@link IOException}
         * 
+        * @return the raw data represented by the given Base64 {@link String}
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static InputStream unbase64(InputStream data, boolean zip)
+                       throws IOException {
+               if (zip) {
+                       data = new java.util.zip.GZIPInputStream(data);
+               }
+
+               return new Base64.InputStream(data, Base64.DECODE);
+       }
+
+       /**
+        * Unconvert the given data from Base64 format back to a raw array of bytes.
+        * <p>
+        * Will automatically detect zipped data and also uncompress it before
+        * returning, unless ZIP is false.
+        * 
+        * @param data
+        *            the data to unconvert
+        * @param offset
+        *            the offset at which to start taking the data (do not take the
+        *            data before it into account)
+        * @param count
+        *            the number of bytes to take into account (do not process after
+        *            this number of bytes has been processed)
+        * @param zip
+        *            TRUE to also uncompress the data from a GZIP format
+        *            automatically; if set to FALSE, zipped data can be returned
+        * 
+        * @return the raw data represented by the given Base64 {@link String}
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static byte[] unbase64(byte[] data, int offset, int count,
+                       boolean zip) throws IOException {
+               return Base64.niki_decode(data, offset, count, zip ? Base64.NO_OPTIONS
+                               : Base64.DONT_GUNZIP);
+       }
+
+       /**
+        * Unonvert the given data from Base64 format back to a {@link String}.
+        * <p>
+        * Will automatically detect zipped data and also uncompress it before
+        * returning, unless ZIP is false.
+        * 
+        * @param data
+        *            the data to unconvert
+        * @param zip
+        *            TRUE to also uncompress the data from a GZIP format
+        *            automatically; if set to FALSE, zipped data can be returned
+        * 
         * @return the {@link String} represented by the given Base64 {@link String}
         *         , optionally compressed with GZIP
         * 
@@ -614,15 +765,34 @@ public class StringUtils {
         *             in case of I/O errors
         */
        public static String unbase64s(String data, boolean zip) throws IOException {
-               ByteArrayInputStream in = new ByteArrayInputStream(unbase64(data, zip));
+               return new String(unbase64(data, zip), "UTF-8");
+       }
 
-               Scanner scan = new Scanner(in);
-               scan.useDelimiter("\\A");
-               try {
-                       return scan.next();
-               } finally {
-                       scan.close();
-               }
+       /**
+        * Unconvert the given data from Base64 format back into a {@link String}.
+        * 
+        * @param data
+        *            the data to unconvert
+        * @param offset
+        *            the offset at which to start taking the data (do not take the
+        *            data before it into account)
+        * @param count
+        *            the number of bytes to take into account (do not process after
+        *            this number of bytes has been processed)
+        * @param zip
+        *            TRUE to also uncompress the data from a GZIP format; take care
+        *            about this flag, as it could easily cause errors in the
+        *            returned content or an {@link IOException}
+        * 
+        * @return the {@link String} represented by the given Base64 {@link String}
+        *         , optionally compressed with GZIP
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        */
+       public static String unbase64s(byte[] data, int offset, int count,
+                       boolean zip) throws IOException {
+               return new String(unbase64(data, offset, count, zip), "UTF-8");
        }
 
        /**