1 package org
.asamk
.signal
.commands
;
3 import net
.sourceforge
.argparse4j
.impl
.Arguments
;
4 import net
.sourceforge
.argparse4j
.inf
.Namespace
;
5 import net
.sourceforge
.argparse4j
.inf
.Subparser
;
7 import org
.asamk
.signal
.commands
.exceptions
.CommandException
;
8 import org
.asamk
.signal
.commands
.exceptions
.UnexpectedErrorException
;
9 import org
.asamk
.signal
.commands
.exceptions
.UserErrorException
;
10 import org
.asamk
.signal
.manager
.AttachmentInvalidException
;
11 import org
.asamk
.signal
.manager
.Manager
;
12 import org
.asamk
.signal
.manager
.api
.SendGroupMessageResults
;
13 import org
.asamk
.signal
.manager
.api
.UpdateGroup
;
14 import org
.asamk
.signal
.manager
.groups
.GroupId
;
15 import org
.asamk
.signal
.manager
.groups
.GroupLinkState
;
16 import org
.asamk
.signal
.manager
.groups
.GroupNotFoundException
;
17 import org
.asamk
.signal
.manager
.groups
.GroupPermission
;
18 import org
.asamk
.signal
.manager
.groups
.GroupSendingNotAllowedException
;
19 import org
.asamk
.signal
.manager
.groups
.NotAGroupMemberException
;
20 import org
.asamk
.signal
.output
.JsonWriter
;
21 import org
.asamk
.signal
.output
.OutputWriter
;
22 import org
.asamk
.signal
.output
.PlainTextWriter
;
23 import org
.asamk
.signal
.util
.CommandUtil
;
24 import org
.asamk
.signal
.util
.SendMessageResultUtils
;
25 import org
.slf4j
.Logger
;
26 import org
.slf4j
.LoggerFactory
;
29 import java
.io
.IOException
;
30 import java
.util
.HashMap
;
31 import java
.util
.stream
.Collectors
;
32 import java
.util
.stream
.Stream
;
34 public class UpdateGroupCommand
implements JsonRpcLocalCommand
{
36 private final static Logger logger
= LoggerFactory
.getLogger(UpdateGroupCommand
.class);
39 public String
getName() {
44 public void attachToSubparser(final Subparser subparser
) {
45 subparser
.help("Create or update a group.");
46 subparser
.addArgument("-g", "--group-id", "--group").help("Specify the group ID.");
47 subparser
.addArgument("-n", "--name").help("Specify the new group name.");
48 subparser
.addArgument("-d", "--description").help("Specify the new group description.");
49 subparser
.addArgument("-a", "--avatar").help("Specify a new group avatar image file");
50 subparser
.addArgument("-m", "--member").nargs("*").help("Specify one or more members to add to the group");
51 subparser
.addArgument("-r", "--remove-member")
53 .help("Specify one or more members to remove from the group");
54 subparser
.addArgument("--admin").nargs("*").help("Specify one or more members to make a group admin");
55 subparser
.addArgument("--remove-admin")
57 .help("Specify one or more members to remove group admin privileges");
59 subparser
.addArgument("--reset-link")
60 .action(Arguments
.storeTrue())
61 .help("Reset group link and create new link password");
62 subparser
.addArgument("--link")
63 .help("Set group link state, with or without admin approval")
64 .choices("enabled", "enabled-with-approval", "disabled");
66 subparser
.addArgument("--set-permission-add-member")
67 .help("Set permission to add new group members")
68 .choices("every-member", "only-admins");
69 subparser
.addArgument("--set-permission-edit-details")
70 .help("Set permission to edit group details")
71 .choices("every-member", "only-admins");
72 subparser
.addArgument("--set-permission-send-messages")
73 .help("Set permission to send messages")
74 .choices("every-member", "only-admins");
76 subparser
.addArgument("-e", "--expiration").type(int.class).help("Set expiration time of messages (seconds)");
79 GroupLinkState
getGroupLinkState(String value
) throws UserErrorException
{
83 return switch (value
) {
84 case "enabled" -> GroupLinkState
.ENABLED
;
85 case "enabled-with-approval", "enabledWithApproval" -> GroupLinkState
.ENABLED_WITH_APPROVAL
;
86 case "disabled" -> GroupLinkState
.DISABLED
;
87 default -> throw new UserErrorException("Invalid group link state: " + value
);
91 GroupPermission
getGroupPermission(String value
) throws UserErrorException
{
95 return switch (value
) {
96 case "every-member", "everyMember" -> GroupPermission
.EVERY_MEMBER
;
97 case "only-admins", "onlyAdmins" -> GroupPermission
.ONLY_ADMINS
;
98 default -> throw new UserErrorException("Invalid group permission: " + value
);
103 public void handleCommand(
104 final Namespace ns
, final Manager m
, final OutputWriter outputWriter
105 ) throws CommandException
{
106 final var groupIdString
= ns
.getString("group-id");
107 var groupId
= CommandUtil
.getGroupId(groupIdString
);
109 final var localNumber
= m
.getSelfNumber();
111 var groupName
= ns
.getString("name");
112 var groupDescription
= ns
.getString("description");
113 var groupMembers
= CommandUtil
.getSingleRecipientIdentifiers(ns
.getList("member"), localNumber
);
114 var groupRemoveMembers
= CommandUtil
.getSingleRecipientIdentifiers(ns
.getList("remove-member"), localNumber
);
115 var groupAdmins
= CommandUtil
.getSingleRecipientIdentifiers(ns
.getList("admin"), localNumber
);
116 var groupRemoveAdmins
= CommandUtil
.getSingleRecipientIdentifiers(ns
.getList("remove-admin"), localNumber
);
117 var groupAvatar
= ns
.getString("avatar");
118 var groupResetLink
= Boolean
.TRUE
.equals(ns
.getBoolean("reset-link"));
119 var groupLinkState
= getGroupLinkState(ns
.getString("link"));
120 var groupExpiration
= ns
.getInt("expiration");
121 var groupAddMemberPermission
= getGroupPermission(ns
.getString("set-permission-add-member"));
122 var groupEditDetailsPermission
= getGroupPermission(ns
.getString("set-permission-edit-details"));
123 var groupSendMessagesPermission
= getGroupPermission(ns
.getString("set-permission-send-messages"));
126 boolean isNewGroup
= false;
127 SendGroupMessageResults groupMessageResults
= null;
128 if (groupId
== null) {
130 var results
= m
.createGroup(groupName
,
132 groupAvatar
== null ?
null : new File(groupAvatar
));
133 groupMessageResults
= results
.second();
134 groupId
= results
.first();
140 var results
= m
.updateGroup(groupId
,
141 UpdateGroup
.newBuilder()
143 .withDescription(groupDescription
)
144 .withMembers(groupMembers
)
145 .withRemoveMembers(groupRemoveMembers
)
146 .withAdmins(groupAdmins
)
147 .withRemoveAdmins(groupRemoveAdmins
)
148 .withResetGroupLink(groupResetLink
)
149 .withGroupLinkState(groupLinkState
)
150 .withAddMemberPermission(groupAddMemberPermission
)
151 .withEditDetailsPermission(groupEditDetailsPermission
)
152 .withAvatarFile(groupAvatar
== null ?
null : new File(groupAvatar
))
153 .withExpirationTimer(groupExpiration
)
154 .withIsAnnouncementGroup(groupSendMessagesPermission
== null
156 : groupSendMessagesPermission
== GroupPermission
.ONLY_ADMINS
)
158 if (results
!= null) {
159 if (groupMessageResults
== null) {
160 groupMessageResults
= results
;
162 groupMessageResults
= new SendGroupMessageResults(results
.timestamp(),
163 Stream
.concat(groupMessageResults
.results().stream(), results
.results().stream())
164 .collect(Collectors
.toList()));
167 outputResult(outputWriter
, groupMessageResults
, isNewGroup ? groupId
: null);
168 } catch (AttachmentInvalidException e
) {
169 throw new UserErrorException("Failed to add avatar attachment for group\": " + e
.getMessage());
170 } catch (GroupNotFoundException
| NotAGroupMemberException
| GroupSendingNotAllowedException e
) {
171 throw new UserErrorException(e
.getMessage());
172 } catch (IOException e
) {
173 throw new UnexpectedErrorException("Failed to send message: " + e
.getMessage() + " (" + e
.getClass()
174 .getSimpleName() + ")", e
);
178 private void outputResult(
179 final OutputWriter outputWriter
, final SendGroupMessageResults results
, final GroupId groupId
181 if (outputWriter
instanceof PlainTextWriter writer
) {
182 if (groupId
!= null) {
183 writer
.println("Created new group: \"{}\"", groupId
.toBase64());
185 if (results
!= null) {
186 var errors
= SendMessageResultUtils
.getErrorMessagesFromSendMessageResults(results
.results());
187 SendMessageResultUtils
.printSendMessageResultErrors(writer
, errors
);
188 writer
.println("{}", results
.timestamp());
191 final var writer
= (JsonWriter
) outputWriter
;
192 final var response
= new HashMap
<>();
193 if (results
!= null) {
194 response
.put("timestamp", results
.timestamp());
195 var jsonResults
= SendMessageResultUtils
.getJsonSendMessageResults(results
.results());
196 response
.put("results", jsonResults
);
198 if (groupId
!= null) {
199 response
.put("groupId", groupId
.toBase64());
201 writer
.write(response
);