1
22
23 package com.liferay.portal.kernel.zip;
24
25 import com.liferay.portal.kernel.util.ByteArrayMaker;
26 import com.liferay.portal.kernel.util.StringPool;
27
28 import java.io.BufferedInputStream;
29 import java.io.BufferedOutputStream;
30 import java.io.ByteArrayInputStream;
31 import java.io.IOException;
32 import java.io.Serializable;
33
34 import java.util.zip.ZipEntry;
35 import java.util.zip.ZipOutputStream;
36
37
43 public class ZipWriter implements Serializable {
44
45 public ZipWriter() {
46 _bam = new ByteArrayMaker();
47 _zos = new ZipOutputStream(new BufferedOutputStream(_bam));
48 }
49
50 public void addEntry(String name, StringBuilder sb) throws IOException {
51 addEntry(name, sb.toString());
52 }
53
54 public void addEntry(String name, String s) throws IOException {
55 addEntry(name, s.getBytes());
56 }
57
58 public void addEntry(String name, byte[] bytes) throws IOException {
59 if (name.startsWith(StringPool.SLASH)) {
60 name = name.substring(1);
61 }
62
63 ZipEntry entry = new ZipEntry(name);
64
65 _zos.putNextEntry(entry);
66
67 BufferedInputStream bis = new BufferedInputStream(
68 new ByteArrayInputStream(bytes), _BUFFER);
69
70 int count;
71
72 while ((count = bis.read(_data, 0, _BUFFER)) != -1) {
73 _zos.write(_data, 0, count);
74 }
75
76 bis.close();
77 }
78
79 public byte[] finish() throws IOException {
80 _zos.close();
81
82 return _bam.toByteArray();
83 }
84
85 private static final int _BUFFER = 2048;
86
87 private ByteArrayMaker _bam;
88 private ZipOutputStream _zos;
89 private byte[] _data = new byte[_BUFFER];
90
91 }