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