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