Zip downloaded algorithms automatically

Author: ctmay4

src/test/java/com/imsweb/staging/cs/CsUpdateFromAPI.java

@@ -1,6 +1,7 @@
 package com.imsweb.staging.cs;
 
 import java.io.IOException;
+import java.nio.file.Paths;
 import java.util.Collections;
 import java.util.HashSet;
 
@@ -12,7 +13,7 @@
 public class CsUpdateFromAPI {
 
     public static void main(String[] args) throws IOException {
-        UpdaterUtils.update("cs", "02.05.50", new HashSet<>(Collections.singletonList("STAGING")));
+        UpdaterUtils.update("cs", "02.05.50", new HashSet<>(Collections.singletonList("STAGING")), Paths.get("c:/dev/tmp"));
     }
 
 }

src/test/java/com/imsweb/staging/pediatric/PediatricUpdateFromAPI.java

@@ -1,6 +1,7 @@
 package com.imsweb.staging.pediatric;
 
 import java.io.IOException;
+import java.nio.file.Paths;
 import java.util.Collections;
 import java.util.HashSet;
 
@@ -12,7 +13,7 @@
 public class PediatricUpdateFromAPI {
 
     public static void main(String[] args) throws IOException {
-        UpdaterUtils.update("pediatric", "1.3", new HashSet<>(Collections.singletonList("STAGING")));
+        UpdaterUtils.update("pediatric", "1.3", new HashSet<>(Collections.singletonList("STAGING")), Paths.get("c:/dev/tmp"));
     }
 
 }

src/test/java/com/imsweb/staging/tnm/TnmUpdateFromAPI.java

@@ -1,6 +1,7 @@
 package com.imsweb.staging.tnm;
 
 import java.io.IOException;
+import java.nio.file.Paths;
 import java.util.Collections;
 import java.util.HashSet;
 
@@ -12,7 +13,7 @@
 public class TnmUpdateFromAPI {
 
     public static void main(String[] args) throws IOException {
-        UpdaterUtils.update("tnm", "2.0", new HashSet<>(Collections.singletonList("STAGING")));
+        UpdaterUtils.update("tnm", "2.0", new HashSet<>(Collections.singletonList("STAGING")), Paths.get("c:/dev/tmp"));
     }
 
 }

src/test/java/com/imsweb/staging/updater/UpdaterUtils.java

@@ -2,6 +2,7 @@
 
 import java.io.File;
 import java.io.IOException;
+import java.io.UncheckedIOException;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Path;
@@ -16,6 +17,9 @@
 import java.util.Set;
 import java.util.regex.Pattern;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -202,6 +206,15 @@ else if (unusedTableIds.contains(id) && !id.startsWith("conversion_"))
         Files.write(Paths.get(glossaryDir + "/terms.txt"), keywords, StandardCharsets.UTF_8);
         _LOG.info("Saved glossary terms to {}/terms.txt", schemaDir);
 
+        // Zip up schemas/, tables/, glossary/
+        Path versionDir = Paths.get(baseDirectory);
+        zipAlgorithm(algorithm, version, versionDir);
+
+        _LOG.info("Created ZIP: {}/{}-{}.zip",
+                versionDir.getParent().getParent(),
+                algorithm,
+                version);
+
         stopwatch.stop();
         _LOG.info("Completed in {}", stopwatch);
     }
@@ -230,4 +243,36 @@ private static int purgeDirectory(File dir) {
         return count;
     }
 
+    /**
+     * Zip the algorithm
+     */
+    private static void zipAlgorithm(String algorithm, String version, Path versionDir) throws IOException {
+        Path algorithmsDir = versionDir.getParent().getParent();   // .../algorithms
+        Path zipPath = algorithmsDir.resolve(algorithm + "-" + version + ".zip");
+
+        try (ZipOutputStream zos = new ZipOutputStream(Files.newOutputStream(zipPath))) {
+            for (String folder : new String[] {"schemas", "tables", "glossary"}) {
+                Path folderPath = versionDir.resolve(folder);
+                if (!Files.exists(folderPath))
+                    continue;
+
+                try (Stream<Path> walk = Files.walk(folderPath)) {
+                    walk.filter(Files::isRegularFile)
+                            .forEach(path -> {
+                                Path relative = versionDir.relativize(path);
+                                ZipEntry entry = new ZipEntry(relative.toString().replace('\\', '/'));
+                                try {
+                                    zos.putNextEntry(entry);
+                                    Files.copy(path, zos);
+                                    zos.closeEntry();
+                                }
+                                catch (IOException e) {
+                                    throw new UncheckedIOException(e);
+                                }
+                            });
+                }
+            }
+        }
+    }
+
 }