/
ChannelOrderActionImpl.java
172 lines (156 loc) · 6.43 KB
/
ChannelOrderActionImpl.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
/*
* Copyright 2015 Austin Keener, Michael Ritter, Florian Spieß, and the JDA contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package net.dv8tion.jda.internal.requests.restaction.order;
import gnu.trove.map.TLongLongMap;
import gnu.trove.map.hash.TLongLongHashMap;
import gnu.trove.set.TLongSet;
import gnu.trove.set.hash.TLongHashSet;
import net.dv8tion.jda.api.Permission;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.channel.attribute.ICategorizableChannel;
import net.dv8tion.jda.api.entities.channel.concrete.Category;
import net.dv8tion.jda.api.entities.channel.middleman.GuildChannel;
import net.dv8tion.jda.api.exceptions.InsufficientPermissionException;
import net.dv8tion.jda.api.requests.Route;
import net.dv8tion.jda.api.requests.restaction.order.ChannelOrderAction;
import net.dv8tion.jda.api.utils.data.DataArray;
import net.dv8tion.jda.api.utils.data.DataObject;
import net.dv8tion.jda.internal.utils.Checks;
import okhttp3.RequestBody;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Collection;
import java.util.stream.Collectors;
public class ChannelOrderActionImpl
extends OrderActionImpl<GuildChannel, ChannelOrderAction>
implements ChannelOrderAction
{
protected final Guild guild;
protected final int bucket;
protected final TLongSet lockPermissions = new TLongHashSet();
protected final TLongLongMap parent = new TLongLongHashMap();
/**
* Creates a new ChannelOrderAction instance
*
* @param guild
* The target {@link net.dv8tion.jda.api.entities.Guild Guild}
* of which to order the channels defined by the specified type
* @param bucket
* The sorting bucket
*/
public ChannelOrderActionImpl(Guild guild, int bucket)
{
this(guild, bucket, getChannelsOfType(guild, bucket));
}
/**
* Creates a new ChannelOrderAction instance using the provided
* {@link net.dv8tion.jda.api.entities.Guild Guild}, as well as the provided
* list of {@link GuildChannel Channels}.
*
* @param guild
* The target {@link net.dv8tion.jda.api.entities.Guild Guild}
* of which to order the channels defined by the specified type
* @param bucket
* The sorting bucket
* @param channels
* The {@link GuildChannel Channels} to order, all of which
* are on the same Guild specified, and all of which are of the same generic type of GuildChannel
* corresponding to the the ChannelType specified.
*
* @throws java.lang.IllegalArgumentException
* If the channels are {@code null}, an empty collection,
* or any of them do not have the same ChannelType as the one
* provided.
*/
public ChannelOrderActionImpl(Guild guild, int bucket, Collection<? extends GuildChannel> channels)
{
super(guild.getJDA(), Route.Guilds.MODIFY_CHANNELS.compile(guild.getId()));
Checks.notNull(channels, "Channels to order");
Checks.notEmpty(channels, "Channels to order");
Checks.check(channels.stream().allMatch(c -> guild.equals(c.getGuild())),
"One or more channels are not from the correct guild");
Checks.check(channels.stream().allMatch(c -> c.getType().getSortBucket() == bucket),
"One or more channels did not match the expected bucket " + bucket);
this.guild = guild;
this.bucket = bucket;
this.orderList.addAll(channels);
}
@Nonnull
@Override
public Guild getGuild()
{
return guild;
}
@Override
public int getSortBucket()
{
return bucket;
}
@Nonnull
@Override
public ChannelOrderAction setCategory(@Nullable Category category, boolean syncPermissions)
{
GuildChannel channel = getSelectedEntity();
if (!(channel instanceof ICategorizableChannel) && category != null)
throw new IllegalStateException("Cannot move channel of type " + channel.getType() + " to category!");
if (category != null)
Checks.check(category.getGuild().equals(getGuild()), "Category is not from the same guild!");
long id = channel.getIdLong();
parent.put(id, category == null ? 0 : category.getIdLong());
if (syncPermissions)
lockPermissions.add(id);
else
lockPermissions.remove(id);
return this;
}
@Override
protected RequestBody finalizeData()
{
final Member self = guild.getSelfMember();
if (!self.hasPermission(Permission.MANAGE_CHANNEL))
throw new InsufficientPermissionException(guild, Permission.MANAGE_CHANNEL);
DataArray array = DataArray.empty();
for (int i = 0; i < orderList.size(); i++)
{
GuildChannel chan = orderList.get(i);
DataObject json = DataObject.empty()
.put("id", chan.getId())
.put("position", i);
if (parent.containsKey(chan.getIdLong()))
{
long parentId = parent.get(chan.getIdLong());
json.put("parent_id", parentId == 0 ? null : parentId);
json.put("lock_permissions", lockPermissions.contains(chan.getIdLong()));
}
array.add(json);
}
return getRequestBody(array);
}
@Override
protected void validateInput(GuildChannel entity)
{
Checks.check(entity.getGuild().equals(guild), "Provided channel is not from this Guild!");
Checks.check(orderList.contains(entity), "Provided channel is not in the list of orderable channels!");
}
protected static Collection<GuildChannel> getChannelsOfType(Guild guild, int bucket)
{
return guild.getChannels().stream()
.filter(it -> it.getType().getSortBucket() == bucket)
.sorted()
.collect(Collectors.toList());
}
}