+package org.asamk.signal.commands;
+
+import java.io.IOException;
+import java.io.File;
+
+import net.sourceforge.argparse4j.impl.Arguments;
+import net.sourceforge.argparse4j.inf.MutuallyExclusiveGroup;
+import net.sourceforge.argparse4j.inf.Namespace;
+import net.sourceforge.argparse4j.inf.Subparser;
+import org.asamk.signal.manager.Manager;
+
+public class UpdateProfileCommand implements LocalCommand {
+
+ @Override
+ public void attachToSubparser(final Subparser subparser) {
+ final MutuallyExclusiveGroup avatarOptions = subparser.addMutuallyExclusiveGroup();
+ avatarOptions.addArgument("--avatar")
+ .help("Path to new profile avatar");
+ avatarOptions.addArgument("--remove-avatar")
+ .action(Arguments.storeTrue());
+
+ subparser.addArgument("--name")
+ .help("New profile name");
+
+ subparser.help("Set a name and/or avatar image for the user profile");
+ }
+
+ @Override
+ public int handleCommand(final Namespace ns, final Manager m) {
+ if (!m.isRegistered()) {
+ System.err.println("User is not registered.");
+ return 1;
+ }
+
+ String name = ns.getString("name");
+
+ if (name != null) {
+ try {
+ m.setProfileName(name);
+ } catch (IOException e) {
+ System.err.println("UpdateAccount error: " + e.getMessage());
+ return 3;
+ }
+ }
+
+ String avatarPath = ns.getString("avatar");
+
+ if (avatarPath != null) {
+ File avatarFile = new File(avatarPath);
+
+ try {
+ m.setProfileAvatar(avatarFile);
+ } catch (IOException e) {
+ System.err.println("UpdateAccount error: " + e.getMessage());
+ return 3;
+ }
+ }
+
+ boolean removeAvatar = ns.getBoolean("remove_avatar");
+
+ if (removeAvatar) {
+ try {
+ m.removeProfileAvatar();
+ } catch (IOException e) {
+ System.err.println("UpdateAccount error: " + e.getMessage());
+ return 3;
+ }
+ }
+
+ return 0;
+ }
+}