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