nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/storage/stickers/StickerStore.java

   1 package org.asamk.signal.manager.storage.stickers;
   2
   3 import java.util.Base64;
   4 import java.util.HashMap;
   5 import java.util.HashSet;
   6 import java.util.List;
   7 import java.util.Map;
   8 import java.util.Objects;
   9 import java.util.stream.Collectors;
  10
  11 public class StickerStore {
  12
  13     private final Map<StickerPackId, Sticker> stickers;
  14
  15     private final Saver saver;
  16
  17     public StickerStore(final Saver saver) {
  18         this.saver = saver;
  19         stickers = new HashMap<>();
  20     }
  21
  22     public StickerStore(final Map<StickerPackId, Sticker> stickers, final Saver saver) {
  23         this.stickers = stickers;
  24         this.saver = saver;
  25     }
  26
  27     public static StickerStore fromStorage(Storage storage, Saver saver) {
  28         final var packIds = new HashSet<StickerPackId>();
  29         final var stickers = storage.stickers.stream().map(s -> {
  30             var packId = StickerPackId.deserialize(Base64.getDecoder().decode(s.packId));
  31             if (packIds.contains(packId)) {
  32                 // Remove legacy duplicate packIds ...
  33                 return null;
  34             }
  35             packIds.add(packId);
  36             var packKey = Base64.getDecoder().decode(s.packKey);
  37             var installed = s.installed;
  38             return new Sticker(packId, packKey, installed);
  39         }).filter(Objects::nonNull).collect(Collectors.toMap(Sticker::getPackId, s -> s));
  40
  41         return new StickerStore(stickers, saver);
  42     }
  43
  44     public Sticker getSticker(StickerPackId packId) {
  45         synchronized (stickers) {
  46             return stickers.get(packId);
  47         }
  48     }
  49
  50     public void updateSticker(Sticker sticker) {
  51         Storage storage;
  52         synchronized (stickers) {
  53             stickers.put(sticker.getPackId(), sticker);
  54             storage = toStorageLocked();
  55         }
  56         saver.save(storage);
  57     }
  58
  59     private Storage toStorageLocked() {
  60         return new Storage(stickers.values()
  61                 .stream()
  62                 .map(s -> new Storage.Sticker(Base64.getEncoder().encodeToString(s.getPackId().serialize()),
  63                         Base64.getEncoder().encodeToString(s.getPackKey()),
  64                         s.isInstalled()))
  65                 .collect(Collectors.toList()));
  66     }
  67
  68     public record Storage(List<Storage.Sticker> stickers) {
  69
  70         private record Sticker(String packId, String packKey, boolean installed) {
  71
  72         }
  73     }
  74
  75     public interface Saver {
  76
  77         void save(Storage storage);
  78     }
  79 }