]> nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/dbus/DbusSignalImpl.java
Include unrestricted unidentified access state in getUserStatus response
[signal-cli] / src / main / java / org / asamk / signal / dbus / DbusSignalImpl.java
1 package org.asamk.signal.dbus;
2
3 import org.asamk.Signal;
4 import org.asamk.signal.BaseConfig;
5 import org.asamk.signal.manager.Manager;
6 import org.asamk.signal.manager.api.AttachmentInvalidException;
7 import org.asamk.signal.manager.api.Identity;
8 import org.asamk.signal.manager.api.InactiveGroupLinkException;
9 import org.asamk.signal.manager.api.InvalidDeviceLinkException;
10 import org.asamk.signal.manager.api.InvalidNumberException;
11 import org.asamk.signal.manager.api.InvalidStickerException;
12 import org.asamk.signal.manager.api.Message;
13 import org.asamk.signal.manager.api.NotMasterDeviceException;
14 import org.asamk.signal.manager.api.Pair;
15 import org.asamk.signal.manager.api.RecipientIdentifier;
16 import org.asamk.signal.manager.api.SendMessageResult;
17 import org.asamk.signal.manager.api.SendMessageResults;
18 import org.asamk.signal.manager.api.StickerPackInvalidException;
19 import org.asamk.signal.manager.api.TypingAction;
20 import org.asamk.signal.manager.api.UnregisteredRecipientException;
21 import org.asamk.signal.manager.api.UpdateGroup;
22 import org.asamk.signal.manager.api.UserStatus;
23 import org.asamk.signal.manager.groups.GroupId;
24 import org.asamk.signal.manager.groups.GroupInviteLinkUrl;
25 import org.asamk.signal.manager.groups.GroupLinkState;
26 import org.asamk.signal.manager.groups.GroupNotFoundException;
27 import org.asamk.signal.manager.groups.GroupPermission;
28 import org.asamk.signal.manager.groups.GroupSendingNotAllowedException;
29 import org.asamk.signal.manager.groups.LastGroupAdminException;
30 import org.asamk.signal.manager.groups.NotAGroupMemberException;
31 import org.asamk.signal.manager.storage.recipients.Profile;
32 import org.asamk.signal.manager.storage.recipients.RecipientAddress;
33 import org.asamk.signal.util.SendMessageResultUtils;
34 import org.freedesktop.dbus.DBusPath;
35 import org.freedesktop.dbus.connections.impl.DBusConnection;
36 import org.freedesktop.dbus.exceptions.DBusException;
37 import org.freedesktop.dbus.exceptions.DBusExecutionException;
38 import org.freedesktop.dbus.interfaces.DBusInterface;
39 import org.freedesktop.dbus.types.Variant;
40 import org.slf4j.Logger;
41 import org.slf4j.LoggerFactory;
42
43 import java.io.File;
44 import java.io.IOException;
45 import java.net.URI;
46 import java.net.URISyntaxException;
47 import java.util.ArrayList;
48 import java.util.Arrays;
49 import java.util.Base64;
50 import java.util.Collection;
51 import java.util.HashSet;
52 import java.util.List;
53 import java.util.Map;
54 import java.util.Objects;
55 import java.util.Optional;
56 import java.util.Set;
57 import java.util.stream.Collectors;
58 import java.util.stream.Stream;
59
60 public class DbusSignalImpl implements Signal {
61
62 private final Manager m;
63 private final DBusConnection connection;
64 private final String objectPath;
65 private final boolean noReceiveOnStart;
66
67 private DBusPath thisDevice;
68 private final List<StructDevice> devices = new ArrayList<>();
69 private final List<StructGroup> groups = new ArrayList<>();
70 private DbusReceiveMessageHandler dbusMessageHandler;
71 private int subscriberCount;
72
73 private final static Logger logger = LoggerFactory.getLogger(DbusSignalImpl.class);
74
75 public DbusSignalImpl(
76 final Manager m, DBusConnection connection, final String objectPath, final boolean noReceiveOnStart
77 ) {
78 this.m = m;
79 this.connection = connection;
80 this.objectPath = objectPath;
81 this.noReceiveOnStart = noReceiveOnStart;
82
83 m.addAddressChangedListener(() -> {
84 unExportObjects();
85 exportObjects();
86 });
87 }
88
89 public void initObjects() {
90 exportObjects();
91 if (!noReceiveOnStart) {
92 subscribeReceive();
93 }
94 }
95
96 private void exportObjects() {
97 exportObject(this);
98
99 updateDevices();
100 updateGroups();
101 updateConfiguration();
102 }
103
104 public void close() {
105 if (dbusMessageHandler != null) {
106 m.removeReceiveHandler(dbusMessageHandler);
107 dbusMessageHandler = null;
108 }
109 unExportObjects();
110 }
111
112 private void unExportObjects() {
113 unExportDevices();
114 unExportGroups();
115 unExportConfiguration();
116 connection.unExportObject(this.objectPath);
117 }
118
119 @Override
120 public String getObjectPath() {
121 return objectPath;
122 }
123
124 @Override
125 public String getSelfNumber() {
126 return m.getSelfNumber();
127 }
128
129 @Override
130 public void subscribeReceive() {
131 if (dbusMessageHandler == null) {
132 dbusMessageHandler = new DbusReceiveMessageHandler(connection, objectPath);
133 m.addReceiveHandler(dbusMessageHandler);
134 }
135 subscriberCount++;
136 }
137
138 @Override
139 public void unsubscribeReceive() {
140 subscriberCount = Math.max(0, subscriberCount - 1);
141 if (subscriberCount == 0 && dbusMessageHandler != null) {
142 m.removeReceiveHandler(dbusMessageHandler);
143 dbusMessageHandler = null;
144 }
145 }
146
147 @Override
148 public void submitRateLimitChallenge(String challenge, String captcha) {
149 try {
150 m.submitRateLimitRecaptchaChallenge(challenge, captcha);
151 } catch (IOException e) {
152 throw new Error.Failure("Submit challenge error: " + e.getMessage());
153 }
154
155 }
156
157 @Override
158 public void unregister() throws Error.Failure {
159 try {
160 m.unregister();
161 } catch (IOException e) {
162 throw new Error.Failure("Failed to unregister: " + e.getMessage());
163 }
164 }
165
166 @Override
167 public void deleteAccount() throws Error.Failure {
168 try {
169 m.deleteAccount();
170 } catch (IOException e) {
171 throw new Error.Failure("Failed to delete account: " + e.getMessage());
172 }
173 }
174
175 @Override
176 public void addDevice(String uri) {
177 try {
178 m.addDeviceLink(new URI(uri));
179 } catch (IOException | InvalidDeviceLinkException e) {
180 throw new Error.Failure(e.getClass().getSimpleName() + " Add device link failed. " + e.getMessage());
181 } catch (URISyntaxException e) {
182 throw new Error.InvalidUri(e.getClass().getSimpleName()
183 + " Device link uri has invalid format: "
184 + e.getMessage());
185 }
186 }
187
188 @Override
189 public DBusPath getDevice(long deviceId) {
190 updateDevices();
191 final var deviceOptional = devices.stream().filter(g -> g.getId().equals(deviceId)).findFirst();
192 if (deviceOptional.isEmpty()) {
193 throw new Error.DeviceNotFound("Device not found");
194 }
195 return deviceOptional.get().getObjectPath();
196 }
197
198 @Override
199 public List<StructDevice> listDevices() {
200 updateDevices();
201 return this.devices;
202 }
203
204 @Override
205 public DBusPath getThisDevice() {
206 updateDevices();
207 return thisDevice;
208 }
209
210 @Override
211 public long sendMessage(final String message, final List<String> attachments, final String recipient) {
212 return sendMessage(message, attachments, List.of(recipient));
213 }
214
215 @Override
216 public long sendMessage(final String message, final List<String> attachments, final List<String> recipients) {
217 try {
218 final var results = m.sendMessage(new Message(message,
219 attachments,
220 List.of(),
221 Optional.empty(),
222 Optional.empty()),
223 getSingleRecipientIdentifiers(recipients, m.getSelfNumber()).stream()
224 .map(RecipientIdentifier.class::cast)
225 .collect(Collectors.toSet()));
226
227 checkSendMessageResults(results);
228 return results.timestamp();
229 } catch (AttachmentInvalidException e) {
230 throw new Error.AttachmentInvalid(e.getMessage());
231 } catch (IOException | InvalidStickerException e) {
232 throw new Error.Failure(e);
233 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
234 throw new Error.GroupNotFound(e.getMessage());
235 } catch (UnregisteredRecipientException e) {
236 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
237 }
238 }
239
240 @Override
241 public long sendRemoteDeleteMessage(
242 final long targetSentTimestamp, final String recipient
243 ) {
244 return sendRemoteDeleteMessage(targetSentTimestamp, List.of(recipient));
245 }
246
247 @Override
248 public long sendRemoteDeleteMessage(
249 final long targetSentTimestamp, final List<String> recipients
250 ) {
251 try {
252 final var results = m.sendRemoteDeleteMessage(targetSentTimestamp,
253 getSingleRecipientIdentifiers(recipients, m.getSelfNumber()).stream()
254 .map(RecipientIdentifier.class::cast)
255 .collect(Collectors.toSet()));
256 checkSendMessageResults(results);
257 return results.timestamp();
258 } catch (IOException e) {
259 throw new Error.Failure(e.getMessage());
260 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
261 throw new Error.GroupNotFound(e.getMessage());
262 }
263 }
264
265 @Override
266 public long sendMessageReaction(
267 final String emoji,
268 final boolean remove,
269 final String targetAuthor,
270 final long targetSentTimestamp,
271 final String recipient
272 ) {
273 return sendMessageReaction(emoji, remove, targetAuthor, targetSentTimestamp, List.of(recipient));
274 }
275
276 @Override
277 public long sendMessageReaction(
278 final String emoji,
279 final boolean remove,
280 final String targetAuthor,
281 final long targetSentTimestamp,
282 final List<String> recipients
283 ) {
284 try {
285 final var results = m.sendMessageReaction(emoji,
286 remove,
287 getSingleRecipientIdentifier(targetAuthor, m.getSelfNumber()),
288 targetSentTimestamp,
289 getSingleRecipientIdentifiers(recipients, m.getSelfNumber()).stream()
290 .map(RecipientIdentifier.class::cast)
291 .collect(Collectors.toSet()));
292 checkSendMessageResults(results);
293 return results.timestamp();
294 } catch (IOException e) {
295 throw new Error.Failure(e.getMessage());
296 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
297 throw new Error.GroupNotFound(e.getMessage());
298 } catch (UnregisteredRecipientException e) {
299 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
300 }
301 }
302
303 @Override
304 public void sendTyping(
305 final String recipient, final boolean stop
306 ) throws Error.Failure, Error.GroupNotFound, Error.UntrustedIdentity {
307 try {
308 final var results = m.sendTypingMessage(stop ? TypingAction.STOP : TypingAction.START,
309 getSingleRecipientIdentifiers(List.of(recipient), m.getSelfNumber()).stream()
310 .map(RecipientIdentifier.class::cast)
311 .collect(Collectors.toSet()));
312 checkSendMessageResults(results);
313 } catch (IOException e) {
314 throw new Error.Failure(e.getMessage());
315 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
316 throw new Error.GroupNotFound(e.getMessage());
317 }
318 }
319
320 @Override
321 public void sendReadReceipt(
322 final String recipient, final List<Long> messageIds
323 ) throws Error.Failure, Error.UntrustedIdentity {
324 try {
325 final var results = m.sendReadReceipt(getSingleRecipientIdentifier(recipient, m.getSelfNumber()),
326 messageIds);
327 checkSendMessageResults(results);
328 } catch (IOException e) {
329 throw new Error.Failure(e.getMessage());
330 }
331 }
332
333 @Override
334 public void sendViewedReceipt(
335 final String recipient, final List<Long> messageIds
336 ) throws Error.Failure, Error.UntrustedIdentity {
337 try {
338 final var results = m.sendViewedReceipt(getSingleRecipientIdentifier(recipient, m.getSelfNumber()),
339 messageIds);
340 checkSendMessageResults(results);
341 } catch (IOException e) {
342 throw new Error.Failure(e.getMessage());
343 }
344 }
345
346 @Override
347 public void sendContacts() {
348 try {
349 m.sendContacts();
350 } catch (IOException e) {
351 throw new Error.Failure("SendContacts error: " + e.getMessage());
352 }
353 }
354
355 @Override
356 public void sendSyncRequest() {
357 try {
358 m.requestAllSyncData();
359 } catch (IOException e) {
360 throw new Error.Failure("Request sync data error: " + e.getMessage());
361 }
362 }
363
364 @Override
365 public long sendNoteToSelfMessage(
366 final String message, final List<String> attachments
367 ) throws Error.AttachmentInvalid, Error.Failure, Error.UntrustedIdentity {
368 try {
369 final var results = m.sendMessage(new Message(message,
370 attachments,
371 List.of(),
372 Optional.empty(),
373 Optional.empty()), Set.of(RecipientIdentifier.NoteToSelf.INSTANCE));
374 checkSendMessageResults(results);
375 return results.timestamp();
376 } catch (AttachmentInvalidException e) {
377 throw new Error.AttachmentInvalid(e.getMessage());
378 } catch (IOException | InvalidStickerException e) {
379 throw new Error.Failure(e.getMessage());
380 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
381 throw new Error.GroupNotFound(e.getMessage());
382 } catch (UnregisteredRecipientException e) {
383 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
384 }
385 }
386
387 @Override
388 public void sendEndSessionMessage(final List<String> recipients) {
389 try {
390 final var results = m.sendEndSessionMessage(getSingleRecipientIdentifiers(recipients, m.getSelfNumber()));
391 checkSendMessageResults(results);
392 } catch (IOException e) {
393 throw new Error.Failure(e.getMessage());
394 }
395 }
396
397 @Override
398 public void deleteRecipient(final String recipient) throws Error.Failure {
399 m.deleteRecipient(getSingleRecipientIdentifier(recipient, m.getSelfNumber()));
400 }
401
402 @Override
403 public void deleteContact(final String recipient) throws Error.Failure {
404 m.deleteContact(getSingleRecipientIdentifier(recipient, m.getSelfNumber()));
405 }
406
407 @Override
408 public long sendGroupMessage(final String message, final List<String> attachments, final byte[] groupId) {
409 try {
410 var results = m.sendMessage(new Message(message,
411 attachments,
412 List.of(),
413 Optional.empty(),
414 Optional.empty()), Set.of(getGroupRecipientIdentifier(groupId)));
415 checkSendMessageResults(results);
416 return results.timestamp();
417 } catch (IOException | InvalidStickerException e) {
418 throw new Error.Failure(e.getMessage());
419 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
420 throw new Error.GroupNotFound(e.getMessage());
421 } catch (AttachmentInvalidException e) {
422 throw new Error.AttachmentInvalid(e.getMessage());
423 } catch (UnregisteredRecipientException e) {
424 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
425 }
426 }
427
428 @Override
429 public void sendGroupTyping(
430 final byte[] groupId, final boolean stop
431 ) throws Error.Failure, Error.GroupNotFound, Error.UntrustedIdentity {
432 try {
433 final var results = m.sendTypingMessage(stop ? TypingAction.STOP : TypingAction.START,
434 Set.of(getGroupRecipientIdentifier(groupId)));
435 checkSendMessageResults(results);
436 } catch (IOException e) {
437 throw new Error.Failure(e.getMessage());
438 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
439 throw new Error.GroupNotFound(e.getMessage());
440 }
441 }
442
443 @Override
444 public long sendGroupRemoteDeleteMessage(
445 final long targetSentTimestamp, final byte[] groupId
446 ) {
447 try {
448 final var results = m.sendRemoteDeleteMessage(targetSentTimestamp,
449 Set.of(getGroupRecipientIdentifier(groupId)));
450 checkSendMessageResults(results);
451 return results.timestamp();
452 } catch (IOException e) {
453 throw new Error.Failure(e.getMessage());
454 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
455 throw new Error.GroupNotFound(e.getMessage());
456 }
457 }
458
459 @Override
460 public long sendGroupMessageReaction(
461 final String emoji,
462 final boolean remove,
463 final String targetAuthor,
464 final long targetSentTimestamp,
465 final byte[] groupId
466 ) {
467 try {
468 final var results = m.sendMessageReaction(emoji,
469 remove,
470 getSingleRecipientIdentifier(targetAuthor, m.getSelfNumber()),
471 targetSentTimestamp,
472 Set.of(getGroupRecipientIdentifier(groupId)));
473 checkSendMessageResults(results);
474 return results.timestamp();
475 } catch (IOException e) {
476 throw new Error.Failure(e.getMessage());
477 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
478 throw new Error.GroupNotFound(e.getMessage());
479 } catch (UnregisteredRecipientException e) {
480 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
481 }
482 }
483
484 // Since contact names might be empty if not defined, also potentially return
485 // the profile name
486 @Override
487 public String getContactName(final String number) {
488 final var name = m.getContactOrProfileName(getSingleRecipientIdentifier(number, m.getSelfNumber()));
489 return name == null ? "" : name;
490 }
491
492 @Override
493 public void setContactName(final String number, final String name) {
494 try {
495 m.setContactName(getSingleRecipientIdentifier(number, m.getSelfNumber()), name);
496 } catch (NotMasterDeviceException e) {
497 throw new Error.Failure("This command doesn't work on linked devices.");
498 } catch (IOException e) {
499 throw new Error.Failure("Contact is not registered.");
500 } catch (UnregisteredRecipientException e) {
501 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
502 }
503 }
504
505 @Override
506 public void setExpirationTimer(final String number, final int expiration) {
507 try {
508 m.setExpirationTimer(getSingleRecipientIdentifier(number, m.getSelfNumber()), expiration);
509 } catch (IOException e) {
510 throw new Error.Failure(e.getMessage());
511 } catch (UnregisteredRecipientException e) {
512 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
513 }
514 }
515
516 @Override
517 public void setContactBlocked(final String number, final boolean blocked) {
518 try {
519 m.setContactBlocked(getSingleRecipientIdentifier(number, m.getSelfNumber()), blocked);
520 } catch (NotMasterDeviceException e) {
521 throw new Error.Failure("This command doesn't work on linked devices.");
522 } catch (IOException e) {
523 throw new Error.Failure(e.getMessage());
524 } catch (UnregisteredRecipientException e) {
525 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
526 }
527 }
528
529 @Override
530 public void setGroupBlocked(final byte[] groupId, final boolean blocked) {
531 try {
532 m.setGroupBlocked(getGroupId(groupId), blocked);
533 } catch (NotMasterDeviceException e) {
534 throw new Error.Failure("This command doesn't work on linked devices.");
535 } catch (GroupNotFoundException e) {
536 throw new Error.GroupNotFound(e.getMessage());
537 } catch (IOException e) {
538 throw new Error.Failure(e.getMessage());
539 }
540 }
541
542 @Override
543 public List<byte[]> getGroupIds() {
544 var groups = m.getGroups();
545 return groups.stream().map(g -> g.groupId().serialize()).toList();
546 }
547
548 @Override
549 public DBusPath getGroup(final byte[] groupId) {
550 updateGroups();
551 final var groupOptional = groups.stream().filter(g -> Arrays.equals(g.getId(), groupId)).findFirst();
552 if (groupOptional.isEmpty()) {
553 throw new Error.GroupNotFound("Group not found");
554 }
555 return groupOptional.get().getObjectPath();
556 }
557
558 @Override
559 public List<StructGroup> listGroups() {
560 updateGroups();
561 return groups;
562 }
563
564 @Override
565 public String getGroupName(final byte[] groupId) {
566 var group = m.getGroup(getGroupId(groupId));
567 if (group == null || group.title() == null) {
568 return "";
569 } else {
570 return group.title();
571 }
572 }
573
574 @Override
575 public List<String> getGroupMembers(final byte[] groupId) {
576 var group = m.getGroup(getGroupId(groupId));
577 if (group == null) {
578 return List.of();
579 } else {
580 final var members = group.members();
581 return getRecipientStrings(members);
582 }
583 }
584
585 @Override
586 public byte[] createGroup(
587 final String name, final List<String> members, final String avatar
588 ) throws Error.AttachmentInvalid, Error.Failure, Error.InvalidNumber {
589 return updateGroup(new byte[0], name, members, avatar);
590 }
591
592 @Override
593 public byte[] updateGroup(byte[] groupId, String name, List<String> members, String avatar) {
594 try {
595 groupId = nullIfEmpty(groupId);
596 name = nullIfEmpty(name);
597 avatar = nullIfEmpty(avatar);
598 final var memberIdentifiers = getSingleRecipientIdentifiers(members, m.getSelfNumber());
599 if (groupId == null) {
600 final var results = m.createGroup(name, memberIdentifiers, avatar == null ? null : new File(avatar));
601 updateGroups();
602 checkGroupSendMessageResults(results.second().timestamp(), results.second().results());
603 return results.first().serialize();
604 } else {
605 final var results = m.updateGroup(getGroupId(groupId),
606 UpdateGroup.newBuilder()
607 .withName(name)
608 .withMembers(memberIdentifiers)
609 .withAvatarFile(avatar == null ? null : new File(avatar))
610 .build());
611 if (results != null) {
612 checkGroupSendMessageResults(results.timestamp(), results.results());
613 }
614 return groupId;
615 }
616 } catch (IOException e) {
617 throw new Error.Failure(e.getMessage());
618 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
619 throw new Error.GroupNotFound(e.getMessage());
620 } catch (AttachmentInvalidException e) {
621 throw new Error.AttachmentInvalid(e.getMessage());
622 } catch (UnregisteredRecipientException e) {
623 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
624 }
625 }
626
627 @Override
628 public boolean isRegistered() {
629 return true;
630 }
631
632 @Override
633 public boolean isRegistered(String number) {
634 var result = isRegistered(List.of(number));
635 return result.get(0);
636 }
637
638 @Override
639 public List<Boolean> isRegistered(List<String> numbers) {
640 if (numbers.isEmpty()) {
641 return List.of();
642 }
643
644 Map<String, UserStatus> registered;
645 try {
646 registered = m.getUserStatus(new HashSet<>(numbers));
647 } catch (IOException e) {
648 throw new Error.Failure(e.getMessage());
649 }
650
651 return numbers.stream().map(number -> registered.get(number).uuid() != null).toList();
652 }
653
654 @Override
655 public void updateProfile(
656 String givenName,
657 String familyName,
658 String about,
659 String aboutEmoji,
660 String avatarPath,
661 final boolean removeAvatar
662 ) {
663 try {
664 givenName = nullIfEmpty(givenName);
665 familyName = nullIfEmpty(familyName);
666 about = nullIfEmpty(about);
667 aboutEmoji = nullIfEmpty(aboutEmoji);
668 avatarPath = nullIfEmpty(avatarPath);
669 Optional<File> avatarFile = removeAvatar
670 ? Optional.empty()
671 : avatarPath == null ? null : Optional.of(new File(avatarPath));
672 m.setProfile(givenName, familyName, about, aboutEmoji, avatarFile);
673 } catch (IOException e) {
674 throw new Error.Failure(e.getMessage());
675 }
676 }
677
678 @Override
679 public void updateProfile(
680 final String name,
681 final String about,
682 final String aboutEmoji,
683 String avatarPath,
684 final boolean removeAvatar
685 ) {
686 updateProfile(name, "", about, aboutEmoji, avatarPath, removeAvatar);
687 }
688
689 @Override
690 public void removePin() {
691 try {
692 m.setRegistrationLockPin(Optional.empty());
693 } catch (IOException e) {
694 throw new Error.Failure("Remove pin error: " + e.getMessage());
695 } catch (NotMasterDeviceException e) {
696 throw new Error.Failure("This command doesn't work on linked devices.");
697 }
698 }
699
700 @Override
701 public void setPin(String registrationLockPin) {
702 try {
703 m.setRegistrationLockPin(Optional.of(registrationLockPin));
704 } catch (IOException e) {
705 throw new Error.Failure("Set pin error: " + e.getMessage());
706 } catch (NotMasterDeviceException e) {
707 throw new Error.Failure("This command doesn't work on linked devices.");
708 }
709 }
710
711 // Provide option to query a version string in order to react on potential
712 // future interface changes
713 @Override
714 public String version() {
715 return BaseConfig.PROJECT_VERSION;
716 }
717
718 // Create a unique list of Numbers from Identities and Contacts to really get
719 // all numbers the system knows
720 @Override
721 public List<String> listNumbers() {
722 return Stream.concat(m.getIdentities().stream().map(Identity::recipient),
723 m.getContacts().stream().map(Pair::first))
724 .map(a -> a.number().orElse(null))
725 .filter(Objects::nonNull)
726 .distinct()
727 .toList();
728 }
729
730 @Override
731 public List<String> getContactNumber(final String name) {
732 // Contact names have precedence.
733 var numbers = new ArrayList<String>();
734 var contacts = m.getContacts();
735 for (var c : contacts) {
736 if (name.equals(c.second().getName())) {
737 numbers.add(c.first().getLegacyIdentifier());
738 }
739 }
740 // Try profiles if no contact name was found
741 for (var identity : m.getIdentities()) {
742 final var address = identity.recipient();
743 var number = address.number().orElse(null);
744 if (number != null) {
745 Profile profile = null;
746 try {
747 profile = m.getRecipientProfile(RecipientIdentifier.Single.fromAddress(address));
748 } catch (IOException | UnregisteredRecipientException ignored) {
749 }
750 if (profile != null && profile.getDisplayName().equals(name)) {
751 numbers.add(number);
752 }
753 }
754 }
755 return numbers;
756 }
757
758 @Override
759 public void quitGroup(final byte[] groupId) {
760 var group = getGroupId(groupId);
761 try {
762 m.quitGroup(group, Set.of());
763 } catch (GroupNotFoundException | NotAGroupMemberException e) {
764 throw new Error.GroupNotFound(e.getMessage());
765 } catch (IOException | LastGroupAdminException e) {
766 throw new Error.Failure(e.getMessage());
767 } catch (UnregisteredRecipientException e) {
768 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
769 }
770 }
771
772 @Override
773 public byte[] joinGroup(final String groupLink) {
774 try {
775 final var linkUrl = GroupInviteLinkUrl.fromUri(groupLink);
776 if (linkUrl == null) {
777 throw new Error.Failure("Group link is invalid:");
778 }
779 final var result = m.joinGroup(linkUrl);
780 return result.first().serialize();
781 } catch (GroupInviteLinkUrl.InvalidGroupLinkException | InactiveGroupLinkException e) {
782 throw new Error.Failure("Group link is invalid: " + e.getMessage());
783 } catch (GroupInviteLinkUrl.UnknownGroupLinkVersionException e) {
784 throw new Error.Failure("Group link was created with an incompatible version: " + e.getMessage());
785 } catch (IOException e) {
786 throw new Error.Failure(e.getMessage());
787 }
788 }
789
790 @Override
791 public boolean isContactBlocked(final String number) {
792 return m.isContactBlocked(getSingleRecipientIdentifier(number, m.getSelfNumber()));
793 }
794
795 @Override
796 public boolean isGroupBlocked(final byte[] groupId) {
797 var group = m.getGroup(getGroupId(groupId));
798 if (group == null) {
799 return false;
800 } else {
801 return group.isBlocked();
802 }
803 }
804
805 @Override
806 public boolean isMember(final byte[] groupId) {
807 var group = m.getGroup(getGroupId(groupId));
808 if (group == null) {
809 return false;
810 } else {
811 return group.isMember();
812 }
813 }
814
815 @Override
816 public String uploadStickerPack(String stickerPackPath) {
817 File path = new File(stickerPackPath);
818 try {
819 return m.uploadStickerPack(path).toString();
820 } catch (IOException e) {
821 throw new Error.Failure("Upload error (maybe image size is too large):" + e.getMessage());
822 } catch (StickerPackInvalidException e) {
823 throw new Error.Failure("Invalid sticker pack: " + e.getMessage());
824 }
825 }
826
827 private static void checkSendMessageResult(long timestamp, SendMessageResult result) throws DBusExecutionException {
828 var error = SendMessageResultUtils.getErrorMessageFromSendMessageResult(result);
829
830 if (error == null) {
831 return;
832 }
833
834 final var message = "\nFailed to send message:\n" + error + '\n' + timestamp;
835
836 if (result.isIdentityFailure()) {
837 throw new Error.UntrustedIdentity(message);
838 } else {
839 throw new Error.Failure(message);
840 }
841 }
842
843 private void checkSendMessageResults(final SendMessageResults results) {
844 final var sendMessageResults = results.results().values().stream().findFirst();
845 if (results.results().size() == 1 && sendMessageResults.get().size() == 1) {
846 checkSendMessageResult(results.timestamp(), sendMessageResults.get().stream().findFirst().get());
847 return;
848 }
849
850 if (results.hasSuccess()) {
851 return;
852 }
853
854 var message = new StringBuilder();
855 message.append("Failed to send messages:\n");
856 var errors = SendMessageResultUtils.getErrorMessagesFromSendMessageResults(results.results());
857 for (var error : errors) {
858 message.append(error).append('\n');
859 }
860 message.append(results.timestamp());
861
862 throw new Error.Failure(message.toString());
863 }
864
865 private static void checkGroupSendMessageResults(
866 long timestamp, Collection<SendMessageResult> results
867 ) throws DBusExecutionException {
868 if (results.size() == 1) {
869 checkSendMessageResult(timestamp, results.stream().findFirst().get());
870 return;
871 }
872
873 var errors = SendMessageResultUtils.getErrorMessagesFromSendMessageResults(results);
874 if (errors.size() == 0 || errors.size() < results.size()) {
875 return;
876 }
877
878 var message = new StringBuilder();
879 message.append("Failed to send message:\n");
880 for (var error : errors) {
881 message.append(error).append('\n');
882 }
883 message.append(timestamp);
884
885 throw new Error.Failure(message.toString());
886 }
887
888 private static List<String> getRecipientStrings(final Set<RecipientAddress> members) {
889 return members.stream().map(RecipientAddress::getLegacyIdentifier).toList();
890 }
891
892 private static Set<RecipientIdentifier.Single> getSingleRecipientIdentifiers(
893 final Collection<String> recipientStrings, final String localNumber
894 ) throws DBusExecutionException {
895 final var identifiers = new HashSet<RecipientIdentifier.Single>();
896 for (var recipientString : recipientStrings) {
897 identifiers.add(getSingleRecipientIdentifier(recipientString, localNumber));
898 }
899 return identifiers;
900 }
901
902 private static RecipientIdentifier.Single getSingleRecipientIdentifier(
903 final String recipientString, final String localNumber
904 ) throws DBusExecutionException {
905 try {
906 return RecipientIdentifier.Single.fromString(recipientString, localNumber);
907 } catch (InvalidNumberException e) {
908 throw new Error.InvalidNumber(e.getMessage());
909 }
910 }
911
912 private RecipientIdentifier.Group getGroupRecipientIdentifier(final byte[] groupId) {
913 return new RecipientIdentifier.Group(getGroupId(groupId));
914 }
915
916 private static GroupId getGroupId(byte[] groupId) throws DBusExecutionException {
917 try {
918 return GroupId.unknownVersion(groupId);
919 } catch (Throwable e) {
920 throw new Error.InvalidGroupId("Invalid group id: " + e.getMessage());
921 }
922 }
923
924 private byte[] nullIfEmpty(final byte[] array) {
925 return array.length == 0 ? null : array;
926 }
927
928 private String nullIfEmpty(final String name) {
929 return name.isEmpty() ? null : name;
930 }
931
932 private String emptyIfNull(final String string) {
933 return string == null ? "" : string;
934 }
935
936 private static String getDeviceObjectPath(String basePath, long deviceId) {
937 return basePath + "/Devices/" + deviceId;
938 }
939
940 private void updateDevices() {
941 List<org.asamk.signal.manager.api.Device> linkedDevices;
942 try {
943 linkedDevices = m.getLinkedDevices();
944 } catch (IOException e) {
945 throw new Error.Failure("Failed to get linked devices: " + e.getMessage());
946 }
947
948 unExportDevices();
949
950 linkedDevices.forEach(d -> {
951 final var object = new DbusSignalDeviceImpl(d);
952 final var deviceObjectPath = object.getObjectPath();
953 exportObject(object);
954 if (d.isThisDevice()) {
955 thisDevice = new DBusPath(deviceObjectPath);
956 }
957 this.devices.add(new StructDevice(new DBusPath(deviceObjectPath), (long) d.id(), emptyIfNull(d.name())));
958 });
959 }
960
961 private void unExportDevices() {
962 this.devices.stream()
963 .map(StructDevice::getObjectPath)
964 .map(DBusPath::getPath)
965 .forEach(connection::unExportObject);
966 this.devices.clear();
967 }
968
969 private static String getGroupObjectPath(String basePath, byte[] groupId) {
970 return basePath + "/Groups/" + Base64.getEncoder()
971 .encodeToString(groupId)
972 .replace("+", "_")
973 .replace("/", "_")
974 .replace("=", "_");
975 }
976
977 private void updateGroups() {
978 List<org.asamk.signal.manager.api.Group> groups;
979 groups = m.getGroups();
980
981 unExportGroups();
982
983 groups.forEach(g -> {
984 final var object = new DbusSignalGroupImpl(g.groupId());
985 exportObject(object);
986 this.groups.add(new StructGroup(new DBusPath(object.getObjectPath()),
987 g.groupId().serialize(),
988 emptyIfNull(g.title())));
989 });
990 }
991
992 private void unExportGroups() {
993 this.groups.stream().map(StructGroup::getObjectPath).map(DBusPath::getPath).forEach(connection::unExportObject);
994 this.groups.clear();
995 }
996
997 private static String getConfigurationObjectPath(String basePath) {
998 return basePath + "/Configuration";
999 }
1000
1001 private void updateConfiguration() {
1002 unExportConfiguration();
1003 final var object = new DbusSignalConfigurationImpl();
1004 exportObject(object);
1005 }
1006
1007 private void unExportConfiguration() {
1008 final var objectPath = getConfigurationObjectPath(this.objectPath);
1009 connection.unExportObject(objectPath);
1010 }
1011
1012 private void exportObject(final DBusInterface object) {
1013 try {
1014 connection.exportObject(object);
1015 logger.debug("Exported dbus object: " + object.getObjectPath());
1016 } catch (DBusException e) {
1017 e.printStackTrace();
1018 }
1019 }
1020
1021 public class DbusSignalDeviceImpl extends DbusProperties implements Signal.Device {
1022
1023 private final org.asamk.signal.manager.api.Device device;
1024
1025 public DbusSignalDeviceImpl(final org.asamk.signal.manager.api.Device device) {
1026 super.addPropertiesHandler(new DbusInterfacePropertiesHandler("org.asamk.Signal.Device",
1027 List.of(new DbusProperty<>("Id", device::id),
1028 new DbusProperty<>("Name", () -> emptyIfNull(device.name()), this::setDeviceName),
1029 new DbusProperty<>("Created", device::created),
1030 new DbusProperty<>("LastSeen", device::lastSeen))));
1031 this.device = device;
1032 }
1033
1034 @Override
1035 public String getObjectPath() {
1036 return getDeviceObjectPath(objectPath, device.id());
1037 }
1038
1039 @Override
1040 public void removeDevice() throws Error.Failure {
1041 try {
1042 m.removeLinkedDevices(device.id());
1043 updateDevices();
1044 } catch (IOException e) {
1045 throw new Error.Failure(e.getMessage());
1046 }
1047 }
1048
1049 private void setDeviceName(String name) {
1050 if (!device.isThisDevice()) {
1051 throw new Error.Failure("Only the name of this device can be changed");
1052 }
1053 try {
1054 m.updateAccountAttributes(name);
1055 // update device list
1056 updateDevices();
1057 } catch (IOException e) {
1058 throw new Error.Failure(e.getMessage());
1059 }
1060 }
1061 }
1062
1063 public class DbusSignalConfigurationImpl extends DbusProperties implements Signal.Configuration {
1064
1065 public DbusSignalConfigurationImpl(
1066 ) {
1067 super.addPropertiesHandler(new DbusInterfacePropertiesHandler("org.asamk.Signal.Configuration",
1068 List.of(new DbusProperty<>("ReadReceipts", this::getReadReceipts, this::setReadReceipts),
1069 new DbusProperty<>("UnidentifiedDeliveryIndicators",
1070 this::getUnidentifiedDeliveryIndicators,
1071 this::setUnidentifiedDeliveryIndicators),
1072 new DbusProperty<>("TypingIndicators",
1073 this::getTypingIndicators,
1074 this::setTypingIndicators),
1075 new DbusProperty<>("LinkPreviews", this::getLinkPreviews, this::setLinkPreviews))));
1076
1077 }
1078
1079 @Override
1080 public String getObjectPath() {
1081 return getConfigurationObjectPath(objectPath);
1082 }
1083
1084 public void setReadReceipts(Boolean readReceipts) {
1085 setConfiguration(readReceipts, null, null, null);
1086 }
1087
1088 public void setUnidentifiedDeliveryIndicators(Boolean unidentifiedDeliveryIndicators) {
1089 setConfiguration(null, unidentifiedDeliveryIndicators, null, null);
1090 }
1091
1092 public void setTypingIndicators(Boolean typingIndicators) {
1093 setConfiguration(null, null, typingIndicators, null);
1094 }
1095
1096 public void setLinkPreviews(Boolean linkPreviews) {
1097 setConfiguration(null, null, null, linkPreviews);
1098 }
1099
1100 private void setConfiguration(
1101 Boolean readReceipts,
1102 Boolean unidentifiedDeliveryIndicators,
1103 Boolean typingIndicators,
1104 Boolean linkPreviews
1105 ) {
1106 try {
1107 m.updateConfiguration(new org.asamk.signal.manager.api.Configuration(Optional.ofNullable(readReceipts),
1108 Optional.ofNullable(unidentifiedDeliveryIndicators),
1109 Optional.ofNullable(typingIndicators),
1110 Optional.ofNullable(linkPreviews)));
1111 } catch (IOException e) {
1112 throw new Error.Failure("UpdateAccount error: " + e.getMessage());
1113 } catch (NotMasterDeviceException e) {
1114 throw new Error.Failure("This command doesn't work on linked devices.");
1115 }
1116 }
1117
1118 private boolean getReadReceipts() {
1119 return m.getConfiguration().readReceipts().orElse(false);
1120 }
1121
1122 private boolean getUnidentifiedDeliveryIndicators() {
1123 return m.getConfiguration().unidentifiedDeliveryIndicators().orElse(false);
1124 }
1125
1126 private boolean getTypingIndicators() {
1127 return m.getConfiguration().typingIndicators().orElse(false);
1128 }
1129
1130 private boolean getLinkPreviews() {
1131 return m.getConfiguration().linkPreviews().orElse(false);
1132 }
1133 }
1134
1135 public class DbusSignalGroupImpl extends DbusProperties implements Signal.Group {
1136
1137 private final GroupId groupId;
1138
1139 public DbusSignalGroupImpl(final GroupId groupId) {
1140 this.groupId = groupId;
1141 super.addPropertiesHandler(new DbusInterfacePropertiesHandler("org.asamk.Signal.Group",
1142 List.of(new DbusProperty<>("Id", groupId::serialize),
1143 new DbusProperty<>("Name", () -> emptyIfNull(getGroup().title()), this::setGroupName),
1144 new DbusProperty<>("Description",
1145 () -> emptyIfNull(getGroup().description()),
1146 this::setGroupDescription),
1147 new DbusProperty<>("Avatar", this::setGroupAvatar),
1148 new DbusProperty<>("IsBlocked", () -> getGroup().isBlocked(), this::setIsBlocked),
1149 new DbusProperty<>("IsMember", () -> getGroup().isMember()),
1150 new DbusProperty<>("IsAdmin", () -> getGroup().isAdmin()),
1151 new DbusProperty<>("MessageExpirationTimer",
1152 () -> getGroup().messageExpirationTimer(),
1153 this::setMessageExpirationTime),
1154 new DbusProperty<>("Members",
1155 () -> new Variant<>(getRecipientStrings(getGroup().members()), "as")),
1156 new DbusProperty<>("PendingMembers",
1157 () -> new Variant<>(getRecipientStrings(getGroup().pendingMembers()), "as")),
1158 new DbusProperty<>("RequestingMembers",
1159 () -> new Variant<>(getRecipientStrings(getGroup().requestingMembers()), "as")),
1160 new DbusProperty<>("Admins",
1161 () -> new Variant<>(getRecipientStrings(getGroup().adminMembers()), "as")),
1162 new DbusProperty<>("Banned",
1163 () -> new Variant<>(getRecipientStrings(getGroup().bannedMembers()), "as")),
1164 new DbusProperty<>("PermissionAddMember",
1165 () -> getGroup().permissionAddMember().name(),
1166 this::setGroupPermissionAddMember),
1167 new DbusProperty<>("PermissionEditDetails",
1168 () -> getGroup().permissionEditDetails().name(),
1169 this::setGroupPermissionEditDetails),
1170 new DbusProperty<>("PermissionSendMessage",
1171 () -> getGroup().permissionSendMessage().name(),
1172 this::setGroupPermissionSendMessage),
1173 new DbusProperty<>("GroupInviteLink", () -> {
1174 final var groupInviteLinkUrl = getGroup().groupInviteLinkUrl();
1175 return groupInviteLinkUrl == null ? "" : groupInviteLinkUrl.getUrl();
1176 }))));
1177 }
1178
1179 @Override
1180 public String getObjectPath() {
1181 return getGroupObjectPath(objectPath, groupId.serialize());
1182 }
1183
1184 @Override
1185 public void quitGroup() throws Error.Failure {
1186 try {
1187 m.quitGroup(groupId, Set.of());
1188 } catch (GroupNotFoundException | NotAGroupMemberException e) {
1189 throw new Error.GroupNotFound(e.getMessage());
1190 } catch (IOException e) {
1191 throw new Error.Failure(e.getMessage());
1192 } catch (LastGroupAdminException e) {
1193 throw new Error.LastGroupAdmin(e.getMessage());
1194 } catch (UnregisteredRecipientException e) {
1195 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
1196 }
1197 }
1198
1199 @Override
1200 public void deleteGroup() throws Error.Failure, Error.LastGroupAdmin {
1201 try {
1202 m.deleteGroup(groupId);
1203 } catch (IOException e) {
1204 throw new Error.Failure(e.getMessage());
1205 }
1206 updateGroups();
1207 }
1208
1209 @Override
1210 public void addMembers(final List<String> recipients) throws Error.Failure {
1211 final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1212 updateGroup(UpdateGroup.newBuilder().withMembers(memberIdentifiers).build());
1213 }
1214
1215 @Override
1216 public void removeMembers(final List<String> recipients) throws Error.Failure {
1217 final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1218 updateGroup(UpdateGroup.newBuilder().withRemoveMembers(memberIdentifiers).build());
1219 }
1220
1221 @Override
1222 public void addAdmins(final List<String> recipients) throws Error.Failure {
1223 final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1224 updateGroup(UpdateGroup.newBuilder().withAdmins(memberIdentifiers).build());
1225 }
1226
1227 @Override
1228 public void removeAdmins(final List<String> recipients) throws Error.Failure {
1229 final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1230 updateGroup(UpdateGroup.newBuilder().withRemoveAdmins(memberIdentifiers).build());
1231 }
1232
1233 @Override
1234 public void resetLink() throws Error.Failure {
1235 updateGroup(UpdateGroup.newBuilder().withResetGroupLink(true).build());
1236 }
1237
1238 @Override
1239 public void disableLink() throws Error.Failure {
1240 updateGroup(UpdateGroup.newBuilder().withGroupLinkState(GroupLinkState.DISABLED).build());
1241 }
1242
1243 @Override
1244 public void enableLink(final boolean requiresApproval) throws Error.Failure {
1245 updateGroup(UpdateGroup.newBuilder()
1246 .withGroupLinkState(requiresApproval
1247 ? GroupLinkState.ENABLED_WITH_APPROVAL
1248 : GroupLinkState.ENABLED)
1249 .build());
1250 }
1251
1252 private org.asamk.signal.manager.api.Group getGroup() {
1253 return m.getGroup(groupId);
1254 }
1255
1256 private void setGroupName(final String name) {
1257 updateGroup(UpdateGroup.newBuilder().withName(name).build());
1258 }
1259
1260 private void setGroupDescription(final String description) {
1261 updateGroup(UpdateGroup.newBuilder().withDescription(description).build());
1262 }
1263
1264 private void setGroupAvatar(final String avatar) {
1265 updateGroup(UpdateGroup.newBuilder().withAvatarFile(new File(avatar)).build());
1266 }
1267
1268 private void setMessageExpirationTime(final int expirationTime) {
1269 updateGroup(UpdateGroup.newBuilder().withExpirationTimer(expirationTime).build());
1270 }
1271
1272 private void setGroupPermissionAddMember(final String permission) {
1273 updateGroup(UpdateGroup.newBuilder().withAddMemberPermission(GroupPermission.valueOf(permission)).build());
1274 }
1275
1276 private void setGroupPermissionEditDetails(final String permission) {
1277 updateGroup(UpdateGroup.newBuilder()
1278 .withEditDetailsPermission(GroupPermission.valueOf(permission))
1279 .build());
1280 }
1281
1282 private void setGroupPermissionSendMessage(final String permission) {
1283 updateGroup(UpdateGroup.newBuilder()
1284 .withIsAnnouncementGroup(GroupPermission.valueOf(permission) == GroupPermission.ONLY_ADMINS)
1285 .build());
1286 }
1287
1288 private void setIsBlocked(final boolean isBlocked) {
1289 try {
1290 m.setGroupBlocked(groupId, isBlocked);
1291 } catch (NotMasterDeviceException e) {
1292 throw new Error.Failure("This command doesn't work on linked devices.");
1293 } catch (GroupNotFoundException e) {
1294 throw new Error.GroupNotFound(e.getMessage());
1295 } catch (IOException e) {
1296 throw new Error.Failure(e.getMessage());
1297 }
1298 }
1299
1300 private void updateGroup(final UpdateGroup updateGroup) {
1301 try {
1302 m.updateGroup(groupId, updateGroup);
1303 } catch (IOException e) {
1304 throw new Error.Failure(e.getMessage());
1305 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
1306 throw new Error.GroupNotFound(e.getMessage());
1307 } catch (AttachmentInvalidException e) {
1308 throw new Error.AttachmentInvalid(e.getMessage());
1309 } catch (UnregisteredRecipientException e) {
1310 throw new Error.UntrustedIdentity(e.getSender().getIdentifier() + " is not registered.");
1311 }
1312 }
1313 }
1314 }