buildSrc/src/main/java/cc/squirreljme/plugin/multivm/NanoCoatBuiltInTaskAction.java

   1 // -*- Mode: Java; indent-tabs-mode: t; tab-width: 4 -*-
   2 // ---------------------------------------------------------------------------
   3 // Multi-Phasic Applications: SquirrelJME
   4 //     Copyright (C) Stephanie Gawroriski <xer@multiphasicapps.net>
   5 // ---------------------------------------------------------------------------
   6 // SquirrelJME is under the Mozilla Public License Version 2.0.
   7 // See license.mkd for licensing and copyright information.
   8 // ---------------------------------------------------------------------------
   9
  10 package cc.squirreljme.plugin.multivm;
  11
  12 import cc.squirreljme.plugin.multivm.ident.SourceTargetClassifier;
  13 import java.io.IOException;
  14 import java.io.InputStream;
  15 import java.io.OutputStream;
  16 import java.nio.file.Files;
  17 import java.nio.file.Path;
  18 import java.nio.file.StandardCopyOption;
  19 import java.nio.file.StandardOpenOption;
  20 import java.util.regex.Pattern;
  21 import java.util.zip.ZipEntry;
  22 import java.util.zip.ZipInputStream;
  23 import org.apache.tools.ant.util.StreamUtils;
  24 import org.gradle.api.Action;
  25 import org.gradle.api.Task;
  26
  27 /**
  28  * Copies over the source file for NanoCoat.
  29  *
  30  * @since 2023/05/31
  31  */
  32 public class NanoCoatBuiltInTaskAction
  33         implements Action<Task>
  34 {
  35         /** The classifier used. */
  36         protected final SourceTargetClassifier classifier;
  37
  38         /**
  39          * Initializes the task.
  40          *
  41          * @param __classifier The classifier used.
  42          * @throws NullPointerException On null arguments.
  43          * @since 2023/05/31
  44          */
  45         public NanoCoatBuiltInTaskAction(SourceTargetClassifier __classifier)
  46                 throws NullPointerException
  47         {
  48                 if (__classifier == null)
  49                         throw new NullPointerException("NARG");
  50
  51                 this.classifier = __classifier;
  52         }
  53
  54         /**
  55          * {@inheritDoc}
  56          * @since 2023/05/31
  57          */
  58         @Override
  59         public void execute(Task __task)
  60         {
  61                 NanoCoatBuiltInTask task = (NanoCoatBuiltInTask)__task;
  62
  63                 // Where is the ROM going?
  64                 Path input = __task.getInputs().getFiles().getSingleFile().toPath();
  65
  66                 // Shared file output
  67                 Path romOutput = task.romBasePath().get();
  68                 Path moduleOutput = task.specificPath().get();
  69                 Path sharedOutput = task.sharedPath().get();
  70
  71                 // This could fail to write
  72                 Path sourceTemp = null;
  73                 try
  74                 {
  75                         // We will be writing to a file then moving it over, so we only
  76                         // need a single temporary file for everything
  77                         sourceTemp = Files.createTempFile("source", ".x");
  78
  79                         // Make sure the target directories exist first, since we just
  80                         // deleted them
  81                         Files.createDirectories(moduleOutput);
  82                         Files.createDirectories(sharedOutput);
  83
  84                         // The ROM is just a ZIP of sources which get copied over
  85                         try (InputStream in = Files.newInputStream(input,
  86                                         StandardOpenOption.READ);
  87                                 ZipInputStream zip = new ZipInputStream(in))
  88                         {
  89                                 for (;;)
  90                                 {
  91                                         // Load in next entry
  92                                         ZipEntry entry = zip.getNextEntry();
  93                                         if (entry == null)
  94                                                 break;
  95
  96                                         // Ignore directories
  97                                         if (entry.isDirectory())
  98                                                 continue;
  99
 100                                         // Dump all the input into the output
 101                                         try (OutputStream out = Files.newOutputStream(
 102                                                 sourceTemp, StandardOpenOption.TRUNCATE_EXISTING,
 103                                                 StandardOpenOption.WRITE,
 104                                                 StandardOpenOption.CREATE))
 105                                         {
 106                                                 VMHelpers.copy(zip, out);
 107                                         }
 108
 109                                         // Determine the target output file
 110                                         Path target = romOutput.resolve(
 111                                                 VMHelpers.stringToPath(entry.getName()));
 112
 113                                         // Make sure the parent directories exist
 114                                         Files.createDirectories(target.getParent());
 115
 116                                         // Move over to target, replace if it exists since
 117                                         // we want to update it
 118                                         Files.move(sourceTemp, target,
 119                                                 StandardCopyOption.REPLACE_EXISTING);
 120                                 }
 121                         }
 122
 123                         // Cleanup after this
 124                         finally
 125                         {
 126                                 if (sourceTemp != null)
 127                                         try
 128                                         {
 129                                                 Files.delete(sourceTemp);
 130                                         }
 131                                         catch (IOException ignored)
 132                                         {
 133                                         }
 134                         }
 135                 }
 136
 137                 // It did fail to write
 138                 catch (IOException e)
 139                 {
 140                         throw new RuntimeException("Could not extract ROM.", e);
 141                 }
 142         }
 143 }