]> nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/UpdateGroupCommand.java
Use Java 17
[signal-cli] / src / main / java / org / asamk / signal / commands / UpdateGroupCommand.java
1 package org.asamk.signal.commands;
2
3 import net.sourceforge.argparse4j.impl.Arguments;
4 import net.sourceforge.argparse4j.inf.Namespace;
5 import net.sourceforge.argparse4j.inf.Subparser;
6
7 import org.asamk.signal.JsonWriter;
8 import org.asamk.signal.OutputWriter;
9 import org.asamk.signal.PlainTextWriter;
10 import org.asamk.signal.commands.exceptions.CommandException;
11 import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
12 import org.asamk.signal.commands.exceptions.UserErrorException;
13 import org.asamk.signal.manager.AttachmentInvalidException;
14 import org.asamk.signal.manager.Manager;
15 import org.asamk.signal.manager.api.UpdateGroup;
16 import org.asamk.signal.manager.groups.GroupId;
17 import org.asamk.signal.manager.groups.GroupLinkState;
18 import org.asamk.signal.manager.groups.GroupNotFoundException;
19 import org.asamk.signal.manager.groups.GroupPermission;
20 import org.asamk.signal.manager.groups.GroupSendingNotAllowedException;
21 import org.asamk.signal.manager.groups.NotAGroupMemberException;
22 import org.asamk.signal.util.CommandUtil;
23 import org.asamk.signal.util.ErrorUtils;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
26
27 import java.io.File;
28 import java.io.IOException;
29 import java.util.HashMap;
30
31 public class UpdateGroupCommand implements JsonRpcLocalCommand {
32
33 private final static Logger logger = LoggerFactory.getLogger(UpdateGroupCommand.class);
34
35 @Override
36 public String getName() {
37 return "updateGroup";
38 }
39
40 @Override
41 public void attachToSubparser(final Subparser subparser) {
42 subparser.help("Create or update a group.");
43 subparser.addArgument("-g", "--group-id", "--group").help("Specify the group ID.");
44 subparser.addArgument("-n", "--name").help("Specify the new group name.");
45 subparser.addArgument("-d", "--description").help("Specify the new group description.");
46 subparser.addArgument("-a", "--avatar").help("Specify a new group avatar image file");
47 subparser.addArgument("-m", "--member").nargs("*").help("Specify one or more members to add to the group");
48 subparser.addArgument("-r", "--remove-member")
49 .nargs("*")
50 .help("Specify one or more members to remove from the group");
51 subparser.addArgument("--admin").nargs("*").help("Specify one or more members to make a group admin");
52 subparser.addArgument("--remove-admin")
53 .nargs("*")
54 .help("Specify one or more members to remove group admin privileges");
55
56 subparser.addArgument("--reset-link")
57 .action(Arguments.storeTrue())
58 .help("Reset group link and create new link password");
59 subparser.addArgument("--link")
60 .help("Set group link state, with or without admin approval")
61 .choices("enabled", "enabled-with-approval", "disabled");
62
63 subparser.addArgument("--set-permission-add-member")
64 .help("Set permission to add new group members")
65 .choices("every-member", "only-admins");
66 subparser.addArgument("--set-permission-edit-details")
67 .help("Set permission to edit group details")
68 .choices("every-member", "only-admins");
69 subparser.addArgument("--set-permission-send-messages")
70 .help("Set permission to send messages")
71 .choices("every-member", "only-admins");
72
73 subparser.addArgument("-e", "--expiration").type(int.class).help("Set expiration time of messages (seconds)");
74 }
75
76 GroupLinkState getGroupLinkState(String value) throws UserErrorException {
77 if (value == null) {
78 return null;
79 }
80 return switch (value) {
81 case "enabled" -> GroupLinkState.ENABLED;
82 case "enabled-with-approval", "enabledWithApproval" -> GroupLinkState.ENABLED_WITH_APPROVAL;
83 case "disabled" -> GroupLinkState.DISABLED;
84 default -> throw new UserErrorException("Invalid group link state: " + value);
85 };
86 }
87
88 GroupPermission getGroupPermission(String value) throws UserErrorException {
89 if (value == null) {
90 return null;
91 }
92 return switch (value) {
93 case "every-member", "everyMember" -> GroupPermission.EVERY_MEMBER;
94 case "only-admins", "onlyAdmins" -> GroupPermission.ONLY_ADMINS;
95 default -> throw new UserErrorException("Invalid group permission: " + value);
96 };
97 }
98
99 @Override
100 public void handleCommand(
101 final Namespace ns, final Manager m, final OutputWriter outputWriter
102 ) throws CommandException {
103 final var groupIdString = ns.getString("group-id");
104 var groupId = CommandUtil.getGroupId(groupIdString);
105
106 final var localNumber = m.getSelfNumber();
107
108 var groupName = ns.getString("name");
109 var groupDescription = ns.getString("description");
110 var groupMembers = CommandUtil.getSingleRecipientIdentifiers(ns.getList("member"), localNumber);
111 var groupRemoveMembers = CommandUtil.getSingleRecipientIdentifiers(ns.getList("remove-member"), localNumber);
112 var groupAdmins = CommandUtil.getSingleRecipientIdentifiers(ns.getList("admin"), localNumber);
113 var groupRemoveAdmins = CommandUtil.getSingleRecipientIdentifiers(ns.getList("remove-admin"), localNumber);
114 var groupAvatar = ns.getString("avatar");
115 var groupResetLink = Boolean.TRUE.equals(ns.getBoolean("reset-link"));
116 var groupLinkState = getGroupLinkState(ns.getString("link"));
117 var groupExpiration = ns.getInt("expiration");
118 var groupAddMemberPermission = getGroupPermission(ns.getString("set-permission-add-member"));
119 var groupEditDetailsPermission = getGroupPermission(ns.getString("set-permission-edit-details"));
120 var groupSendMessagesPermission = getGroupPermission(ns.getString("set-permission-send-messages"));
121
122 try {
123 boolean isNewGroup = false;
124 Long timestamp = null;
125 if (groupId == null) {
126 isNewGroup = true;
127 var results = m.createGroup(groupName,
128 groupMembers,
129 groupAvatar == null ? null : new File(groupAvatar));
130 timestamp = results.second().getTimestamp();
131 ErrorUtils.handleSendMessageResults(results.second().getResults());
132 groupId = results.first();
133 groupName = null;
134 groupMembers = null;
135 groupAvatar = null;
136 }
137
138 var results = m.updateGroup(groupId,
139 UpdateGroup.newBuilder()
140 .withName(groupName)
141 .withDescription(groupDescription)
142 .withMembers(groupMembers)
143 .withRemoveMembers(groupRemoveMembers)
144 .withAdmins(groupAdmins)
145 .withRemoveAdmins(groupRemoveAdmins)
146 .withResetGroupLink(groupResetLink)
147 .withGroupLinkState(groupLinkState)
148 .withAddMemberPermission(groupAddMemberPermission)
149 .withEditDetailsPermission(groupEditDetailsPermission)
150 .withAvatarFile(groupAvatar == null ? null : new File(groupAvatar))
151 .withExpirationTimer(groupExpiration)
152 .withIsAnnouncementGroup(groupSendMessagesPermission == null
153 ? null
154 : groupSendMessagesPermission == GroupPermission.ONLY_ADMINS)
155 .build());
156 if (results != null) {
157 timestamp = results.getTimestamp();
158 ErrorUtils.handleSendMessageResults(results.getResults());
159 }
160 outputResult(outputWriter, timestamp, isNewGroup ? groupId : null);
161 } catch (AttachmentInvalidException e) {
162 throw new UserErrorException("Failed to add avatar attachment for group\": " + e.getMessage());
163 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
164 throw new UserErrorException(e.getMessage());
165 } catch (IOException e) {
166 throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
167 .getSimpleName() + ")", e);
168 }
169 }
170
171 private void outputResult(final OutputWriter outputWriter, final Long timestamp, final GroupId groupId) {
172 if (outputWriter instanceof PlainTextWriter writer) {
173 if (groupId != null) {
174 writer.println("Created new group: \"{}\"", groupId.toBase64());
175 }
176 if (timestamp != null) {
177 writer.println("{}", timestamp);
178 }
179 } else {
180 final var writer = (JsonWriter) outputWriter;
181 final var result = new HashMap<>();
182 if (timestamp != null) {
183 result.put("timestamp", timestamp);
184 }
185 if (groupId != null) {
186 result.put("groupId", groupId.toBase64());
187 }
188 writer.write(result);
189 }
190 }
191 }