nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/dbus/DbusManagerImpl.java

   1 package org.asamk.signal.dbus;
   2
   3 import org.asamk.Signal;
   4 import org.asamk.signal.manager.AttachmentInvalidException;
   5 import org.asamk.signal.manager.Manager;
   6 import org.asamk.signal.manager.NotMasterDeviceException;
   7 import org.asamk.signal.manager.StickerPackInvalidException;
   8 import org.asamk.signal.manager.UntrustedIdentityException;
   9 import org.asamk.signal.manager.api.Device;
  10 import org.asamk.signal.manager.api.Group;
  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.SendGroupMessageResults;
  15 import org.asamk.signal.manager.api.SendMessageResults;
  16 import org.asamk.signal.manager.api.TypingAction;
  17 import org.asamk.signal.manager.groups.GroupId;
  18 import org.asamk.signal.manager.groups.GroupInviteLinkUrl;
  19 import org.asamk.signal.manager.groups.GroupLinkState;
  20 import org.asamk.signal.manager.groups.GroupNotFoundException;
  21 import org.asamk.signal.manager.groups.GroupPermission;
  22 import org.asamk.signal.manager.groups.GroupSendingNotAllowedException;
  23 import org.asamk.signal.manager.groups.LastGroupAdminException;
  24 import org.asamk.signal.manager.groups.NotAGroupMemberException;
  25 import org.asamk.signal.manager.storage.recipients.Contact;
  26 import org.asamk.signal.manager.storage.recipients.Profile;
  27 import org.asamk.signal.manager.storage.recipients.RecipientAddress;
  28 import org.whispersystems.libsignal.IdentityKey;
  29 import org.whispersystems.libsignal.InvalidKeyException;
  30 import org.whispersystems.libsignal.util.Pair;
  31 import org.whispersystems.libsignal.util.guava.Optional;
  32 import org.whispersystems.signalservice.api.groupsv2.GroupLinkNotActiveException;
  33 import org.whispersystems.signalservice.api.messages.SignalServiceAttachmentRemoteId;
  34 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
  35 import org.whispersystems.signalservice.api.push.exceptions.UnregisteredUserException;
  36 import org.whispersystems.signalservice.api.util.UuidUtil;
  37 import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
  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.HashMap;
  45 import java.util.List;
  46 import java.util.Map;
  47 import java.util.Set;
  48 import java.util.UUID;
  49 import java.util.concurrent.TimeUnit;
  50 import java.util.function.Function;
  51 import java.util.function.Supplier;
  52 import java.util.stream.Collectors;
  53
  54 /**
  55  * This class implements the Manager interface using the DBus Signal interface, where possible.
  56  * It's used for the signal-cli dbus client mode (--dbus, --dbus-system)
  57  */
  58 public class DbusManagerImpl implements Manager {
  59
  60     private final Signal signal;
  61
  62     public DbusManagerImpl(final Signal signal) {
  63         this.signal = signal;
  64     }
  65
  66     @Override
  67     public String getSelfNumber() {
  68         return signal.getSelfNumber();
  69     }
  70
  71     @Override
  72     public void checkAccountState() throws IOException {
  73         throw new UnsupportedOperationException();
  74     }
  75
  76     @Override
  77     public Map<String, Pair<String, UUID>> areUsersRegistered(final Set<String> numbers) throws IOException {
  78         final var numbersList = new ArrayList<>(numbers);
  79         final var registered = signal.isRegistered(numbersList);
  80
  81         final var result = new HashMap<String, Pair<String, UUID>>();
  82         for (var i = 0; i < numbersList.size(); i++) {
  83             result.put(numbersList.get(i),
  84                     new Pair<>(numbersList.get(i), registered.get(i) ? UuidUtil.UNKNOWN_UUID : null));
  85         }
  86         return result;
  87     }
  88
  89     @Override
  90     public void updateAccountAttributes(final String deviceName) throws IOException {
  91         if (deviceName != null) {
  92             signal.updateDeviceName(deviceName);
  93         }
  94     }
  95
  96     @Override
  97     public void setProfile(
  98             final String givenName,
  99             final String familyName,
 100             final String about,
 101             final String aboutEmoji,
 102             final Optional<File> avatar
 103     ) throws IOException {
 104         signal.updateProfile(emptyIfNull(givenName),
 105                 emptyIfNull(familyName),
 106                 emptyIfNull(about),
 107                 emptyIfNull(aboutEmoji),
 108                 avatar == null ? "" : avatar.transform(File::getPath).or(""),
 109                 avatar != null && !avatar.isPresent());
 110     }
 111
 112     @Override
 113     public void unregister() throws IOException {
 114         throw new UnsupportedOperationException();
 115     }
 116
 117     @Override
 118     public void deleteAccount() throws IOException {
 119         throw new UnsupportedOperationException();
 120     }
 121
 122     @Override
 123     public void submitRateLimitRecaptchaChallenge(final String challenge, final String captcha) throws IOException {
 124         throw new UnsupportedOperationException();
 125     }
 126
 127     @Override
 128     public List<Device> getLinkedDevices() throws IOException {
 129         return signal.listDevices()
 130                 .stream()
 131                 .map(name -> new Device(-1, name, 0, 0, false))
 132                 .collect(Collectors.toList());
 133     }
 134
 135     @Override
 136     public void removeLinkedDevices(final int deviceId) throws IOException {
 137         signal.removeDevice(deviceId);
 138     }
 139
 140     @Override
 141     public void addDeviceLink(final URI linkUri) throws IOException, InvalidKeyException {
 142         signal.addDevice(linkUri.toString());
 143     }
 144
 145     @Override
 146     public void setRegistrationLockPin(final Optional<String> pin) throws IOException, UnauthenticatedResponseException {
 147         if (pin.isPresent()) {
 148             signal.setPin(pin.get());
 149         } else {
 150             signal.removePin();
 151         }
 152     }
 153
 154     @Override
 155     public Profile getRecipientProfile(final RecipientIdentifier.Single recipient) throws UnregisteredUserException {
 156         throw new UnsupportedOperationException();
 157     }
 158
 159     @Override
 160     public List<Group> getGroups() {
 161         final var groupIds = signal.getGroupIds();
 162         return groupIds.stream().map(id -> getGroup(GroupId.unknownVersion(id))).collect(Collectors.toList());
 163     }
 164
 165     @Override
 166     public SendGroupMessageResults quitGroup(
 167             final GroupId groupId, final Set<RecipientIdentifier.Single> groupAdmins
 168     ) throws GroupNotFoundException, IOException, NotAGroupMemberException, LastGroupAdminException {
 169         if (groupAdmins.size() > 0) {
 170             throw new UnsupportedOperationException();
 171         }
 172         signal.quitGroup(groupId.serialize());
 173         return new SendGroupMessageResults(0, List.of());
 174     }
 175
 176     @Override
 177     public void deleteGroup(final GroupId groupId) throws IOException {
 178         throw new UnsupportedOperationException();
 179     }
 180
 181     @Override
 182     public Pair<GroupId, SendGroupMessageResults> createGroup(
 183             final String name, final Set<RecipientIdentifier.Single> members, final File avatarFile
 184     ) throws IOException, AttachmentInvalidException {
 185         final var newGroupId = signal.updateGroup(new byte[0],
 186                 emptyIfNull(name),
 187                 members.stream().map(RecipientIdentifier.Single::getIdentifier).collect(Collectors.toList()),
 188                 avatarFile == null ? "" : avatarFile.getPath());
 189         return new Pair<>(GroupId.unknownVersion(newGroupId), new SendGroupMessageResults(0, List.of()));
 190     }
 191
 192     @Override
 193     public SendGroupMessageResults updateGroup(
 194             final GroupId groupId,
 195             final String name,
 196             final String description,
 197             final Set<RecipientIdentifier.Single> members,
 198             final Set<RecipientIdentifier.Single> removeMembers,
 199             final Set<RecipientIdentifier.Single> admins,
 200             final Set<RecipientIdentifier.Single> removeAdmins,
 201             final boolean resetGroupLink,
 202             final GroupLinkState groupLinkState,
 203             final GroupPermission addMemberPermission,
 204             final GroupPermission editDetailsPermission,
 205             final File avatarFile,
 206             final Integer expirationTimer,
 207             final Boolean isAnnouncementGroup
 208     ) throws IOException, GroupNotFoundException, AttachmentInvalidException, NotAGroupMemberException, GroupSendingNotAllowedException {
 209         signal.updateGroup(groupId.serialize(),
 210                 emptyIfNull(name),
 211                 members.stream().map(RecipientIdentifier.Single::getIdentifier).collect(Collectors.toList()),
 212                 avatarFile == null ? "" : avatarFile.getPath());
 213         return new SendGroupMessageResults(0, List.of());
 214     }
 215
 216     @Override
 217     public Pair<GroupId, SendGroupMessageResults> joinGroup(final GroupInviteLinkUrl inviteLinkUrl) throws IOException, GroupLinkNotActiveException {
 218         final var newGroupId = signal.joinGroup(inviteLinkUrl.getUrl());
 219         return new Pair<>(GroupId.unknownVersion(newGroupId), new SendGroupMessageResults(0, List.of()));
 220     }
 221
 222     @Override
 223     public void sendTypingMessage(
 224             final TypingAction action, final Set<RecipientIdentifier> recipients
 225     ) throws IOException, UntrustedIdentityException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 226         for (final var recipient : recipients) {
 227             if (recipient instanceof RecipientIdentifier.Single) {
 228                 signal.sendTyping(((RecipientIdentifier.Single) recipient).getIdentifier(),
 229                         action == TypingAction.STOP);
 230             } else if (recipient instanceof RecipientIdentifier.Group) {
 231                 throw new UnsupportedOperationException();
 232             }
 233         }
 234     }
 235
 236     @Override
 237     public void sendReadReceipt(
 238             final RecipientIdentifier.Single sender, final List<Long> messageIds
 239     ) throws IOException, UntrustedIdentityException {
 240         signal.sendReadReceipt(sender.getIdentifier(), messageIds);
 241     }
 242
 243     @Override
 244     public void sendViewedReceipt(
 245             final RecipientIdentifier.Single sender, final List<Long> messageIds
 246     ) throws IOException, UntrustedIdentityException {
 247         throw new UnsupportedOperationException();
 248     }
 249
 250     @Override
 251     public SendMessageResults sendMessage(
 252             final Message message, final Set<RecipientIdentifier> recipients
 253     ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 254         return handleMessage(recipients,
 255                 numbers -> signal.sendMessage(message.getMessageText(), message.getAttachments(), numbers),
 256                 () -> signal.sendNoteToSelfMessage(message.getMessageText(), message.getAttachments()),
 257                 groupId -> signal.sendGroupMessage(message.getMessageText(), message.getAttachments(), groupId));
 258     }
 259
 260     @Override
 261     public SendMessageResults sendRemoteDeleteMessage(
 262             final long targetSentTimestamp, final Set<RecipientIdentifier> recipients
 263     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 264         return handleMessage(recipients,
 265                 numbers -> signal.sendRemoteDeleteMessage(targetSentTimestamp, numbers),
 266                 () -> signal.sendRemoteDeleteMessage(targetSentTimestamp, signal.getSelfNumber()),
 267                 groupId -> signal.sendGroupRemoteDeleteMessage(targetSentTimestamp, groupId));
 268     }
 269
 270     @Override
 271     public SendMessageResults sendMessageReaction(
 272             final String emoji,
 273             final boolean remove,
 274             final RecipientIdentifier.Single targetAuthor,
 275             final long targetSentTimestamp,
 276             final Set<RecipientIdentifier> recipients
 277     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 278         return handleMessage(recipients,
 279                 numbers -> signal.sendMessageReaction(emoji,
 280                         remove,
 281                         targetAuthor.getIdentifier(),
 282                         targetSentTimestamp,
 283                         numbers),
 284                 () -> signal.sendMessageReaction(emoji,
 285                         remove,
 286                         targetAuthor.getIdentifier(),
 287                         targetSentTimestamp,
 288                         signal.getSelfNumber()),
 289                 groupId -> signal.sendGroupMessageReaction(emoji,
 290                         remove,
 291                         targetAuthor.getIdentifier(),
 292                         targetSentTimestamp,
 293                         groupId));
 294     }
 295
 296     @Override
 297     public SendMessageResults sendEndSessionMessage(final Set<RecipientIdentifier.Single> recipients) throws IOException {
 298         signal.sendEndSessionMessage(recipients.stream()
 299                 .map(RecipientIdentifier.Single::getIdentifier)
 300                 .collect(Collectors.toList()));
 301         return new SendMessageResults(0, Map.of());
 302     }
 303
 304     @Override
 305     public void setContactName(
 306             final RecipientIdentifier.Single recipient, final String name
 307     ) throws NotMasterDeviceException, UnregisteredUserException {
 308         signal.setContactName(recipient.getIdentifier(), name);
 309     }
 310
 311     @Override
 312     public void setContactBlocked(
 313             final RecipientIdentifier.Single recipient, final boolean blocked
 314     ) throws NotMasterDeviceException, IOException {
 315         signal.setContactBlocked(recipient.getIdentifier(), blocked);
 316     }
 317
 318     @Override
 319     public void setGroupBlocked(
 320             final GroupId groupId, final boolean blocked
 321     ) throws GroupNotFoundException, IOException {
 322         signal.setGroupBlocked(groupId.serialize(), blocked);
 323     }
 324
 325     @Override
 326     public void setExpirationTimer(
 327             final RecipientIdentifier.Single recipient, final int messageExpirationTimer
 328     ) throws IOException {
 329         signal.setExpirationTimer(recipient.getIdentifier(), messageExpirationTimer);
 330     }
 331
 332     @Override
 333     public URI uploadStickerPack(final File path) throws IOException, StickerPackInvalidException {
 334         try {
 335             return new URI(signal.uploadStickerPack(path.getPath()));
 336         } catch (URISyntaxException e) {
 337             throw new AssertionError(e);
 338         }
 339     }
 340
 341     @Override
 342     public void requestAllSyncData() throws IOException {
 343         signal.sendSyncRequest();
 344     }
 345
 346     @Override
 347     public void receiveMessages(
 348             final long timeout,
 349             final TimeUnit unit,
 350             final boolean returnOnTimeout,
 351             final boolean ignoreAttachments,
 352             final ReceiveMessageHandler handler
 353     ) throws IOException {
 354         throw new UnsupportedOperationException();
 355     }
 356
 357     @Override
 358     public boolean hasCaughtUpWithOldMessages() {
 359         throw new UnsupportedOperationException();
 360     }
 361
 362     @Override
 363     public boolean isContactBlocked(final RecipientIdentifier.Single recipient) {
 364         return signal.isContactBlocked(recipient.getIdentifier());
 365     }
 366
 367     @Override
 368     public File getAttachmentFile(final SignalServiceAttachmentRemoteId attachmentId) {
 369         throw new UnsupportedOperationException();
 370     }
 371
 372     @Override
 373     public void sendContacts() throws IOException {
 374         signal.sendContacts();
 375     }
 376
 377     @Override
 378     public List<Pair<RecipientAddress, Contact>> getContacts() {
 379         throw new UnsupportedOperationException();
 380     }
 381
 382     @Override
 383     public String getContactOrProfileName(final RecipientIdentifier.Single recipient) {
 384         return signal.getContactName(recipient.getIdentifier());
 385     }
 386
 387     @Override
 388     public Group getGroup(final GroupId groupId) {
 389         final var id = groupId.serialize();
 390         return new Group(groupId,
 391                 signal.getGroupName(id),
 392                 null,
 393                 null,
 394                 signal.getGroupMembers(id).stream().map(m -> new RecipientAddress(null, m)).collect(Collectors.toSet()),
 395                 Set.of(),
 396                 Set.of(),
 397                 Set.of(),
 398                 signal.isGroupBlocked(id),
 399                 0,
 400                 false,
 401                 signal.isMember(id));
 402     }
 403
 404     @Override
 405     public List<Identity> getIdentities() {
 406         throw new UnsupportedOperationException();
 407     }
 408
 409     @Override
 410     public List<Identity> getIdentities(final RecipientIdentifier.Single recipient) {
 411         throw new UnsupportedOperationException();
 412     }
 413
 414     @Override
 415     public boolean trustIdentityVerified(final RecipientIdentifier.Single recipient, final byte[] fingerprint) {
 416         throw new UnsupportedOperationException();
 417     }
 418
 419     @Override
 420     public boolean trustIdentityVerifiedSafetyNumber(
 421             final RecipientIdentifier.Single recipient, final String safetyNumber
 422     ) {
 423         throw new UnsupportedOperationException();
 424     }
 425
 426     @Override
 427     public boolean trustIdentityVerifiedSafetyNumber(
 428             final RecipientIdentifier.Single recipient, final byte[] safetyNumber
 429     ) {
 430         throw new UnsupportedOperationException();
 431     }
 432
 433     @Override
 434     public boolean trustIdentityAllKeys(final RecipientIdentifier.Single recipient) {
 435         throw new UnsupportedOperationException();
 436     }
 437
 438     @Override
 439     public String computeSafetyNumber(
 440             final SignalServiceAddress theirAddress, final IdentityKey theirIdentityKey
 441     ) {
 442         throw new UnsupportedOperationException();
 443     }
 444
 445     @Override
 446     public SignalServiceAddress resolveSignalServiceAddress(final SignalServiceAddress address) {
 447         return address;
 448     }
 449
 450     @Override
 451     public void close() throws IOException {
 452     }
 453
 454     private SendMessageResults handleMessage(
 455             Set<RecipientIdentifier> recipients,
 456             Function<List<String>, Long> recipientsHandler,
 457             Supplier<Long> noteToSelfHandler,
 458             Function<byte[], Long> groupHandler
 459     ) {
 460         long timestamp = 0;
 461         final var singleRecipients = recipients.stream()
 462                 .filter(r -> r instanceof RecipientIdentifier.Single)
 463                 .map(RecipientIdentifier.Single.class::cast)
 464                 .map(RecipientIdentifier.Single::getIdentifier)
 465                 .collect(Collectors.toList());
 466         if (singleRecipients.size() > 0) {
 467             timestamp = recipientsHandler.apply(singleRecipients);
 468         }
 469
 470         if (recipients.contains(RecipientIdentifier.NoteToSelf.INSTANCE)) {
 471             timestamp = noteToSelfHandler.get();
 472         }
 473         final var groupRecipients = recipients.stream()
 474                 .filter(r -> r instanceof RecipientIdentifier.Group)
 475                 .map(RecipientIdentifier.Group.class::cast)
 476                 .map(g -> g.groupId)
 477                 .collect(Collectors.toList());
 478         for (final var groupId : groupRecipients) {
 479             timestamp = groupHandler.apply(groupId.serialize());
 480         }
 481         return new SendMessageResults(timestamp, Map.of());
 482     }
 483
 484     private String emptyIfNull(final String string) {
 485         return string == null ? "" : string;
 486     }
 487 }