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