-
Notifications
You must be signed in to change notification settings - Fork 17
/
ScheduleModule.java
249 lines (198 loc) · 8.63 KB
/
ScheduleModule.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
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
/**
* This file is part of LWC (https://github.com/Hidendra/LWC)
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.griefcraft.modules.schedule;
import com.griefcraft.jobs.IJobHandler;
import com.griefcraft.lwc.LWC;
import com.griefcraft.model.Job;
import com.griefcraft.scripting.JavaModule;
import com.griefcraft.scripting.event.LWCCommandEvent;
import com.griefcraft.util.Colors;
import com.griefcraft.util.StringUtils;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;
import java.util.Date;
import java.util.Set;
public class ScheduleModule extends JavaModule {
@Override
public void onCommand(LWCCommandEvent event) {
if (event.isCancelled()) {
return;
}
if (!event.hasFlag("schedule") && !event.hasFlag("tasks") && !event.hasFlag("task")) {
return;
}
LWC lwc = LWC.getInstance();
CommandSender sender = event.getSender();
String[] args = event.getArgs();
// We are using the command from here on out!
event.setCancelled(true);
if (args.length < 1) {
lwc.sendSimpleUsage(sender, "/lwc schedule <Action> [JobName] [...]");
return;
}
String action = args[0].toLowerCase();
String name = args.length > 1 ? args[1].toLowerCase() : "";
String[] jobArgs = new String[0];
String joinedArguments = ""; // equivilent to StringUtils.join(jobArgs);
// get the job specific arguments found if they are there
if (args.length > 2) {
jobArgs = new String[args.length - 2];
System.arraycopy(args, 2, jobArgs, 0, jobArgs.length);
joinedArguments = StringUtils.join(jobArgs).trim();
}
// Attempt to load the job they named
Job job = lwc.getJobManager().getJob(name);
// Look for actions .. !
if (action.equals("create")) {
if (args.length < 2) {
lwc.sendSimpleUsage(sender, "/lwc schedule <Action> [JobName] [...]");
return;
}
if (job != null) {
sender.sendMessage(Colors.Red + "Job " + name + " already exists.");
return;
}
if (jobArgs.length == 0) {
lwc.sendSimpleUsage(sender, "/lwc schedule create " + name + " HANDLER_NAME");
return;
}
// Check for the job handler
IJobHandler handler = lwc.getJobManager().getJobHandler(joinedArguments);
if (handler == null) {
sender.sendMessage(Colors.Red + "Job handler " + joinedArguments + " does not exist!");
return;
}
// create the job
job = new Job();
job.setName(name);
job.setType(handler.getType());
// add who created the Job
if (sender instanceof Player) {
job.getData().put("creator", ((Player) sender).getName());
} else {
job.getData().put("creator", "Console");
}
// save it to the database
job.save();
sender.sendMessage(Colors.Green + "Created the job " + name + " successfully using the handler " + handler.getName());
}
else if (action.equals("run")) {
if (args.length < 2) {
lwc.sendSimpleUsage(sender, "/lwc schedule <Action> [JobName] [...]");
return;
}
if (job == null) {
sender.sendMessage(Colors.Red + "Invalid job specified.");
return;
}
sender.sendMessage("Running job " + job.getName());
long start = System.currentTimeMillis();
// run the job!
lwc.getJobManager().execute(job);
// calculate how long it took
long timeMillis = System.currentTimeMillis() - start;
sender.sendMessage("Successfully ran the job (" + timeMillis + " ms)");
}
else if (action.equals("remove") || action.equals("delete")) {
if (args.length < 2) {
lwc.sendSimpleUsage(sender, "/lwc schedule <Action> [JobName] [...]");
return;
}
if (job == null) {
sender.sendMessage(Colors.Red + "Invalid job specified.");
return;
}
job.remove();
sender.sendMessage(Colors.Green + "Removed job successfully!");
}
else if (action.equals("list")) {
// force a reload of jobs
lwc.getJobManager().load();
// Get the jobs
Set<Job> jobs = lwc.getJobManager().getJobs();
if (jobs.size() == 0) {
sender.sendMessage(Colors.Red + "No jobs have yet been created.");
return;
}
// No pages as of yet (TODO)
String format = "%16s%16s%16s";
sender.sendMessage(Colors.Blue + String.format(format, "Name", "Handler", "Creator"));
sender.sendMessage(" ");
for(Job foundJob : jobs) {
IJobHandler handler = foundJob.getJobHandler();
String lineColour = "";
// check the time remaining, if applicable
if (foundJob.getNextRun() <= 0) {
lineColour = Colors.Yellow;
} else {
if (foundJob.getTimeRemaining() <= 0) {
lineColour = Colors.Red;
} else {
lineColour = Colors.Green;
}
}
sender.sendMessage(lineColour + String.format(format, foundJob.getName(), handler.getName(), foundJob.getData().get("creator")));
}
}
else if (action.equals("autorun")) {
if (args.length < 2) {
lwc.sendSimpleUsage(sender, "/lwc schedule <Action> [JobName] [...]");
return;
}
if (job == null) {
sender.sendMessage(Colors.Red + "Invalid job specified.");
return;
}
if (jobArgs.length == 0) {
lwc.sendSimpleUsage(sender, "/lwc schedule autoRun " + name + " Time (e.g 1 week)");
return;
}
// Attempt to parse the time ...
long parsedTimeSeconds = StringUtils.parseTime(joinedArguments);
long parsedTimeMillis = parsedTimeSeconds * 1000L;
if (parsedTimeSeconds == 0) {
sender.sendMessage(Colors.Red + "Invalid time: " + joinedArguments);
return;
}
// calculate the time to next run at
job.setNextRun(System.currentTimeMillis() + parsedTimeMillis);
// store the time interval for the next time it runs so it can recalculate
job.getData().put("autoRun", parsedTimeMillis);
// save it to the database and notify the player
job.save();
sender.sendMessage(Colors.Green + "The job will " + job.getName() + " run again in " + StringUtils.timeToString(parsedTimeSeconds));
sender.sendMessage(Colors.Green + "In server time, that is at " + new Date(job.getNextRun()).toString());
}
else if (action.equals("check")) {
if (job == null) {
sender.sendMessage(Colors.Red + "Invalid job specified.");
return;
}
long nextRun = job.getNextRun();
if (nextRun <= 0) {
sender.sendMessage(Colors.Red + "The job " + job.getName() + " is not set to automatically run.");
} else {
long timeRemaining = job.getTimeRemaining();
if (job.shouldRun()) {
sender.sendMessage(Colors.Green + "The job " + job.getName() + " is waiting to be executed.");
return;
}
sender.sendMessage(Colors.Green + "The job " + job.getName() + " will be executed in " + StringUtils.timeToString(timeRemaining / 1000L));
}
}
}
}