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()
  18                 .required(true);
  19         avatarOptions.addArgument("--avatar")
  20                 .help("Path to new profile avatar");
  21         avatarOptions.addArgument("--remove-avatar")
  22                 .action(Arguments.storeTrue());
  23
  24         subparser.addArgument("--name")
  25                 .required(true)
  26                 .help("New profile name");
  27
  28         subparser.help("Set a name and avatar image for the user profile");
  29     }
  30
  31     @Override
  32     public int handleCommand(final Namespace ns, final Manager m) {
  33         if (!m.isRegistered()) {
  34             System.err.println("User is not registered.");
  35             return 1;
  36         }
  37
  38         String name = ns.getString("name");
  39         String avatarPath = ns.getString("avatar");
  40         boolean removeAvatar = ns.getBoolean("remove_avatar");
  41
  42         try {
  43             File avatarFile = removeAvatar ? null : new File(avatarPath);
  44             m.setProfile(name, avatarFile);
  45         } catch (IOException e) {
  46             System.err.println("UpdateAccount error: " + e.getMessage());
  47             return 3;
  48         }
  49
  50         return 0;
  51     }
  52 }