/*
* Copyright 2019 John Grosh (john.a.grosh@gmail.com).
*
* 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 com.jagrosh.vortex.commands.owner;
import com.jagrosh.jdautilities.command.Command;
import com.jagrosh.jdautilities.command.CommandEvent;
import com.jagrosh.vortex.Vortex;
import com.jagrosh.vortex.database.managers.PremiumManager;
import com.jagrosh.vortex.database.managers.PremiumManager.PremiumInfo;
import com.jagrosh.vortex.utils.OtherUtil;
import java.time.temporal.ChronoUnit;
import java.util.stream.Collectors;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.User;
/**
*
* @author John Grosh (john.a.grosh@gmail.com)
*/
public class PremiumCmd extends Command
{
private final Vortex vortex;
public PremiumCmd(Vortex vortex)
{
this.vortex = vortex;
this.name = "premium";
this.help = "root command";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
this.children = new Command[]{ new PremiumAddCmd(), new PremiumInfoCmd(), new PremiumUserCmd(), new PremiumRemoveCmd(), new PremiumAddUltraCmd(), new PremiumAddPlusCmd() };
}
@Override
protected void execute(CommandEvent event)
{
String str = "Premium Commands";
for(Command c: children)
str += "\n`" + event.getClient().getPrefix() + name + " " + c.getName() + " " + c.getArguments() + "` - " + c.getHelp();
event.reply(str);
}
private class PremiumUserCmd extends Command
{
private PremiumUserCmd()
{
this.name = "user";
this.help = "sets user for guild";
this.arguments = "<guildId> <userId>";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
}
@Override
protected void execute(CommandEvent event)
{
String[] parts = event.getArgs().split("\\s+", 2);
if(parts.length < 2)
{
event.replyError("Too few arguments");
return;
}
long guildid, userid;
try
{
guildid = Long.parseLong(parts[0]);
userid = Long.parseLong(parts[1]);
}
catch(NumberFormatException ex)
{
event.replyError("Invalid guild or user ID");
return;
}
Guild guild = vortex.getShardManager().getGuildById(guildid);
if(guild == null)
{
event.replyError("Guild " + guildid + " not found.");
return;
}
User user = vortex.getShardManager().getUserById(userid);
if(user == null)
{
event.replyError("User " + userid + " not found.");
return;
}
boolean result = vortex.getDatabase().premium.setUserForGuild(user, guild);
if(result)
event.replySuccess("Successfully set " + user.getAsMention() + " on " + guild.getName());
else
event.replyError("Could not set " + user.getAsMention() + " on " + guild.getName());
}
}
private class PremiumAddCmd extends Command
{
private PremiumAddCmd()
{
this.name = "add";
this.help = "adds premium time";
this.arguments = "<guildId> <time>";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
}
@Override
protected void execute(CommandEvent event)
{
String[] parts = event.getArgs().split("\\s+", 2);
if(parts.length < 2)
{
event.replyError("Too few arguments");
return;
}
int seconds = OtherUtil.parseTime(parts[1]);
if(seconds <= 0)
{
event.replyError("Invalid time");
return;
}
Guild guild;
try
{
guild = vortex.getShardManager().getGuildById(Long.parseLong(parts[0]));
}
catch(NumberFormatException ex)
{
event.replyError("Invalid guild ID");
return;
}
if(guild == null)
{
event.replyError("No guild found with ID `" + parts[0] + "`");
return;
}
PremiumInfo before = vortex.getDatabase().premium.getPremiumInfo(guild);
vortex.getDatabase().premium.addPremium(guild, PremiumManager.Level.PRO, seconds, ChronoUnit.SECONDS);
PremiumInfo after = vortex.getDatabase().premium.getPremiumInfo(guild);
event.replySuccess("Before: " + before + "\n" + event.getClient().getSuccess() + " After: " + after);
}
}
private class PremiumAddUltraCmd extends Command
{
private PremiumAddUltraCmd()
{
this.name = "addultra";
this.help = "adds premium ultra time";
this.arguments = "<guildId> <time>";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
}
@Override
protected void execute(CommandEvent event)
{
String[] parts = event.getArgs().split("\\s+", 2);
if(parts.length < 2)
{
event.replyError("Too few arguments");
return;
}
int seconds = OtherUtil.parseTime(parts[1]);
if(seconds <= 0)
{
event.replyError("Invalid time");
return;
}
Guild guild;
try
{
guild = vortex.getShardManager().getGuildById(Long.parseLong(parts[0]));
}
catch(NumberFormatException ex)
{
event.replyError("Invalid guild ID");
return;
}
if(guild == null)
{
event.replyError("No guild found with ID `" + parts[0] + "`");
return;
}
PremiumInfo before = vortex.getDatabase().premium.getPremiumInfo(guild);
vortex.getDatabase().premium.addPremium(guild, PremiumManager.Level.ULTRA, seconds, ChronoUnit.SECONDS);
PremiumInfo after = vortex.getDatabase().premium.getPremiumInfo(guild);
event.replySuccess("Before: " + before + "\n" + event.getClient().getSuccess() + " After: " + after);
}
}
private class PremiumAddPlusCmd extends Command
{
private PremiumAddPlusCmd()
{
this.name = "addplus";
this.help = "adds premium plus time";
this.arguments = "<guildId> <time>";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
}
@Override
protected void execute(CommandEvent event)
{
String[] parts = event.getArgs().split("\\s+", 2);
if(parts.length < 2)
{
event.replyError("Too few arguments");
return;
}
int seconds = OtherUtil.parseTime(parts[1]);
if(seconds <= 0)
{
event.replyError("Invalid time");
return;
}
Guild guild;
try
{
guild = vortex.getShardManager().getGuildById(Long.parseLong(parts[0]));
}
catch(NumberFormatException ex)
{
event.replyError("Invalid guild ID");
return;
}
if(guild == null)
{
event.replyError("No guild found with ID `" + parts[0] + "`");
return;
}
PremiumInfo before = vortex.getDatabase().premium.getPremiumInfo(guild);
vortex.getDatabase().premium.addPremium(guild, PremiumManager.Level.PLUS, seconds, ChronoUnit.SECONDS);
PremiumInfo after = vortex.getDatabase().premium.getPremiumInfo(guild);
event.replySuccess("Before: " + before + "\n" + event.getClient().getSuccess() + " After: " + after);
}
}
private class PremiumInfoCmd extends Command
{
private PremiumInfoCmd()
{
this.name = "info";
this.help = "shows premium info";
this.arguments = "";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
}
@Override
protected void execute(CommandEvent event)
{
event.reply(vortex.getDatabase().premium.getPremiumGuildsInfo().entrySet().stream()
.sorted((a,b) -> a.getValue().until.compareTo(b.getValue().until))
.map(pinfo ->
{
Guild guild = vortex.getShardManager().getGuildById(pinfo.getKey());
return pinfo.getKey() + " (" + (guild == null ? "unknown" : "**" + guild.getName() + "**") + ") " + pinfo.getValue().toString();
}).collect(Collectors.joining("\n")));
}
}
private class PremiumRemoveCmd extends Command
{
public PremiumRemoveCmd()
{
this.name = "remove";
this.help = "removes premium";
this.arguments = "<guildId>";
this.ownerCommand = true;
this.guildOnly = false;
this.hidden = true;
}
@Override
protected void execute(CommandEvent event)
{
String[] parts = event.getArgs().split("\\s+", 1);
if(parts.length < 1)
{
event.replyError("Too few arguments");
return;
}
Guild guild;
try
{
guild = vortex.getShardManager().getGuildById(Long.parseLong(parts[0]));
}
catch(NumberFormatException ex)
{
event.replyError("Invalid guild ID");
return;
}
if(guild == null)
{
event.replyError("No guild found with ID `" + parts[0] + "`");
return;
}
PremiumInfo before = vortex.getDatabase().premium.getPremiumInfo(guild);
if(before == null || before.level == PremiumManager.Level.NONE)
{
event.replyError("Guild `" + guild.getName() + "` does not have premium!");
return;
}
// remove 5 years, so that the cleanup will still happen correctly
// tbh there should be a better cleanup system
vortex.getDatabase().premium.addPremium(guild, PremiumManager.Level.NONE, -5 * 365, ChronoUnit.DAYS);
PremiumInfo after = vortex.getDatabase().premium.getPremiumInfo(guild);
event.replySuccess("Before: " + before + "\n" + event.getClient().getSuccess() + " After: " + after);
}
}
}
