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

   1 /*
   2   Copyright (C) 2015-2022 AsamK and contributors
   3
   4   This program is free software: you can redistribute it and/or modify
   5   it under the terms of the GNU General Public License as published by
   6   the Free Software Foundation, either version 3 of the License, or
   7   (at your option) any later version.
   8
   9   This program is distributed in the hope that it will be useful,
  10   but WITHOUT ANY WARRANTY; without even the implied warranty of
  11   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12   GNU General Public License for more details.
  13
  14   You should have received a copy of the GNU General Public License
  15   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17 package org.asamk.signal.manager.internal;
  18
  19 import org.asamk.signal.manager.Manager;
  20 import org.asamk.signal.manager.api.AlreadyReceivingException;
  21 import org.asamk.signal.manager.api.AttachmentInvalidException;
  22 import org.asamk.signal.manager.api.CaptchaRequiredException;
  23 import org.asamk.signal.manager.api.Configuration;
  24 import org.asamk.signal.manager.api.Device;
  25 import org.asamk.signal.manager.api.DeviceLinkUrl;
  26 import org.asamk.signal.manager.api.Group;
  27 import org.asamk.signal.manager.api.GroupId;
  28 import org.asamk.signal.manager.api.GroupInviteLinkUrl;
  29 import org.asamk.signal.manager.api.GroupNotFoundException;
  30 import org.asamk.signal.manager.api.GroupSendingNotAllowedException;
  31 import org.asamk.signal.manager.api.Identity;
  32 import org.asamk.signal.manager.api.IdentityVerificationCode;
  33 import org.asamk.signal.manager.api.InactiveGroupLinkException;
  34 import org.asamk.signal.manager.api.IncorrectPinException;
  35 import org.asamk.signal.manager.api.InvalidDeviceLinkException;
  36 import org.asamk.signal.manager.api.InvalidStickerException;
  37 import org.asamk.signal.manager.api.InvalidUsernameException;
  38 import org.asamk.signal.manager.api.LastGroupAdminException;
  39 import org.asamk.signal.manager.api.Message;
  40 import org.asamk.signal.manager.api.MessageEnvelope;
  41 import org.asamk.signal.manager.api.NonNormalizedPhoneNumberException;
  42 import org.asamk.signal.manager.api.NotAGroupMemberException;
  43 import org.asamk.signal.manager.api.NotPrimaryDeviceException;
  44 import org.asamk.signal.manager.api.Pair;
  45 import org.asamk.signal.manager.api.PendingAdminApprovalException;
  46 import org.asamk.signal.manager.api.PinLockedException;
  47 import org.asamk.signal.manager.api.Profile;
  48 import org.asamk.signal.manager.api.RateLimitException;
  49 import org.asamk.signal.manager.api.ReceiveConfig;
  50 import org.asamk.signal.manager.api.Recipient;
  51 import org.asamk.signal.manager.api.RecipientIdentifier;
  52 import org.asamk.signal.manager.api.SendGroupMessageResults;
  53 import org.asamk.signal.manager.api.SendMessageResult;
  54 import org.asamk.signal.manager.api.SendMessageResults;
  55 import org.asamk.signal.manager.api.StickerPackId;
  56 import org.asamk.signal.manager.api.StickerPackInvalidException;
  57 import org.asamk.signal.manager.api.StickerPackUrl;
  58 import org.asamk.signal.manager.api.TextStyle;
  59 import org.asamk.signal.manager.api.TypingAction;
  60 import org.asamk.signal.manager.api.UnregisteredRecipientException;
  61 import org.asamk.signal.manager.api.UpdateGroup;
  62 import org.asamk.signal.manager.api.UpdateProfile;
  63 import org.asamk.signal.manager.api.UserStatus;
  64 import org.asamk.signal.manager.api.UsernameLinkUrl;
  65 import org.asamk.signal.manager.config.ServiceEnvironmentConfig;
  66 import org.asamk.signal.manager.helper.AccountFileUpdater;
  67 import org.asamk.signal.manager.helper.Context;
  68 import org.asamk.signal.manager.helper.RecipientHelper.RegisteredUser;
  69 import org.asamk.signal.manager.storage.AttachmentStore;
  70 import org.asamk.signal.manager.storage.AvatarStore;
  71 import org.asamk.signal.manager.storage.SignalAccount;
  72 import org.asamk.signal.manager.storage.groups.GroupInfo;
  73 import org.asamk.signal.manager.storage.identities.IdentityInfo;
  74 import org.asamk.signal.manager.storage.recipients.RecipientAddress;
  75 import org.asamk.signal.manager.storage.recipients.RecipientId;
  76 import org.asamk.signal.manager.storage.stickerPacks.JsonStickerPack;
  77 import org.asamk.signal.manager.storage.stickerPacks.StickerPackStore;
  78 import org.asamk.signal.manager.storage.stickers.StickerPack;
  79 import org.asamk.signal.manager.util.AttachmentUtils;
  80 import org.asamk.signal.manager.util.KeyUtils;
  81 import org.asamk.signal.manager.util.MimeUtils;
  82 import org.asamk.signal.manager.util.StickerUtils;
  83 import org.signal.libsignal.protocol.InvalidMessageException;
  84 import org.signal.libsignal.usernames.BaseUsernameException;
  85 import org.slf4j.Logger;
  86 import org.slf4j.LoggerFactory;
  87 import org.whispersystems.signalservice.api.SignalSessionLock;
  88 import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
  89 import org.whispersystems.signalservice.api.messages.SignalServicePreview;
  90 import org.whispersystems.signalservice.api.messages.SignalServiceReceiptMessage;
  91 import org.whispersystems.signalservice.api.messages.SignalServiceTypingMessage;
  92 import org.whispersystems.signalservice.api.push.ServiceId;
  93 import org.whispersystems.signalservice.api.push.ServiceId.ACI;
  94 import org.whispersystems.signalservice.api.push.ServiceIdType;
  95 import org.whispersystems.signalservice.api.push.exceptions.CdsiResourceExhaustedException;
  96 import org.whispersystems.signalservice.api.util.DeviceNameUtil;
  97 import org.whispersystems.signalservice.api.util.InvalidNumberException;
  98 import org.whispersystems.signalservice.api.util.PhoneNumberFormatter;
  99 import org.whispersystems.signalservice.api.util.StreamDetails;
 100 import org.whispersystems.signalservice.internal.util.Hex;
 101 import org.whispersystems.signalservice.internal.util.Util;
 102
 103 import java.io.ByteArrayInputStream;
 104 import java.io.File;
 105 import java.io.IOException;
 106 import java.io.InputStream;
 107 import java.nio.charset.StandardCharsets;
 108 import java.time.Duration;
 109 import java.util.ArrayList;
 110 import java.util.Collection;
 111 import java.util.HashMap;
 112 import java.util.HashSet;
 113 import java.util.List;
 114 import java.util.Map;
 115 import java.util.Objects;
 116 import java.util.Optional;
 117 import java.util.Set;
 118 import java.util.concurrent.ExecutorService;
 119 import java.util.concurrent.Executors;
 120 import java.util.concurrent.TimeUnit;
 121 import java.util.concurrent.atomic.AtomicInteger;
 122 import java.util.concurrent.locks.ReentrantLock;
 123 import java.util.function.Function;
 124 import java.util.stream.Collectors;
 125 import java.util.stream.Stream;
 126
 127 import io.reactivex.rxjava3.disposables.CompositeDisposable;
 128
 129 public class ManagerImpl implements Manager {
 130
 131     private static final Logger logger = LoggerFactory.getLogger(ManagerImpl.class);
 132
 133     private SignalAccount account;
 134     private final SignalDependencies dependencies;
 135     private final Context context;
 136
 137     private final ExecutorService executor = Executors.newCachedThreadPool();
 138
 139     private Thread receiveThread;
 140     private boolean isReceivingSynchronous;
 141     private final Set<ReceiveMessageHandler> weakHandlers = new HashSet<>();
 142     private final Set<ReceiveMessageHandler> messageHandlers = new HashSet<>();
 143     private final List<Runnable> closedListeners = new ArrayList<>();
 144     private final List<Runnable> addressChangedListeners = new ArrayList<>();
 145     private final CompositeDisposable disposable = new CompositeDisposable();
 146
 147     public ManagerImpl(
 148             SignalAccount account,
 149             PathConfig pathConfig,
 150             AccountFileUpdater accountFileUpdater,
 151             ServiceEnvironmentConfig serviceEnvironmentConfig,
 152             String userAgent
 153     ) {
 154         this.account = account;
 155
 156         final var sessionLock = new SignalSessionLock() {
 157             private final ReentrantLock LEGACY_LOCK = new ReentrantLock();
 158
 159             @Override
 160             public Lock acquire() {
 161                 LEGACY_LOCK.lock();
 162                 return LEGACY_LOCK::unlock;
 163             }
 164         };
 165         this.dependencies = new SignalDependencies(serviceEnvironmentConfig,
 166                 userAgent,
 167                 account.getCredentialsProvider(),
 168                 account.getSignalServiceDataStore(),
 169                 executor,
 170                 sessionLock);
 171         final var avatarStore = new AvatarStore(pathConfig.avatarsPath());
 172         final var attachmentStore = new AttachmentStore(pathConfig.attachmentsPath());
 173         final var stickerPackStore = new StickerPackStore(pathConfig.stickerPacksPath());
 174
 175         this.context = new Context(account, new AccountFileUpdater() {
 176             @Override
 177             public void updateAccountIdentifiers(final String number, final ACI aci) {
 178                 accountFileUpdater.updateAccountIdentifiers(number, aci);
 179                 synchronized (addressChangedListeners) {
 180                     addressChangedListeners.forEach(Runnable::run);
 181                 }
 182             }
 183
 184             @Override
 185             public void removeAccount() {
 186                 accountFileUpdater.removeAccount();
 187             }
 188         }, dependencies, avatarStore, attachmentStore, stickerPackStore);
 189         this.context.getAccountHelper().setUnregisteredListener(this::close);
 190         this.context.getReceiveHelper().setAuthenticationFailureListener(this::close);
 191         this.context.getReceiveHelper().setCaughtUpWithOldMessagesListener(() -> {
 192             synchronized (this) {
 193                 this.notifyAll();
 194             }
 195         });
 196         disposable.add(account.getIdentityKeyStore().getIdentityChanges().subscribe(serviceId -> {
 197             logger.trace("Archiving old sessions for {}", serviceId);
 198             account.getAccountData(ServiceIdType.ACI).getSessionStore().archiveSessions(serviceId);
 199             account.getAccountData(ServiceIdType.PNI).getSessionStore().archiveSessions(serviceId);
 200             account.getSenderKeyStore().deleteSharedWith(serviceId);
 201             final var recipientId = account.getRecipientResolver().resolveRecipient(serviceId);
 202             final var profile = account.getProfileStore().getProfile(recipientId);
 203             if (profile != null) {
 204                 account.getProfileStore()
 205                         .storeProfile(recipientId,
 206                                 Profile.newBuilder(profile)
 207                                         .withUnidentifiedAccessMode(Profile.UnidentifiedAccessMode.UNKNOWN)
 208                                         .withLastUpdateTimestamp(0)
 209                                         .build());
 210             }
 211         }));
 212     }
 213
 214     @Override
 215     public String getSelfNumber() {
 216         return account.getNumber();
 217     }
 218
 219     public void checkAccountState() throws IOException {
 220         context.getAccountHelper().checkAccountState();
 221         final var lastRecipientsRefresh = account.getLastRecipientsRefresh();
 222         if (lastRecipientsRefresh == null
 223                 || lastRecipientsRefresh < System.currentTimeMillis() - TimeUnit.DAYS.toMillis(1)) {
 224             try {
 225                 context.getRecipientHelper().refreshUsers();
 226             } catch (Exception e) {
 227                 logger.warn("Full CDSI recipients refresh failed, ignoring: {} ({})",
 228                         e.getMessage(),
 229                         e.getClass().getSimpleName());
 230                 logger.debug("Full CDSI refresh failed", e);
 231             }
 232             context.getAccountHelper().checkWhoAmiI();
 233         }
 234     }
 235
 236     @Override
 237     public Map<String, UserStatus> getUserStatus(Set<String> numbers) throws IOException, RateLimitException {
 238         final var canonicalizedNumbers = numbers.stream().collect(Collectors.toMap(n -> n, n -> {
 239             try {
 240                 final var canonicalizedNumber = PhoneNumberFormatter.formatNumber(n, account.getNumber());
 241                 if (!canonicalizedNumber.equals(n)) {
 242                     logger.debug("Normalized number {} to {}.", n, canonicalizedNumber);
 243                 }
 244                 return canonicalizedNumber;
 245             } catch (InvalidNumberException e) {
 246                 return "";
 247             }
 248         }));
 249
 250         // Note "registeredUsers" has no optionals. It only gives us info on users who are registered
 251         final var canonicalizedNumbersSet = canonicalizedNumbers.values()
 252                 .stream()
 253                 .filter(s -> !s.isEmpty())
 254                 .collect(Collectors.toSet());
 255
 256         final Map<String, RegisteredUser> registeredUsers;
 257         try {
 258             registeredUsers = context.getRecipientHelper().getRegisteredUsers(canonicalizedNumbersSet);
 259         } catch (CdsiResourceExhaustedException e) {
 260             logger.debug("CDSI resource exhausted: {}", e.getMessage());
 261             throw new RateLimitException(System.currentTimeMillis() + e.getRetryAfterSeconds() * 1000L);
 262         }
 263
 264         return numbers.stream().collect(Collectors.toMap(n -> n, n -> {
 265             final var number = canonicalizedNumbers.get(n);
 266             final var user = registeredUsers.get(number);
 267             final var serviceId = user == null ? null : user.getServiceId();
 268             final var profile = serviceId == null
 269                     ? null
 270                     : context.getProfileHelper()
 271                             .getRecipientProfile(account.getRecipientResolver().resolveRecipient(serviceId));
 272             return new UserStatus(number.isEmpty() ? null : number,
 273                     serviceId == null ? null : serviceId.getRawUuid(),
 274                     profile != null
 275                             && profile.getUnidentifiedAccessMode() == Profile.UnidentifiedAccessMode.UNRESTRICTED);
 276         }));
 277     }
 278
 279     @Override
 280     public void updateAccountAttributes(String deviceName) throws IOException {
 281         if (deviceName != null) {
 282             context.getAccountHelper().setDeviceName(deviceName);
 283         }
 284         context.getAccountHelper().updateAccountAttributes();
 285         context.getAccountHelper().checkWhoAmiI();
 286     }
 287
 288     @Override
 289     public Configuration getConfiguration() {
 290         final var configurationStore = account.getConfigurationStore();
 291         return Configuration.from(configurationStore);
 292     }
 293
 294     @Override
 295     public void updateConfiguration(
 296             Configuration configuration
 297     ) throws NotPrimaryDeviceException {
 298         if (!account.isPrimaryDevice()) {
 299             throw new NotPrimaryDeviceException();
 300         }
 301
 302         final var configurationStore = account.getConfigurationStore();
 303         if (configuration.readReceipts().isPresent()) {
 304             configurationStore.setReadReceipts(configuration.readReceipts().get());
 305         }
 306         if (configuration.unidentifiedDeliveryIndicators().isPresent()) {
 307             configurationStore.setUnidentifiedDeliveryIndicators(configuration.unidentifiedDeliveryIndicators().get());
 308         }
 309         if (configuration.typingIndicators().isPresent()) {
 310             configurationStore.setTypingIndicators(configuration.typingIndicators().get());
 311         }
 312         if (configuration.linkPreviews().isPresent()) {
 313             configurationStore.setLinkPreviews(configuration.linkPreviews().get());
 314         }
 315         context.getSyncHelper().sendConfigurationMessage();
 316     }
 317
 318     @Override
 319     public void updateProfile(UpdateProfile updateProfile) throws IOException {
 320         context.getProfileHelper()
 321                 .setProfile(updateProfile.getGivenName(),
 322                         updateProfile.getFamilyName(),
 323                         updateProfile.getAbout(),
 324                         updateProfile.getAboutEmoji(),
 325                         updateProfile.isDeleteAvatar()
 326                                 ? Optional.empty()
 327                                 : updateProfile.getAvatar() == null ? null : Optional.of(updateProfile.getAvatar()),
 328                         updateProfile.getMobileCoinAddress());
 329         context.getSyncHelper().sendSyncFetchProfileMessage();
 330     }
 331
 332     void refreshCurrentUsername() throws IOException, BaseUsernameException {
 333         context.getAccountHelper().refreshCurrentUsername();
 334     }
 335
 336     @Override
 337     public String getUsername() {
 338         return account.getUsername();
 339     }
 340
 341     @Override
 342     public UsernameLinkUrl getUsernameLink() {
 343         return new UsernameLinkUrl(account.getUsernameLink());
 344     }
 345
 346     @Override
 347     public void setUsername(final String username) throws IOException, InvalidUsernameException {
 348         try {
 349             context.getAccountHelper().reserveUsername(username);
 350         } catch (BaseUsernameException e) {
 351             throw new InvalidUsernameException(e.getMessage() + " (" + e.getClass().getSimpleName() + ")", e);
 352         }
 353     }
 354
 355     @Override
 356     public void deleteUsername() throws IOException {
 357         context.getAccountHelper().deleteUsername();
 358     }
 359
 360     @Override
 361     public void startChangeNumber(
 362             String newNumber, boolean voiceVerification, String captcha
 363     ) throws RateLimitException, IOException, CaptchaRequiredException, NonNormalizedPhoneNumberException, NotPrimaryDeviceException {
 364         if (!account.isPrimaryDevice()) {
 365             throw new NotPrimaryDeviceException();
 366         }
 367         context.getAccountHelper().startChangeNumber(newNumber, voiceVerification, captcha);
 368     }
 369
 370     @Override
 371     public void finishChangeNumber(
 372             String newNumber, String verificationCode, String pin
 373     ) throws IncorrectPinException, PinLockedException, IOException, NotPrimaryDeviceException {
 374         if (!account.isPrimaryDevice()) {
 375             throw new NotPrimaryDeviceException();
 376         }
 377         context.getAccountHelper().finishChangeNumber(newNumber, verificationCode, pin);
 378     }
 379
 380     @Override
 381     public void unregister() throws IOException {
 382         context.getAccountHelper().unregister();
 383     }
 384
 385     @Override
 386     public void deleteAccount() throws IOException {
 387         context.getAccountHelper().deleteAccount();
 388     }
 389
 390     @Override
 391     public void submitRateLimitRecaptchaChallenge(String challenge, String captcha) throws IOException {
 392         captcha = captcha == null ? null : captcha.replace("signalcaptcha://", "");
 393
 394         dependencies.getAccountManager().submitRateLimitRecaptchaChallenge(challenge, captcha);
 395     }
 396
 397     @Override
 398     public List<Device> getLinkedDevices() throws IOException {
 399         var devices = dependencies.getAccountManager().getDevices();
 400         account.setMultiDevice(devices.size() > 1);
 401         var identityKey = account.getAciIdentityKeyPair().getPrivateKey();
 402         return devices.stream().map(d -> {
 403             String deviceName = d.getName();
 404             if (deviceName != null) {
 405                 try {
 406                     deviceName = DeviceNameUtil.decryptDeviceName(deviceName, identityKey);
 407                 } catch (IOException e) {
 408                     logger.debug("Failed to decrypt device name, maybe plain text?", e);
 409                 }
 410             }
 411             return new Device(d.getId(),
 412                     deviceName,
 413                     d.getCreated(),
 414                     d.getLastSeen(),
 415                     d.getId() == account.getDeviceId());
 416         }).toList();
 417     }
 418
 419     @Override
 420     public void removeLinkedDevices(int deviceId) throws IOException {
 421         context.getAccountHelper().removeLinkedDevices(deviceId);
 422     }
 423
 424     @Override
 425     public void addDeviceLink(DeviceLinkUrl linkUrl) throws IOException, InvalidDeviceLinkException, NotPrimaryDeviceException {
 426         if (!account.isPrimaryDevice()) {
 427             throw new NotPrimaryDeviceException();
 428         }
 429         context.getAccountHelper().addDevice(linkUrl);
 430     }
 431
 432     @Override
 433     public void setRegistrationLockPin(Optional<String> pin) throws IOException, NotPrimaryDeviceException {
 434         if (!account.isPrimaryDevice()) {
 435             throw new NotPrimaryDeviceException();
 436         }
 437         if (pin.isPresent()) {
 438             context.getAccountHelper().setRegistrationPin(pin.get());
 439         } else {
 440             context.getAccountHelper().removeRegistrationPin();
 441         }
 442     }
 443
 444     void refreshPreKeys() throws IOException {
 445         context.getPreKeyHelper().refreshPreKeysIfNecessary();
 446     }
 447
 448     @Override
 449     public List<Group> getGroups() {
 450         return account.getGroupStore().getGroups().stream().map(this::toGroup).toList();
 451     }
 452
 453     private Group toGroup(final GroupInfo groupInfo) {
 454         if (groupInfo == null) {
 455             return null;
 456         }
 457
 458         return Group.from(groupInfo, account.getRecipientAddressResolver(), account.getSelfRecipientId());
 459     }
 460
 461     @Override
 462     public SendGroupMessageResults quitGroup(
 463             GroupId groupId, Set<RecipientIdentifier.Single> groupAdmins
 464     ) throws GroupNotFoundException, IOException, NotAGroupMemberException, LastGroupAdminException, UnregisteredRecipientException {
 465         final var newAdmins = context.getRecipientHelper().resolveRecipients(groupAdmins);
 466         return context.getGroupHelper().quitGroup(groupId, newAdmins);
 467     }
 468
 469     @Override
 470     public void deleteGroup(GroupId groupId) throws IOException {
 471         final var group = context.getGroupHelper().getGroup(groupId);
 472         if (group.isMember(account.getSelfRecipientId())) {
 473             throw new IOException(
 474                     "The local group information cannot be removed, as the user is still a member of the group");
 475         }
 476         context.getGroupHelper().deleteGroup(groupId);
 477     }
 478
 479     @Override
 480     public Pair<GroupId, SendGroupMessageResults> createGroup(
 481             String name, Set<RecipientIdentifier.Single> members, String avatarFile
 482     ) throws IOException, AttachmentInvalidException, UnregisteredRecipientException {
 483         return context.getGroupHelper()
 484                 .createGroup(name,
 485                         members == null ? null : context.getRecipientHelper().resolveRecipients(members),
 486                         avatarFile);
 487     }
 488
 489     @Override
 490     public SendGroupMessageResults updateGroup(
 491             final GroupId groupId, final UpdateGroup updateGroup
 492     ) throws IOException, GroupNotFoundException, AttachmentInvalidException, NotAGroupMemberException, GroupSendingNotAllowedException, UnregisteredRecipientException {
 493         return context.getGroupHelper()
 494                 .updateGroup(groupId,
 495                         updateGroup.getName(),
 496                         updateGroup.getDescription(),
 497                         updateGroup.getMembers() == null
 498                                 ? null
 499                                 : context.getRecipientHelper().resolveRecipients(updateGroup.getMembers()),
 500                         updateGroup.getRemoveMembers() == null
 501                                 ? null
 502                                 : context.getRecipientHelper().resolveRecipients(updateGroup.getRemoveMembers()),
 503                         updateGroup.getAdmins() == null
 504                                 ? null
 505                                 : context.getRecipientHelper().resolveRecipients(updateGroup.getAdmins()),
 506                         updateGroup.getRemoveAdmins() == null
 507                                 ? null
 508                                 : context.getRecipientHelper().resolveRecipients(updateGroup.getRemoveAdmins()),
 509                         updateGroup.getBanMembers() == null
 510                                 ? null
 511                                 : context.getRecipientHelper().resolveRecipients(updateGroup.getBanMembers()),
 512                         updateGroup.getUnbanMembers() == null
 513                                 ? null
 514                                 : context.getRecipientHelper().resolveRecipients(updateGroup.getUnbanMembers()),
 515                         updateGroup.isResetGroupLink(),
 516                         updateGroup.getGroupLinkState(),
 517                         updateGroup.getAddMemberPermission(),
 518                         updateGroup.getEditDetailsPermission(),
 519                         updateGroup.getAvatarFile(),
 520                         updateGroup.getExpirationTimer(),
 521                         updateGroup.getIsAnnouncementGroup());
 522     }
 523
 524     @Override
 525     public Pair<GroupId, SendGroupMessageResults> joinGroup(
 526             GroupInviteLinkUrl inviteLinkUrl
 527     ) throws IOException, InactiveGroupLinkException, PendingAdminApprovalException {
 528         return context.getGroupHelper().joinGroup(inviteLinkUrl);
 529     }
 530
 531     private SendMessageResults sendMessage(
 532             SignalServiceDataMessage.Builder messageBuilder, Set<RecipientIdentifier> recipients, boolean notifySelf
 533     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 534         return sendMessage(messageBuilder, recipients, notifySelf, Optional.empty());
 535     }
 536
 537     private SendMessageResults sendMessage(
 538             SignalServiceDataMessage.Builder messageBuilder,
 539             Set<RecipientIdentifier> recipients,
 540             boolean notifySelf,
 541             Optional<Long> editTargetTimestamp
 542     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 543         var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
 544         long timestamp = System.currentTimeMillis();
 545         messageBuilder.withTimestamp(timestamp);
 546         for (final var recipient : recipients) {
 547             if (recipient instanceof RecipientIdentifier.NoteToSelf || (
 548                     recipient instanceof RecipientIdentifier.Single single
 549                             && new RecipientAddress(single.toPartialRecipientAddress()).matches(account.getSelfRecipientAddress())
 550             )) {
 551                 final var result = notifySelf
 552                         ? context.getSendHelper()
 553                         .sendMessage(messageBuilder, account.getSelfRecipientId(), editTargetTimestamp)
 554                         : context.getSendHelper().sendSelfMessage(messageBuilder, editTargetTimestamp);
 555                 results.put(recipient, List.of(toSendMessageResult(result)));
 556             } else if (recipient instanceof RecipientIdentifier.Single single) {
 557                 try {
 558                     final var recipientId = context.getRecipientHelper().resolveRecipient(single);
 559                     final var result = context.getSendHelper()
 560                             .sendMessage(messageBuilder, recipientId, editTargetTimestamp);
 561                     results.put(recipient, List.of(toSendMessageResult(result)));
 562                 } catch (UnregisteredRecipientException e) {
 563                     results.put(recipient,
 564                             List.of(SendMessageResult.unregisteredFailure(single.toPartialRecipientAddress())));
 565                 }
 566             } else if (recipient instanceof RecipientIdentifier.Group group) {
 567                 final var result = context.getSendHelper()
 568                         .sendAsGroupMessage(messageBuilder, group.groupId(), notifySelf, editTargetTimestamp);
 569                 results.put(recipient, result.stream().map(this::toSendMessageResult).toList());
 570             }
 571         }
 572         return new SendMessageResults(timestamp, results);
 573     }
 574
 575     private SendMessageResult toSendMessageResult(final org.whispersystems.signalservice.api.messages.SendMessageResult result) {
 576         return SendMessageResult.from(result, account.getRecipientResolver(), account.getRecipientAddressResolver());
 577     }
 578
 579     private SendMessageResults sendTypingMessage(
 580             SignalServiceTypingMessage.Action action, Set<RecipientIdentifier> recipients
 581     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 582         var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
 583         final var timestamp = System.currentTimeMillis();
 584         for (var recipient : recipients) {
 585             if (recipient instanceof RecipientIdentifier.Single single) {
 586                 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.empty());
 587                 try {
 588                     final var recipientId = context.getRecipientHelper().resolveRecipient(single);
 589                     final var result = context.getSendHelper().sendTypingMessage(message, recipientId);
 590                     results.put(recipient, List.of(toSendMessageResult(result)));
 591                 } catch (UnregisteredRecipientException e) {
 592                     results.put(recipient,
 593                             List.of(SendMessageResult.unregisteredFailure(single.toPartialRecipientAddress())));
 594                 }
 595             } else if (recipient instanceof RecipientIdentifier.Group) {
 596                 final var groupId = ((RecipientIdentifier.Group) recipient).groupId();
 597                 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.of(groupId.serialize()));
 598                 final var result = context.getSendHelper().sendGroupTypingMessage(message, groupId);
 599                 results.put(recipient, result.stream().map(this::toSendMessageResult).toList());
 600             }
 601         }
 602         return new SendMessageResults(timestamp, results);
 603     }
 604
 605     @Override
 606     public SendMessageResults sendTypingMessage(
 607             TypingAction action, Set<RecipientIdentifier> recipients
 608     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 609         return sendTypingMessage(action.toSignalService(), recipients);
 610     }
 611
 612     @Override
 613     public SendMessageResults sendReadReceipt(
 614             RecipientIdentifier.Single sender, List<Long> messageIds
 615     ) {
 616         final var timestamp = System.currentTimeMillis();
 617         var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.READ,
 618                 messageIds,
 619                 timestamp);
 620
 621         return sendReceiptMessage(sender, timestamp, receiptMessage);
 622     }
 623
 624     @Override
 625     public SendMessageResults sendViewedReceipt(
 626             RecipientIdentifier.Single sender, List<Long> messageIds
 627     ) {
 628         final var timestamp = System.currentTimeMillis();
 629         var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.VIEWED,
 630                 messageIds,
 631                 timestamp);
 632
 633         return sendReceiptMessage(sender, timestamp, receiptMessage);
 634     }
 635
 636     private SendMessageResults sendReceiptMessage(
 637             final RecipientIdentifier.Single sender,
 638             final long timestamp,
 639             final SignalServiceReceiptMessage receiptMessage
 640     ) {
 641         try {
 642             final var result = context.getSendHelper()
 643                     .sendReceiptMessage(receiptMessage, context.getRecipientHelper().resolveRecipient(sender));
 644             return new SendMessageResults(timestamp, Map.of(sender, List.of(toSendMessageResult(result))));
 645         } catch (UnregisteredRecipientException e) {
 646             return new SendMessageResults(timestamp,
 647                     Map.of(sender, List.of(SendMessageResult.unregisteredFailure(sender.toPartialRecipientAddress()))));
 648         }
 649     }
 650
 651     @Override
 652     public SendMessageResults sendMessage(
 653             Message message, Set<RecipientIdentifier> recipients, boolean notifySelf
 654     ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException, InvalidStickerException {
 655         final var selfProfile = context.getProfileHelper().getSelfProfile();
 656         if (selfProfile == null || selfProfile.getDisplayName().isEmpty()) {
 657             logger.warn(
 658                     "No profile name set. When sending a message it's recommended to set a profile name with the updateProfile command. This may become mandatory in the future.");
 659         }
 660         final var messageBuilder = SignalServiceDataMessage.newBuilder();
 661         applyMessage(messageBuilder, message);
 662         return sendMessage(messageBuilder, recipients, notifySelf);
 663     }
 664
 665     @Override
 666     public SendMessageResults sendEditMessage(
 667             Message message, Set<RecipientIdentifier> recipients, long editTargetTimestamp
 668     ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException, InvalidStickerException {
 669         final var messageBuilder = SignalServiceDataMessage.newBuilder();
 670         applyMessage(messageBuilder, message);
 671         return sendMessage(messageBuilder, recipients, false, Optional.of(editTargetTimestamp));
 672     }
 673
 674     private void applyMessage(
 675             final SignalServiceDataMessage.Builder messageBuilder, final Message message
 676     ) throws AttachmentInvalidException, IOException, UnregisteredRecipientException, InvalidStickerException {
 677         if (message.messageText().length() > 2000) {
 678             final var messageBytes = message.messageText().getBytes(StandardCharsets.UTF_8);
 679             final var textAttachment = AttachmentUtils.createAttachmentStream(new StreamDetails(new ByteArrayInputStream(
 680                     messageBytes), MimeUtils.LONG_TEXT, messageBytes.length), Optional.empty());
 681             messageBuilder.withBody(message.messageText().substring(0, 2000));
 682             messageBuilder.withAttachment(context.getAttachmentHelper().uploadAttachment(textAttachment));
 683         } else {
 684             messageBuilder.withBody(message.messageText());
 685         }
 686         if (!message.attachments().isEmpty()) {
 687             messageBuilder.withAttachments(context.getAttachmentHelper().uploadAttachments(message.attachments()));
 688         }
 689         if (!message.mentions().isEmpty()) {
 690             messageBuilder.withMentions(resolveMentions(message.mentions()));
 691         }
 692         if (!message.textStyles().isEmpty()) {
 693             messageBuilder.withBodyRanges(message.textStyles().stream().map(TextStyle::toBodyRange).toList());
 694         }
 695         if (message.quote().isPresent()) {
 696             final var quote = message.quote().get();
 697             final var quotedAttachments = new ArrayList<SignalServiceDataMessage.Quote.QuotedAttachment>();
 698             for (final var a : quote.attachments()) {
 699                 final var quotedAttachment = new SignalServiceDataMessage.Quote.QuotedAttachment(a.contentType(),
 700                         a.filename(),
 701                         a.preview() == null ? null : context.getAttachmentHelper().uploadAttachment(a.preview()));
 702                 quotedAttachments.add(quotedAttachment);
 703             }
 704             messageBuilder.withQuote(new SignalServiceDataMessage.Quote(quote.timestamp(),
 705                     context.getRecipientHelper()
 706                             .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(quote.author()))
 707                             .getServiceId(),
 708                     quote.message(),
 709                     quotedAttachments,
 710                     resolveMentions(quote.mentions()),
 711                     SignalServiceDataMessage.Quote.Type.NORMAL,
 712                     quote.textStyles().stream().map(TextStyle::toBodyRange).toList()));
 713         }
 714         if (message.sticker().isPresent()) {
 715             final var sticker = message.sticker().get();
 716             final var packId = StickerPackId.deserialize(sticker.packId());
 717             final var stickerId = sticker.stickerId();
 718
 719             final var stickerPack = context.getAccount().getStickerStore().getStickerPack(packId);
 720             if (stickerPack == null) {
 721                 throw new InvalidStickerException("Sticker pack not found");
 722             }
 723             final var manifest = context.getStickerHelper().getOrRetrieveStickerPack(packId, stickerPack.packKey());
 724             if (manifest.stickers().size() <= stickerId) {
 725                 throw new InvalidStickerException("Sticker id not part of this pack");
 726             }
 727             final var manifestSticker = manifest.stickers().get(stickerId);
 728             final var streamDetails = context.getStickerPackStore().retrieveSticker(packId, stickerId);
 729             if (streamDetails == null) {
 730                 throw new InvalidStickerException("Missing local sticker file");
 731             }
 732             messageBuilder.withSticker(new SignalServiceDataMessage.Sticker(packId.serialize(),
 733                     stickerPack.packKey(),
 734                     stickerId,
 735                     manifestSticker.emoji(),
 736                     AttachmentUtils.createAttachmentStream(streamDetails, Optional.empty())));
 737         }
 738         if (!message.previews().isEmpty()) {
 739             final var previews = new ArrayList<SignalServicePreview>(message.previews().size());
 740             for (final var p : message.previews()) {
 741                 final var image = p.image().isPresent() ? context.getAttachmentHelper()
 742                         .uploadAttachment(p.image().get()) : null;
 743                 previews.add(new SignalServicePreview(p.url(),
 744                         p.title(),
 745                         p.description(),
 746                         0,
 747                         Optional.ofNullable(image)));
 748             }
 749             messageBuilder.withPreviews(previews);
 750         }
 751         if (message.storyReply().isPresent()) {
 752             final var storyReply = message.storyReply().get();
 753             final var authorServiceId = context.getRecipientHelper()
 754                     .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(storyReply.author()))
 755                     .getServiceId();
 756             messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
 757                     storyReply.timestamp()));
 758         }
 759     }
 760
 761     private ArrayList<SignalServiceDataMessage.Mention> resolveMentions(final List<Message.Mention> mentionList) throws UnregisteredRecipientException {
 762         final var mentions = new ArrayList<SignalServiceDataMessage.Mention>();
 763         for (final var m : mentionList) {
 764             final var recipientId = context.getRecipientHelper().resolveRecipient(m.recipient());
 765             mentions.add(new SignalServiceDataMessage.Mention(context.getRecipientHelper()
 766                     .resolveSignalServiceAddress(recipientId)
 767                     .getServiceId(), m.start(), m.length()));
 768         }
 769         return mentions;
 770     }
 771
 772     @Override
 773     public SendMessageResults sendRemoteDeleteMessage(
 774             long targetSentTimestamp, Set<RecipientIdentifier> recipients
 775     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 776         var delete = new SignalServiceDataMessage.RemoteDelete(targetSentTimestamp);
 777         final var messageBuilder = SignalServiceDataMessage.newBuilder().withRemoteDelete(delete);
 778         for (final var recipient : recipients) {
 779             if (recipient instanceof RecipientIdentifier.Uuid u) {
 780                 account.getMessageSendLogStore()
 781                         .deleteEntryForRecipientNonGroup(targetSentTimestamp, ACI.from(u.uuid()));
 782             } else if (recipient instanceof RecipientIdentifier.Single r) {
 783                 try {
 784                     final var recipientId = context.getRecipientHelper().resolveRecipient(r);
 785                     final var address = account.getRecipientAddressResolver().resolveRecipientAddress(recipientId);
 786                     if (address.serviceId().isPresent()) {
 787                         account.getMessageSendLogStore()
 788                                 .deleteEntryForRecipientNonGroup(targetSentTimestamp, address.serviceId().get());
 789                     }
 790                 } catch (UnregisteredRecipientException ignored) {
 791                 }
 792             } else if (recipient instanceof RecipientIdentifier.Group r) {
 793                 account.getMessageSendLogStore().deleteEntryForGroup(targetSentTimestamp, r.groupId());
 794             }
 795         }
 796         return sendMessage(messageBuilder, recipients, false);
 797     }
 798
 799     @Override
 800     public SendMessageResults sendMessageReaction(
 801             String emoji,
 802             boolean remove,
 803             RecipientIdentifier.Single targetAuthor,
 804             long targetSentTimestamp,
 805             Set<RecipientIdentifier> recipients,
 806             final boolean isStory
 807     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException {
 808         var targetAuthorRecipientId = context.getRecipientHelper().resolveRecipient(targetAuthor);
 809         final var authorServiceId = context.getRecipientHelper()
 810                 .resolveSignalServiceAddress(targetAuthorRecipientId)
 811                 .getServiceId();
 812         var reaction = new SignalServiceDataMessage.Reaction(emoji, remove, authorServiceId, targetSentTimestamp);
 813         final var messageBuilder = SignalServiceDataMessage.newBuilder().withReaction(reaction);
 814         if (isStory) {
 815             messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
 816                     targetSentTimestamp));
 817         }
 818         return sendMessage(messageBuilder, recipients, false);
 819     }
 820
 821     @Override
 822     public SendMessageResults sendPaymentNotificationMessage(
 823             byte[] receipt, String note, RecipientIdentifier.Single recipient
 824     ) throws IOException {
 825         final var paymentNotification = new SignalServiceDataMessage.PaymentNotification(receipt, note);
 826         final var payment = new SignalServiceDataMessage.Payment(paymentNotification, null);
 827         final var messageBuilder = SignalServiceDataMessage.newBuilder().withPayment(payment);
 828         try {
 829             return sendMessage(messageBuilder, Set.of(recipient), false);
 830         } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
 831             throw new AssertionError(e);
 832         }
 833     }
 834
 835     @Override
 836     public SendMessageResults sendEndSessionMessage(Set<RecipientIdentifier.Single> recipients) throws IOException {
 837         var messageBuilder = SignalServiceDataMessage.newBuilder().asEndSessionMessage();
 838
 839         try {
 840             return sendMessage(messageBuilder,
 841                     recipients.stream().map(RecipientIdentifier.class::cast).collect(Collectors.toSet()),
 842                     false);
 843         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
 844             throw new AssertionError(e);
 845         } finally {
 846             for (var recipient : recipients) {
 847                 final RecipientId recipientId;
 848                 try {
 849                     recipientId = context.getRecipientHelper().resolveRecipient(recipient);
 850                 } catch (UnregisteredRecipientException e) {
 851                     continue;
 852                 }
 853                 final var serviceId = context.getAccount()
 854                         .getRecipientAddressResolver()
 855                         .resolveRecipientAddress(recipientId)
 856                         .serviceId();
 857                 if (serviceId.isPresent()) {
 858                     account.getAccountData(ServiceIdType.ACI).getSessionStore().deleteAllSessions(serviceId.get());
 859                 }
 860             }
 861         }
 862     }
 863
 864     @Override
 865     public void hideRecipient(final RecipientIdentifier.Single recipient) {
 866         final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
 867         if (recipientIdOptional.isPresent()) {
 868             context.getContactHelper().setContactHidden(recipientIdOptional.get(), true);
 869             account.removeRecipient(recipientIdOptional.get());
 870         }
 871     }
 872
 873     @Override
 874     public void deleteRecipient(final RecipientIdentifier.Single recipient) {
 875         final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
 876         if (recipientIdOptional.isPresent()) {
 877             account.removeRecipient(recipientIdOptional.get());
 878         }
 879     }
 880
 881     @Override
 882     public void deleteContact(final RecipientIdentifier.Single recipient) {
 883         final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
 884         if (recipientIdOptional.isPresent()) {
 885             account.getContactStore().deleteContact(recipientIdOptional.get());
 886         }
 887     }
 888
 889     @Override
 890     public void setContactName(
 891             RecipientIdentifier.Single recipient, String givenName, final String familyName
 892     ) throws NotPrimaryDeviceException, UnregisteredRecipientException {
 893         if (!account.isPrimaryDevice()) {
 894             throw new NotPrimaryDeviceException();
 895         }
 896         context.getContactHelper()
 897                 .setContactName(context.getRecipientHelper().resolveRecipient(recipient), givenName, familyName);
 898     }
 899
 900     @Override
 901     public void setContactsBlocked(
 902             Collection<RecipientIdentifier.Single> recipients, boolean blocked
 903     ) throws NotPrimaryDeviceException, IOException, UnregisteredRecipientException {
 904         if (!account.isPrimaryDevice()) {
 905             throw new NotPrimaryDeviceException();
 906         }
 907         if (recipients.isEmpty()) {
 908             return;
 909         }
 910         final var recipientIds = context.getRecipientHelper().resolveRecipients(recipients);
 911         final var selfRecipientId = account.getSelfRecipientId();
 912         boolean shouldRotateProfileKey = false;
 913         for (final var recipientId : recipientIds) {
 914             if (context.getContactHelper().isContactBlocked(recipientId) == blocked) {
 915                 continue;
 916             }
 917             context.getContactHelper().setContactBlocked(recipientId, blocked);
 918             // if we don't have a common group with the blocked contact we need to rotate the profile key
 919             shouldRotateProfileKey = blocked && (
 920                     shouldRotateProfileKey || account.getGroupStore()
 921                             .getGroups()
 922                             .stream()
 923                             .noneMatch(g -> g.isMember(selfRecipientId) && g.isMember(recipientId))
 924             );
 925         }
 926         if (shouldRotateProfileKey) {
 927             context.getProfileHelper().rotateProfileKey();
 928         }
 929         context.getSyncHelper().sendBlockedList();
 930     }
 931
 932     @Override
 933     public void setGroupsBlocked(
 934             final Collection<GroupId> groupIds, final boolean blocked
 935     ) throws GroupNotFoundException, NotPrimaryDeviceException, IOException {
 936         if (!account.isPrimaryDevice()) {
 937             throw new NotPrimaryDeviceException();
 938         }
 939         if (groupIds.isEmpty()) {
 940             return;
 941         }
 942         boolean shouldRotateProfileKey = false;
 943         for (final var groupId : groupIds) {
 944             if (context.getGroupHelper().isGroupBlocked(groupId) == blocked) {
 945                 continue;
 946             }
 947             context.getGroupHelper().setGroupBlocked(groupId, blocked);
 948             shouldRotateProfileKey = blocked;
 949         }
 950         if (shouldRotateProfileKey) {
 951             context.getProfileHelper().rotateProfileKey();
 952         }
 953         context.getSyncHelper().sendBlockedList();
 954     }
 955
 956     @Override
 957     public void setExpirationTimer(
 958             RecipientIdentifier.Single recipient, int messageExpirationTimer
 959     ) throws IOException, UnregisteredRecipientException {
 960         var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
 961         context.getContactHelper().setExpirationTimer(recipientId, messageExpirationTimer);
 962         final var messageBuilder = SignalServiceDataMessage.newBuilder().asExpirationUpdate();
 963         try {
 964             sendMessage(messageBuilder, Set.of(recipient), false);
 965         } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
 966             throw new AssertionError(e);
 967         }
 968     }
 969
 970     @Override
 971     public StickerPackUrl uploadStickerPack(File path) throws IOException, StickerPackInvalidException {
 972         var manifest = StickerUtils.getSignalServiceStickerManifestUpload(path);
 973
 974         var messageSender = dependencies.getMessageSender();
 975
 976         var packKey = KeyUtils.createStickerUploadKey();
 977         var packIdString = messageSender.uploadStickerManifest(manifest, packKey);
 978         var packId = StickerPackId.deserialize(Hex.fromStringCondensed(packIdString));
 979
 980         var sticker = new StickerPack(packId, packKey);
 981         account.getStickerStore().addStickerPack(sticker);
 982         context.getSyncHelper().sendStickerOperationsMessage(List.of(sticker), List.of());
 983
 984         return new StickerPackUrl(packId, packKey);
 985     }
 986
 987     @Override
 988     public void installStickerPack(StickerPackUrl url) throws IOException {
 989         final var packId = url.packId();
 990         final var packKey = url.packKey();
 991         try {
 992             context.getStickerHelper().retrieveStickerPack(packId, packKey);
 993         } catch (InvalidMessageException e) {
 994             throw new IOException(e);
 995         }
 996
 997         final var sticker = context.getStickerHelper().addOrUpdateStickerPack(packId, packKey, true);
 998         context.getSyncHelper().sendStickerOperationsMessage(List.of(sticker), List.of());
 999     }
1000
1001     @Override
1002     public List<org.asamk.signal.manager.api.StickerPack> getStickerPacks() {
1003         final var stickerPackStore = context.getStickerPackStore();
1004         return account.getStickerStore().getStickerPacks().stream().map(pack -> {
1005             if (stickerPackStore.existsStickerPack(pack.packId())) {
1006                 try {
1007                     final var manifest = stickerPackStore.retrieveManifest(pack.packId());
1008                     return new org.asamk.signal.manager.api.StickerPack(pack.packId(),
1009                             new StickerPackUrl(pack.packId(), pack.packKey()),
1010                             pack.isInstalled(),
1011                             manifest.title(),
1012                             manifest.author(),
1013                             Optional.ofNullable(manifest.cover() == null ? null : manifest.cover().toApi()),
1014                             manifest.stickers().stream().map(JsonStickerPack.JsonSticker::toApi).toList());
1015                 } catch (Exception e) {
1016                     logger.warn("Failed to read local sticker pack manifest: {}", e.getMessage(), e);
1017                 }
1018             }
1019
1020             return new org.asamk.signal.manager.api.StickerPack(pack.packId(), pack.packKey(), pack.isInstalled());
1021         }).toList();
1022     }
1023
1024     @Override
1025     public void requestAllSyncData() throws IOException {
1026         context.getSyncHelper().requestAllSyncData();
1027         retrieveRemoteStorage();
1028     }
1029
1030     void retrieveRemoteStorage() throws IOException {
1031         context.getStorageHelper().readDataFromStorage();
1032     }
1033
1034     @Override
1035     public void addReceiveHandler(final ReceiveMessageHandler handler, final boolean isWeakListener) {
1036         synchronized (messageHandlers) {
1037             if (isWeakListener) {
1038                 weakHandlers.add(handler);
1039             } else {
1040                 messageHandlers.add(handler);
1041                 startReceiveThreadIfRequired();
1042             }
1043         }
1044     }
1045
1046     private static final AtomicInteger threadNumber = new AtomicInteger(0);
1047
1048     private void startReceiveThreadIfRequired() {
1049         if (receiveThread != null || isReceivingSynchronous) {
1050             return;
1051         }
1052         receiveThread = Thread.ofPlatform().name("receive-" + threadNumber.getAndIncrement()).start(() -> {
1053             logger.debug("Starting receiving messages");
1054             context.getReceiveHelper().receiveMessagesContinuously(this::passReceivedMessageToHandlers);
1055             logger.debug("Finished receiving messages");
1056             synchronized (messageHandlers) {
1057                 receiveThread = null;
1058
1059                 // Check if in the meantime another handler has been registered
1060                 if (!messageHandlers.isEmpty()) {
1061                     logger.debug("Another handler has been registered, starting receive thread again");
1062                     startReceiveThreadIfRequired();
1063                 }
1064             }
1065         });
1066     }
1067
1068     private void passReceivedMessageToHandlers(MessageEnvelope envelope, Throwable e) {
1069         synchronized (messageHandlers) {
1070             Stream.concat(messageHandlers.stream(), weakHandlers.stream()).forEach(h -> {
1071                 try {
1072                     h.handleMessage(envelope, e);
1073                 } catch (Throwable ex) {
1074                     logger.warn("Message handler failed, ignoring", ex);
1075                 }
1076             });
1077         }
1078     }
1079
1080     @Override
1081     public void removeReceiveHandler(final ReceiveMessageHandler handler) {
1082         final Thread thread;
1083         synchronized (messageHandlers) {
1084             weakHandlers.remove(handler);
1085             messageHandlers.remove(handler);
1086             if (!messageHandlers.isEmpty() || receiveThread == null || isReceivingSynchronous) {
1087                 return;
1088             }
1089             thread = receiveThread;
1090             receiveThread = null;
1091         }
1092
1093         stopReceiveThread(thread);
1094     }
1095
1096     private void stopReceiveThread(final Thread thread) {
1097         if (context.getReceiveHelper().requestStopReceiveMessages()) {
1098             logger.debug("Receive stop requested, interrupting read from server.");
1099             thread.interrupt();
1100         }
1101         try {
1102             thread.join();
1103         } catch (InterruptedException ignored) {
1104         }
1105     }
1106
1107     @Override
1108     public boolean isReceiving() {
1109         if (isReceivingSynchronous) {
1110             return true;
1111         }
1112         synchronized (messageHandlers) {
1113             return !messageHandlers.isEmpty();
1114         }
1115     }
1116
1117     @Override
1118     public void receiveMessages(
1119             Optional<Duration> timeout, Optional<Integer> maxMessages, ReceiveMessageHandler handler
1120     ) throws IOException, AlreadyReceivingException {
1121         receiveMessages(timeout.orElse(Duration.ofMinutes(1)), timeout.isPresent(), maxMessages.orElse(null), handler);
1122     }
1123
1124     @Override
1125     public void stopReceiveMessages() {
1126         Thread thread = null;
1127         synchronized (messageHandlers) {
1128             if (isReceivingSynchronous) {
1129                 thread = receiveThread;
1130                 receiveThread = null;
1131             }
1132         }
1133         if (thread != null) {
1134             stopReceiveThread(thread);
1135         }
1136     }
1137
1138     private void receiveMessages(
1139             Duration timeout, boolean returnOnTimeout, Integer maxMessages, ReceiveMessageHandler handler
1140     ) throws IOException, AlreadyReceivingException {
1141         synchronized (messageHandlers) {
1142             if (isReceiving()) {
1143                 throw new AlreadyReceivingException("Already receiving message.");
1144             }
1145             isReceivingSynchronous = true;
1146             receiveThread = Thread.currentThread();
1147         }
1148         try {
1149             context.getReceiveHelper().receiveMessages(timeout, returnOnTimeout, maxMessages, (envelope, e) -> {
1150                 passReceivedMessageToHandlers(envelope, e);
1151                 handler.handleMessage(envelope, e);
1152             });
1153         } finally {
1154             synchronized (messageHandlers) {
1155                 receiveThread = null;
1156                 isReceivingSynchronous = false;
1157                 if (!messageHandlers.isEmpty()) {
1158                     startReceiveThreadIfRequired();
1159                 }
1160             }
1161         }
1162     }
1163
1164     @Override
1165     public void setReceiveConfig(final ReceiveConfig receiveConfig) {
1166         context.getReceiveHelper().setReceiveConfig(receiveConfig);
1167     }
1168
1169     @Override
1170     public boolean isContactBlocked(final RecipientIdentifier.Single recipient) {
1171         final RecipientId recipientId;
1172         try {
1173             recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1174         } catch (UnregisteredRecipientException e) {
1175             return false;
1176         }
1177         return context.getContactHelper().isContactBlocked(recipientId);
1178     }
1179
1180     @Override
1181     public void sendContacts() throws IOException {
1182         context.getSyncHelper().sendContacts();
1183     }
1184
1185     @Override
1186     public List<Recipient> getRecipients(
1187             boolean onlyContacts,
1188             Optional<Boolean> blocked,
1189             Collection<RecipientIdentifier.Single> recipients,
1190             Optional<String> name
1191     ) {
1192         final var recipientIds = recipients.stream().map(a -> {
1193             try {
1194                 return context.getRecipientHelper().resolveRecipient(a);
1195             } catch (UnregisteredRecipientException e) {
1196                 return null;
1197             }
1198         }).filter(Objects::nonNull).collect(Collectors.toSet());
1199         if (!recipients.isEmpty() && recipientIds.isEmpty()) {
1200             return List.of();
1201         }
1202         // refresh profiles of explicitly given recipients
1203         context.getProfileHelper().refreshRecipientProfiles(recipientIds);
1204         return account.getRecipientStore()
1205                 .getRecipients(onlyContacts, blocked, recipientIds, name)
1206                 .stream()
1207                 .map(s -> new Recipient(s.getRecipientId(),
1208                         s.getAddress().toApiRecipientAddress(),
1209                         s.getContact(),
1210                         s.getProfileKey(),
1211                         s.getExpiringProfileKeyCredential(),
1212                         s.getProfile()))
1213                 .toList();
1214     }
1215
1216     @Override
1217     public String getContactOrProfileName(RecipientIdentifier.Single recipient) {
1218         final RecipientId recipientId;
1219         try {
1220             recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1221         } catch (UnregisteredRecipientException e) {
1222             return null;
1223         }
1224
1225         final var contact = account.getContactStore().getContact(recipientId);
1226         if (contact != null && !Util.isEmpty(contact.getName())) {
1227             return contact.getName();
1228         }
1229
1230         final var profile = context.getProfileHelper().getRecipientProfile(recipientId);
1231         if (profile != null) {
1232             return profile.getDisplayName();
1233         }
1234
1235         return null;
1236     }
1237
1238     @Override
1239     public Group getGroup(GroupId groupId) {
1240         return toGroup(context.getGroupHelper().getGroup(groupId));
1241     }
1242
1243     @Override
1244     public List<Identity> getIdentities() {
1245         return account.getIdentityKeyStore()
1246                 .getIdentities()
1247                 .stream()
1248                 .map(this::toIdentity)
1249                 .filter(Objects::nonNull)
1250                 .toList();
1251     }
1252
1253     private Identity toIdentity(final IdentityInfo identityInfo) {
1254         if (identityInfo == null) {
1255             return null;
1256         }
1257
1258         final var address = account.getRecipientAddressResolver()
1259                 .resolveRecipientAddress(account.getRecipientResolver().resolveRecipient(identityInfo.getServiceId()));
1260         if (address.serviceId().isPresent() && !Objects.equals(address.serviceId().get(),
1261                 identityInfo.getServiceId())) {
1262             return null;
1263         }
1264         final var scannableFingerprint = context.getIdentityHelper()
1265                 .computeSafetyNumberForScanning(identityInfo.getServiceId(), identityInfo.getIdentityKey());
1266         return new Identity(address.toApiRecipientAddress(),
1267                 identityInfo.getIdentityKey(),
1268                 context.getIdentityHelper()
1269                         .computeSafetyNumber(identityInfo.getServiceId(), identityInfo.getIdentityKey()),
1270                 scannableFingerprint == null ? null : scannableFingerprint.getSerialized(),
1271                 identityInfo.getTrustLevel(),
1272                 identityInfo.getDateAddedTimestamp());
1273     }
1274
1275     @Override
1276     public List<Identity> getIdentities(RecipientIdentifier.Single recipient) {
1277         ServiceId serviceId;
1278         try {
1279             final var address = account.getRecipientAddressResolver()
1280                     .resolveRecipientAddress(context.getRecipientHelper().resolveRecipient(recipient));
1281             if (address.serviceId().isEmpty()) {
1282                 return List.of();
1283             }
1284             serviceId = address.serviceId().get();
1285         } catch (UnregisteredRecipientException e) {
1286             return List.of();
1287         }
1288         final var identity = account.getIdentityKeyStore().getIdentityInfo(serviceId);
1289         return identity == null ? List.of() : List.of(toIdentity(identity));
1290     }
1291
1292     @Override
1293     public boolean trustIdentityVerified(
1294             RecipientIdentifier.Single recipient, IdentityVerificationCode verificationCode
1295     ) throws UnregisteredRecipientException {
1296         return switch (verificationCode) {
1297             case IdentityVerificationCode.Fingerprint fingerprint -> trustIdentity(recipient,
1298                     r -> context.getIdentityHelper().trustIdentityVerified(r, fingerprint.fingerprint()));
1299             case IdentityVerificationCode.SafetyNumber safetyNumber -> trustIdentity(recipient,
1300                     r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber.safetyNumber()));
1301             case IdentityVerificationCode.ScannableSafetyNumber safetyNumber -> trustIdentity(recipient,
1302                     r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber.safetyNumber()));
1303             case null -> throw new AssertionError("Invalid verification code type");
1304         };
1305     }
1306
1307     @Override
1308     public boolean trustIdentityAllKeys(RecipientIdentifier.Single recipient) throws UnregisteredRecipientException {
1309         return trustIdentity(recipient, r -> context.getIdentityHelper().trustIdentityAllKeys(r));
1310     }
1311
1312     private boolean trustIdentity(
1313             RecipientIdentifier.Single recipient, Function<RecipientId, Boolean> trustMethod
1314     ) throws UnregisteredRecipientException {
1315         final var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1316         final var updated = trustMethod.apply(recipientId);
1317         if (updated && this.isReceiving()) {
1318             context.getReceiveHelper().setNeedsToRetryFailedMessages(true);
1319         }
1320         return updated;
1321     }
1322
1323     @Override
1324     public void addAddressChangedListener(final Runnable listener) {
1325         synchronized (addressChangedListeners) {
1326             addressChangedListeners.add(listener);
1327         }
1328     }
1329
1330     @Override
1331     public void addClosedListener(final Runnable listener) {
1332         synchronized (closedListeners) {
1333             closedListeners.add(listener);
1334         }
1335     }
1336
1337     @Override
1338     public InputStream retrieveAttachment(final String id) throws IOException {
1339         return context.getAttachmentHelper().retrieveAttachment(id).getStream();
1340     }
1341
1342     @Override
1343     public void close() {
1344         Thread thread;
1345         synchronized (messageHandlers) {
1346             weakHandlers.clear();
1347             messageHandlers.clear();
1348             thread = receiveThread;
1349             receiveThread = null;
1350         }
1351         if (thread != null) {
1352             stopReceiveThread(thread);
1353         }
1354         executor.close();
1355         context.close();
1356
1357         dependencies.getSignalWebSocket().disconnect();
1358         dependencies.getPushServiceSocket().close();
1359         disposable.dispose();
1360
1361         if (account != null) {
1362             account.close();
1363         }
1364
1365         synchronized (closedListeners) {
1366             closedListeners.forEach(Runnable::run);
1367             closedListeners.clear();
1368         }
1369
1370         account = null;
1371     }
1372 }