Skip to content

Commit

Permalink
merge: Create ApplicationHandler (#21)
Browse files Browse the repository at this point in the history
  • Loading branch information
@gmitch215
gmitch215 authored Dec 3, 2024
2 parents f3e05ef + 8f81967 commit 94bd102
Show file tree
Hide file tree
Showing 4 changed files with 337 additions and 180 deletions.
179 changes: 3 additions & 176 deletions src/main/java/io/codemc/bot/commands/CmdApplication.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,39 +20,20 @@

import com.jagrosh.jdautilities.command.SlashCommand;
import com.jagrosh.jdautilities.command.SlashCommandEvent;
import io.codemc.api.database.DatabaseAPI;
import io.codemc.bot.CodeMCBot;
import io.codemc.bot.utils.APIUtil;
import io.codemc.bot.utils.ApplicationHandler;
import io.codemc.bot.utils.CommandUtil;
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.MessageEmbed;
import net.dv8tion.jda.api.entities.Role;
import net.dv8tion.jda.api.entities.User;
import net.dv8tion.jda.api.entities.channel.concrete.TextChannel;
import net.dv8tion.jda.api.entities.channel.concrete.ThreadChannel;
import net.dv8tion.jda.api.exceptions.ErrorHandler;
import net.dv8tion.jda.api.interactions.InteractionHook;
import net.dv8tion.jda.api.interactions.commands.OptionMapping;
import net.dv8tion.jda.api.interactions.commands.OptionType;
import net.dv8tion.jda.api.interactions.commands.build.OptionData;
import net.dv8tion.jda.api.requests.ErrorResponse;
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
import net.dv8tion.jda.api.utils.messages.MessageCreateData;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class CmdApplication extends BotCommand{

public static final Pattern GITHUB_URL_PATTERN = Pattern.compile("^https://github\\.com/([a-zA-Z0-9-]+)/([a-zA-Z0-9-_.]+?)(?:\\.git)?(?:/.*)?$");

private static final Logger LOGGER = LoggerFactory.getLogger(CmdApplication.class);

public CmdApplication(CodeMCBot bot){
super(bot);

Expand All @@ -73,160 +54,6 @@ public void withModalReply(SlashCommandEvent event){}
@Override
public void withHookReply(InteractionHook hook, SlashCommandEvent event, Guild guild, Member member){}

public static void handle(CodeMCBot bot, InteractionHook hook, Guild guild, long messageId, String str, boolean accepted){
TextChannel requestChannel = guild.getTextChannelById(bot.getConfigHandler().getLong("channels", "request_access"));
if(requestChannel == null){
CommandUtil.EmbedReply.from(hook).error("Unable to retrieve `request-access` channel.").send();
return;
}

requestChannel.retrieveMessageById(messageId).queue(message -> {
List<MessageEmbed> embeds = message.getEmbeds();
if(embeds.isEmpty()){
CommandUtil.EmbedReply.from(hook).error("Provided message does not have any embeds.").send();
return;
}

MessageEmbed embed = embeds.get(0);
if(embed.getFooter() == null || embed.getFields().isEmpty()){
CommandUtil.EmbedReply.from(hook).error("Embed does not have a footer or any Embed Fields.").send();
return;
}

String userId = embed.getFooter().getText().trim();
if(userId == null || userId.isEmpty()){
CommandUtil.EmbedReply.from(hook).error("Embed does not have a valid footer.").send();
return;
}

String userLink = null;
String repoLink = null;
for(MessageEmbed.Field field : embed.getFields()){
if(field.getName() == null || field.getValue() == null)
continue;

if(field.getName().equalsIgnoreCase("user/organisation:")){
userLink = field.getValue();
}else
if(field.getName().equalsIgnoreCase("repository:")){
String link = field.getValue();
String url = link.substring(link.indexOf("(") + 1, link.indexOf(")"));

repoLink = url.isEmpty() ? link : url;
}
}

if(userLink == null || repoLink == null){
CommandUtil.EmbedReply.from(hook).error("Embed does not have any valid Fields.").send();
return;
}

TextChannel channel = guild.getTextChannelById(accepted
? bot.getConfigHandler().getLong("channels", "accepted_requests")
: bot.getConfigHandler().getLong("channels", "rejected_requests")
);
if(channel == null){
CommandUtil.EmbedReply.from(hook)
.error("Unable to retrieve `" + (accepted ? "accepted" : "rejected") + "-requests` channel.")
.send();
return;
}

Matcher matcher = GITHUB_URL_PATTERN.matcher(repoLink);
if (!matcher.matches()) {
CommandUtil.EmbedReply.from(hook)
.error("The user/organisation or repository name is invalid!")
.send();
return;
}

String username = matcher.group(1);
String project = matcher.group(2);
String jenkinsUrl = bot.getConfigHandler().getString("jenkins", "url") + "/job/" + username + "/job/" + project + "/";
Member member = guild.getMemberById(userId);

if (accepted) {
String password = APIUtil.newPassword();
boolean jenkinsSuccess = APIUtil.createJenkinsJob(hook, username, password, project, repoLink);
boolean nexusSuccess = APIUtil.createNexus(hook, username, password);
if (!nexusSuccess || !jenkinsSuccess) return;

if (member == null)
LOGGER.warn("Member with ID '{}' not found!", userId);
else {
if (DatabaseAPI.getUser(username) == null)
DatabaseAPI.addUser(username, member.getIdLong());
}
}

channel.sendMessage(getMessage(bot, userId, userLink, repoLink, str == null ? jenkinsUrl : str, hook.getInteraction().getUser(), accepted)).queue(m -> {
ThreadChannel thread = message.getStartedThread();
if(thread != null && !thread.isArchived()){
thread.getManager().setArchived(true)
.reason("Archiving Thread of deleted Request message.")
.queue();
}

message.delete().queue();

if(!accepted){
CommandUtil.EmbedReply.from(hook)
.success("Denied Application of " + (member == null ? "Unknown" : member.getUser().getEffectiveName()) + "!")
.send();
return;
}

Role authorRole = guild.getRoleById(bot.getConfigHandler().getLong("author_role"));
if(authorRole == null){
CommandUtil.EmbedReply.from(hook)
.error("Unable to retrieve Author Role!")
.send();
return;
}

if(member == null){
CommandUtil.EmbedReply.from(hook)
.error("Unable to apply Role. Member not found!")
.send();
return;
}

guild.addRoleToMember(member, authorRole)
.reason("[Access Request] Application accepted.")
.queue(
v -> CommandUtil.EmbedReply.from(hook)
.success("Accepted application of " + member.getUser().getEffectiveName() + "!")
.send(),
new ErrorHandler()
.handle(
ErrorResponse.MISSING_PERMISSIONS,
e -> CommandUtil.EmbedReply.from(hook)
.appendWarning("I lack the `Manage Roles` permission to apply the role.")
.send()
)
);
});
});
}

private static MessageCreateData getMessage(CodeMCBot bot, String userId, String userLink, String repoLink, String str, User reviewer, boolean accepted){
String msg = String.join("\n", bot.getConfigHandler().getStringList("messages", (accepted ? "accepted" : "denied")));

MessageEmbed embed = new EmbedBuilder()
.setColor(accepted ? 0x00FF00 : 0xFF0000)
.setDescription(msg)
.addField("User/Organisation:", userLink, true)
.addField("Repository:", repoLink, true)
.addField("Reviewer:", reviewer.getAsMention(), true)
.addField(accepted ? "New Project:" : "Reason:", str, false)
.build();

return new MessageCreateBuilder()
.addContent("<@" + userId + ">")
.setEmbeds(embed)
.build();
}

private static class Accept extends BotCommand{

public Accept(CodeMCBot bot){
Expand Down Expand Up @@ -265,7 +92,7 @@ public void withHookReply(InteractionHook hook, SlashCommandEvent event, Guild g
return;
}

handle(bot, hook, guild, messageId, null, true);
ApplicationHandler.handle(bot, hook, guild, messageId, null, true);
} catch (NumberFormatException e) {
CommandUtil.EmbedReply.from(hook).error("Invalid message ID!").send();
}
Expand Down Expand Up @@ -313,7 +140,7 @@ public void withHookReply(InteractionHook hook, SlashCommandEvent event, Guild g
return;
}

handle(bot, hook, guild, messageId, reason, false);
ApplicationHandler.handle(bot, hook, guild, messageId, reason, false);
} catch (NumberFormatException e) {
CommandUtil.EmbedReply.from(hook).error("Invalid message ID!").send();
}
Expand Down
4 changes: 2 additions & 2 deletions src/main/java/io/codemc/bot/listeners/ButtonListener.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
package io.codemc.bot.listeners;

import io.codemc.bot.CodeMCBot;
import io.codemc.bot.commands.CmdApplication;
import io.codemc.bot.utils.ApplicationHandler;
import io.codemc.bot.utils.CommandUtil;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Member;
Expand Down Expand Up @@ -95,7 +95,7 @@ public void onButtonInteraction(@NotNull ButtonInteractionEvent event){
}

event.deferReply(true).queue(
hook -> CmdApplication.handle(bot, hook, guild, event.getMessageIdLong(), null, true)
hook -> ApplicationHandler.handle(bot, hook, guild, event.getMessageIdLong(), null, true)
);
}else{
if(lacksRole(roleIds, denyApplicationRoles)){
Expand Down
4 changes: 2 additions & 2 deletions src/main/java/io/codemc/bot/listeners/ModalListener.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@

import io.codemc.api.jenkins.JenkinsAPI;
import io.codemc.bot.CodeMCBot;
import io.codemc.bot.commands.CmdApplication;
import io.codemc.bot.utils.ApplicationHandler;
import io.codemc.bot.utils.CommandUtil;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Message;
Expand Down Expand Up @@ -253,7 +253,7 @@ public void onModalInteraction(@NotNull ModalInteractionEvent event){
if(reason == null || reason.isEmpty())
reason = "*No reason provided*";

CmdApplication.handle(this.bot, hook, guild, messageId, reason, false);
ApplicationHandler.handle(this.bot, hook, guild, messageId, reason, false);
});

default -> CommandUtil.EmbedReply.from(event)
Expand Down
Loading

0 comments on commit 94bd102

Please sign in to comment.