package com.guardianbot.config; import org.yaml.snakeyaml.Yaml; import java.io.*; import java.nio.file.Files; import java.util.HashMap; import java.util.Map; public class ConfigManager { private final File configFile; private Map config; private final Yaml yaml; public ConfigManager(File configFile) { this.configFile = configFile; this.yaml = new Yaml(); this.config = new HashMap<>(); } public void load() throws IOException { if (!configFile.exists()) { InputStream resource = getClass().getClassLoader().getResourceAsStream("config.yml"); if (resource != null) { Files.copy(resource, configFile.toPath()); resource.close(); } else { configFile.createNewFile(); } } try (FileInputStream fis = new FileInputStream(configFile)) { config = yaml.load(fis); if (config == null) { config = new HashMap<>(); } } } public void save() throws IOException { try (FileWriter writer = new FileWriter(configFile)) { yaml.dump(config, writer); } } public String getBotToken() { return getString("bot-token", "YOUR_BOT_TOKEN_HERE"); } public boolean isWelcomeEnabled() { return getBoolean("welcome.enabled", true); } public String getWelcomeChannelId() { return getString("welcome.channel-id", "0"); } public String getWelcomeTitle() { return getString("welcome.embed.title", "Welcome!"); } public String getWelcomeDescription() { return getString("welcome.embed.description", "Welcome to the server!"); } public String getWelcomeColor() { return getString("welcome.embed.color", "#00FF00"); } public boolean isWelcomeThumbnailEnabled() { return getBoolean("welcome.embed.thumbnail", true); } public String getWelcomeFooter() { return getString("welcome.embed.footer", "Member #{count}"); } public boolean isRaidProtectionEnabled() { return getBoolean("raid-protection.enabled", true); } public int getMaxJoinsPerMinute() { return getInt("raid-protection.max-joins-per-minute", 10); } public int getLockdownDuration() { return getInt("raid-protection.lockdown-duration", 300); } public String getRaidAlertChannelId() { return getString("raid-protection.alert-channel-id", "0"); } public boolean isNukeProtectionEnabled() { return getBoolean("nuke-protection.enabled", true); } public int getMaxDeletionsPerMinute() { return getInt("nuke-protection.max-deletions-per-minute", 5); } public int getMaxBansPerMinute() { return getInt("nuke-protection.max-bans-per-minute", 3); } public String getNukeAlertChannelId() { return getString("nuke-protection.alert-channel-id", "0"); } public boolean isAutoRemovePermissions() { return getBoolean("nuke-protection.auto-remove-permissions", true); } public boolean isTicketsEnabled() { return getBoolean("tickets.enabled", true); } public String getTicketCategoryId() { return getString("tickets.category-id", "0"); } public String getSupportRoleId() { return getString("tickets.support-role-id", "0"); } public String getTranscriptChannelId() { return getString("tickets.transcript-channel-id", "0"); } public String getTicketPanelChannelId() { return getString("tickets.ticket-panel-channel-id", "0"); } public boolean isCountingEnabled() { return getBoolean("counting.enabled", true); } public String getCountingChannelId() { return getString("counting.channel-id", "0"); } public int getCurrentNumber() { return getInt("counting.current-number", 0); } public void setCurrentNumber(int number) { setInt("counting.current-number", number); } public String getLastUserId() { return getString("counting.last-user-id", "0"); } public void setLastUserId(String userId) { setString("counting.last-user-id", userId); } public boolean isResetOnMistake() { return getBoolean("counting.reset-on-mistake", true); } public boolean isAFKEnabled() { return getBoolean("afk.enabled", true); } public boolean isAutoRemoveOnMessage() { return getBoolean("afk.auto-remove-on-message", true); } public String getAFKPrefix() { return getString("afk.nickname-prefix", "[AFK] "); } private String getString(String path, String defaultValue) { Object value = getValueAtPath(path); return value != null ? String.valueOf(value) : defaultValue; } private boolean getBoolean(String path, boolean defaultValue) { Object value = getValueAtPath(path); return value instanceof Boolean ? (Boolean) value : defaultValue; } private int getInt(String path, int defaultValue) { Object value = getValueAtPath(path); if (value instanceof Number) { return ((Number) value).intValue(); } return defaultValue; } private void setString(String path, String value) { setValueAtPath(path, value); } private void setInt(String path, int value) { setValueAtPath(path, value); } private Object getValueAtPath(String path) { String[] parts = path.split("\\."); Map current = config; for (int i = 0; i < parts.length - 1; i++) { Object next = current.get(parts[i]); if (next instanceof Map) { current = (Map) next; } else { return null; } } return current.get(parts[parts.length - 1]); } private void setValueAtPath(String path, Object value) { String[] parts = path.split("\\."); Map current = config; for (int i = 0; i < parts.length - 1; i++) { Object next = current.get(parts[i]); if (!(next instanceof Map)) { next = new HashMap(); current.put(parts[i], next); } current = (Map) next; } current.put(parts[parts.length - 1], value); } }