1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.maven.plugins.shade.resource;
20
21 import java.io.ByteArrayOutputStream;
22 import java.io.IOException;
23 import java.io.InputStream;
24 import java.util.HashMap;
25 import java.util.List;
26 import java.util.Map;
27 import java.util.jar.JarEntry;
28 import java.util.jar.JarOutputStream;
29 import java.util.regex.Pattern;
30
31 import org.apache.maven.plugins.shade.relocation.Relocator;
32 import org.codehaus.plexus.util.IOUtil;
33
34
35
36
37
38
39
40 public class ResourceBundleAppendingTransformer extends AbstractCompatibilityTransformer {
41 private Map<String, ByteArrayOutputStream> dataMap = new HashMap<>();
42
43 private Pattern resourceBundlePattern;
44
45 private long time = Long.MIN_VALUE;
46
47
48
49
50
51 public void setBasename(String basename) {
52 resourceBundlePattern = Pattern.compile(basename + "(_[a-zA-Z]+){0,3}\\.properties");
53 }
54
55 @Override
56 public boolean canTransformResource(String r) {
57 return resourceBundlePattern != null && resourceBundlePattern.matcher(r).matches();
58 }
59
60 @Override
61 public void processResource(String resource, InputStream is, List<Relocator> relocators, long time)
62 throws IOException {
63 ByteArrayOutputStream data = dataMap.get(resource);
64 if (data == null) {
65 data = new ByteArrayOutputStream();
66 dataMap.put(resource, data);
67 }
68
69 IOUtil.copy(is, data);
70 data.write('\n');
71
72 if (time > this.time) {
73 this.time = time;
74 }
75 }
76
77 @Override
78 public boolean hasTransformedResource() {
79 return !dataMap.isEmpty();
80 }
81
82 @Override
83 public void modifyOutputStream(JarOutputStream jos) throws IOException {
84 for (Map.Entry<String, ByteArrayOutputStream> dataEntry : dataMap.entrySet()) {
85 JarEntry jarEntry = new JarEntry(dataEntry.getKey());
86 jarEntry.setTime(time);
87 jos.putNextEntry(jarEntry);
88
89 jos.write(dataEntry.getValue().toByteArray());
90 dataEntry.getValue().reset();
91 }
92 }
93 }