]> nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/internal/ManagerImpl.java
Document listStickerPacks command
[signal-cli] / lib / src / main / java / org / asamk / signal / manager / internal / ManagerImpl.java
1 /*
2 Copyright (C) 2015-2022 AsamK and contributors
3
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
16 */
17 package org.asamk.signal.manager.internal;
18
19 import org.asamk.signal.manager.Manager;
20 import org.asamk.signal.manager.api.AlreadyReceivingException;
21 import org.asamk.signal.manager.api.AttachmentInvalidException;
22 import org.asamk.signal.manager.api.Configuration;
23 import org.asamk.signal.manager.api.Device;
24 import org.asamk.signal.manager.api.DeviceLinkUrl;
25 import org.asamk.signal.manager.api.Group;
26 import org.asamk.signal.manager.api.GroupId;
27 import org.asamk.signal.manager.api.GroupInviteLinkUrl;
28 import org.asamk.signal.manager.api.GroupNotFoundException;
29 import org.asamk.signal.manager.api.GroupSendingNotAllowedException;
30 import org.asamk.signal.manager.api.Identity;
31 import org.asamk.signal.manager.api.IdentityVerificationCode;
32 import org.asamk.signal.manager.api.InactiveGroupLinkException;
33 import org.asamk.signal.manager.api.InvalidDeviceLinkException;
34 import org.asamk.signal.manager.api.InvalidStickerException;
35 import org.asamk.signal.manager.api.InvalidUsernameException;
36 import org.asamk.signal.manager.api.LastGroupAdminException;
37 import org.asamk.signal.manager.api.Message;
38 import org.asamk.signal.manager.api.MessageEnvelope;
39 import org.asamk.signal.manager.api.NotAGroupMemberException;
40 import org.asamk.signal.manager.api.NotPrimaryDeviceException;
41 import org.asamk.signal.manager.api.Pair;
42 import org.asamk.signal.manager.api.PendingAdminApprovalException;
43 import org.asamk.signal.manager.api.Profile;
44 import org.asamk.signal.manager.api.ReceiveConfig;
45 import org.asamk.signal.manager.api.Recipient;
46 import org.asamk.signal.manager.api.RecipientIdentifier;
47 import org.asamk.signal.manager.api.SendGroupMessageResults;
48 import org.asamk.signal.manager.api.SendMessageResult;
49 import org.asamk.signal.manager.api.SendMessageResults;
50 import org.asamk.signal.manager.api.StickerPackId;
51 import org.asamk.signal.manager.api.StickerPackInvalidException;
52 import org.asamk.signal.manager.api.StickerPackUrl;
53 import org.asamk.signal.manager.api.TextStyle;
54 import org.asamk.signal.manager.api.TypingAction;
55 import org.asamk.signal.manager.api.UnregisteredRecipientException;
56 import org.asamk.signal.manager.api.UpdateGroup;
57 import org.asamk.signal.manager.api.UpdateProfile;
58 import org.asamk.signal.manager.api.UserStatus;
59 import org.asamk.signal.manager.config.ServiceEnvironmentConfig;
60 import org.asamk.signal.manager.helper.AccountFileUpdater;
61 import org.asamk.signal.manager.helper.Context;
62 import org.asamk.signal.manager.storage.AttachmentStore;
63 import org.asamk.signal.manager.storage.AvatarStore;
64 import org.asamk.signal.manager.storage.SignalAccount;
65 import org.asamk.signal.manager.storage.groups.GroupInfo;
66 import org.asamk.signal.manager.storage.identities.IdentityInfo;
67 import org.asamk.signal.manager.storage.recipients.RecipientId;
68 import org.asamk.signal.manager.storage.stickerPacks.JsonStickerPack;
69 import org.asamk.signal.manager.storage.stickerPacks.StickerPackStore;
70 import org.asamk.signal.manager.storage.stickers.StickerPack;
71 import org.asamk.signal.manager.util.AttachmentUtils;
72 import org.asamk.signal.manager.util.KeyUtils;
73 import org.asamk.signal.manager.util.MimeUtils;
74 import org.asamk.signal.manager.util.StickerUtils;
75 import org.signal.libsignal.usernames.BaseUsernameException;
76 import org.slf4j.Logger;
77 import org.slf4j.LoggerFactory;
78 import org.whispersystems.signalservice.api.SignalSessionLock;
79 import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
80 import org.whispersystems.signalservice.api.messages.SignalServicePreview;
81 import org.whispersystems.signalservice.api.messages.SignalServiceReceiptMessage;
82 import org.whispersystems.signalservice.api.messages.SignalServiceTypingMessage;
83 import org.whispersystems.signalservice.api.push.ServiceId;
84 import org.whispersystems.signalservice.api.push.ServiceId.ACI;
85 import org.whispersystems.signalservice.api.push.ServiceIdType;
86 import org.whispersystems.signalservice.api.util.DeviceNameUtil;
87 import org.whispersystems.signalservice.api.util.InvalidNumberException;
88 import org.whispersystems.signalservice.api.util.PhoneNumberFormatter;
89 import org.whispersystems.signalservice.api.util.StreamDetails;
90 import org.whispersystems.signalservice.internal.util.Hex;
91 import org.whispersystems.signalservice.internal.util.Util;
92
93 import java.io.ByteArrayInputStream;
94 import java.io.File;
95 import java.io.IOException;
96 import java.io.InputStream;
97 import java.nio.charset.StandardCharsets;
98 import java.time.Duration;
99 import java.util.ArrayList;
100 import java.util.Collection;
101 import java.util.HashMap;
102 import java.util.HashSet;
103 import java.util.List;
104 import java.util.Map;
105 import java.util.Objects;
106 import java.util.Optional;
107 import java.util.Set;
108 import java.util.concurrent.ExecutorService;
109 import java.util.concurrent.Executors;
110 import java.util.concurrent.atomic.AtomicInteger;
111 import java.util.concurrent.locks.ReentrantLock;
112 import java.util.function.Function;
113 import java.util.stream.Collectors;
114 import java.util.stream.Stream;
115
116 import io.reactivex.rxjava3.disposables.CompositeDisposable;
117
118 public class ManagerImpl implements Manager {
119
120 private final static Logger logger = LoggerFactory.getLogger(ManagerImpl.class);
121
122 private SignalAccount account;
123 private final SignalDependencies dependencies;
124 private final Context context;
125
126 private final ExecutorService executor = Executors.newCachedThreadPool();
127
128 private Thread receiveThread;
129 private boolean isReceivingSynchronous;
130 private final Set<ReceiveMessageHandler> weakHandlers = new HashSet<>();
131 private final Set<ReceiveMessageHandler> messageHandlers = new HashSet<>();
132 private final List<Runnable> closedListeners = new ArrayList<>();
133 private final List<Runnable> addressChangedListeners = new ArrayList<>();
134 private final CompositeDisposable disposable = new CompositeDisposable();
135
136 public ManagerImpl(
137 SignalAccount account,
138 PathConfig pathConfig,
139 AccountFileUpdater accountFileUpdater,
140 ServiceEnvironmentConfig serviceEnvironmentConfig,
141 String userAgent
142 ) {
143 this.account = account;
144
145 final var sessionLock = new SignalSessionLock() {
146 private final ReentrantLock LEGACY_LOCK = new ReentrantLock();
147
148 @Override
149 public Lock acquire() {
150 LEGACY_LOCK.lock();
151 return LEGACY_LOCK::unlock;
152 }
153 };
154 this.dependencies = new SignalDependencies(serviceEnvironmentConfig,
155 userAgent,
156 account.getCredentialsProvider(),
157 account.getSignalServiceDataStore(),
158 executor,
159 sessionLock);
160 final var avatarStore = new AvatarStore(pathConfig.avatarsPath());
161 final var attachmentStore = new AttachmentStore(pathConfig.attachmentsPath());
162 final var stickerPackStore = new StickerPackStore(pathConfig.stickerPacksPath());
163
164 this.context = new Context(account, new AccountFileUpdater() {
165 @Override
166 public void updateAccountIdentifiers(final String number, final ACI aci) {
167 accountFileUpdater.updateAccountIdentifiers(number, aci);
168 synchronized (addressChangedListeners) {
169 addressChangedListeners.forEach(Runnable::run);
170 }
171 }
172
173 @Override
174 public void removeAccount() {
175 accountFileUpdater.removeAccount();
176 }
177 }, dependencies, avatarStore, attachmentStore, stickerPackStore);
178 this.context.getAccountHelper().setUnregisteredListener(this::close);
179 this.context.getReceiveHelper().setAuthenticationFailureListener(this::close);
180 this.context.getReceiveHelper().setCaughtUpWithOldMessagesListener(() -> {
181 synchronized (this) {
182 this.notifyAll();
183 }
184 });
185 disposable.add(account.getIdentityKeyStore().getIdentityChanges().subscribe(serviceId -> {
186 logger.trace("Archiving old sessions for {}", serviceId);
187 account.getAccountData(ServiceIdType.ACI).getSessionStore().archiveSessions(serviceId);
188 account.getAccountData(ServiceIdType.PNI).getSessionStore().archiveSessions(serviceId);
189 account.getSenderKeyStore().deleteSharedWith(serviceId);
190 final var recipientId = account.getRecipientResolver().resolveRecipient(serviceId);
191 final var profile = account.getProfileStore().getProfile(recipientId);
192 if (profile != null) {
193 account.getProfileStore()
194 .storeProfile(recipientId,
195 Profile.newBuilder(profile)
196 .withUnidentifiedAccessMode(Profile.UnidentifiedAccessMode.UNKNOWN)
197 .withLastUpdateTimestamp(0)
198 .build());
199 }
200 }));
201 }
202
203 @Override
204 public String getSelfNumber() {
205 return account.getNumber();
206 }
207
208 public void checkAccountState() throws IOException {
209 context.getAccountHelper().checkAccountState();
210 }
211
212 @Override
213 public Map<String, UserStatus> getUserStatus(Set<String> numbers) throws IOException {
214 final var canonicalizedNumbers = numbers.stream().collect(Collectors.toMap(n -> n, n -> {
215 try {
216 final var canonicalizedNumber = PhoneNumberFormatter.formatNumber(n, account.getNumber());
217 if (!canonicalizedNumber.equals(n)) {
218 logger.debug("Normalized number {} to {}.", n, canonicalizedNumber);
219 }
220 return canonicalizedNumber;
221 } catch (InvalidNumberException e) {
222 return "";
223 }
224 }));
225
226 // Note "registeredUsers" has no optionals. It only gives us info on users who are registered
227 final var canonicalizedNumbersSet = canonicalizedNumbers.values()
228 .stream()
229 .filter(s -> !s.isEmpty())
230 .collect(Collectors.toSet());
231 final var registeredUsers = context.getRecipientHelper().getRegisteredUsers(canonicalizedNumbersSet);
232
233 return numbers.stream().collect(Collectors.toMap(n -> n, n -> {
234 final var number = canonicalizedNumbers.get(n);
235 final var user = registeredUsers.get(number);
236 final var serviceId = user == null ? null : user.getServiceId();
237 final var profile = serviceId == null
238 ? null
239 : context.getProfileHelper()
240 .getRecipientProfile(account.getRecipientResolver().resolveRecipient(serviceId));
241 return new UserStatus(number.isEmpty() ? null : number,
242 serviceId == null ? null : serviceId.getRawUuid(),
243 profile != null
244 && profile.getUnidentifiedAccessMode() == Profile.UnidentifiedAccessMode.UNRESTRICTED);
245 }));
246 }
247
248 @Override
249 public void updateAccountAttributes(String deviceName) throws IOException {
250 if (deviceName != null) {
251 context.getAccountHelper().setDeviceName(deviceName);
252 }
253 context.getAccountHelper().updateAccountAttributes();
254 context.getAccountHelper().checkWhoAmiI();
255 }
256
257 @Override
258 public Configuration getConfiguration() {
259 final var configurationStore = account.getConfigurationStore();
260 return Configuration.from(configurationStore);
261 }
262
263 @Override
264 public void updateConfiguration(
265 Configuration configuration
266 ) throws NotPrimaryDeviceException {
267 if (!account.isPrimaryDevice()) {
268 throw new NotPrimaryDeviceException();
269 }
270
271 final var configurationStore = account.getConfigurationStore();
272 if (configuration.readReceipts().isPresent()) {
273 configurationStore.setReadReceipts(configuration.readReceipts().get());
274 }
275 if (configuration.unidentifiedDeliveryIndicators().isPresent()) {
276 configurationStore.setUnidentifiedDeliveryIndicators(configuration.unidentifiedDeliveryIndicators().get());
277 }
278 if (configuration.typingIndicators().isPresent()) {
279 configurationStore.setTypingIndicators(configuration.typingIndicators().get());
280 }
281 if (configuration.linkPreviews().isPresent()) {
282 configurationStore.setLinkPreviews(configuration.linkPreviews().get());
283 }
284 context.getSyncHelper().sendConfigurationMessage();
285 }
286
287 @Override
288 public void updateProfile(UpdateProfile updateProfile) throws IOException {
289 context.getProfileHelper()
290 .setProfile(updateProfile.getGivenName(),
291 updateProfile.getFamilyName(),
292 updateProfile.getAbout(),
293 updateProfile.getAboutEmoji(),
294 updateProfile.isDeleteAvatar()
295 ? Optional.empty()
296 : updateProfile.getAvatar() == null ? null : Optional.of(updateProfile.getAvatar()),
297 updateProfile.getMobileCoinAddress());
298 context.getSyncHelper().sendSyncFetchProfileMessage();
299 }
300
301 void refreshCurrentUsername() throws IOException, BaseUsernameException {
302 context.getAccountHelper().refreshCurrentUsername();
303 }
304
305 @Override
306 public String setUsername(final String username) throws IOException, InvalidUsernameException {
307 try {
308 return context.getAccountHelper().reserveUsername(username);
309 } catch (BaseUsernameException e) {
310 throw new InvalidUsernameException(e.getMessage() + " (" + e.getClass().getSimpleName() + ")", e);
311 }
312 }
313
314 @Override
315 public void deleteUsername() throws IOException {
316 context.getAccountHelper().deleteUsername();
317 }
318
319 @Override
320 public void unregister() throws IOException {
321 context.getAccountHelper().unregister();
322 }
323
324 @Override
325 public void deleteAccount() throws IOException {
326 context.getAccountHelper().deleteAccount();
327 }
328
329 @Override
330 public void submitRateLimitRecaptchaChallenge(String challenge, String captcha) throws IOException {
331 captcha = captcha == null ? null : captcha.replace("signalcaptcha://", "");
332
333 dependencies.getAccountManager().submitRateLimitRecaptchaChallenge(challenge, captcha);
334 }
335
336 @Override
337 public List<Device> getLinkedDevices() throws IOException {
338 var devices = dependencies.getAccountManager().getDevices();
339 account.setMultiDevice(devices.size() > 1);
340 var identityKey = account.getAciIdentityKeyPair().getPrivateKey();
341 return devices.stream().map(d -> {
342 String deviceName = d.getName();
343 if (deviceName != null) {
344 try {
345 deviceName = DeviceNameUtil.decryptDeviceName(deviceName, identityKey);
346 } catch (IOException e) {
347 logger.debug("Failed to decrypt device name, maybe plain text?", e);
348 }
349 }
350 return new Device(d.getId(),
351 deviceName,
352 d.getCreated(),
353 d.getLastSeen(),
354 d.getId() == account.getDeviceId());
355 }).toList();
356 }
357
358 @Override
359 public void removeLinkedDevices(int deviceId) throws IOException {
360 context.getAccountHelper().removeLinkedDevices(deviceId);
361 }
362
363 @Override
364 public void addDeviceLink(DeviceLinkUrl linkUrl) throws IOException, InvalidDeviceLinkException {
365 context.getAccountHelper().addDevice(linkUrl);
366 }
367
368 @Override
369 public void setRegistrationLockPin(Optional<String> pin) throws IOException, NotPrimaryDeviceException {
370 if (!account.isPrimaryDevice()) {
371 throw new NotPrimaryDeviceException();
372 }
373 if (pin.isPresent()) {
374 context.getAccountHelper().setRegistrationPin(pin.get());
375 } else {
376 context.getAccountHelper().removeRegistrationPin();
377 }
378 }
379
380 void refreshPreKeys() throws IOException {
381 context.getPreKeyHelper().refreshPreKeysIfNecessary();
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.textStyles().size() > 0) {
622 messageBuilder.withBodyRanges(message.textStyles().stream().map(TextStyle::toBodyRange).toList());
623 }
624 if (message.quote().isPresent()) {
625 final var quote = message.quote().get();
626 messageBuilder.withQuote(new SignalServiceDataMessage.Quote(quote.timestamp(),
627 context.getRecipientHelper()
628 .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(quote.author()))
629 .getServiceId(),
630 quote.message(),
631 List.of(),
632 resolveMentions(quote.mentions()),
633 SignalServiceDataMessage.Quote.Type.NORMAL,
634 quote.textStyles().stream().map(TextStyle::toBodyRange).toList()));
635 }
636 if (message.sticker().isPresent()) {
637 final var sticker = message.sticker().get();
638 final var packId = StickerPackId.deserialize(sticker.packId());
639 final var stickerId = sticker.stickerId();
640
641 final var stickerPack = context.getAccount().getStickerStore().getStickerPack(packId);
642 if (stickerPack == null) {
643 throw new InvalidStickerException("Sticker pack not found");
644 }
645 final var manifest = context.getStickerHelper().getOrRetrieveStickerPack(packId, stickerPack.packKey());
646 if (manifest.stickers().size() <= stickerId) {
647 throw new InvalidStickerException("Sticker id not part of this pack");
648 }
649 final var manifestSticker = manifest.stickers().get(stickerId);
650 final var streamDetails = context.getStickerPackStore().retrieveSticker(packId, stickerId);
651 if (streamDetails == null) {
652 throw new InvalidStickerException("Missing local sticker file");
653 }
654 messageBuilder.withSticker(new SignalServiceDataMessage.Sticker(packId.serialize(),
655 stickerPack.packKey(),
656 stickerId,
657 manifestSticker.emoji(),
658 AttachmentUtils.createAttachmentStream(streamDetails, Optional.empty())));
659 }
660 if (message.previews().size() > 0) {
661 final var previews = new ArrayList<SignalServicePreview>(message.previews().size());
662 for (final var p : message.previews()) {
663 final var image = p.image().isPresent() ? context.getAttachmentHelper()
664 .uploadAttachment(p.image().get()) : null;
665 previews.add(new SignalServicePreview(p.url(),
666 p.title(),
667 p.description(),
668 0,
669 Optional.ofNullable(image)));
670 }
671 messageBuilder.withPreviews(previews);
672 }
673 if (message.storyReply().isPresent()) {
674 final var storyReply = message.storyReply().get();
675 final var authorServiceId = context.getRecipientHelper()
676 .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(storyReply.author()))
677 .getServiceId();
678 messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
679 storyReply.timestamp()));
680 }
681 }
682
683 private ArrayList<SignalServiceDataMessage.Mention> resolveMentions(final List<Message.Mention> mentionList) throws UnregisteredRecipientException {
684 final var mentions = new ArrayList<SignalServiceDataMessage.Mention>();
685 for (final var m : mentionList) {
686 final var recipientId = context.getRecipientHelper().resolveRecipient(m.recipient());
687 mentions.add(new SignalServiceDataMessage.Mention(context.getRecipientHelper()
688 .resolveSignalServiceAddress(recipientId)
689 .getServiceId(), m.start(), m.length()));
690 }
691 return mentions;
692 }
693
694 @Override
695 public SendMessageResults sendRemoteDeleteMessage(
696 long targetSentTimestamp, Set<RecipientIdentifier> recipients
697 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
698 var delete = new SignalServiceDataMessage.RemoteDelete(targetSentTimestamp);
699 final var messageBuilder = SignalServiceDataMessage.newBuilder().withRemoteDelete(delete);
700 for (final var recipient : recipients) {
701 if (recipient instanceof RecipientIdentifier.Uuid u) {
702 account.getMessageSendLogStore()
703 .deleteEntryForRecipientNonGroup(targetSentTimestamp, ACI.from(u.uuid()));
704 } else if (recipient instanceof RecipientIdentifier.Single r) {
705 try {
706 final var recipientId = context.getRecipientHelper().resolveRecipient(r);
707 final var address = account.getRecipientAddressResolver().resolveRecipientAddress(recipientId);
708 if (address.serviceId().isPresent()) {
709 account.getMessageSendLogStore()
710 .deleteEntryForRecipientNonGroup(targetSentTimestamp, address.serviceId().get());
711 }
712 } catch (UnregisteredRecipientException ignored) {
713 }
714 } else if (recipient instanceof RecipientIdentifier.Group r) {
715 account.getMessageSendLogStore().deleteEntryForGroup(targetSentTimestamp, r.groupId());
716 }
717 }
718 return sendMessage(messageBuilder, recipients);
719 }
720
721 @Override
722 public SendMessageResults sendMessageReaction(
723 String emoji,
724 boolean remove,
725 RecipientIdentifier.Single targetAuthor,
726 long targetSentTimestamp,
727 Set<RecipientIdentifier> recipients,
728 final boolean isStory
729 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException {
730 var targetAuthorRecipientId = context.getRecipientHelper().resolveRecipient(targetAuthor);
731 final var authorServiceId = context.getRecipientHelper()
732 .resolveSignalServiceAddress(targetAuthorRecipientId)
733 .getServiceId();
734 var reaction = new SignalServiceDataMessage.Reaction(emoji, remove, authorServiceId, targetSentTimestamp);
735 final var messageBuilder = SignalServiceDataMessage.newBuilder().withReaction(reaction);
736 if (isStory) {
737 messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
738 targetSentTimestamp));
739 }
740 return sendMessage(messageBuilder, recipients);
741 }
742
743 @Override
744 public SendMessageResults sendPaymentNotificationMessage(
745 byte[] receipt, String note, RecipientIdentifier.Single recipient
746 ) throws IOException {
747 final var paymentNotification = new SignalServiceDataMessage.PaymentNotification(receipt, note);
748 final var payment = new SignalServiceDataMessage.Payment(paymentNotification, null);
749 final var messageBuilder = SignalServiceDataMessage.newBuilder().withPayment(payment);
750 try {
751 return sendMessage(messageBuilder, Set.of(recipient));
752 } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
753 throw new AssertionError(e);
754 }
755 }
756
757 @Override
758 public SendMessageResults sendEndSessionMessage(Set<RecipientIdentifier.Single> recipients) throws IOException {
759 var messageBuilder = SignalServiceDataMessage.newBuilder().asEndSessionMessage();
760
761 try {
762 return sendMessage(messageBuilder,
763 recipients.stream().map(RecipientIdentifier.class::cast).collect(Collectors.toSet()));
764 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
765 throw new AssertionError(e);
766 } finally {
767 for (var recipient : recipients) {
768 final RecipientId recipientId;
769 try {
770 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
771 } catch (UnregisteredRecipientException e) {
772 continue;
773 }
774 final var serviceId = context.getAccount()
775 .getRecipientAddressResolver()
776 .resolveRecipientAddress(recipientId)
777 .serviceId();
778 if (serviceId.isPresent()) {
779 account.getAccountData(ServiceIdType.ACI).getSessionStore().deleteAllSessions(serviceId.get());
780 }
781 }
782 }
783 }
784
785 @Override
786 public void deleteRecipient(final RecipientIdentifier.Single recipient) {
787 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
788 if (recipientIdOptional.isPresent()) {
789 account.removeRecipient(recipientIdOptional.get());
790 }
791 }
792
793 @Override
794 public void deleteContact(final RecipientIdentifier.Single recipient) {
795 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
796 if (recipientIdOptional.isPresent()) {
797 account.getContactStore().deleteContact(recipientIdOptional.get());
798 }
799 }
800
801 @Override
802 public void setContactName(
803 RecipientIdentifier.Single recipient, String givenName, final String familyName
804 ) throws NotPrimaryDeviceException, UnregisteredRecipientException {
805 if (!account.isPrimaryDevice()) {
806 throw new NotPrimaryDeviceException();
807 }
808 context.getContactHelper()
809 .setContactName(context.getRecipientHelper().resolveRecipient(recipient), givenName, familyName);
810 }
811
812 @Override
813 public void setContactsBlocked(
814 Collection<RecipientIdentifier.Single> recipients, boolean blocked
815 ) throws NotPrimaryDeviceException, IOException, UnregisteredRecipientException {
816 if (!account.isPrimaryDevice()) {
817 throw new NotPrimaryDeviceException();
818 }
819 if (recipients.size() == 0) {
820 return;
821 }
822 final var recipientIds = context.getRecipientHelper().resolveRecipients(recipients);
823 final var selfRecipientId = account.getSelfRecipientId();
824 boolean shouldRotateProfileKey = false;
825 for (final var recipientId : recipientIds) {
826 if (context.getContactHelper().isContactBlocked(recipientId) == blocked) {
827 continue;
828 }
829 context.getContactHelper().setContactBlocked(recipientId, blocked);
830 // if we don't have a common group with the blocked contact we need to rotate the profile key
831 shouldRotateProfileKey = blocked && (
832 shouldRotateProfileKey || account.getGroupStore()
833 .getGroups()
834 .stream()
835 .noneMatch(g -> g.isMember(selfRecipientId) && g.isMember(recipientId))
836 );
837 }
838 if (shouldRotateProfileKey) {
839 context.getProfileHelper().rotateProfileKey();
840 }
841 context.getSyncHelper().sendBlockedList();
842 }
843
844 @Override
845 public void setGroupsBlocked(
846 final Collection<GroupId> groupIds, final boolean blocked
847 ) throws GroupNotFoundException, NotPrimaryDeviceException, IOException {
848 if (!account.isPrimaryDevice()) {
849 throw new NotPrimaryDeviceException();
850 }
851 if (groupIds.size() == 0) {
852 return;
853 }
854 boolean shouldRotateProfileKey = false;
855 for (final var groupId : groupIds) {
856 if (context.getGroupHelper().isGroupBlocked(groupId) == blocked) {
857 continue;
858 }
859 context.getGroupHelper().setGroupBlocked(groupId, blocked);
860 shouldRotateProfileKey = blocked;
861 }
862 if (shouldRotateProfileKey) {
863 context.getProfileHelper().rotateProfileKey();
864 }
865 context.getSyncHelper().sendBlockedList();
866 }
867
868 @Override
869 public void setExpirationTimer(
870 RecipientIdentifier.Single recipient, int messageExpirationTimer
871 ) throws IOException, UnregisteredRecipientException {
872 var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
873 context.getContactHelper().setExpirationTimer(recipientId, messageExpirationTimer);
874 final var messageBuilder = SignalServiceDataMessage.newBuilder().asExpirationUpdate();
875 try {
876 sendMessage(messageBuilder, Set.of(recipient));
877 } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
878 throw new AssertionError(e);
879 }
880 }
881
882 @Override
883 public StickerPackUrl uploadStickerPack(File path) throws IOException, StickerPackInvalidException {
884 var manifest = StickerUtils.getSignalServiceStickerManifestUpload(path);
885
886 var messageSender = dependencies.getMessageSender();
887
888 var packKey = KeyUtils.createStickerUploadKey();
889 var packIdString = messageSender.uploadStickerManifest(manifest, packKey);
890 var packId = StickerPackId.deserialize(Hex.fromStringCondensed(packIdString));
891
892 var sticker = new StickerPack(packId, packKey);
893 account.getStickerStore().addStickerPack(sticker);
894
895 return new StickerPackUrl(packId, packKey);
896 }
897
898 @Override
899 public List<org.asamk.signal.manager.api.StickerPack> getStickerPacks() {
900 final var stickerPackStore = context.getStickerPackStore();
901 return account.getStickerStore().getStickerPacks().stream().map(pack -> {
902 if (stickerPackStore.existsStickerPack(pack.packId())) {
903 try {
904 final var manifest = stickerPackStore.retrieveManifest(pack.packId());
905 return new org.asamk.signal.manager.api.StickerPack(pack.packId(),
906 new StickerPackUrl(pack.packId(), pack.packKey()),
907 pack.isInstalled(),
908 manifest.title(),
909 manifest.author(),
910 Optional.ofNullable(manifest.cover() == null ? null : manifest.cover().toApi()),
911 manifest.stickers().stream().map(JsonStickerPack.JsonSticker::toApi).toList());
912 } catch (Exception e) {
913 logger.warn("Failed to read local sticker pack manifest: {}", e.getMessage(), e);
914 }
915 }
916
917 return new org.asamk.signal.manager.api.StickerPack(pack.packId(), pack.packKey(), pack.isInstalled());
918 }).toList();
919 }
920
921 @Override
922 public void requestAllSyncData() throws IOException {
923 context.getSyncHelper().requestAllSyncData();
924 retrieveRemoteStorage();
925 }
926
927 void retrieveRemoteStorage() throws IOException {
928 context.getStorageHelper().readDataFromStorage();
929 }
930
931 @Override
932 public void addReceiveHandler(final ReceiveMessageHandler handler, final boolean isWeakListener) {
933 synchronized (messageHandlers) {
934 if (isWeakListener) {
935 weakHandlers.add(handler);
936 } else {
937 messageHandlers.add(handler);
938 startReceiveThreadIfRequired();
939 }
940 }
941 }
942
943 private static final AtomicInteger threadNumber = new AtomicInteger(0);
944
945 private void startReceiveThreadIfRequired() {
946 if (receiveThread != null || isReceivingSynchronous) {
947 return;
948 }
949 receiveThread = new Thread(() -> {
950 logger.debug("Starting receiving messages");
951 context.getReceiveHelper().receiveMessagesContinuously(this::passReceivedMessageToHandlers);
952 logger.debug("Finished receiving messages");
953 synchronized (messageHandlers) {
954 receiveThread = null;
955
956 // Check if in the meantime another handler has been registered
957 if (!messageHandlers.isEmpty()) {
958 logger.debug("Another handler has been registered, starting receive thread again");
959 startReceiveThreadIfRequired();
960 }
961 }
962 });
963 receiveThread.setName("receive-" + threadNumber.getAndIncrement());
964
965 receiveThread.start();
966 }
967
968 private void passReceivedMessageToHandlers(MessageEnvelope envelope, Throwable e) {
969 synchronized (messageHandlers) {
970 Stream.concat(messageHandlers.stream(), weakHandlers.stream()).forEach(h -> {
971 try {
972 h.handleMessage(envelope, e);
973 } catch (Throwable ex) {
974 logger.warn("Message handler failed, ignoring", ex);
975 }
976 });
977 }
978 }
979
980 @Override
981 public void removeReceiveHandler(final ReceiveMessageHandler handler) {
982 final Thread thread;
983 synchronized (messageHandlers) {
984 weakHandlers.remove(handler);
985 messageHandlers.remove(handler);
986 if (!messageHandlers.isEmpty() || receiveThread == null || isReceivingSynchronous) {
987 return;
988 }
989 thread = receiveThread;
990 receiveThread = null;
991 }
992
993 stopReceiveThread(thread);
994 }
995
996 private void stopReceiveThread(final Thread thread) {
997 if (context.getReceiveHelper().requestStopReceiveMessages()) {
998 logger.debug("Receive stop requested, interrupting read from server.");
999 thread.interrupt();
1000 }
1001 try {
1002 thread.join();
1003 } catch (InterruptedException ignored) {
1004 }
1005 }
1006
1007 @Override
1008 public boolean isReceiving() {
1009 if (isReceivingSynchronous) {
1010 return true;
1011 }
1012 synchronized (messageHandlers) {
1013 return messageHandlers.size() > 0;
1014 }
1015 }
1016
1017 @Override
1018 public void receiveMessages(
1019 Optional<Duration> timeout, Optional<Integer> maxMessages, ReceiveMessageHandler handler
1020 ) throws IOException, AlreadyReceivingException {
1021 receiveMessages(timeout.orElse(Duration.ofMinutes(1)), timeout.isPresent(), maxMessages.orElse(null), handler);
1022 }
1023
1024 private void receiveMessages(
1025 Duration timeout, boolean returnOnTimeout, Integer maxMessages, ReceiveMessageHandler handler
1026 ) throws IOException, AlreadyReceivingException {
1027 synchronized (messageHandlers) {
1028 if (isReceiving()) {
1029 throw new AlreadyReceivingException("Already receiving message.");
1030 }
1031 isReceivingSynchronous = true;
1032 receiveThread = Thread.currentThread();
1033 }
1034 try {
1035 context.getReceiveHelper().receiveMessages(timeout, returnOnTimeout, maxMessages, (envelope, e) -> {
1036 passReceivedMessageToHandlers(envelope, e);
1037 handler.handleMessage(envelope, e);
1038 });
1039 } finally {
1040 synchronized (messageHandlers) {
1041 receiveThread = null;
1042 isReceivingSynchronous = false;
1043 if (messageHandlers.size() > 0) {
1044 startReceiveThreadIfRequired();
1045 }
1046 }
1047 }
1048 }
1049
1050 @Override
1051 public void setReceiveConfig(final ReceiveConfig receiveConfig) {
1052 context.getReceiveHelper().setReceiveConfig(receiveConfig);
1053 }
1054
1055 @Override
1056 public boolean isContactBlocked(final RecipientIdentifier.Single recipient) {
1057 final RecipientId recipientId;
1058 try {
1059 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1060 } catch (UnregisteredRecipientException e) {
1061 return false;
1062 }
1063 return context.getContactHelper().isContactBlocked(recipientId);
1064 }
1065
1066 @Override
1067 public void sendContacts() throws IOException {
1068 context.getSyncHelper().sendContacts();
1069 }
1070
1071 @Override
1072 public List<Recipient> getRecipients(
1073 boolean onlyContacts,
1074 Optional<Boolean> blocked,
1075 Collection<RecipientIdentifier.Single> recipients,
1076 Optional<String> name
1077 ) {
1078 final var recipientIds = recipients.stream().map(a -> {
1079 try {
1080 return context.getRecipientHelper().resolveRecipient(a);
1081 } catch (UnregisteredRecipientException e) {
1082 return null;
1083 }
1084 }).filter(Objects::nonNull).collect(Collectors.toSet());
1085 if (!recipients.isEmpty() && recipientIds.isEmpty()) {
1086 return List.of();
1087 }
1088 // refresh profiles of explicitly given recipients
1089 context.getProfileHelper().refreshRecipientProfiles(recipientIds);
1090 return account.getRecipientStore()
1091 .getRecipients(onlyContacts, blocked, recipientIds, name)
1092 .stream()
1093 .map(s -> new Recipient(s.getRecipientId(),
1094 s.getAddress().toApiRecipientAddress(),
1095 s.getContact(),
1096 s.getProfileKey(),
1097 s.getExpiringProfileKeyCredential(),
1098 s.getProfile()))
1099 .toList();
1100 }
1101
1102 @Override
1103 public String getContactOrProfileName(RecipientIdentifier.Single recipient) {
1104 final RecipientId recipientId;
1105 try {
1106 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1107 } catch (UnregisteredRecipientException e) {
1108 return null;
1109 }
1110
1111 final var contact = account.getContactStore().getContact(recipientId);
1112 if (contact != null && !Util.isEmpty(contact.getName())) {
1113 return contact.getName();
1114 }
1115
1116 final var profile = context.getProfileHelper().getRecipientProfile(recipientId);
1117 if (profile != null) {
1118 return profile.getDisplayName();
1119 }
1120
1121 return null;
1122 }
1123
1124 @Override
1125 public Group getGroup(GroupId groupId) {
1126 return toGroup(context.getGroupHelper().getGroup(groupId));
1127 }
1128
1129 @Override
1130 public List<Identity> getIdentities() {
1131 return account.getIdentityKeyStore().getIdentities().stream().map(this::toIdentity).toList();
1132 }
1133
1134 private Identity toIdentity(final IdentityInfo identityInfo) {
1135 if (identityInfo == null) {
1136 return null;
1137 }
1138
1139 final var address = account.getRecipientAddressResolver()
1140 .resolveRecipientAddress(account.getRecipientResolver().resolveRecipient(identityInfo.getServiceId()));
1141 final var scannableFingerprint = context.getIdentityHelper()
1142 .computeSafetyNumberForScanning(identityInfo.getServiceId(), identityInfo.getIdentityKey());
1143 return new Identity(address.toApiRecipientAddress(),
1144 identityInfo.getIdentityKey(),
1145 context.getIdentityHelper()
1146 .computeSafetyNumber(identityInfo.getServiceId(), identityInfo.getIdentityKey()),
1147 scannableFingerprint == null ? null : scannableFingerprint.getSerialized(),
1148 identityInfo.getTrustLevel(),
1149 identityInfo.getDateAddedTimestamp());
1150 }
1151
1152 @Override
1153 public List<Identity> getIdentities(RecipientIdentifier.Single recipient) {
1154 ServiceId serviceId;
1155 try {
1156 final var address = account.getRecipientAddressResolver()
1157 .resolveRecipientAddress(context.getRecipientHelper().resolveRecipient(recipient));
1158 if (address.serviceId().isEmpty()) {
1159 return List.of();
1160 }
1161 serviceId = address.serviceId().get();
1162 } catch (UnregisteredRecipientException e) {
1163 return List.of();
1164 }
1165 final var identity = account.getIdentityKeyStore().getIdentityInfo(serviceId);
1166 return identity == null ? List.of() : List.of(toIdentity(identity));
1167 }
1168
1169 @Override
1170 public boolean trustIdentityVerified(
1171 RecipientIdentifier.Single recipient, IdentityVerificationCode verificationCode
1172 ) throws UnregisteredRecipientException {
1173 if (verificationCode instanceof IdentityVerificationCode.Fingerprint fingerprint) {
1174 return trustIdentity(recipient,
1175 r -> context.getIdentityHelper().trustIdentityVerified(r, fingerprint.fingerprint()));
1176 } else if (verificationCode instanceof IdentityVerificationCode.SafetyNumber safetyNumber) {
1177 return trustIdentity(recipient,
1178 r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber.safetyNumber()));
1179 } else if (verificationCode instanceof IdentityVerificationCode.ScannableSafetyNumber safetyNumber) {
1180 return trustIdentity(recipient,
1181 r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber.safetyNumber()));
1182 } else {
1183 throw new AssertionError("Invalid verification code type");
1184 }
1185 }
1186
1187 @Override
1188 public boolean trustIdentityAllKeys(RecipientIdentifier.Single recipient) throws UnregisteredRecipientException {
1189 return trustIdentity(recipient, r -> context.getIdentityHelper().trustIdentityAllKeys(r));
1190 }
1191
1192 private boolean trustIdentity(
1193 RecipientIdentifier.Single recipient, Function<RecipientId, Boolean> trustMethod
1194 ) throws UnregisteredRecipientException {
1195 final var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1196 final var updated = trustMethod.apply(recipientId);
1197 if (updated && this.isReceiving()) {
1198 context.getReceiveHelper().setNeedsToRetryFailedMessages(true);
1199 }
1200 return updated;
1201 }
1202
1203 @Override
1204 public void addAddressChangedListener(final Runnable listener) {
1205 synchronized (addressChangedListeners) {
1206 addressChangedListeners.add(listener);
1207 }
1208 }
1209
1210 @Override
1211 public void addClosedListener(final Runnable listener) {
1212 synchronized (closedListeners) {
1213 closedListeners.add(listener);
1214 }
1215 }
1216
1217 @Override
1218 public InputStream retrieveAttachment(final String id) throws IOException {
1219 return context.getAttachmentHelper().retrieveAttachment(id).getStream();
1220 }
1221
1222 @Override
1223 public void close() {
1224 Thread thread;
1225 synchronized (messageHandlers) {
1226 weakHandlers.clear();
1227 messageHandlers.clear();
1228 thread = receiveThread;
1229 receiveThread = null;
1230 }
1231 if (thread != null) {
1232 stopReceiveThread(thread);
1233 }
1234 executor.shutdown();
1235
1236 dependencies.getSignalWebSocket().disconnect();
1237 disposable.dispose();
1238
1239 if (account != null) {
1240 account.close();
1241 }
1242
1243 synchronized (closedListeners) {
1244 closedListeners.forEach(Runnable::run);
1245 closedListeners.clear();
1246 }
1247
1248 account = null;
1249 }
1250 }