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         avatarOptions.addArgument("--avatar")
  19                 .help("Path to new profile avatar");
  20         avatarOptions.addArgument("--remove-avatar")
  21                 .action(Arguments.storeTrue());
  22
  23         subparser.addArgument("--name")
  24                 .help("New profile name");
  25
  26         subparser.help("Set a name and/or avatar image for the user profile");
  27     }
  28
  29     @Override
  30     public int handleCommand(final Namespace ns, final Manager m) {
  31         if (!m.isRegistered()) {
  32             System.err.println("User is not registered.");
  33             return 1;
  34         }
  35
  36         String name = ns.getString("name");
  37
  38         if (name != null) {
  39             try {
  40                 m.setProfileName(name);
  41             } catch (IOException e) {
  42                 System.err.println("UpdateAccount error: " + e.getMessage());
  43                 return 3;
  44             }
  45         }
  46
  47         String avatarPath = ns.getString("avatar");
  48
  49         if (avatarPath != null) {
  50             File avatarFile = new File(avatarPath);
  51
  52             try {
  53                 m.setProfileAvatar(avatarFile);
  54             } catch (IOException e) {
  55                 System.err.println("UpdateAccount error: " + e.getMessage());
  56                 return 3;
  57             }
  58         }
  59
  60         boolean removeAvatar = ns.getBoolean("remove_avatar");
  61
  62         if (removeAvatar) {
  63             try {
  64                 m.removeProfileAvatar();
  65             } catch (IOException e) {
  66                 System.err.println("UpdateAccount error: " + e.getMessage());
  67                 return 3;
  68             }
  69         }
  70
  71         return 0;
  72     }
  73 }