Skip to content

Commit

Permalink
Add --quote-attachment paramter to send command
Browse files Browse the repository at this point in the history
Fixes #1318
  • Loading branch information
AsamK committed Oct 10, 2023
1 parent 91ab0b1 commit 7b5b577
Show file tree
Hide file tree
Showing 7 changed files with 64 additions and 22 deletions.
3 changes: 3 additions & 0 deletions client/src/cli.rs
Original file line number Diff line number Diff line change
Expand Up @@ -164,6 +164,9 @@ pub enum CliCommands {
#[arg(long)]
quote_mention: Vec<String>,

#[arg(long)]
quote_attachment: Vec<String>,

#[arg(long)]
sticker: Option<String>,

Expand Down
1 change: 1 addition & 0 deletions client/src/jsonrpc.rs
Original file line number Diff line number Diff line change
Expand Up @@ -143,6 +143,7 @@ pub trait Rpc {
#[allow(non_snake_case)] quoteAuthor: Option<String>,
#[allow(non_snake_case)] quoteMessage: Option<String>,
#[allow(non_snake_case)] quoteMention: Vec<String>,
#[allow(non_snake_case)] quoteAttachment: Vec<String>,
sticker: Option<String>,
#[allow(non_snake_case)] storyTimestamp: Option<u64>,
#[allow(non_snake_case)] storyAuthor: Option<String>,
Expand Down
2 changes: 2 additions & 0 deletions client/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,7 @@ async fn handle_command(
quote_author,
quote_message,
quote_mention,
quote_attachment,
sticker,
story_timestamp,
story_author,
Expand All @@ -158,6 +159,7 @@ async fn handle_command(
quote_author,
quote_message,
quote_mention,
quote_attachment,
sticker,
story_timestamp,
story_author,
Expand Down
6 changes: 4 additions & 2 deletions graalvm-config-dir/reflect-config.json
Original file line number Diff line number Diff line change
Expand Up @@ -544,7 +544,8 @@
{
"name":"org.asamk.Signal",
"allDeclaredMethods":true,
"allDeclaredClasses":true
"allDeclaredClasses":true,
"methods":[{"name":"getSelfNumber","parameterTypes":[] }, {"name":"sendGroupMessageReaction","parameterTypes":["java.lang.String","boolean","java.lang.String","long","byte[]"] }, {"name":"sendMessage","parameterTypes":["java.lang.String","java.util.List","java.lang.String"] }, {"name":"sendMessageReaction","parameterTypes":["java.lang.String","boolean","java.lang.String","long","java.util.List"] }]
},
{
"name":"org.asamk.Signal$Configuration",
Expand Down Expand Up @@ -1747,7 +1748,8 @@
{
"name":"org.freedesktop.dbus.interfaces.Introspectable",
"allDeclaredMethods":true,
"allDeclaredClasses":true
"allDeclaredClasses":true,
"methods":[{"name":"Introspect","parameterTypes":[] }]
},
{
"name":"org.freedesktop.dbus.interfaces.Peer",
Expand Down
8 changes: 6 additions & 2 deletions lib/src/main/java/org/asamk/signal/manager/api/Message.java
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,12 @@ public record Quote(
RecipientIdentifier.Single author,
String message,
List<Mention> mentions,
List<TextStyle> textStyles
) {}
List<TextStyle> textStyles,
List<Attachment> attachments
) {

public record Attachment(String contentType, String filename, String preview) {}
}

public record Sticker(byte[] packId, int stickerId) {}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -627,12 +627,19 @@ private void applyMessage(
}
if (message.quote().isPresent()) {
final var quote = message.quote().get();
final var quotedAttachments = new ArrayList<SignalServiceDataMessage.Quote.QuotedAttachment>();
for (final var a : quote.attachments()) {
final var quotedAttachment = new SignalServiceDataMessage.Quote.QuotedAttachment(a.contentType(),
a.filename(),
a.preview() == null ? null : context.getAttachmentHelper().uploadAttachment(a.preview()));
quotedAttachments.add(quotedAttachment);
}
messageBuilder.withQuote(new SignalServiceDataMessage.Quote(quote.timestamp(),
context.getRecipientHelper()
.resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(quote.author()))
.getServiceId(),
quote.message(),
List.of(),
quotedAttachments,
resolveMentions(quote.mentions()),
SignalServiceDataMessage.Quote.Type.NORMAL,
quote.textStyles().stream().map(TextStyle::toBodyRange).toList()));
Expand Down
57 changes: 40 additions & 17 deletions src/main/java/org/asamk/signal/commands/SendCommand.java
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,9 @@ public void attachToSubparser(final Subparser subparser) {
subparser.addArgument("--quote-mention")
.nargs("*")
.help("Quote with mention of another group member (syntax: start:length:recipientNumber)");
subparser.addArgument("--quote-attachment")
.nargs("*")
.help("Specify the attachments of the original message (syntax: contentType[:filename[:previewFile]], e.g. 'audio/aac' or 'image/png:test.png:/tmp/preview.jpg'.");
subparser.addArgument("--quote-text-style")
.nargs("*")
.help("Quote with style parts of the message text (syntax: start:length:STYLE)");
Expand Down Expand Up @@ -145,49 +148,54 @@ public void handleCommand(
messageText = "";
}

List<String> attachments = ns.getList("attachment");
var attachments = ns.<String>getList("attachment");
if (attachments == null) {
attachments = List.of();
}

final var selfNumber = m.getSelfNumber();

List<String> mentionStrings = ns.getList("mention");
final var mentionStrings = ns.<String>getList("mention");
final var mentions = mentionStrings == null
? List.<Message.Mention>of()
: parseMentions(selfNumber, mentionStrings);

List<String> textStyleStrings = ns.getList("text-style");
final var textStyleStrings = ns.<String>getList("text-style");
final var textStyles = textStyleStrings == null ? List.<TextStyle>of() : parseTextStyles(textStyleStrings);

final Message.Quote quote;
final var quoteTimestamp = ns.getLong("quote-timestamp");
if (quoteTimestamp != null) {
final var quoteAuthor = ns.getString("quote-author");
final var quoteMessage = ns.getString("quote-message");
List<String> quoteMentionStrings = ns.getList("quote-mention");
final var quoteMentionStrings = ns.<String>getList("quote-mention");
final var quoteMentions = quoteMentionStrings == null
? List.<Message.Mention>of()
: parseMentions(selfNumber, quoteMentionStrings);
List<String> quoteTextStyleStrings = ns.getList("quote-text-style");
final var quoteTextStyleStrings = ns.<String>getList("quote-text-style");
final var quoteAttachmentStrings = ns.<String>getList("quote-attachment");
final var quoteTextStyles = quoteTextStyleStrings == null
? List.<TextStyle>of()
: parseTextStyles(quoteTextStyleStrings);
final var quoteAttachments = quoteAttachmentStrings == null
? List.<Message.Quote.Attachment>of()
: parseQuoteAttachments(quoteAttachmentStrings);
quote = new Message.Quote(quoteTimestamp,
CommandUtil.getSingleRecipientIdentifier(quoteAuthor, selfNumber),
quoteMessage == null ? "" : quoteMessage,
quoteMentions,
quoteTextStyles);
quoteTextStyles,
quoteAttachments);
} else {
quote = null;
}

final List<Message.Preview> previews;
String previewUrl = ns.getString("preview-url");
final var previewUrl = ns.getString("preview-url");
if (previewUrl != null) {
String previewTitle = ns.getString("preview-title");
String previewDescription = ns.getString("preview-description");
String previewImage = ns.getString("preview-image");
final var previewTitle = ns.getString("preview-title");
final var previewDescription = ns.getString("preview-description");
final var previewImage = ns.getString("preview-image");
previews = List.of(new Message.Preview(previewUrl,
Optional.ofNullable(previewTitle).orElse(""),
Optional.ofNullable(previewDescription).orElse(""),
Expand Down Expand Up @@ -241,9 +249,8 @@ public void handleCommand(
private List<Message.Mention> parseMentions(
final String selfNumber, final List<String> mentionStrings
) throws UserErrorException {
List<Message.Mention> mentions;
final Pattern mentionPattern = Pattern.compile("(\\d+):(\\d+):(.+)");
mentions = new ArrayList<>();
final var mentionPattern = Pattern.compile("(\\d+):(\\d+):(.+)");
final var mentions = new ArrayList<Message.Mention>();
for (final var mention : mentionStrings) {
final var matcher = mentionPattern.matcher(mention);
if (!matcher.matches()) {
Expand All @@ -261,9 +268,8 @@ private List<Message.Mention> parseMentions(
private List<TextStyle> parseTextStyles(
final List<String> textStylesStrings
) throws UserErrorException {
List<TextStyle> textStyles;
final Pattern textStylePattern = Pattern.compile("(\\d+):(\\d+):(.+)");
textStyles = new ArrayList<>();
final var textStylePattern = Pattern.compile("(\\d+):(\\d+):(.+)");
final var textStyles = new ArrayList<TextStyle>();
for (final var textStyle : textStylesStrings) {
final var matcher = textStylePattern.matcher(textStyle);
if (!matcher.matches()) {
Expand All @@ -283,7 +289,7 @@ private List<TextStyle> parseTextStyles(
}

private Message.Sticker parseSticker(final String stickerString) throws UserErrorException {
final Pattern stickerPattern = Pattern.compile("([\\da-f]+):(\\d+)");
final var stickerPattern = Pattern.compile("([\\da-f]+):(\\d+)");
final var matcher = stickerPattern.matcher(stickerString);
if (!matcher.matches() || matcher.group(1).length() % 2 != 0) {
throw new UserErrorException("Invalid sticker syntax ("
Expand All @@ -292,4 +298,21 @@ private Message.Sticker parseSticker(final String stickerString) throws UserErro
}
return new Message.Sticker(Hex.toByteArray(matcher.group(1)), Integer.parseInt(matcher.group(2)));
}

private List<Message.Quote.Attachment> parseQuoteAttachments(
final List<String> attachmentStrings
) throws UserErrorException {
final var attachmentPattern = Pattern.compile("([^:]+)(:([^:]+)(:(.+))?)?");
final var attachments = new ArrayList<Message.Quote.Attachment>();
for (final var attachment : attachmentStrings) {
final var matcher = attachmentPattern.matcher(attachment);
if (!matcher.matches()) {
throw new UserErrorException("Invalid attachment syntax ("
+ attachment
+ ") expected 'contentType[:filename[:previewFile]]'");
}
attachments.add(new Message.Quote.Attachment(matcher.group(1), matcher.group(3), matcher.group(5)));
}
return attachments;
}
}

0 comments on commit 7b5b577

Please sign in to comment.