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