#107 Extract entries by prefix exclude prefix path

Author: oleg-cherednik

src/main/java/ru/olegcherednik/zip4jvm/engine/unzip/UnzipExtractEngine.java

@@ -37,6 +37,7 @@
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.io.FilenameUtils;
 import org.apache.commons.io.IOUtils;
+import org.apache.commons.lang3.StringUtils;
 
 import java.io.IOException;
 import java.io.OutputStream;
@@ -86,10 +87,10 @@ protected Map<String, String> getEntriesByPrefix(Set<String> fileNames) {
             if (zipModel.hasEntry(entryName)) {
                 ZipEntry zipEntry = zipModel.getZipEntryByFileName(entryName);
                 map.put(entryName, FilenameUtils.getName(zipEntry.getFileName()));
-            } else {
-                for (ZipEntry zipEntry : getEntriesByPrefix(entryName + '/'))
-                    map.put(zipEntry.getFileName(), zipEntry.getFileName());
             }
+
+            for (ZipEntry zipEntry : getEntriesByPrefix(entryName + '/'))
+                map.put(zipEntry.getFileName(), StringUtils.substring(zipEntry.getFileName(), fileName.length() + 1));
         }
 
         return map.isEmpty() ? Collections.emptyMap() : Collections.unmodifiableMap(map);

src/test/java/ru/olegcherednik/zip4jvm/UnzipItSolidTest.java

@@ -86,9 +86,7 @@ public void shouldUnzipOneFileIgnorePath() throws IOException {
     public void shouldUnzipFolder() throws IOException {
         Path dstDir = Zip4jvmSuite.subDirNameAsMethodNameWithTime(rootDir);
         UnzipIt.zip(zipDeflateSolid).dstDir(dstDir).extract(dirNameBikes);
-
-        assertThatDirectory(dstDir).exists().hasEntries(1).hasDirectories(1);
-        assertThatDirectory(dstDir.resolve(dirNameBikes)).matches(dirBikesAssert);
+        assertThatDirectory(dstDir).matches(dirBikesAssert);
     }
 
     public void shouldExtractZipArchiveWhenEntryNameWithCustomCharset() throws IOException {

src/test/java/ru/olegcherednik/zip4jvm/compression/CompressionDeflateTest.java

@@ -132,8 +132,7 @@ public void shouldUnzipWhenWhenDeflateCompressionAndPkwareEncryption() throws IO
         Path dstDir = Zip4jvmSuite.subDirNameAsMethodName(rootDir);
 
         UnzipIt.zip(zipDeflateSolidPkware).dstDir(dstDir).password(password).extract(dirNameCars);
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
     public void shouldUnzipWhenWhenDeflateCompressionAndAesEncryption() throws IOException {
@@ -144,8 +143,7 @@ public void shouldUnzipWhenWhenDeflateCompressionAndAesEncryption() throws IOExc
                                               .build();
 
         UnzipIt.zip(zipDeflateSolidAes).dstDir(dstDir).settings(settings).extract(dirNameCars);
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
     public void shouldUseCompressStoreWhenFileEmpty() throws IOException {

src/test/java/ru/olegcherednik/zip4jvm/compression/CompressionStoreTest.java

@@ -123,8 +123,7 @@ public void shouldUnzipWhenWhenStoreCompressionAndPkwareEncryption() throws IOEx
         Path dstDir = Zip4jvmSuite.subDirNameAsMethodName(rootDir);
 
         UnzipIt.zip(zipStoreSolidPkware).dstDir(dstDir).password(password).extract(dirNameCars);
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
     public void shouldUnzipWhenWhenStoreCompressionAndAesEncryption() throws IOException {
@@ -133,8 +132,7 @@ public void shouldUnzipWhenWhenStoreCompressionAndAesEncryption() throws IOExcep
         UnzipSettings settings = UnzipSettings.builder().passwordProvider(fileNamePasswordProvider).build();
 
         UnzipIt.zip(zipStoreSolidAes).dstDir(dstDir).settings(settings).extract(dirNameCars);
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
 }

src/test/java/ru/olegcherednik/zip4jvm/engine/UnzipEngineTest.java

@@ -70,29 +70,23 @@ public void shouldUnzipZipFileIntoDestinationFolderWhenDeflateSolid() throws IOE
         Path dstDir = Zip4jvmSuite.subDirNameAsMethodName(rootDir);
 
         UnzipIt.zip(zipDeflateSolid).dstDir(dstDir).extract(dirNameCars);
-
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
     public void shouldUnzipZipFileIntoDestinationFolderWhenDeflateSolidPkware() throws IOException {
         Path dstDir = Zip4jvmSuite.subDirNameAsMethodName(rootDir);
         UnzipSettings settings = UnzipSettings.builder().password(password).build();
 
         UnzipIt.zip(zipDeflateSolid).settings(settings).dstDir(dstDir).extract(dirNameCars);
-
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
     public void shouldUnzipZipFileIntoDestinationFolderWhenDeflateSolidAes() throws IOException {
         Path dstDir = Zip4jvmSuite.subDirNameAsMethodName(rootDir);
         UnzipSettings settings = UnzipSettings.builder().passwordProvider(fileNamePasswordProvider).build();
 
         UnzipIt.zip(zipDeflateSolid).settings(settings).dstDir(dstDir).extract(dirNameCars);
-
-        assertThatDirectory(dstDir).exists().hasDirectories(1).hasRegularFiles(0);
-        assertThatDirectory(dstDir.resolve(dirNameCars)).matches(dirCarsAssert);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
     }
 
     public void shouldCorrectlySetLastTimeStampWhenUnzip() throws IOException, ParseException {
@@ -113,6 +107,13 @@ public void shouldCorrectlySetLastTimeStampWhenUnzip() throws IOException, Parse
         assertThat(convert(Files.getLastModifiedTime(fileFooUnzip).toMillis())).isEqualTo(str);
     }
 
+    public void shouldUnzipZipFileIntoDestinationFolderWhenStoreSolid() throws IOException {
+        Path dstDir = Zip4jvmSuite.subDirNameAsMethodName(rootDir);
+
+        UnzipIt.zip(zipDeflateSolid).dstDir(dstDir).extract(dirNameCars);
+        assertThatDirectory(dstDir).matches(dirCarsAssert);
+    }
+
     private static long convert(String str) throws ParseException {
         return new SimpleDateFormat("yyyy.MM.dd'T'HH:mm:ss", Locale.ENGLISH).parse(str).getTime();
     }