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