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

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.impl.Arguments;
   4 import net.sourceforge.argparse4j.inf.Namespace;
   5 import net.sourceforge.argparse4j.inf.Subparser;
   6
   7 import org.asamk.signal.commands.exceptions.CommandException;
   8 import org.asamk.signal.commands.exceptions.IOErrorException;
   9 import org.asamk.signal.manager.Manager;
  10 import org.asamk.signal.manager.api.UpdateProfile;
  11 import org.asamk.signal.output.OutputWriter;
  12
  13 import java.io.IOException;
  14 import java.util.Base64;
  15
  16 public class UpdateProfileCommand implements JsonRpcLocalCommand {
  17
  18     @Override
  19     public String getName() {
  20         return "updateProfile";
  21     }
  22
  23     @Override
  24     public void attachToSubparser(final Subparser subparser) {
  25         subparser.help("Set a name, about and avatar image for the user profile");
  26         subparser.addArgument("--given-name", "--name").help("New profile (given) name");
  27         subparser.addArgument("--family-name").help("New profile family name (optional)");
  28         subparser.addArgument("--about").help("New profile about text");
  29         subparser.addArgument("--about-emoji").help("New profile about emoji");
  30         subparser.addArgument("--mobile-coin-address", "--mobilecoin-address")
  31                 .help("New MobileCoin address (Base64 encoded public address)");
  32
  33         final var avatarOptions = subparser.addMutuallyExclusiveGroup();
  34         avatarOptions.addArgument("--avatar").help("Path to new profile avatar");
  35         avatarOptions.addArgument("--remove-avatar").action(Arguments.storeTrue());
  36     }
  37
  38     @Override
  39     public void handleCommand(
  40             final Namespace ns,
  41             final Manager m,
  42             final OutputWriter outputWriter
  43     ) throws CommandException {
  44         var givenName = ns.getString("given-name");
  45         var familyName = ns.getString("family-name");
  46         var about = ns.getString("about");
  47         var aboutEmoji = ns.getString("about-emoji");
  48         var mobileCoinAddressString = ns.getString("mobile-coin-address");
  49         var mobileCoinAddress = mobileCoinAddressString == null
  50                 ? null
  51                 : Base64.getDecoder().decode(mobileCoinAddressString);
  52
  53         var avatarPath = ns.getString("avatar");
  54         boolean removeAvatar = Boolean.TRUE.equals(ns.getBoolean("remove-avatar"));
  55         String avatarFile = removeAvatar || avatarPath == null ? null : avatarPath;
  56
  57         try {
  58             m.updateProfile(UpdateProfile.newBuilder()
  59                     .withGivenName(givenName)
  60                     .withFamilyName(familyName)
  61                     .withAbout(about)
  62                     .withAboutEmoji(aboutEmoji)
  63                     .withMobileCoinAddress(mobileCoinAddress)
  64                     .withAvatar(avatarFile)
  65                     .withDeleteAvatar(removeAvatar)
  66                     .build());
  67         } catch (IOException e) {
  68             throw new IOErrorException("Update profile error: " + e.getMessage(), e);
  69         }
  70     }
  71 }