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