1 files changed, 87 insertions, 0 deletions
diff --git a/src/main/java/at/gv/egiz/moazs/process/SaveResponseToFileSink.java b/src/main/java/at/gv/egiz/moazs/process/SaveResponseToFileSink.java
new file mode 100644
index 0000000..09d8570
--- /dev/null
+++ b/src/main/java/at/gv/egiz/moazs/process/SaveResponseToFileSink.java
@@ -0,0 +1,87 @@
+package at.gv.egiz.moazs.process;
+
+import at.gv.egiz.moazs.repository.DeliveryRepository;
+import at.gv.egiz.moazs.scheme.Marshaller;
+import at.gv.egiz.moazs.scheme.MsgResponse;
+import org.apache.commons.io.FileUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.concurrent.CompletableFuture;
+
+import static at.gv.egiz.moazs.MoaZSException.moaZSException;
+import static java.lang.String.format;
+import static java.util.concurrent.CompletableFuture.allOf;
+import static java.util.concurrent.CompletableFuture.supplyAsync;
+
+@Component
+public class SaveResponseToFileSink implements MsgResponseSink {
+
+    private static final Logger log = LoggerFactory.getLogger(SaveResponseToFileSink.class);
+    private static final String SAVING_FAILED_MSG = "Could not save response with AppDeliveryId=%s.";
+
+    private final SimpleDateFormat isoFormatter;
+    private final Marshaller msgMarshaller;
+    private final DeliveryRepository repository;
+    private final String root;
+
+    @Autowired
+    public SaveResponseToFileSink(Marshaller msgMarshaller, DeliveryRepository repository, String root) {
+        this.msgMarshaller = msgMarshaller;
+        this.repository = repository;
+        this.root = root;
+        this.isoFormatter = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss");
+    }
+
+    @Override
+    public CompletableFuture<Void> handle(MsgResponse response) {
+
+        var responseID = response.getResponseID();
+
+        var responsePath = generatePath(responseID, "xml");
+        var storeResponseToFileSystemFuture = supplyAsync(() -> msgMarshaller.marshallXml(response.getResponse()))
+                .thenApply(responseString -> responseString.getBytes(StandardCharsets.UTF_8))
+                .thenAccept(responseByteArray -> storeToFile(responsePath, responseByteArray))
+                .exceptionally(ex -> logException(ex, responseID));
+
+        var binaryResponsePath = generatePath(responseID, "binary.xml");
+        var storeBinaryResponseToFileSystemFuture = supplyAsync(() -> repository.retrieveBinaryResponse(responseID).get())
+                .thenAccept(binaryResponseByteArray -> storeToFile(binaryResponsePath, binaryResponseByteArray))
+                .exceptionally(ex -> logException(ex, responseID));
+
+        return allOf(storeResponseToFileSystemFuture, storeBinaryResponseToFileSystemFuture);
+
+    }
+
+    private String generatePath(String id, String suffix) {
+        var folder = sanitizeFileString(id);
+        var nowInIso8601 = isoFormatter.format(new Date());
+        return format("%s/%s/%s.%s", root, folder, nowInIso8601,  suffix);
+    }
+
+    private String sanitizeFileString(String fileString) {
+        return fileString.replaceAll("[^a-zA-Z0-9\\._\\-]", "");
+    }
+
+    private Void logException(Throwable ex, String appDeliveryID) {
+        if(log.isErrorEnabled()) {
+            log.error(format(SAVING_FAILED_MSG, appDeliveryID), ex);
+        }
+        return null;
+    }
+
+    private void storeToFile(String path, byte[] content) {
+        try {
+            FileUtils.writeByteArrayToFile(new File(path), content);
+        } catch (IOException e) {
+            throw moaZSException(e.getMessage(), e);
+        }
+    }
+}