package org.asamk.signal.commands;
import net.sourceforge.argparse4j.impl.Arguments;
-import net.sourceforge.argparse4j.inf.MutuallyExclusiveGroup;
import net.sourceforge.argparse4j.inf.Namespace;
import net.sourceforge.argparse4j.inf.Subparser;
import org.asamk.Signal;
-import org.asamk.signal.manager.groups.GroupIdFormatException;
+import org.asamk.signal.JsonWriter;
+import org.asamk.signal.OutputWriter;
+import org.asamk.signal.PlainTextWriter;
+import org.asamk.signal.commands.exceptions.CommandException;
+import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
+import org.asamk.signal.commands.exceptions.UntrustedKeyErrorException;
+import org.asamk.signal.commands.exceptions.UserErrorException;
+import org.asamk.signal.manager.AttachmentInvalidException;
+import org.asamk.signal.manager.Manager;
+import org.asamk.signal.manager.api.Message;
+import org.asamk.signal.manager.api.RecipientIdentifier;
+import org.asamk.signal.manager.groups.GroupNotFoundException;
+import org.asamk.signal.manager.groups.GroupSendingNotAllowedException;
+import org.asamk.signal.manager.groups.NotAGroupMemberException;
+import org.asamk.signal.util.CommandUtil;
+import org.asamk.signal.util.ErrorUtils;
import org.asamk.signal.util.IOUtils;
-import org.asamk.signal.util.Util;
import org.freedesktop.dbus.errors.UnknownObject;
import org.freedesktop.dbus.exceptions.DBusExecutionException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
-import static org.asamk.signal.util.ErrorUtils.handleAssertionError;
-import static org.asamk.signal.util.ErrorUtils.handleGroupIdFormatException;
+public class SendCommand implements DbusCommand, JsonRpcLocalCommand {
-public class SendCommand implements DbusCommand {
+ private final static Logger logger = LoggerFactory.getLogger(SendCommand.class);
+
+ @Override
+ public String getName() {
+ return "send";
+ }
@Override
public void attachToSubparser(final Subparser subparser) {
+ subparser.help("Send a message to another user or group.");
subparser.addArgument("recipient").help("Specify the recipients' phone number.").nargs("*");
- final MutuallyExclusiveGroup mutuallyExclusiveGroup = subparser.addMutuallyExclusiveGroup();
- mutuallyExclusiveGroup.addArgument("-g", "--group").help("Specify the recipient group ID.");
- mutuallyExclusiveGroup.addArgument("--note-to-self")
+ subparser.addArgument("-g", "--group-id", "--group").help("Specify the recipient group ID.").nargs("*");
+ subparser.addArgument("--note-to-self")
.help("Send the message to self without notification.")
.action(Arguments.storeTrue());
subparser.addArgument("-m", "--message").help("Specify the message, if missing standard input is used.");
subparser.addArgument("-a", "--attachment").nargs("*").help("Add file as attachment");
- subparser.addArgument("-e", "--endsession")
+ subparser.addArgument("-e", "--end-session", "--endsession")
.help("Clear session state and send end session message.")
.action(Arguments.storeTrue());
}
@Override
- public int handleCommand(final Namespace ns, final Signal signal) {
- final List<String> recipients = ns.getList("recipient");
- final Boolean isEndSession = ns.getBoolean("endsession");
- final String groupIdString = ns.getString("group");
- final Boolean isNoteToSelf = ns.getBoolean("note_to_self");
-
- final boolean noRecipients = recipients == null || recipients.isEmpty();
- if ((noRecipients && isEndSession) || (noRecipients && groupIdString == null && !isNoteToSelf)) {
- System.err.println("No recipients given");
- System.err.println("Aborting sending.");
- return 1;
+ public void handleCommand(
+ final Namespace ns, final Manager m, final OutputWriter outputWriter
+ ) throws CommandException {
+ final var isNoteToSelf = ns.getBoolean("note-to-self");
+ final var recipientStrings = ns.<String>getList("recipient");
+ final var groupIdStrings = ns.<String>getList("group-id");
+
+ final var recipientIdentifiers = CommandUtil.getRecipientIdentifiers(m,
+ isNoteToSelf,
+ recipientStrings,
+ groupIdStrings);
+
+ final var isEndSession = ns.getBoolean("end-session");
+ if (isEndSession) {
+ final var singleRecipients = recipientIdentifiers.stream()
+ .filter(r -> r instanceof RecipientIdentifier.Single)
+ .map(RecipientIdentifier.Single.class::cast)
+ .collect(Collectors.toSet());
+ if (singleRecipients.isEmpty()) {
+ throw new UserErrorException("No recipients given");
+ }
+
+ try {
+ m.sendEndSessionMessage(singleRecipients);
+ return;
+ } catch (IOException e) {
+ throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
+ }
}
- if (!noRecipients && groupIdString != null) {
- System.err.println("You cannot specify recipients by phone number and groups at the same time");
- return 1;
+
+ var messageText = ns.getString("message");
+ if (messageText == null) {
+ try {
+ messageText = IOUtils.readAll(System.in, Charset.defaultCharset());
+ } catch (IOException e) {
+ throw new UserErrorException("Failed to read message from stdin: " + e.getMessage());
+ }
+ }
+
+ List<String> attachments = ns.getList("attachment");
+ if (attachments == null) {
+ attachments = List.of();
+ }
+
+ try {
+ var results = m.sendMessage(new Message(messageText, attachments), recipientIdentifiers);
+ outputResult(outputWriter, results.getTimestamp());
+ ErrorUtils.handleSendMessageResults(results.getResults());
+ } catch (AttachmentInvalidException | IOException e) {
+ throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
+ } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
+ throw new UserErrorException(e.getMessage());
+ }
+ }
+
+ @Override
+ public void handleCommand(
+ final Namespace ns, final Signal signal, final OutputWriter outputWriter
+ ) throws CommandException {
+ final var recipients = ns.<String>getList("recipient");
+ final var isEndSession = ns.getBoolean("end-session");
+ final var groupIdStrings = ns.<String>getList("group-id");
+ final var isNoteToSelf = ns.getBoolean("note-to-self");
+
+ final var noRecipients = recipients == null || recipients.isEmpty();
+ final var noGroups = groupIdStrings == null || groupIdStrings.isEmpty();
+ if ((noRecipients && isEndSession) || (noRecipients && noGroups && !isNoteToSelf)) {
+ throw new UserErrorException("No recipients given");
+ }
+ if (!noRecipients && !noGroups) {
+ throw new UserErrorException("You cannot specify recipients by phone number and groups at the same time");
}
if (!noRecipients && isNoteToSelf) {
- System.err.println("You cannot specify recipients by phone number and not to self at the same time");
- return 1;
+ throw new UserErrorException(
+ "You cannot specify recipients by phone number and note to self at the same time");
}
if (isEndSession) {
try {
signal.sendEndSessionMessage(recipients);
- return 0;
- } catch (AssertionError e) {
- handleAssertionError(e);
- return 1;
+ return;
+ } catch (Signal.Error.UntrustedIdentity e) {
+ throw new UntrustedKeyErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
} catch (DBusExecutionException e) {
- System.err.println("Failed to send message: " + e.getMessage());
- return 2;
+ throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
}
}
- String messageText = ns.getString("message");
+ var messageText = ns.getString("message");
if (messageText == null) {
try {
messageText = IOUtils.readAll(System.in, Charset.defaultCharset());
} catch (IOException e) {
- System.err.println("Failed to read message from stdin: " + e.getMessage());
- System.err.println("Aborting sending.");
- return 1;
+ throw new UserErrorException("Failed to read message from stdin: " + e.getMessage());
}
}
attachments = List.of();
}
- if (groupIdString != null) {
+ if (!noGroups) {
+ final var groupIds = CommandUtil.getGroupIds(groupIdStrings);
+
try {
- byte[] groupId;
- try {
- groupId = Util.decodeGroupId(groupIdString).serialize();
- } catch (GroupIdFormatException e) {
- handleGroupIdFormatException(e);
- return 1;
+ long timestamp = 0;
+ for (final var groupId : groupIds) {
+ timestamp = signal.sendGroupMessage(messageText, attachments, groupId.serialize());
}
-
- long timestamp = signal.sendGroupMessage(messageText, attachments, groupId);
- System.out.println(timestamp);
- return 0;
- } catch (AssertionError e) {
- handleAssertionError(e);
- return 1;
+ outputResult(outputWriter, timestamp);
+ return;
} catch (DBusExecutionException e) {
- System.err.println("Failed to send group message: " + e.getMessage());
- return 2;
+ throw new UnexpectedErrorException("Failed to send group message: " + e.getMessage());
}
}
if (isNoteToSelf) {
try {
- long timestamp = signal.sendNoteToSelfMessage(messageText, attachments);
- System.out.println(timestamp);
- return 0;
- } catch (AssertionError e) {
- handleAssertionError(e);
- return 1;
+ var timestamp = signal.sendNoteToSelfMessage(messageText, attachments);
+ outputResult(outputWriter, timestamp);
+ return;
+ } catch (Signal.Error.UntrustedIdentity e) {
+ throw new UntrustedKeyErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
} catch (DBusExecutionException e) {
- System.err.println("Failed to send note to self message: " + e.getMessage());
- return 2;
+ throw new UnexpectedErrorException("Failed to send note to self message: " + e.getMessage());
}
}
try {
- long timestamp = signal.sendMessage(messageText, attachments, recipients);
- System.out.println(timestamp);
- return 0;
- } catch (AssertionError e) {
- handleAssertionError(e);
- return 1;
+ var timestamp = signal.sendMessage(messageText, attachments, recipients);
+ outputResult(outputWriter, timestamp);
} catch (UnknownObject e) {
- System.err.println("Failed to find dbus object, maybe missing the -u flag: " + e.getMessage());
- return 1;
+ throw new UserErrorException("Failed to find dbus object, maybe missing the -u flag: " + e.getMessage());
+ } catch (Signal.Error.UntrustedIdentity e) {
+ throw new UntrustedKeyErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
} catch (DBusExecutionException e) {
- System.err.println("Failed to send message: " + e.getMessage());
- return 2;
+ throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
+ .getSimpleName() + ")");
+ }
+ }
+
+ private void outputResult(final OutputWriter outputWriter, final long timestamp) {
+ if (outputWriter instanceof PlainTextWriter) {
+ final var writer = (PlainTextWriter) outputWriter;
+ writer.println("{}", timestamp);
+ } else {
+ final var writer = (JsonWriter) outputWriter;
+ writer.write(Map.of("timestamp", timestamp));
}
}
}
git.nmode.ca / signal-cli / blobdiff
