Implement sticker pack retrieval

[signal-cli] / lib / src / main / java / org / asamk / signal / manager / StickerPackStore.java

diff --git a/lib/src/main/java/org/asamk/signal/manager/StickerPackStore.java b/lib/src/main/java/org/asamk/signal/manager/StickerPackStore.java
new file mode 100644 (file)
index 0000000..51ebece
--- /dev/null
+++ b/lib/src/main/java/org/asamk/signal/manager/StickerPackStore.java
@@ -0,0 +1,65 @@
+package org.asamk.signal.manager;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import org.asamk.signal.manager.storage.stickers.StickerPackId;
+import org.asamk.signal.manager.util.IOUtils;
+import org.whispersystems.signalservice.internal.util.Hex;
+
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.io.OutputStreamWriter;
+import java.nio.charset.StandardCharsets;
+
+public class StickerPackStore {
+
+    private final File stickersPath;
+
+    public StickerPackStore(final File stickersPath) {
+        this.stickersPath = stickersPath;
+    }
+
+    public boolean existsStickerPack(StickerPackId stickerPackId) {
+        return getStickerPackManifestFile(stickerPackId).exists();
+    }
+
+    public void storeManifest(StickerPackId stickerPackId, JsonStickerPack manifest) throws IOException {
+        try (OutputStream output = new FileOutputStream(getStickerPackManifestFile(stickerPackId))) {
+            try (var writer = new BufferedWriter(new OutputStreamWriter(output, StandardCharsets.UTF_8))) {
+                new ObjectMapper().writeValue(writer, manifest);
+            }
+        }
+    }
+
+    public void storeSticker(StickerPackId stickerPackId, int stickerId, StickerStorer storer) throws IOException {
+        createStickerPackDir(stickerPackId);
+        try (OutputStream output = new FileOutputStream(getStickerPackStickerFile(stickerPackId, stickerId))) {
+            storer.store(output);
+        }
+    }
+
+    private File getStickerPackManifestFile(StickerPackId stickerPackId) {
+        return new File(getStickerPackPath(stickerPackId), "manifest.json");
+    }
+
+    private File getStickerPackStickerFile(StickerPackId stickerPackId, int stickerId) {
+        return new File(getStickerPackPath(stickerPackId), String.valueOf(stickerId));
+    }
+
+    private File getStickerPackPath(StickerPackId stickerPackId) {
+        return new File(stickersPath, Hex.toStringCondensed(stickerPackId.serialize()));
+    }
+
+    private void createStickerPackDir(StickerPackId stickerPackId) throws IOException {
+        IOUtils.createPrivateDirectories(getStickerPackPath(stickerPackId));
+    }
+
+    @FunctionalInterface
+    public interface StickerStorer {
+
+        void store(OutputStream outputStream) throws IOException;
+    }
+}