]> nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/ManagerImpl.java
Update README.md
[signal-cli] / 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 @Override
296 public String setUsername(final String username) throws IOException, InvalidUsernameException {
297 try {
298 return context.getAccountHelper().reserveUsername(username);
299 } catch (BaseUsernameException e) {
300 throw new InvalidUsernameException(e.getMessage() + " (" + e.getClass().getSimpleName() + ")", e);
301 }
302 }
303
304 @Override
305 public void deleteUsername() throws IOException {
306 context.getAccountHelper().deleteUsername();
307 }
308
309 @Override
310 public void unregister() throws IOException {
311 context.getAccountHelper().unregister();
312 }
313
314 @Override
315 public void deleteAccount() throws IOException {
316 context.getAccountHelper().deleteAccount();
317 }
318
319 @Override
320 public void submitRateLimitRecaptchaChallenge(String challenge, String captcha) throws IOException {
321 captcha = captcha == null ? null : captcha.replace("signalcaptcha://", "");
322
323 dependencies.getAccountManager().submitRateLimitRecaptchaChallenge(challenge, captcha);
324 }
325
326 @Override
327 public List<Device> getLinkedDevices() throws IOException {
328 var devices = dependencies.getAccountManager().getDevices();
329 account.setMultiDevice(devices.size() > 1);
330 var identityKey = account.getAciIdentityKeyPair().getPrivateKey();
331 return devices.stream().map(d -> {
332 String deviceName = d.getName();
333 if (deviceName != null) {
334 try {
335 deviceName = DeviceNameUtil.decryptDeviceName(deviceName, identityKey);
336 } catch (IOException e) {
337 logger.debug("Failed to decrypt device name, maybe plain text?", e);
338 }
339 }
340 return new Device(d.getId(),
341 deviceName,
342 d.getCreated(),
343 d.getLastSeen(),
344 d.getId() == account.getDeviceId());
345 }).toList();
346 }
347
348 @Override
349 public void removeLinkedDevices(int deviceId) throws IOException {
350 context.getAccountHelper().removeLinkedDevices(deviceId);
351 }
352
353 @Override
354 public void addDeviceLink(URI linkUri) throws IOException, InvalidDeviceLinkException {
355 var deviceLinkInfo = DeviceLinkInfo.parseDeviceLinkUri(linkUri);
356 context.getAccountHelper().addDevice(deviceLinkInfo);
357 }
358
359 @Override
360 public void setRegistrationLockPin(Optional<String> pin) throws IOException, NotPrimaryDeviceException {
361 if (!account.isPrimaryDevice()) {
362 throw new NotPrimaryDeviceException();
363 }
364 if (pin.isPresent()) {
365 context.getAccountHelper().setRegistrationPin(pin.get());
366 } else {
367 context.getAccountHelper().removeRegistrationPin();
368 }
369 }
370
371 void refreshPreKeys() throws IOException {
372 context.getPreKeyHelper().refreshPreKeys();
373 }
374
375 @Override
376 public Profile getRecipientProfile(RecipientIdentifier.Single recipient) throws UnregisteredRecipientException {
377 return context.getProfileHelper().getRecipientProfile(context.getRecipientHelper().resolveRecipient(recipient));
378 }
379
380 @Override
381 public List<Group> getGroups() {
382 return account.getGroupStore().getGroups().stream().map(this::toGroup).toList();
383 }
384
385 private Group toGroup(final GroupInfo groupInfo) {
386 if (groupInfo == null) {
387 return null;
388 }
389
390 return Group.from(groupInfo, account.getRecipientAddressResolver(), account.getSelfRecipientId());
391 }
392
393 @Override
394 public SendGroupMessageResults quitGroup(
395 GroupId groupId, Set<RecipientIdentifier.Single> groupAdmins
396 ) throws GroupNotFoundException, IOException, NotAGroupMemberException, LastGroupAdminException, UnregisteredRecipientException {
397 final var newAdmins = context.getRecipientHelper().resolveRecipients(groupAdmins);
398 return context.getGroupHelper().quitGroup(groupId, newAdmins);
399 }
400
401 @Override
402 public void deleteGroup(GroupId groupId) throws IOException {
403 final var group = context.getGroupHelper().getGroup(groupId);
404 if (group.isMember(account.getSelfRecipientId())) {
405 throw new IOException(
406 "The local group information cannot be removed, as the user is still a member of the group");
407 }
408 context.getGroupHelper().deleteGroup(groupId);
409 }
410
411 @Override
412 public Pair<GroupId, SendGroupMessageResults> createGroup(
413 String name, Set<RecipientIdentifier.Single> members, String avatarFile
414 ) throws IOException, AttachmentInvalidException, UnregisteredRecipientException {
415 return context.getGroupHelper()
416 .createGroup(name,
417 members == null ? null : context.getRecipientHelper().resolveRecipients(members),
418 avatarFile);
419 }
420
421 @Override
422 public SendGroupMessageResults updateGroup(
423 final GroupId groupId, final UpdateGroup updateGroup
424 ) throws IOException, GroupNotFoundException, AttachmentInvalidException, NotAGroupMemberException, GroupSendingNotAllowedException, UnregisteredRecipientException {
425 return context.getGroupHelper()
426 .updateGroup(groupId,
427 updateGroup.getName(),
428 updateGroup.getDescription(),
429 updateGroup.getMembers() == null
430 ? null
431 : context.getRecipientHelper().resolveRecipients(updateGroup.getMembers()),
432 updateGroup.getRemoveMembers() == null
433 ? null
434 : context.getRecipientHelper().resolveRecipients(updateGroup.getRemoveMembers()),
435 updateGroup.getAdmins() == null
436 ? null
437 : context.getRecipientHelper().resolveRecipients(updateGroup.getAdmins()),
438 updateGroup.getRemoveAdmins() == null
439 ? null
440 : context.getRecipientHelper().resolveRecipients(updateGroup.getRemoveAdmins()),
441 updateGroup.getBanMembers() == null
442 ? null
443 : context.getRecipientHelper().resolveRecipients(updateGroup.getBanMembers()),
444 updateGroup.getUnbanMembers() == null
445 ? null
446 : context.getRecipientHelper().resolveRecipients(updateGroup.getUnbanMembers()),
447 updateGroup.isResetGroupLink(),
448 updateGroup.getGroupLinkState(),
449 updateGroup.getAddMemberPermission(),
450 updateGroup.getEditDetailsPermission(),
451 updateGroup.getAvatarFile(),
452 updateGroup.getExpirationTimer(),
453 updateGroup.getIsAnnouncementGroup());
454 }
455
456 @Override
457 public Pair<GroupId, SendGroupMessageResults> joinGroup(
458 GroupInviteLinkUrl inviteLinkUrl
459 ) throws IOException, InactiveGroupLinkException, PendingAdminApprovalException {
460 return context.getGroupHelper().joinGroup(inviteLinkUrl);
461 }
462
463 private SendMessageResults sendMessage(
464 SignalServiceDataMessage.Builder messageBuilder, Set<RecipientIdentifier> recipients
465 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
466 var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
467 long timestamp = System.currentTimeMillis();
468 messageBuilder.withTimestamp(timestamp);
469 for (final var recipient : recipients) {
470 if (recipient instanceof RecipientIdentifier.Single single) {
471 try {
472 final var recipientId = context.getRecipientHelper().resolveRecipient(single);
473 final var result = context.getSendHelper().sendMessage(messageBuilder, recipientId);
474 results.put(recipient, List.of(toSendMessageResult(result)));
475 } catch (UnregisteredRecipientException e) {
476 results.put(recipient,
477 List.of(SendMessageResult.unregisteredFailure(single.toPartialRecipientAddress())));
478 }
479 } else if (recipient instanceof RecipientIdentifier.NoteToSelf) {
480 final var result = context.getSendHelper().sendSelfMessage(messageBuilder);
481 results.put(recipient, List.of(toSendMessageResult(result)));
482 } else if (recipient instanceof RecipientIdentifier.Group group) {
483 final var result = context.getSendHelper().sendAsGroupMessage(messageBuilder, group.groupId());
484 results.put(recipient, result.stream().map(this::toSendMessageResult).toList());
485 }
486 }
487 return new SendMessageResults(timestamp, results);
488 }
489
490 private SendMessageResult toSendMessageResult(final org.whispersystems.signalservice.api.messages.SendMessageResult result) {
491 return SendMessageResult.from(result, account.getRecipientResolver(), account.getRecipientAddressResolver());
492 }
493
494 private SendMessageResults sendTypingMessage(
495 SignalServiceTypingMessage.Action action, Set<RecipientIdentifier> recipients
496 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
497 var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
498 final var timestamp = System.currentTimeMillis();
499 for (var recipient : recipients) {
500 if (recipient instanceof RecipientIdentifier.Single single) {
501 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.empty());
502 try {
503 final var recipientId = context.getRecipientHelper().resolveRecipient(single);
504 final var result = context.getSendHelper().sendTypingMessage(message, recipientId);
505 results.put(recipient, List.of(toSendMessageResult(result)));
506 } catch (UnregisteredRecipientException e) {
507 results.put(recipient,
508 List.of(SendMessageResult.unregisteredFailure(single.toPartialRecipientAddress())));
509 }
510 } else if (recipient instanceof RecipientIdentifier.Group) {
511 final var groupId = ((RecipientIdentifier.Group) recipient).groupId();
512 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.of(groupId.serialize()));
513 final var result = context.getSendHelper().sendGroupTypingMessage(message, groupId);
514 results.put(recipient, result.stream().map(this::toSendMessageResult).toList());
515 }
516 }
517 return new SendMessageResults(timestamp, results);
518 }
519
520 @Override
521 public SendMessageResults sendTypingMessage(
522 TypingAction action, Set<RecipientIdentifier> recipients
523 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
524 return sendTypingMessage(action.toSignalService(), recipients);
525 }
526
527 @Override
528 public SendMessageResults sendReadReceipt(
529 RecipientIdentifier.Single sender, List<Long> messageIds
530 ) {
531 final var timestamp = System.currentTimeMillis();
532 var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.READ,
533 messageIds,
534 timestamp);
535
536 return sendReceiptMessage(sender, timestamp, receiptMessage);
537 }
538
539 @Override
540 public SendMessageResults sendViewedReceipt(
541 RecipientIdentifier.Single sender, List<Long> messageIds
542 ) {
543 final var timestamp = System.currentTimeMillis();
544 var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.VIEWED,
545 messageIds,
546 timestamp);
547
548 return sendReceiptMessage(sender, timestamp, receiptMessage);
549 }
550
551 private SendMessageResults sendReceiptMessage(
552 final RecipientIdentifier.Single sender,
553 final long timestamp,
554 final SignalServiceReceiptMessage receiptMessage
555 ) {
556 try {
557 final var result = context.getSendHelper()
558 .sendReceiptMessage(receiptMessage, context.getRecipientHelper().resolveRecipient(sender));
559 return new SendMessageResults(timestamp, Map.of(sender, List.of(toSendMessageResult(result))));
560 } catch (UnregisteredRecipientException e) {
561 return new SendMessageResults(timestamp,
562 Map.of(sender, List.of(SendMessageResult.unregisteredFailure(sender.toPartialRecipientAddress()))));
563 }
564 }
565
566 @Override
567 public SendMessageResults sendMessage(
568 Message message, Set<RecipientIdentifier> recipients
569 ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException, InvalidStickerException {
570 final var selfProfile = context.getProfileHelper().getSelfProfile();
571 if (selfProfile == null || selfProfile.getDisplayName().isEmpty()) {
572 logger.warn(
573 "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.");
574 }
575 final var messageBuilder = SignalServiceDataMessage.newBuilder();
576 applyMessage(messageBuilder, message);
577 return sendMessage(messageBuilder, recipients);
578 }
579
580 private void applyMessage(
581 final SignalServiceDataMessage.Builder messageBuilder, final Message message
582 ) throws AttachmentInvalidException, IOException, UnregisteredRecipientException, InvalidStickerException {
583 if (message.messageText().length() > 2000) {
584 final var messageBytes = message.messageText().getBytes(StandardCharsets.UTF_8);
585 final var textAttachment = AttachmentUtils.createAttachmentStream(new StreamDetails(new ByteArrayInputStream(
586 messageBytes), MimeUtils.LONG_TEXT, messageBytes.length), Optional.empty());
587 messageBuilder.withBody(message.messageText().substring(0, 2000));
588 messageBuilder.withAttachment(context.getAttachmentHelper().uploadAttachment(textAttachment));
589 } else {
590 messageBuilder.withBody(message.messageText());
591 }
592 if (message.attachments().size() > 0) {
593 messageBuilder.withAttachments(context.getAttachmentHelper().uploadAttachments(message.attachments()));
594 }
595 if (message.mentions().size() > 0) {
596 messageBuilder.withMentions(resolveMentions(message.mentions()));
597 }
598 if (message.quote().isPresent()) {
599 final var quote = message.quote().get();
600 messageBuilder.withQuote(new SignalServiceDataMessage.Quote(quote.timestamp(),
601 context.getRecipientHelper()
602 .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(quote.author()))
603 .getServiceId(),
604 quote.message(),
605 List.of(),
606 resolveMentions(quote.mentions()),
607 SignalServiceDataMessage.Quote.Type.NORMAL,
608 List.of()));
609 }
610 if (message.sticker().isPresent()) {
611 final var sticker = message.sticker().get();
612 final var packId = StickerPackId.deserialize(sticker.packId());
613 final var stickerId = sticker.stickerId();
614
615 final var stickerPack = context.getAccount().getStickerStore().getStickerPack(packId);
616 if (stickerPack == null) {
617 throw new InvalidStickerException("Sticker pack not found");
618 }
619 final var manifest = context.getStickerHelper().getOrRetrieveStickerPack(packId, stickerPack.packKey());
620 if (manifest.stickers().size() <= stickerId) {
621 throw new InvalidStickerException("Sticker id not part of this pack");
622 }
623 final var manifestSticker = manifest.stickers().get(stickerId);
624 final var streamDetails = context.getStickerPackStore().retrieveSticker(packId, stickerId);
625 if (streamDetails == null) {
626 throw new InvalidStickerException("Missing local sticker file");
627 }
628 messageBuilder.withSticker(new SignalServiceDataMessage.Sticker(packId.serialize(),
629 stickerPack.packKey(),
630 stickerId,
631 manifestSticker.emoji(),
632 AttachmentUtils.createAttachmentStream(streamDetails, Optional.empty())));
633 }
634 if (message.previews().size() > 0) {
635 final var previews = new ArrayList<SignalServicePreview>(message.previews().size());
636 for (final var p : message.previews()) {
637 final var image = p.image().isPresent() ? context.getAttachmentHelper()
638 .uploadAttachment(p.image().get()) : null;
639 previews.add(new SignalServicePreview(p.url(),
640 p.title(),
641 p.description(),
642 0,
643 Optional.ofNullable(image)));
644 }
645 messageBuilder.withPreviews(previews);
646 }
647 if (message.storyReply().isPresent()) {
648 final var storyReply = message.storyReply().get();
649 final var authorServiceId = context.getRecipientHelper()
650 .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(storyReply.author()))
651 .getServiceId();
652 messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
653 storyReply.timestamp()));
654 }
655 }
656
657 private ArrayList<SignalServiceDataMessage.Mention> resolveMentions(final List<Message.Mention> mentionList) throws UnregisteredRecipientException {
658 final var mentions = new ArrayList<SignalServiceDataMessage.Mention>();
659 for (final var m : mentionList) {
660 final var recipientId = context.getRecipientHelper().resolveRecipient(m.recipient());
661 mentions.add(new SignalServiceDataMessage.Mention(context.getRecipientHelper()
662 .resolveSignalServiceAddress(recipientId)
663 .getServiceId(), m.start(), m.length()));
664 }
665 return mentions;
666 }
667
668 @Override
669 public SendMessageResults sendRemoteDeleteMessage(
670 long targetSentTimestamp, Set<RecipientIdentifier> recipients
671 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
672 var delete = new SignalServiceDataMessage.RemoteDelete(targetSentTimestamp);
673 final var messageBuilder = SignalServiceDataMessage.newBuilder().withRemoteDelete(delete);
674 for (final var recipient : recipients) {
675 if (recipient instanceof RecipientIdentifier.Single r) {
676 try {
677 final var recipientId = context.getRecipientHelper().resolveRecipient(r);
678 account.getMessageSendLogStore()
679 .deleteEntryForRecipientNonGroup(targetSentTimestamp,
680 account.getRecipientAddressResolver()
681 .resolveRecipientAddress(recipientId)
682 .getServiceId());
683 } catch (UnregisteredRecipientException ignored) {
684 }
685 } else if (recipient instanceof RecipientIdentifier.Group r) {
686 account.getMessageSendLogStore().deleteEntryForGroup(targetSentTimestamp, r.groupId());
687 }
688 }
689 return sendMessage(messageBuilder, recipients);
690 }
691
692 @Override
693 public SendMessageResults sendMessageReaction(
694 String emoji,
695 boolean remove,
696 RecipientIdentifier.Single targetAuthor,
697 long targetSentTimestamp,
698 Set<RecipientIdentifier> recipients,
699 final boolean isStory
700 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException {
701 var targetAuthorRecipientId = context.getRecipientHelper().resolveRecipient(targetAuthor);
702 final var authorServiceId = context.getRecipientHelper()
703 .resolveSignalServiceAddress(targetAuthorRecipientId)
704 .getServiceId();
705 var reaction = new SignalServiceDataMessage.Reaction(emoji, remove, authorServiceId, targetSentTimestamp);
706 final var messageBuilder = SignalServiceDataMessage.newBuilder().withReaction(reaction);
707 if (isStory) {
708 messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
709 targetSentTimestamp));
710 }
711 return sendMessage(messageBuilder, recipients);
712 }
713
714 @Override
715 public SendMessageResults sendPaymentNotificationMessage(
716 byte[] receipt, String note, RecipientIdentifier.Single recipient
717 ) throws IOException {
718 final var paymentNotification = new SignalServiceDataMessage.PaymentNotification(receipt, note);
719 final var payment = new SignalServiceDataMessage.Payment(paymentNotification, null);
720 final var messageBuilder = SignalServiceDataMessage.newBuilder().withPayment(payment);
721 try {
722 return sendMessage(messageBuilder, Set.of(recipient));
723 } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
724 throw new AssertionError(e);
725 }
726 }
727
728 @Override
729 public SendMessageResults sendEndSessionMessage(Set<RecipientIdentifier.Single> recipients) throws IOException {
730 var messageBuilder = SignalServiceDataMessage.newBuilder().asEndSessionMessage();
731
732 try {
733 return sendMessage(messageBuilder,
734 recipients.stream().map(RecipientIdentifier.class::cast).collect(Collectors.toSet()));
735 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
736 throw new AssertionError(e);
737 } finally {
738 for (var recipient : recipients) {
739 final RecipientId recipientId;
740 try {
741 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
742 } catch (UnregisteredRecipientException e) {
743 continue;
744 }
745 final var serviceId = context.getAccount()
746 .getRecipientAddressResolver()
747 .resolveRecipientAddress(recipientId)
748 .getServiceId();
749 account.getAciSessionStore().deleteAllSessions(serviceId);
750 }
751 }
752 }
753
754 @Override
755 public void deleteRecipient(final RecipientIdentifier.Single recipient) {
756 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
757 if (recipientIdOptional.isPresent()) {
758 account.removeRecipient(recipientIdOptional.get());
759 }
760 }
761
762 @Override
763 public void deleteContact(final RecipientIdentifier.Single recipient) {
764 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
765 if (recipientIdOptional.isPresent()) {
766 account.getContactStore().deleteContact(recipientIdOptional.get());
767 }
768 }
769
770 @Override
771 public void setContactName(
772 RecipientIdentifier.Single recipient, String givenName, final String familyName
773 ) throws NotPrimaryDeviceException, UnregisteredRecipientException {
774 if (!account.isPrimaryDevice()) {
775 throw new NotPrimaryDeviceException();
776 }
777 context.getContactHelper()
778 .setContactName(context.getRecipientHelper().resolveRecipient(recipient), givenName, familyName);
779 }
780
781 @Override
782 public void setContactsBlocked(
783 Collection<RecipientIdentifier.Single> recipients, boolean blocked
784 ) throws NotPrimaryDeviceException, IOException, UnregisteredRecipientException {
785 if (!account.isPrimaryDevice()) {
786 throw new NotPrimaryDeviceException();
787 }
788 if (recipients.size() == 0) {
789 return;
790 }
791 final var recipientIds = context.getRecipientHelper().resolveRecipients(recipients);
792 final var selfRecipientId = account.getSelfRecipientId();
793 boolean shouldRotateProfileKey = false;
794 for (final var recipientId : recipientIds) {
795 if (context.getContactHelper().isContactBlocked(recipientId) == blocked) {
796 continue;
797 }
798 context.getContactHelper().setContactBlocked(recipientId, blocked);
799 // if we don't have a common group with the blocked contact we need to rotate the profile key
800 shouldRotateProfileKey = blocked && (
801 shouldRotateProfileKey || account.getGroupStore()
802 .getGroups()
803 .stream()
804 .noneMatch(g -> g.isMember(selfRecipientId) && g.isMember(recipientId))
805 );
806 }
807 if (shouldRotateProfileKey) {
808 context.getProfileHelper().rotateProfileKey();
809 }
810 context.getSyncHelper().sendBlockedList();
811 }
812
813 @Override
814 public void setGroupsBlocked(
815 final Collection<GroupId> groupIds, final boolean blocked
816 ) throws GroupNotFoundException, NotPrimaryDeviceException, IOException {
817 if (!account.isPrimaryDevice()) {
818 throw new NotPrimaryDeviceException();
819 }
820 if (groupIds.size() == 0) {
821 return;
822 }
823 boolean shouldRotateProfileKey = false;
824 for (final var groupId : groupIds) {
825 if (context.getGroupHelper().isGroupBlocked(groupId) == blocked) {
826 continue;
827 }
828 context.getGroupHelper().setGroupBlocked(groupId, blocked);
829 shouldRotateProfileKey = blocked;
830 }
831 if (shouldRotateProfileKey) {
832 context.getProfileHelper().rotateProfileKey();
833 }
834 context.getSyncHelper().sendBlockedList();
835 }
836
837 @Override
838 public void setExpirationTimer(
839 RecipientIdentifier.Single recipient, int messageExpirationTimer
840 ) throws IOException, UnregisteredRecipientException {
841 var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
842 context.getContactHelper().setExpirationTimer(recipientId, messageExpirationTimer);
843 final var messageBuilder = SignalServiceDataMessage.newBuilder().asExpirationUpdate();
844 try {
845 sendMessage(messageBuilder, Set.of(recipient));
846 } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
847 throw new AssertionError(e);
848 }
849 }
850
851 @Override
852 public StickerPackUrl uploadStickerPack(File path) throws IOException, StickerPackInvalidException {
853 var manifest = StickerUtils.getSignalServiceStickerManifestUpload(path);
854
855 var messageSender = dependencies.getMessageSender();
856
857 var packKey = KeyUtils.createStickerUploadKey();
858 var packIdString = messageSender.uploadStickerManifest(manifest, packKey);
859 var packId = StickerPackId.deserialize(Hex.fromStringCondensed(packIdString));
860
861 var sticker = new StickerPack(packId, packKey);
862 account.getStickerStore().addStickerPack(sticker);
863
864 return new StickerPackUrl(packId, packKey);
865 }
866
867 @Override
868 public List<org.asamk.signal.manager.api.StickerPack> getStickerPacks() {
869 final var stickerPackStore = context.getStickerPackStore();
870 return account.getStickerStore().getStickerPacks().stream().map(pack -> {
871 if (stickerPackStore.existsStickerPack(pack.packId())) {
872 try {
873 final var manifest = stickerPackStore.retrieveManifest(pack.packId());
874 return new org.asamk.signal.manager.api.StickerPack(pack.packId(),
875 new StickerPackUrl(pack.packId(), pack.packKey()),
876 pack.isInstalled(),
877 manifest.title(),
878 manifest.author(),
879 Optional.ofNullable(manifest.cover() == null ? null : manifest.cover().toApi()),
880 manifest.stickers().stream().map(JsonStickerPack.JsonSticker::toApi).toList());
881 } catch (Exception e) {
882 logger.warn("Failed to read local sticker pack manifest: {}", e.getMessage(), e);
883 }
884 }
885
886 return new org.asamk.signal.manager.api.StickerPack(pack.packId(), pack.packKey(), pack.isInstalled());
887 }).toList();
888 }
889
890 @Override
891 public void requestAllSyncData() throws IOException {
892 context.getSyncHelper().requestAllSyncData();
893 retrieveRemoteStorage();
894 }
895
896 void retrieveRemoteStorage() throws IOException {
897 context.getStorageHelper().readDataFromStorage();
898 }
899
900 @Override
901 public void addReceiveHandler(final ReceiveMessageHandler handler, final boolean isWeakListener) {
902 synchronized (messageHandlers) {
903 if (isWeakListener) {
904 weakHandlers.add(handler);
905 } else {
906 messageHandlers.add(handler);
907 startReceiveThreadIfRequired();
908 }
909 }
910 }
911
912 private static final AtomicInteger threadNumber = new AtomicInteger(0);
913
914 private void startReceiveThreadIfRequired() {
915 if (receiveThread != null || isReceivingSynchronous) {
916 return;
917 }
918 receiveThread = new Thread(() -> {
919 logger.debug("Starting receiving messages");
920 context.getReceiveHelper().receiveMessagesContinuously(this::passReceivedMessageToHandlers);
921 logger.debug("Finished receiving messages");
922 synchronized (messageHandlers) {
923 receiveThread = null;
924
925 // Check if in the meantime another handler has been registered
926 if (!messageHandlers.isEmpty()) {
927 logger.debug("Another handler has been registered, starting receive thread again");
928 startReceiveThreadIfRequired();
929 }
930 }
931 });
932 receiveThread.setName("receive-" + threadNumber.getAndIncrement());
933
934 receiveThread.start();
935 }
936
937 private void passReceivedMessageToHandlers(MessageEnvelope envelope, Throwable e) {
938 synchronized (messageHandlers) {
939 Stream.concat(messageHandlers.stream(), weakHandlers.stream()).forEach(h -> {
940 try {
941 h.handleMessage(envelope, e);
942 } catch (Throwable ex) {
943 logger.warn("Message handler failed, ignoring", ex);
944 }
945 });
946 }
947 }
948
949 @Override
950 public void removeReceiveHandler(final ReceiveMessageHandler handler) {
951 final Thread thread;
952 synchronized (messageHandlers) {
953 weakHandlers.remove(handler);
954 messageHandlers.remove(handler);
955 if (!messageHandlers.isEmpty() || receiveThread == null || isReceivingSynchronous) {
956 return;
957 }
958 thread = receiveThread;
959 receiveThread = null;
960 }
961
962 stopReceiveThread(thread);
963 }
964
965 private void stopReceiveThread(final Thread thread) {
966 if (context.getReceiveHelper().requestStopReceiveMessages()) {
967 logger.debug("Receive stop requested, interrupting read from server.");
968 thread.interrupt();
969 }
970 try {
971 thread.join();
972 } catch (InterruptedException ignored) {
973 }
974 }
975
976 @Override
977 public boolean isReceiving() {
978 if (isReceivingSynchronous) {
979 return true;
980 }
981 synchronized (messageHandlers) {
982 return messageHandlers.size() > 0;
983 }
984 }
985
986 @Override
987 public void receiveMessages(
988 Optional<Duration> timeout, Optional<Integer> maxMessages, ReceiveMessageHandler handler
989 ) throws IOException, AlreadyReceivingException {
990 receiveMessages(timeout.orElse(Duration.ofMinutes(1)), timeout.isPresent(), maxMessages.orElse(null), handler);
991 }
992
993 private void receiveMessages(
994 Duration timeout, boolean returnOnTimeout, Integer maxMessages, ReceiveMessageHandler handler
995 ) throws IOException, AlreadyReceivingException {
996 synchronized (messageHandlers) {
997 if (isReceiving()) {
998 throw new AlreadyReceivingException("Already receiving message.");
999 }
1000 isReceivingSynchronous = true;
1001 receiveThread = Thread.currentThread();
1002 }
1003 try {
1004 context.getReceiveHelper().receiveMessages(timeout, returnOnTimeout, maxMessages, (envelope, e) -> {
1005 passReceivedMessageToHandlers(envelope, e);
1006 handler.handleMessage(envelope, e);
1007 });
1008 } finally {
1009 synchronized (messageHandlers) {
1010 receiveThread = null;
1011 isReceivingSynchronous = false;
1012 if (messageHandlers.size() > 0) {
1013 startReceiveThreadIfRequired();
1014 }
1015 }
1016 }
1017 }
1018
1019 @Override
1020 public void setReceiveConfig(final ReceiveConfig receiveConfig) {
1021 context.getReceiveHelper().setReceiveConfig(receiveConfig);
1022 }
1023
1024 @Override
1025 public boolean hasCaughtUpWithOldMessages() {
1026 return context.getReceiveHelper().hasCaughtUpWithOldMessages();
1027 }
1028
1029 @Override
1030 public boolean isContactBlocked(final RecipientIdentifier.Single recipient) {
1031 final RecipientId recipientId;
1032 try {
1033 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1034 } catch (UnregisteredRecipientException e) {
1035 return false;
1036 }
1037 return context.getContactHelper().isContactBlocked(recipientId);
1038 }
1039
1040 @Override
1041 public void sendContacts() throws IOException {
1042 context.getSyncHelper().sendContacts();
1043 }
1044
1045 @Override
1046 public List<Recipient> getRecipients(
1047 boolean onlyContacts,
1048 Optional<Boolean> blocked,
1049 Collection<RecipientIdentifier.Single> recipients,
1050 Optional<String> name
1051 ) {
1052 final var recipientIds = recipients.stream().map(a -> {
1053 try {
1054 return context.getRecipientHelper().resolveRecipient(a);
1055 } catch (UnregisteredRecipientException e) {
1056 return null;
1057 }
1058 }).filter(Objects::nonNull).collect(Collectors.toSet());
1059 if (!recipients.isEmpty() && recipientIds.isEmpty()) {
1060 return List.of();
1061 }
1062 // refresh profiles of explicitly given recipients
1063 context.getProfileHelper().refreshRecipientProfiles(recipientIds);
1064 return account.getRecipientStore()
1065 .getRecipients(onlyContacts, blocked, recipientIds, name)
1066 .stream()
1067 .map(s -> new Recipient(s.getRecipientId(),
1068 s.getAddress().toApiRecipientAddress(),
1069 s.getContact(),
1070 s.getProfileKey(),
1071 s.getExpiringProfileKeyCredential(),
1072 s.getProfile()))
1073 .toList();
1074 }
1075
1076 @Override
1077 public String getContactOrProfileName(RecipientIdentifier.Single recipient) {
1078 final RecipientId recipientId;
1079 try {
1080 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1081 } catch (UnregisteredRecipientException e) {
1082 return null;
1083 }
1084
1085 final var contact = account.getContactStore().getContact(recipientId);
1086 if (contact != null && !Util.isEmpty(contact.getName())) {
1087 return contact.getName();
1088 }
1089
1090 final var profile = context.getProfileHelper().getRecipientProfile(recipientId);
1091 if (profile != null) {
1092 return profile.getDisplayName();
1093 }
1094
1095 return null;
1096 }
1097
1098 @Override
1099 public Group getGroup(GroupId groupId) {
1100 return toGroup(context.getGroupHelper().getGroup(groupId));
1101 }
1102
1103 @Override
1104 public List<Identity> getIdentities() {
1105 return account.getIdentityKeyStore().getIdentities().stream().map(this::toIdentity).toList();
1106 }
1107
1108 private Identity toIdentity(final IdentityInfo identityInfo) {
1109 if (identityInfo == null) {
1110 return null;
1111 }
1112
1113 final var address = account.getRecipientAddressResolver()
1114 .resolveRecipientAddress(account.getRecipientResolver().resolveRecipient(identityInfo.getServiceId()));
1115 final var scannableFingerprint = context.getIdentityHelper()
1116 .computeSafetyNumberForScanning(identityInfo.getServiceId(), identityInfo.getIdentityKey());
1117 return new Identity(address.toApiRecipientAddress(),
1118 identityInfo.getIdentityKey(),
1119 context.getIdentityHelper()
1120 .computeSafetyNumber(identityInfo.getServiceId(), identityInfo.getIdentityKey()),
1121 scannableFingerprint == null ? null : scannableFingerprint.getSerialized(),
1122 identityInfo.getTrustLevel(),
1123 identityInfo.getDateAddedTimestamp());
1124 }
1125
1126 @Override
1127 public List<Identity> getIdentities(RecipientIdentifier.Single recipient) {
1128 ServiceId serviceId;
1129 try {
1130 serviceId = account.getRecipientAddressResolver()
1131 .resolveRecipientAddress(context.getRecipientHelper().resolveRecipient(recipient))
1132 .getServiceId();
1133 } catch (UnregisteredRecipientException e) {
1134 return List.of();
1135 }
1136 final var identity = account.getIdentityKeyStore().getIdentityInfo(serviceId);
1137 return identity == null ? List.of() : List.of(toIdentity(identity));
1138 }
1139
1140 @Override
1141 public boolean trustIdentityVerified(
1142 RecipientIdentifier.Single recipient, byte[] fingerprint
1143 ) throws UnregisteredRecipientException {
1144 return trustIdentity(recipient, r -> context.getIdentityHelper().trustIdentityVerified(r, fingerprint));
1145 }
1146
1147 @Override
1148 public boolean trustIdentityVerifiedSafetyNumber(
1149 RecipientIdentifier.Single recipient, String safetyNumber
1150 ) throws UnregisteredRecipientException {
1151 return trustIdentity(recipient,
1152 r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber));
1153 }
1154
1155 @Override
1156 public boolean trustIdentityVerifiedSafetyNumber(
1157 RecipientIdentifier.Single recipient, byte[] safetyNumber
1158 ) throws UnregisteredRecipientException {
1159 return trustIdentity(recipient,
1160 r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber));
1161 }
1162
1163 @Override
1164 public boolean trustIdentityAllKeys(RecipientIdentifier.Single recipient) throws UnregisteredRecipientException {
1165 return trustIdentity(recipient, r -> context.getIdentityHelper().trustIdentityAllKeys(r));
1166 }
1167
1168 private boolean trustIdentity(
1169 RecipientIdentifier.Single recipient, Function<RecipientId, Boolean> trustMethod
1170 ) throws UnregisteredRecipientException {
1171 final var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1172 final var updated = trustMethod.apply(recipientId);
1173 if (updated && this.isReceiving()) {
1174 context.getReceiveHelper().setNeedsToRetryFailedMessages(true);
1175 }
1176 return updated;
1177 }
1178
1179 @Override
1180 public void addAddressChangedListener(final Runnable listener) {
1181 synchronized (addressChangedListeners) {
1182 addressChangedListeners.add(listener);
1183 }
1184 }
1185
1186 @Override
1187 public void addClosedListener(final Runnable listener) {
1188 synchronized (closedListeners) {
1189 closedListeners.add(listener);
1190 }
1191 }
1192
1193 @Override
1194 public InputStream retrieveAttachment(final String id) throws IOException {
1195 return context.getAttachmentHelper().retrieveAttachment(id).getStream();
1196 }
1197
1198 @Override
1199 public void close() {
1200 Thread thread;
1201 synchronized (messageHandlers) {
1202 weakHandlers.clear();
1203 messageHandlers.clear();
1204 thread = receiveThread;
1205 receiveThread = null;
1206 }
1207 if (thread != null) {
1208 stopReceiveThread(thread);
1209 }
1210 executor.shutdown();
1211
1212 dependencies.getSignalWebSocket().disconnect();
1213 disposable.dispose();
1214
1215 if (account != null) {
1216 account.close();
1217 }
1218
1219 synchronized (closedListeners) {
1220 closedListeners.forEach(Runnable::run);
1221 closedListeners.clear();
1222 }
1223
1224 account = null;
1225 }
1226 }