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.MutuallyExclusiveGroup;
   5 import net.sourceforge.argparse4j.inf.Namespace;
   6 import net.sourceforge.argparse4j.inf.Subparser;
   7
   8 import org.asamk.signal.manager.Manager;
   9
  10 import java.io.File;
  11 import java.io.IOException;
  12
  13 public class UpdateProfileCommand implements LocalCommand {
  14
  15     @Override
  16     public void attachToSubparser(final Subparser subparser) {
  17         final MutuallyExclusiveGroup avatarOptions = subparser.addMutuallyExclusiveGroup().required(true);
  18         avatarOptions.addArgument("--avatar").help("Path to new profile avatar");
  19         avatarOptions.addArgument("--remove-avatar").action(Arguments.storeTrue());
  20
  21         subparser.addArgument("--name").required(true).help("New profile name");
  22
  23         subparser.help("Set a name and avatar image for the user profile");
  24     }
  25
  26     @Override
  27     public int handleCommand(final Namespace ns, final Manager m) {
  28         if (!m.isRegistered()) {
  29             System.err.println("User is not registered.");
  30             return 1;
  31         }
  32
  33         String name = ns.getString("name");
  34         String avatarPath = ns.getString("avatar");
  35         boolean removeAvatar = ns.getBoolean("remove_avatar");
  36
  37         try {
  38             File avatarFile = removeAvatar ? null : new File(avatarPath);
  39             m.setProfile(name, avatarFile);
  40         } catch (IOException e) {
  41             System.err.println("UpdateAccount error: " + e.getMessage());
  42             return 3;
  43         }
  44
  45         return 0;
  46     }
  47 }