Initial commit

2016-05-22 01:57:10 +01:00
commit dc1e06ebce
70 changed files with 4044 additions and 0 deletions
@@ -0,0 +1,78 @@
+package me.lucko.luckperms;
+
+import me.lucko.luckperms.utils.LPConfiguration;
+import net.md_5.bungee.config.Configuration;
+import net.md_5.bungee.config.ConfigurationProvider;
+import net.md_5.bungee.config.YamlConfiguration;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.file.Files;
+
+public class BungeeConfig implements LPConfiguration {
+    private final LPBungeePlugin plugin;
+    private Configuration configuration;
+
+    public BungeeConfig(LPBungeePlugin plugin) {
+        this.plugin = plugin;
+        reload();
+    }
+
+    private void reload() {
+        try {
+            configuration = ConfigurationProvider.getProvider(YamlConfiguration.class).load(makeFile("config.yml"));
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private File makeFile(String file) throws IOException {
+        File cfg = new File(plugin.getDataFolder(), file);
+
+        if (!cfg.exists()) {
+            plugin.getDataFolder().mkdir();
+            try (InputStream is = plugin.getResourceAsStream(file)) {
+                Files.copy(is, cfg.toPath());
+            }
+        }
+
+        return cfg;
+    }
+
+
+    @Override
+    public String getServer() {
+        return configuration.getString("server");
+    }
+
+    @Override
+    public String getPrefix() {
+        return configuration.getString("prefix");
+    }
+
+    @Override
+    public int getSyncTime() {
+        return configuration.getInt("sql.sync-minutes");
+    }
+
+    @Override
+    public String getDefaultGroupNode() {
+        return "luckperms.group." + configuration.getString("default-group");
+    }
+
+    @Override
+    public String getDefaultGroupName() {
+        return configuration.getString("default-group");
+    }
+
+    @Override
+    public boolean getIncludeGlobalPerms() {
+        return configuration.getBoolean("include-global");
+    }
+
+    @Override
+    public String getDatabaseValue(String value) {
+        return configuration.getString("sql." + value);
+    }
+}
@@ -0,0 +1,112 @@
+package me.lucko.luckperms;
+
+import lombok.Getter;
+import me.lucko.luckperms.commands.CommandManager;
+import me.lucko.luckperms.data.Datastore;
+import me.lucko.luckperms.data.DatastoreConfiguration;
+import me.lucko.luckperms.data.HikariDatastore;
+import me.lucko.luckperms.groups.GroupManager;
+import me.lucko.luckperms.listeners.PlayerListener;
+import me.lucko.luckperms.runnables.UpdateTask;
+import me.lucko.luckperms.users.BungeeUserManager;
+import me.lucko.luckperms.users.UserManager;
+import me.lucko.luckperms.utils.LPConfiguration;
+import net.md_5.bungee.api.plugin.Plugin;
+
+import java.util.UUID;
+import java.util.concurrent.TimeUnit;
+
+public class LPBungeePlugin extends Plugin implements LuckPermsPlugin {
+    public static final String VERSION = "v1.0";
+
+    @Getter
+    LPConfiguration configuration;
+
+    @Getter
+    UserManager userManager;
+
+    @Getter
+    GroupManager groupManager;
+
+    @Getter
+    Datastore datastore;
+
+    @Override
+    public void onEnable() {
+        configuration = new BungeeConfig(this);
+
+        // register events
+        getProxy().getPluginManager().registerListener(this, new PlayerListener(this));
+
+        // register commands
+        getProxy().getPluginManager().registerCommand(this, new MainCommand(new CommandManager(this)));
+
+        datastore = new HikariDatastore(this);
+        datastore.init(new DatastoreConfiguration(
+                configuration.getDatabaseValue("address"),
+                configuration.getDatabaseValue("database"),
+                configuration.getDatabaseValue("username"),
+                configuration.getDatabaseValue("password")
+        ));
+
+        userManager = new BungeeUserManager(this);
+        groupManager = new GroupManager(this);
+
+        int mins = getConfiguration().getSyncTime();
+        if (mins > 0) {
+            getProxy().getScheduler().schedule(this, new UpdateTask(this), mins, mins, TimeUnit.MINUTES);
+        }
+    }
+
+
+    @Override
+    public UserManager getUserManager() {
+        return userManager;
+    }
+
+    @Override
+    public GroupManager getGroupManager() {
+        return groupManager;
+    }
+
+    @Override
+    public String getVersion() {
+        return VERSION;
+    }
+
+    @Override
+    public String getPlayerStatus(UUID uuid) {
+        if (getProxy().getPlayer(uuid) != null) return "&aOnline";
+        return "&cOffline";
+    }
+
+    @Override
+    public int getPlayerCount() {
+        return getProxy().getOnlineCount();
+    }
+
+    @Override
+    public LPConfiguration getConfiguration() {
+        return configuration;
+    }
+
+    @Override
+    public Datastore getDatastore() {
+        return datastore;
+    }
+
+    @Override
+    public void runUpdateTask() {
+        new UpdateTask(this).run();
+    }
+
+    @Override
+    public void doAsync(Runnable r) {
+        getProxy().getScheduler().runAsync(this, r);
+    }
+
+    @Override
+    public void doSync(Runnable r) {
+        r.run();
+    }
+}
@@ -0,0 +1,34 @@
+package me.lucko.luckperms;
+
+import me.lucko.luckperms.commands.CommandManager;
+import me.lucko.luckperms.commands.Sender;
+import net.md_5.bungee.api.CommandSender;
+import net.md_5.bungee.api.chat.TextComponent;
+import net.md_5.bungee.api.plugin.Command;
+
+import java.util.Arrays;
+
+public class MainCommand extends Command {
+    private final CommandManager manager;
+
+    public MainCommand(CommandManager manager) {
+        super("luckpermsbungee", "luckperms.use", "bperms", "lpb", "bpermissions", "bp", "bperm");
+        this.manager = manager;
+
+    }
+
+    @Override
+    public void execute(CommandSender sender, String[] args) {
+        manager.onCommand(new Sender() {
+            @Override
+            public void sendMessage(String s) {
+                sender.sendMessage(new TextComponent(s));
+            }
+
+            @Override
+            public boolean hasPermission(String node) {
+                return sender.hasPermission(node);
+            }
+        }, Arrays.asList(args));
+    }
+}
@@ -0,0 +1,44 @@
+package me.lucko.luckperms.listeners;
+
+import lombok.AllArgsConstructor;
+import me.lucko.luckperms.LPBungeePlugin;
+import me.lucko.luckperms.commands.Util;
+import me.lucko.luckperms.data.Datastore;
+import me.lucko.luckperms.users.User;
+import net.md_5.bungee.api.chat.TextComponent;
+import net.md_5.bungee.api.connection.ProxiedPlayer;
+import net.md_5.bungee.api.event.PlayerDisconnectEvent;
+import net.md_5.bungee.api.event.PostLoginEvent;
+import net.md_5.bungee.api.plugin.Listener;
+import net.md_5.bungee.event.EventHandler;
+
+@AllArgsConstructor
+public class PlayerListener implements Listener {
+
+    private final LPBungeePlugin plugin;
+
+    @EventHandler
+    public void onPlayerPostLogin(PostLoginEvent e) {
+        final ProxiedPlayer player = e.getPlayer();
+
+        plugin.getDatastore().loadOrCreateUser(player.getUniqueId(), player.getName(), success -> {
+            if (!success) {
+                e.getPlayer().sendMessage(new TextComponent(Util.color("&e&l[LP] &cPermissions data could not be loaded. Please contact an administrator.")));
+            } else {
+                User user = plugin.getUserManager().getUser(player.getUniqueId());
+                user.refreshPermissions();
+            }
+        });
+
+        plugin.getDatastore().saveUUIDData(e.getPlayer().getName(), e.getPlayer().getUniqueId(), success -> {});
+    }
+
+    @EventHandler
+    public void onPlayerQuit(PlayerDisconnectEvent e) {
+        ProxiedPlayer player = e.getPlayer();
+
+        // Unload the user from memory when they disconnect
+        User user = plugin.getUserManager().getUser(player.getUniqueId());
+        plugin.getUserManager().unloadUser(user);
+    }
+}
@@ -0,0 +1,23 @@
+package me.lucko.luckperms.runnables;
+
+import lombok.AllArgsConstructor;
+import me.lucko.luckperms.LPBungeePlugin;
+import net.md_5.bungee.api.connection.ProxiedPlayer;
+
+@AllArgsConstructor
+public class UpdateTask implements Runnable {
+    private final LPBungeePlugin plugin;
+
+    @Override
+    public void run() {
+        plugin.getLogger().info("Running update task.");
+
+        // Re-load all of the groups
+        plugin.getGroupManager().loadAllGroups();
+
+        // Refresh all online users.
+        for (ProxiedPlayer p : plugin.getProxy().getPlayers()) {
+            plugin.getDatastore().loadUser(p.getUniqueId(), success -> {});
+        }
+    }
+}
@@ -0,0 +1,42 @@
+package me.lucko.luckperms.users;
+
+import me.lucko.luckperms.LPBungeePlugin;
+import net.md_5.bungee.api.connection.ProxiedPlayer;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Map;
+import java.util.UUID;
+
+public class BungeeUser extends User {
+
+    private final LPBungeePlugin plugin;
+
+    public BungeeUser(UUID uuid, LPBungeePlugin plugin) {
+        super(uuid, plugin);
+        this.plugin = plugin;
+    }
+
+    public BungeeUser(UUID uuid, String username, LPBungeePlugin plugin) {
+        super(uuid, username, plugin);
+        this.plugin = plugin;
+    }
+
+    @Override
+    public void refreshPermissions() {
+        ProxiedPlayer player = plugin.getProxy().getPlayer(getUuid());
+        if (player == null) return;
+
+        // Clear existing permissions
+        Collection<String> perms = new ArrayList<>(player.getPermissions());
+        for (String p : perms) {
+            player.setPermission(p, false);
+        }
+
+        // Re-add all defined permissions for the user
+        Map<String, Boolean> local = getLocalPermissions(getPlugin().getConfiguration().getServer(), null);
+        for (String node : local.keySet()) {
+            player.setPermission(node, local.get(node));
+        }
+    }
+}
@@ -0,0 +1,44 @@
+package me.lucko.luckperms.users;
+
+import me.lucko.luckperms.LPBungeePlugin;
+import net.md_5.bungee.api.connection.ProxiedPlayer;
+
+import java.util.UUID;
+
+public class BungeeUserManager extends UserManager {
+    private final LPBungeePlugin plugin;
+
+    public BungeeUserManager(LPBungeePlugin plugin) {
+        this.plugin = plugin;
+    }
+
+    @Override
+    public void unloadUser(User user) {
+        if (user != null) {
+
+            ProxiedPlayer player = plugin.getProxy().getPlayer(user.getUuid());
+            if (player != null) {
+                player.getPermissions().clear();
+            }
+
+            getUsers().remove(user.getUuid());
+        }
+    }
+
+    @Override
+    public void cleanupUser(User user) {
+        if (plugin.getProxy().getPlayer(user.getUuid()) == null) {
+            unloadUser(user);
+        }
+    }
+
+    @Override
+    public User makeUser(UUID uuid) {
+        return new BungeeUser(uuid, plugin);
+    }
+
+    @Override
+    public User makeUser(UUID uuid, String username) {
+        return new BungeeUser(uuid, username, plugin);
+    }
+}