Extract getTextSecureAttachments method

This commit is contained in:
AsamK 2015-12-10 21:42:44 +01:00
parent c1abc12907
commit 4b5bfcba80

View file

@ -119,20 +119,14 @@ public class Main {
if (ns.getBoolean("endsession")) { if (ns.getBoolean("endsession")) {
sendEndSessionMessage(m, recipients); sendEndSessionMessage(m, recipients);
} else { } else {
final List<String> attachments = ns.getList("attachment");
List<TextSecureAttachment> textSecureAttachments = null; List<TextSecureAttachment> textSecureAttachments = null;
if (attachments != null) {
textSecureAttachments = new ArrayList<>(attachments.size());
for (String attachment : attachments) {
try { try {
textSecureAttachments.add(createAttachment(attachment)); textSecureAttachments = getTextSecureAttachments(ns.<String>getList("attachment"));
} catch (IOException e) { } catch (IOException e) {
System.err.println("Failed to add attachment \"" + attachment + "\": " + e.getMessage()); System.err.println("Failed to add attachment: " + e.getMessage());
System.err.println("Aborting sending."); System.err.println("Aborting sending.");
System.exit(1); System.exit(1);
} }
}
}
String messageText = ns.getString("message"); String messageText = ns.getString("message");
if (messageText == null) { if (messageText == null) {
@ -270,6 +264,18 @@ public class Main {
System.exit(0); System.exit(0);
} }
private static List<TextSecureAttachment> getTextSecureAttachments(List<String> attachments) {
private static List<TextSecureAttachment> getTextSecureAttachments(List<String> attachments) throws IOException {
List<TextSecureAttachment> textSecureAttachments = null;
if (attachments != null) {
textSecureAttachments = new ArrayList<>(attachments.size());
for (String attachment : attachments) {
textSecureAttachments.add(createAttachment(attachment));
}
}
return textSecureAttachments;
}
private static TextSecureAttachmentStream createAttachment(String attachment) throws IOException { private static TextSecureAttachmentStream createAttachment(String attachment) throws IOException {
File attachmentFile = new File(attachment); File attachmentFile = new File(attachment);
InputStream attachmentStream = new FileInputStream(attachmentFile); InputStream attachmentStream = new FileInputStream(attachmentFile);