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

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.inf.Namespace;
   4 import net.sourceforge.argparse4j.inf.Subparser;
   5
   6 import org.asamk.signal.commands.exceptions.CommandException;
   7 import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
   8 import org.asamk.signal.commands.exceptions.UserErrorException;
   9 import org.asamk.signal.json.JsonAttachmentData;
  10 import org.asamk.signal.manager.Manager;
  11 import org.asamk.signal.output.JsonWriter;
  12 import org.asamk.signal.output.OutputWriter;
  13 import org.asamk.signal.output.PlainTextWriter;
  14
  15 import java.io.FileNotFoundException;
  16 import java.io.IOException;
  17 import java.io.InputStream;
  18 import java.util.Base64;
  19
  20 public class GetAttachmentCommand implements JsonRpcLocalCommand {
  21
  22     @Override
  23     public String getName() {
  24         return "getAttachment";
  25     }
  26
  27     @Override
  28     public void attachToSubparser(final Subparser subparser) {
  29         subparser.addArgument("--id")
  30                 .required(true)
  31                 .help("The ID of the attachment file.");
  32         var mut = subparser.addMutuallyExclusiveGroup()
  33                 .required(true);
  34         mut.addArgument("--recipient")
  35                 .help("Sender of the attachment");
  36         mut.addArgument("-g", "--group-id")
  37                 .help("Group in which the attachment was received");
  38     }
  39
  40     @Override
  41     public void handleCommand(
  42             final Namespace ns,
  43             final Manager m,
  44             final OutputWriter outputWriter
  45     ) throws CommandException {
  46
  47         final var id = ns.getString("id");
  48
  49         try(InputStream attachment = m.retrieveAttachment(id)) {
  50             if (outputWriter instanceof PlainTextWriter writer) {
  51                 final var bytes = attachment.readAllBytes();
  52                 final var base64 = Base64.getEncoder().encodeToString(bytes);
  53                 writer.println(base64);
  54             } else if (outputWriter instanceof JsonWriter writer) {
  55                 writer.write(new JsonAttachmentData(attachment));
  56             }
  57         } catch (FileNotFoundException ex) {
  58             throw new UserErrorException("Could not find attachment with ID: " + id, ex);
  59         } catch (IOException ex) {
  60             throw new UnexpectedErrorException("An error occurred reading attachment: " + id, ex);
  61         }
  62     }
  63 }