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