1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19
20 package org.apache.commons.compress.compressors.pack200;
21
22 import java.io.IOException;
23 import java.io.OutputStream;
24 import java.util.Map;
25 import java.util.jar.JarInputStream;
26
27 import org.apache.commons.compress.compressors.CompressorOutputStream;
28 import org.apache.commons.compress.java.util.jar.Pack200;
29
30 /**
31 * An output stream that compresses using the Pack200 format.
32 *
33 * @NotThreadSafe
34 * @since 1.3
35 */
36 public class Pack200CompressorOutputStream extends CompressorOutputStream<OutputStream> {
37 private boolean finished;
38 private final AbstractStreamBridge abstractStreamBridge;
39 private final Map<String, String> properties;
40
41 /**
42 * Compresses the given stream, caching the compressed data in memory.
43 *
44 * @param out the stream to write to
45 * @throws IOException if writing fails
46 */
47 public Pack200CompressorOutputStream(final OutputStream out) throws IOException {
48 this(out, Pack200Strategy.IN_MEMORY);
49 }
50
51 /**
52 * Compresses the given stream, caching the compressed data in memory and using the given properties.
53 *
54 * @param out the stream to write to
55 * @param props Pack200 properties to use
56 * @throws IOException if writing fails
57 */
58 public Pack200CompressorOutputStream(final OutputStream out, final Map<String, String> props) throws IOException {
59 this(out, Pack200Strategy.IN_MEMORY, props);
60 }
61
62 /**
63 * Compresses the given stream using the given strategy to cache the results.
64 *
65 * @param out the stream to write to
66 * @param mode the strategy to use
67 * @throws IOException if writing fails
68 */
69 public Pack200CompressorOutputStream(final OutputStream out, final Pack200Strategy mode) throws IOException {
70 this(out, mode, null);
71 }
72
73 /**
74 * Compresses the given stream using the given strategy to cache the results and the given properties.
75 *
76 * @param out the stream to write to
77 * @param mode the strategy to use
78 * @param props Pack200 properties to use
79 * @throws IOException if writing fails
80 */
81 public Pack200CompressorOutputStream(final OutputStream out, final Pack200Strategy mode, final Map<String, String> props) throws IOException {
82 super(out);
83 abstractStreamBridge = mode.newStreamBridge();
84 properties = props;
85 }
86
87 @Override
88 public void close() throws IOException {
89 try {
90 finish();
91 } finally {
92 try {
93 abstractStreamBridge.stop();
94 } finally {
95 out.close();
96 }
97 }
98 }
99
100 public void finish() throws IOException {
101 if (!finished) {
102 finished = true;
103 final Pack200.Packer p = Pack200.newPacker();
104 if (properties != null) {
105 p.properties().putAll(properties);
106 }
107 try (JarInputStream ji = new JarInputStream(abstractStreamBridge.getInputStream())) {
108 p.pack(ji, out);
109 }
110 }
111 }
112
113 @Override
114 public void write(final byte[] b) throws IOException {
115 abstractStreamBridge.write(b);
116 }
117
118 @Override
119 public void write(final byte[] b, final int from, final int length) throws IOException {
120 abstractStreamBridge.write(b, from, length);
121 }
122
123 @Override
124 public void write(final int b) throws IOException {
125 abstractStreamBridge.write(b);
126 }
127 }