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.timestamp(), results.results());
 157             return results.timestamp();
 158         } catch (AttachmentInvalidException e) {
 159             throw new Error.AttachmentInvalid(e.getMessage());
 160         } catch (IOException e) {
 161             throw new Error.Failure(e);
 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.timestamp(), results.results());
 186             return results.timestamp();
 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.timestamp(), results.results());
 202             return results.timestamp();
 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.timestamp(), results.results());
 240             return results.timestamp();
 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 sendViewedReceipt(
 283             final String recipient, final List<Long> messageIds
 284     ) throws Error.Failure, Error.UntrustedIdentity {
 285         try {
 286             m.sendViewedReceipt(getSingleRecipientIdentifier(recipient, m.getSelfNumber()), messageIds);
 287         } catch (IOException e) {
 288             throw new Error.Failure(e.getMessage());
 289         } catch (UntrustedIdentityException e) {
 290             throw new Error.UntrustedIdentity(e.getMessage());
 291         }
 292     }
 293
 294     @Override
 295     public void sendContacts() {
 296         try {
 297             m.sendContacts();
 298         } catch (IOException e) {
 299             throw new Error.Failure("SendContacts error: " + e.getMessage());
 300         }
 301     }
 302
 303     @Override
 304     public void sendSyncRequest() {
 305         try {
 306             m.requestAllSyncData();
 307         } catch (IOException e) {
 308             throw new Error.Failure("Request sync data error: " + e.getMessage());
 309         }
 310     }
 311
 312     @Override
 313     public long sendNoteToSelfMessage(
 314             final String message, final List<String> attachments
 315     ) throws Error.AttachmentInvalid, Error.Failure, Error.UntrustedIdentity {
 316         try {
 317             final var results = m.sendMessage(new Message(message, attachments),
 318                     Set.of(RecipientIdentifier.NoteToSelf.INSTANCE));
 319             checkSendMessageResults(results.timestamp(), results.results());
 320             return results.timestamp();
 321         } catch (AttachmentInvalidException e) {
 322             throw new Error.AttachmentInvalid(e.getMessage());
 323         } catch (IOException e) {
 324             throw new Error.Failure(e.getMessage());
 325         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
 326             throw new Error.GroupNotFound(e.getMessage());
 327         }
 328     }
 329
 330     @Override
 331     public void sendEndSessionMessage(final List<String> recipients) {
 332         try {
 333             final var results = m.sendEndSessionMessage(getSingleRecipientIdentifiers(recipients, m.getSelfNumber()));
 334             checkSendMessageResults(results.timestamp(), results.results());
 335         } catch (IOException e) {
 336             throw new Error.Failure(e.getMessage());
 337         }
 338     }
 339
 340     @Override
 341     public long sendGroupMessage(final String message, final List<String> attachments, final byte[] groupId) {
 342         try {
 343             var results = m.sendMessage(new Message(message, attachments),
 344                     Set.of(new RecipientIdentifier.Group(getGroupId(groupId))));
 345             checkSendMessageResults(results.timestamp(), results.results());
 346             return results.timestamp();
 347         } catch (IOException e) {
 348             throw new Error.Failure(e.getMessage());
 349         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
 350             throw new Error.GroupNotFound(e.getMessage());
 351         } catch (AttachmentInvalidException e) {
 352             throw new Error.AttachmentInvalid(e.getMessage());
 353         }
 354     }
 355
 356     @Override
 357     public long sendGroupMessageReaction(
 358             final String emoji,
 359             final boolean remove,
 360             final String targetAuthor,
 361             final long targetSentTimestamp,
 362             final byte[] groupId
 363     ) {
 364         try {
 365             final var results = m.sendMessageReaction(emoji,
 366                     remove,
 367                     getSingleRecipientIdentifier(targetAuthor, m.getSelfNumber()),
 368                     targetSentTimestamp,
 369                     Set.of(new RecipientIdentifier.Group(getGroupId(groupId))));
 370             checkSendMessageResults(results.timestamp(), results.results());
 371             return results.timestamp();
 372         } catch (IOException e) {
 373             throw new Error.Failure(e.getMessage());
 374         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
 375             throw new Error.GroupNotFound(e.getMessage());
 376         }
 377     }
 378
 379     // Since contact names might be empty if not defined, also potentially return
 380     // the profile name
 381     @Override
 382     public String getContactName(final String number) {
 383         final var name = m.getContactOrProfileName(getSingleRecipientIdentifier(number, m.getSelfNumber()));
 384         return name == null ? "" : name;
 385     }
 386
 387     @Override
 388     public void setContactName(final String number, final String name) {
 389         try {
 390             m.setContactName(getSingleRecipientIdentifier(number, m.getSelfNumber()), name);
 391         } catch (NotMasterDeviceException e) {
 392             throw new Error.Failure("This command doesn't work on linked devices.");
 393         } catch (UnregisteredUserException e) {
 394             throw new Error.Failure("Contact is not registered.");
 395         }
 396     }
 397
 398     @Override
 399     public void setExpirationTimer(final String number, final int expiration) {
 400         try {
 401             m.setExpirationTimer(getSingleRecipientIdentifier(number, m.getSelfNumber()), expiration);
 402         } catch (IOException e) {
 403             throw new Error.Failure(e.getMessage());
 404         }
 405     }
 406
 407     @Override
 408     public void setContactBlocked(final String number, final boolean blocked) {
 409         try {
 410             m.setContactBlocked(getSingleRecipientIdentifier(number, m.getSelfNumber()), blocked);
 411         } catch (NotMasterDeviceException e) {
 412             throw new Error.Failure("This command doesn't work on linked devices.");
 413         } catch (IOException e) {
 414             throw new Error.Failure(e.getMessage());
 415         }
 416     }
 417
 418     @Override
 419     public void setGroupBlocked(final byte[] groupId, final boolean blocked) {
 420         try {
 421             m.setGroupBlocked(getGroupId(groupId), blocked);
 422         } catch (NotMasterDeviceException e) {
 423             throw new Error.Failure("This command doesn't work on linked devices.");
 424         } catch (GroupNotFoundException e) {
 425             throw new Error.GroupNotFound(e.getMessage());
 426         } catch (IOException e) {
 427             throw new Error.Failure(e.getMessage());
 428         }
 429     }
 430
 431     @Override
 432     public List<byte[]> getGroupIds() {
 433         var groups = m.getGroups();
 434         var ids = new ArrayList<byte[]>(groups.size());
 435         for (var group : groups) {
 436             ids.add(group.groupId().serialize());
 437         }
 438         return ids;
 439     }
 440
 441     @Override
 442     public DBusPath getGroup(final byte[] groupId) {
 443         updateGroups();
 444         final var groupOptional = groups.stream().filter(g -> Arrays.equals(g.getId(), groupId)).findFirst();
 445         if (groupOptional.isEmpty()) {
 446             throw new Error.GroupNotFound("Group not found");
 447         }
 448         return groupOptional.get().getObjectPath();
 449     }
 450
 451     @Override
 452     public List<StructGroup> listGroups() {
 453         updateGroups();
 454         return groups;
 455     }
 456
 457     @Override
 458     public String getGroupName(final byte[] groupId) {
 459         var group = m.getGroup(getGroupId(groupId));
 460         if (group == null || group.title() == null) {
 461             return "";
 462         } else {
 463             return group.title();
 464         }
 465     }
 466
 467     @Override
 468     public List<String> getGroupMembers(final byte[] groupId) {
 469         var group = m.getGroup(getGroupId(groupId));
 470         if (group == null) {
 471             return List.of();
 472         } else {
 473             final var members = group.members();
 474             return getRecipientStrings(members);
 475         }
 476     }
 477
 478     @Override
 479     public byte[] createGroup(
 480             final String name, final List<String> members, final String avatar
 481     ) throws Error.AttachmentInvalid, Error.Failure, Error.InvalidNumber {
 482         return updateGroup(new byte[0], name, members, avatar);
 483     }
 484
 485     @Override
 486     public byte[] updateGroup(byte[] groupId, String name, List<String> members, String avatar) {
 487         try {
 488             groupId = nullIfEmpty(groupId);
 489             name = nullIfEmpty(name);
 490             avatar = nullIfEmpty(avatar);
 491             final var memberIdentifiers = getSingleRecipientIdentifiers(members, m.getSelfNumber());
 492             if (groupId == null) {
 493                 final var results = m.createGroup(name, memberIdentifiers, avatar == null ? null : new File(avatar));
 494                 checkSendMessageResults(results.second().timestamp(), results.second().results());
 495                 return results.first().serialize();
 496             } else {
 497                 final var results = m.updateGroup(getGroupId(groupId),
 498                         UpdateGroup.newBuilder()
 499                                 .withName(name)
 500                                 .withMembers(memberIdentifiers)
 501                                 .withAvatarFile(avatar == null ? null : new File(avatar))
 502                                 .build());
 503                 if (results != null) {
 504                     checkSendMessageResults(results.timestamp(), results.results());
 505                 }
 506                 return groupId;
 507             }
 508         } catch (IOException e) {
 509             throw new Error.Failure(e.getMessage());
 510         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
 511             throw new Error.GroupNotFound(e.getMessage());
 512         } catch (AttachmentInvalidException e) {
 513             throw new Error.AttachmentInvalid(e.getMessage());
 514         }
 515     }
 516
 517     @Override
 518     public boolean isRegistered() {
 519         return true;
 520     }
 521
 522     @Override
 523     public boolean isRegistered(String number) {
 524         var result = isRegistered(List.of(number));
 525         return result.get(0);
 526     }
 527
 528     @Override
 529     public List<Boolean> isRegistered(List<String> numbers) {
 530         var results = new ArrayList<Boolean>();
 531         if (numbers.isEmpty()) {
 532             return results;
 533         }
 534
 535         Map<String, Pair<String, UUID>> registered;
 536         try {
 537             registered = m.areUsersRegistered(new HashSet<>(numbers));
 538         } catch (IOException e) {
 539             throw new Error.Failure(e.getMessage());
 540         }
 541
 542         return numbers.stream().map(number -> {
 543             var uuid = registered.get(number).second();
 544             return uuid != null;
 545         }).collect(Collectors.toList());
 546     }
 547
 548     @Override
 549     public void updateProfile(
 550             String givenName,
 551             String familyName,
 552             String about,
 553             String aboutEmoji,
 554             String avatarPath,
 555             final boolean removeAvatar
 556     ) {
 557         try {
 558             givenName = nullIfEmpty(givenName);
 559             familyName = nullIfEmpty(familyName);
 560             about = nullIfEmpty(about);
 561             aboutEmoji = nullIfEmpty(aboutEmoji);
 562             avatarPath = nullIfEmpty(avatarPath);
 563             Optional<File> avatarFile = removeAvatar
 564                     ? Optional.absent()
 565                     : avatarPath == null ? null : Optional.of(new File(avatarPath));
 566             m.setProfile(givenName, familyName, about, aboutEmoji, avatarFile);
 567         } catch (IOException e) {
 568             throw new Error.Failure(e.getMessage());
 569         }
 570     }
 571
 572     @Override
 573     public void updateProfile(
 574             final String name,
 575             final String about,
 576             final String aboutEmoji,
 577             String avatarPath,
 578             final boolean removeAvatar
 579     ) {
 580         updateProfile(name, "", about, aboutEmoji, avatarPath, removeAvatar);
 581     }
 582
 583     @Override
 584     public void removePin() {
 585         try {
 586             m.setRegistrationLockPin(Optional.absent());
 587         } catch (UnauthenticatedResponseException e) {
 588             throw new Error.Failure("Remove pin failed with unauthenticated response: " + e.getMessage());
 589         } catch (IOException e) {
 590             throw new Error.Failure("Remove pin error: " + e.getMessage());
 591         }
 592     }
 593
 594     @Override
 595     public void setPin(String registrationLockPin) {
 596         try {
 597             m.setRegistrationLockPin(Optional.of(registrationLockPin));
 598         } catch (UnauthenticatedResponseException e) {
 599             throw new Error.Failure("Set pin error failed with unauthenticated response: " + e.getMessage());
 600         } catch (IOException e) {
 601             throw new Error.Failure("Set pin error: " + e.getMessage());
 602         }
 603     }
 604
 605     // Provide option to query a version string in order to react on potential
 606     // future interface changes
 607     @Override
 608     public String version() {
 609         return BaseConfig.PROJECT_VERSION;
 610     }
 611
 612     // Create a unique list of Numbers from Identities and Contacts to really get
 613     // all numbers the system knows
 614     @Override
 615     public List<String> listNumbers() {
 616         return Stream.concat(m.getIdentities().stream().map(Identity::recipient),
 617                         m.getContacts().stream().map(Pair::first))
 618                 .map(a -> a.getNumber().orElse(null))
 619                 .filter(Objects::nonNull)
 620                 .distinct()
 621                 .collect(Collectors.toList());
 622     }
 623
 624     @Override
 625     public List<String> getContactNumber(final String name) {
 626         // Contact names have precedence.
 627         var numbers = new ArrayList<String>();
 628         var contacts = m.getContacts();
 629         for (var c : contacts) {
 630             if (name.equals(c.second().getName())) {
 631                 numbers.add(c.first().getLegacyIdentifier());
 632             }
 633         }
 634         // Try profiles if no contact name was found
 635         for (var identity : m.getIdentities()) {
 636             final var address = identity.recipient();
 637             var number = address.getNumber().orElse(null);
 638             if (number != null) {
 639                 Profile profile = null;
 640                 try {
 641                     profile = m.getRecipientProfile(RecipientIdentifier.Single.fromAddress(address));
 642                 } catch (UnregisteredUserException ignored) {
 643                 }
 644                 if (profile != null && profile.getDisplayName().equals(name)) {
 645                     numbers.add(number);
 646                 }
 647             }
 648         }
 649         return numbers;
 650     }
 651
 652     @Override
 653     public void quitGroup(final byte[] groupId) {
 654         var group = getGroupId(groupId);
 655         try {
 656             m.quitGroup(group, Set.of());
 657         } catch (GroupNotFoundException | NotAGroupMemberException e) {
 658             throw new Error.GroupNotFound(e.getMessage());
 659         } catch (IOException | LastGroupAdminException e) {
 660             throw new Error.Failure(e.getMessage());
 661         }
 662     }
 663
 664     @Override
 665     public byte[] joinGroup(final String groupLink) {
 666         try {
 667             final var linkUrl = GroupInviteLinkUrl.fromUri(groupLink);
 668             if (linkUrl == null) {
 669                 throw new Error.Failure("Group link is invalid:");
 670             }
 671             final var result = m.joinGroup(linkUrl);
 672             return result.first().serialize();
 673         } catch (GroupInviteLinkUrl.InvalidGroupLinkException | GroupLinkNotActiveException e) {
 674             throw new Error.Failure("Group link is invalid: " + e.getMessage());
 675         } catch (GroupInviteLinkUrl.UnknownGroupLinkVersionException e) {
 676             throw new Error.Failure("Group link was created with an incompatible version: " + e.getMessage());
 677         } catch (IOException e) {
 678             throw new Error.Failure(e.getMessage());
 679         }
 680     }
 681
 682     @Override
 683     public boolean isContactBlocked(final String number) {
 684         return m.isContactBlocked(getSingleRecipientIdentifier(number, m.getSelfNumber()));
 685     }
 686
 687     @Override
 688     public boolean isGroupBlocked(final byte[] groupId) {
 689         var group = m.getGroup(getGroupId(groupId));
 690         if (group == null) {
 691             return false;
 692         } else {
 693             return group.isBlocked();
 694         }
 695     }
 696
 697     @Override
 698     public boolean isMember(final byte[] groupId) {
 699         var group = m.getGroup(getGroupId(groupId));
 700         if (group == null) {
 701             return false;
 702         } else {
 703             return group.isMember();
 704         }
 705     }
 706
 707     @Override
 708     public String uploadStickerPack(String stickerPackPath) {
 709         File path = new File(stickerPackPath);
 710         try {
 711             return m.uploadStickerPack(path).toString();
 712         } catch (IOException e) {
 713             throw new Error.Failure("Upload error (maybe image size is too large):" + e.getMessage());
 714         } catch (StickerPackInvalidException e) {
 715             throw new Error.Failure("Invalid sticker pack: " + e.getMessage());
 716         }
 717     }
 718
 719     private static void checkSendMessageResult(long timestamp, SendMessageResult result) throws DBusExecutionException {
 720         var error = ErrorUtils.getErrorMessageFromSendMessageResult(result);
 721
 722         if (error == null) {
 723             return;
 724         }
 725
 726         final var message = timestamp + "\nFailed to send message:\n" + error + '\n';
 727
 728         if (result.getIdentityFailure() != null) {
 729             throw new Error.UntrustedIdentity(message);
 730         } else {
 731             throw new Error.Failure(message);
 732         }
 733     }
 734
 735     private static void checkSendMessageResults(
 736             long timestamp, Map<RecipientIdentifier, List<SendMessageResult>> results
 737     ) throws DBusExecutionException {
 738         final var sendMessageResults = results.values().stream().findFirst();
 739         if (results.size() == 1 && sendMessageResults.get().size() == 1) {
 740             checkSendMessageResult(timestamp, sendMessageResults.get().stream().findFirst().get());
 741             return;
 742         }
 743
 744         var errors = ErrorUtils.getErrorMessagesFromSendMessageResults(results);
 745         if (errors.size() == 0) {
 746             return;
 747         }
 748
 749         var message = new StringBuilder();
 750         message.append(timestamp).append('\n');
 751         message.append("Failed to send (some) messages:\n");
 752         for (var error : errors) {
 753             message.append(error).append('\n');
 754         }
 755
 756         throw new Error.Failure(message.toString());
 757     }
 758
 759     private static void checkSendMessageResults(
 760             long timestamp, Collection<SendMessageResult> results
 761     ) throws DBusExecutionException {
 762         if (results.size() == 1) {
 763             checkSendMessageResult(timestamp, results.stream().findFirst().get());
 764             return;
 765         }
 766
 767         var errors = ErrorUtils.getErrorMessagesFromSendMessageResults(results);
 768         if (errors.size() == 0) {
 769             return;
 770         }
 771
 772         var message = new StringBuilder();
 773         message.append(timestamp).append('\n');
 774         message.append("Failed to send (some) messages:\n");
 775         for (var error : errors) {
 776             message.append(error).append('\n');
 777         }
 778
 779         throw new Error.Failure(message.toString());
 780     }
 781
 782     private static List<String> getRecipientStrings(final Set<RecipientAddress> members) {
 783         return members.stream().map(RecipientAddress::getLegacyIdentifier).collect(Collectors.toList());
 784     }
 785
 786     private static Set<RecipientIdentifier.Single> getSingleRecipientIdentifiers(
 787             final Collection<String> recipientStrings, final String localNumber
 788     ) throws DBusExecutionException {
 789         final var identifiers = new HashSet<RecipientIdentifier.Single>();
 790         for (var recipientString : recipientStrings) {
 791             identifiers.add(getSingleRecipientIdentifier(recipientString, localNumber));
 792         }
 793         return identifiers;
 794     }
 795
 796     private static RecipientIdentifier.Single getSingleRecipientIdentifier(
 797             final String recipientString, final String localNumber
 798     ) throws DBusExecutionException {
 799         try {
 800             return RecipientIdentifier.Single.fromString(recipientString, localNumber);
 801         } catch (InvalidNumberException e) {
 802             throw new Error.InvalidNumber(e.getMessage());
 803         }
 804     }
 805
 806     private static GroupId getGroupId(byte[] groupId) throws DBusExecutionException {
 807         try {
 808             return GroupId.unknownVersion(groupId);
 809         } catch (Throwable e) {
 810             throw new Error.InvalidGroupId("Invalid group id: " + e.getMessage());
 811         }
 812     }
 813
 814     private byte[] nullIfEmpty(final byte[] array) {
 815         return array.length == 0 ? null : array;
 816     }
 817
 818     private String nullIfEmpty(final String name) {
 819         return name.isEmpty() ? null : name;
 820     }
 821
 822     private String emptyIfNull(final String string) {
 823         return string == null ? "" : string;
 824     }
 825
 826     private static String getDeviceObjectPath(String basePath, long deviceId) {
 827         return basePath + "/Devices/" + deviceId;
 828     }
 829
 830     private void updateDevices() {
 831         List<org.asamk.signal.manager.api.Device> linkedDevices;
 832         try {
 833             linkedDevices = m.getLinkedDevices();
 834         } catch (IOException e) {
 835             throw new Error.Failure("Failed to get linked devices: " + e.getMessage());
 836         }
 837
 838         unExportDevices();
 839
 840         linkedDevices.forEach(d -> {
 841             final var object = new DbusSignalDeviceImpl(d);
 842             final var deviceObjectPath = object.getObjectPath();
 843             try {
 844                 connection.exportObject(object);
 845             } catch (DBusException e) {
 846                 e.printStackTrace();
 847             }
 848             if (d.isThisDevice()) {
 849                 thisDevice = new DBusPath(deviceObjectPath);
 850             }
 851             this.devices.add(new StructDevice(new DBusPath(deviceObjectPath), d.id(), emptyIfNull(d.name())));
 852         });
 853     }
 854
 855     private void unExportDevices() {
 856         this.devices.stream()
 857                 .map(StructDevice::getObjectPath)
 858                 .map(DBusPath::getPath)
 859                 .forEach(connection::unExportObject);
 860         this.devices.clear();
 861     }
 862
 863     private static String getGroupObjectPath(String basePath, byte[] groupId) {
 864         return basePath + "/Groups/" + Base64.getEncoder()
 865                 .encodeToString(groupId)
 866                 .replace("+", "_")
 867                 .replace("/", "_")
 868                 .replace("=", "_");
 869     }
 870
 871     private void updateGroups() {
 872         List<org.asamk.signal.manager.api.Group> groups;
 873         groups = m.getGroups();
 874
 875         unExportGroups();
 876
 877         groups.forEach(g -> {
 878             final var object = new DbusSignalGroupImpl(g.groupId());
 879             try {
 880                 connection.exportObject(object);
 881             } catch (DBusException e) {
 882                 e.printStackTrace();
 883             }
 884             this.groups.add(new StructGroup(new DBusPath(object.getObjectPath()),
 885                     g.groupId().serialize(),
 886                     emptyIfNull(g.title())));
 887         });
 888     }
 889
 890     private void unExportGroups() {
 891         this.groups.stream().map(StructGroup::getObjectPath).map(DBusPath::getPath).forEach(connection::unExportObject);
 892         this.groups.clear();
 893     }
 894
 895     public class DbusSignalDeviceImpl extends DbusProperties implements Signal.Device {
 896
 897         private final org.asamk.signal.manager.api.Device device;
 898
 899         public DbusSignalDeviceImpl(final org.asamk.signal.manager.api.Device device) {
 900             super.addPropertiesHandler(new DbusInterfacePropertiesHandler("org.asamk.Signal.Device",
 901                     List.of(new DbusProperty<>("Id", device::id),
 902                             new DbusProperty<>("Name", () -> emptyIfNull(device.name()), this::setDeviceName),
 903                             new DbusProperty<>("Created", device::created),
 904                             new DbusProperty<>("LastSeen", device::lastSeen))));
 905             this.device = device;
 906         }
 907
 908         @Override
 909         public String getObjectPath() {
 910             return getDeviceObjectPath(objectPath, device.id());
 911         }
 912
 913         @Override
 914         public void removeDevice() throws Error.Failure {
 915             try {
 916                 m.removeLinkedDevices(device.id());
 917                 updateDevices();
 918             } catch (IOException e) {
 919                 throw new Error.Failure(e.getMessage());
 920             }
 921         }
 922
 923         private void setDeviceName(String name) {
 924             if (!device.isThisDevice()) {
 925                 throw new Error.Failure("Only the name of this device can be changed");
 926             }
 927             try {
 928                 m.updateAccountAttributes(name);
 929                 // update device list
 930                 updateDevices();
 931             } catch (IOException e) {
 932                 throw new Error.Failure(e.getMessage());
 933             }
 934         }
 935     }
 936
 937     public class DbusSignalGroupImpl extends DbusProperties implements Signal.Group {
 938
 939         private final GroupId groupId;
 940
 941         public DbusSignalGroupImpl(final GroupId groupId) {
 942             this.groupId = groupId;
 943             super.addPropertiesHandler(new DbusInterfacePropertiesHandler("org.asamk.Signal.Group",
 944                     List.of(new DbusProperty<>("Id", groupId::serialize),
 945                             new DbusProperty<>("Name", () -> emptyIfNull(getGroup().title()), this::setGroupName),
 946                             new DbusProperty<>("Description",
 947                                     () -> emptyIfNull(getGroup().description()),
 948                                     this::setGroupDescription),
 949                             new DbusProperty<>("Avatar", this::setGroupAvatar),
 950                             new DbusProperty<>("IsBlocked", () -> getGroup().isBlocked(), this::setIsBlocked),
 951                             new DbusProperty<>("IsMember", () -> getGroup().isMember()),
 952                             new DbusProperty<>("IsAdmin", () -> getGroup().isAdmin()),
 953                             new DbusProperty<>("MessageExpirationTimer",
 954                                     () -> getGroup().messageExpirationTimer(),
 955                                     this::setMessageExpirationTime),
 956                             new DbusProperty<>("Members",
 957                                     () -> new Variant<>(getRecipientStrings(getGroup().members()), "as")),
 958                             new DbusProperty<>("PendingMembers",
 959                                     () -> new Variant<>(getRecipientStrings(getGroup().pendingMembers()), "as")),
 960                             new DbusProperty<>("RequestingMembers",
 961                                     () -> new Variant<>(getRecipientStrings(getGroup().requestingMembers()), "as")),
 962                             new DbusProperty<>("Admins",
 963                                     () -> new Variant<>(getRecipientStrings(getGroup().adminMembers()), "as")),
 964                             new DbusProperty<>("PermissionAddMember",
 965                                     () -> getGroup().permissionAddMember().name(),
 966                                     this::setGroupPermissionAddMember),
 967                             new DbusProperty<>("PermissionEditDetails",
 968                                     () -> getGroup().permissionEditDetails().name(),
 969                                     this::setGroupPermissionEditDetails),
 970                             new DbusProperty<>("PermissionSendMessage",
 971                                     () -> getGroup().permissionSendMessage().name(),
 972                                     this::setGroupPermissionSendMessage),
 973                             new DbusProperty<>("GroupInviteLink", () -> {
 974                                 final var groupInviteLinkUrl = getGroup().groupInviteLinkUrl();
 975                                 return groupInviteLinkUrl == null ? "" : groupInviteLinkUrl.getUrl();
 976                             }))));
 977         }
 978
 979         @Override
 980         public String getObjectPath() {
 981             return getGroupObjectPath(objectPath, groupId.serialize());
 982         }
 983
 984         @Override
 985         public void quitGroup() throws Error.Failure {
 986             try {
 987                 m.quitGroup(groupId, Set.of());
 988             } catch (GroupNotFoundException | NotAGroupMemberException e) {
 989                 throw new Error.GroupNotFound(e.getMessage());
 990             } catch (IOException e) {
 991                 throw new Error.Failure(e.getMessage());
 992             } catch (LastGroupAdminException e) {
 993                 throw new Error.LastGroupAdmin(e.getMessage());
 994             }
 995         }
 996
 997         @Override
 998         public void addMembers(final List<String> recipients) throws Error.Failure {
 999             final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1000             updateGroup(UpdateGroup.newBuilder().withMembers(memberIdentifiers).build());
1001         }
1002
1003         @Override
1004         public void removeMembers(final List<String> recipients) throws Error.Failure {
1005             final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1006             updateGroup(UpdateGroup.newBuilder().withRemoveMembers(memberIdentifiers).build());
1007         }
1008
1009         @Override
1010         public void addAdmins(final List<String> recipients) throws Error.Failure {
1011             final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1012             updateGroup(UpdateGroup.newBuilder().withAdmins(memberIdentifiers).build());
1013         }
1014
1015         @Override
1016         public void removeAdmins(final List<String> recipients) throws Error.Failure {
1017             final var memberIdentifiers = getSingleRecipientIdentifiers(recipients, m.getSelfNumber());
1018             updateGroup(UpdateGroup.newBuilder().withRemoveAdmins(memberIdentifiers).build());
1019         }
1020
1021         @Override
1022         public void resetLink() throws Error.Failure {
1023             updateGroup(UpdateGroup.newBuilder().withResetGroupLink(true).build());
1024         }
1025
1026         @Override
1027         public void disableLink() throws Error.Failure {
1028             updateGroup(UpdateGroup.newBuilder().withGroupLinkState(GroupLinkState.DISABLED).build());
1029         }
1030
1031         @Override
1032         public void enableLink(final boolean requiresApproval) throws Error.Failure {
1033             updateGroup(UpdateGroup.newBuilder()
1034                     .withGroupLinkState(requiresApproval
1035                             ? GroupLinkState.ENABLED_WITH_APPROVAL
1036                             : GroupLinkState.ENABLED)
1037                     .build());
1038         }
1039
1040         private org.asamk.signal.manager.api.Group getGroup() {
1041             return m.getGroup(groupId);
1042         }
1043
1044         private void setGroupName(final String name) {
1045             updateGroup(UpdateGroup.newBuilder().withName(name).build());
1046         }
1047
1048         private void setGroupDescription(final String description) {
1049             updateGroup(UpdateGroup.newBuilder().withDescription(description).build());
1050         }
1051
1052         private void setGroupAvatar(final String avatar) {
1053             updateGroup(UpdateGroup.newBuilder().withAvatarFile(new File(avatar)).build());
1054         }
1055
1056         private void setMessageExpirationTime(final int expirationTime) {
1057             updateGroup(UpdateGroup.newBuilder().withExpirationTimer(expirationTime).build());
1058         }
1059
1060         private void setGroupPermissionAddMember(final String permission) {
1061             updateGroup(UpdateGroup.newBuilder().withAddMemberPermission(GroupPermission.valueOf(permission)).build());
1062         }
1063
1064         private void setGroupPermissionEditDetails(final String permission) {
1065             updateGroup(UpdateGroup.newBuilder()
1066                     .withEditDetailsPermission(GroupPermission.valueOf(permission))
1067                     .build());
1068         }
1069
1070         private void setGroupPermissionSendMessage(final String permission) {
1071             updateGroup(UpdateGroup.newBuilder()
1072                     .withIsAnnouncementGroup(GroupPermission.valueOf(permission) == GroupPermission.ONLY_ADMINS)
1073                     .build());
1074         }
1075
1076         private void setIsBlocked(final boolean isBlocked) {
1077             try {
1078                 m.setGroupBlocked(groupId, isBlocked);
1079             } catch (NotMasterDeviceException e) {
1080                 throw new Error.Failure("This command doesn't work on linked devices.");
1081             } catch (GroupNotFoundException e) {
1082                 throw new Error.GroupNotFound(e.getMessage());
1083             } catch (IOException e) {
1084                 throw new Error.Failure(e.getMessage());
1085             }
1086         }
1087
1088         private void updateGroup(final UpdateGroup updateGroup) {
1089             try {
1090                 m.updateGroup(groupId, updateGroup);
1091             } catch (IOException e) {
1092                 throw new Error.Failure(e.getMessage());
1093             } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
1094                 throw new Error.GroupNotFound(e.getMessage());
1095             } catch (AttachmentInvalidException e) {
1096                 throw new Error.AttachmentInvalid(e.getMessage());
1097             }
1098         }
1099     }
1100 }