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

   1 package org.asamk.signal.manager;
   2
   3 import org.asamk.signal.manager.groups.GroupId;
   4 import org.asamk.signal.manager.util.IOUtils;
   5 import org.asamk.signal.manager.util.Utils;
   6 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
   7 import org.whispersystems.signalservice.api.util.StreamDetails;
   8
   9 import java.io.File;
  10 import java.io.FileOutputStream;
  11 import java.io.IOException;
  12 import java.io.OutputStream;
  13 import java.nio.file.Files;
  14
  15 public class AvatarStore {
  16
  17     private final File avatarsPath;
  18
  19     public AvatarStore(final File avatarsPath) {
  20         this.avatarsPath = avatarsPath;
  21     }
  22
  23     public StreamDetails retrieveContactAvatar(SignalServiceAddress address) throws IOException {
  24         return retrieveAvatar(getContactAvatarFile(address));
  25     }
  26
  27     public StreamDetails retrieveProfileAvatar(SignalServiceAddress address) throws IOException {
  28         return retrieveAvatar(getProfileAvatarFile(address));
  29     }
  30
  31     public StreamDetails retrieveGroupAvatar(GroupId groupId) throws IOException {
  32         final File groupAvatarFile = getGroupAvatarFile(groupId);
  33         return retrieveAvatar(groupAvatarFile);
  34     }
  35
  36     public void storeContactAvatar(SignalServiceAddress address, AvatarStorer storer) throws IOException {
  37         storeAvatar(getContactAvatarFile(address), storer);
  38     }
  39
  40     public void storeProfileAvatar(SignalServiceAddress address, AvatarStorer storer) throws IOException {
  41         storeAvatar(getProfileAvatarFile(address), storer);
  42     }
  43
  44     public void storeGroupAvatar(GroupId groupId, AvatarStorer storer) throws IOException {
  45         storeAvatar(getGroupAvatarFile(groupId), storer);
  46     }
  47
  48     public void deleteProfileAvatar(SignalServiceAddress address) throws IOException {
  49         deleteAvatar(getProfileAvatarFile(address));
  50     }
  51
  52     private StreamDetails retrieveAvatar(final File avatarFile) throws IOException {
  53         if (!avatarFile.exists()) {
  54             return null;
  55         }
  56         return Utils.createStreamDetailsFromFile(avatarFile);
  57     }
  58
  59     private void storeAvatar(final File avatarFile, final AvatarStorer storer) throws IOException {
  60         createAvatarsDir();
  61         try (OutputStream output = new FileOutputStream(avatarFile)) {
  62             storer.store(output);
  63         }
  64     }
  65
  66     private void deleteAvatar(final File avatarFile) throws IOException {
  67         if (avatarFile.exists()) {
  68             Files.delete(avatarFile.toPath());
  69         }
  70     }
  71
  72     private File getGroupAvatarFile(GroupId groupId) {
  73         return new File(avatarsPath, "group-" + groupId.toBase64().replace("/", "_"));
  74     }
  75
  76     private File getContactAvatarFile(SignalServiceAddress address) {
  77         return new File(avatarsPath, "contact-" + address.getLegacyIdentifier());
  78     }
  79
  80     private File getProfileAvatarFile(SignalServiceAddress address) {
  81         return new File(avatarsPath, "profile-" + address.getLegacyIdentifier());
  82     }
  83
  84     private void createAvatarsDir() throws IOException {
  85         IOUtils.createPrivateDirectories(avatarsPath);
  86     }
  87
  88     @FunctionalInterface
  89     public interface AvatarStorer {
  90
  91         void store(OutputStream outputStream) throws IOException;
  92     }
  93 }