Spigot / Bukkit / 0b440145d31

Commits

Thinkofname authored 0b440145d31 27 Dec 2014

Implement version checking into the /version command

src/main/java/org/bukkit/command/defaults/VersionCommand.java

Modified

-1
+package org.bukkit.command.defaults;
-2
-+
+import com.google.common.base.Charsets;
-4
+import java.util.ArrayList;
-5
+import java.util.Arrays;
-6
+import java.util.List;
-7
-8
+import org.apache.commons.lang.Validate;
-9
+import org.bukkit.Bukkit;
-10
+import org.bukkit.ChatColor;
-11
+import org.bukkit.command.CommandSender;
-12
+import org.bukkit.plugin.Plugin;
-13
+import org.bukkit.plugin.PluginDescriptionFile;
-14
+import org.bukkit.util.StringUtil;
-15
-16
+import com.google.common.collect.ImmutableList;
-+
+import com.google.common.io.Resources;
-+
+import java.io.BufferedReader;
-+
+import java.io.IOException;
-+
+import java.net.URL;
-+
+import java.net.URLEncoder;
-+
+import java.util.HashSet;
-+
+import java.util.Set;
-+
+import java.util.concurrent.locks.ReentrantLock;
-+
+import java.util.logging.Level;
-+
+import java.util.logging.Logger;
-+
+import org.json.simple.JSONObject;
-+
+import org.json.simple.parser.JSONParser;
-+
+import org.json.simple.parser.ParseException;
-30
-31
+public class VersionCommand extends BukkitCommand {
-32
+    public VersionCommand(String name) {
-33
+        super(name);
-34
-35
+        this.description = "Gets the version of this server including any plugins in use";
-36
+        this.usageMessage = "/version [plugin name]";
-37
+        this.setPermission("bukkit.command.version");
-38
+        this.setAliases(Arrays.asList("ver", "about"));
-39
+    }
-40
-41
+    @Override
-42
+    public boolean execute(CommandSender sender, String currentAlias, String[] args) {
-43
+        if (!testPermission(sender)) return true;
-44
-45
+        if (args.length == 0) {
-46
+            sender.sendMessage("This server is running " + Bukkit.getName() + " version " + Bukkit.getVersion() + " (Implementing API version " + Bukkit.getBukkitVersion() + ")");
-+
+            sendVersion(sender);
-48
+        } else {
-49
+            StringBuilder name = new StringBuilder();
-50
-51
+            for (String arg : args) {
-52
+                if (name.length() > 0) {
-53
+                    name.append(' ');
-54
+                }
-55
-56
+                name.append(arg);
-57
+            }
-134
+            String toComplete = args[0].toLowerCase();
-135
+            for (Plugin plugin : Bukkit.getPluginManager().getPlugins()) {
-136
+                if (StringUtil.startsWithIgnoreCase(plugin.getName(), toComplete)) {
-137
+                    completions.add(plugin.getName());
-138
+                }
-139
+            }
-140
+            return completions;
-141
+        }
-142
+        return ImmutableList.of();
-143
+    }
-+
-+
+    private final ReentrantLock versionLock = new ReentrantLock();
-+
+    private boolean hasVersion = false;
-+
+    private String versionMessage = null;
-+
+    private final Set<CommandSender> versionWaiters = new HashSet<CommandSender>();
-+
+    private boolean versionTaskStarted = false;
-+
+    private long lastCheck = 0;
-+
-+
+    private void sendVersion(CommandSender sender) {
-+
+        if (hasVersion) {
-+
+            if (System.currentTimeMillis() - lastCheck > 21600000) {
-+
+                lastCheck = System.currentTimeMillis();
-+
+                hasVersion = false;
-+
+            } else {
-+
+                sender.sendMessage(versionMessage);
-+
+                return;
-+
+            }
-+
+        }
-+
+        versionLock.lock();
-+
+        try {
-+
+            if (hasVersion) {
-+
+                sender.sendMessage(versionMessage);
-+
+                return;
-+
+            }
-+
+            versionWaiters.add(sender);
-+
+            sender.sendMessage("Checking version, please wait...");
-+
+            if (!versionTaskStarted) {
-+
+                versionTaskStarted = true;
-+
+                new Thread(new Runnable() {
-+
-+
+                    @Override
-+
+                    public void run() {
-+
+                        obtainVersion();
-+
+                    }
-+
+                }).start();
-+
+            }
-+
+        } finally {
-+
+            versionLock.unlock();
-+
+        }
-+
+    }
-+
-+
+    private void obtainVersion() {
-+
+        String version = Bukkit.getVersion();
-+
+        if (version == null) version = "Custom";
-+
+        if (version.startsWith("git-Spigot-")) {
-+
+            String[] parts = version.substring("git-Spigot-".length()).split("-");
-+
+            int cbVersions = getDistance("craftbukkit", parts[0]);
-+
+            int spigotVersions = getDistance("spigot", parts[1]);
-+
+            if (cbVersions == -1 || spigotVersions == -1) {
-+
+                setVersionMessage("Error obtaining version information");
-+
+            } else {
-+
+                if (cbVersions != 0 && spigotVersions != 0) {
-+
+                    setVersionMessage("You are running the latest version");
-+
+                } else {
-+
+                    setVersionMessage("You are " + (cbVersions + spigotVersions) + " behind");
-+
+                }
-+
+            }
-+
-+
+        } else if (version.startsWith("git-Bukkit-")) {
-+
+            int cbVersions = getDistance("craftbukkit", version.substring("git-Bukkit-".length()));
-+
+            if (cbVersions == -1) {
-+
+                setVersionMessage("Error obtaining version information");
-+
+            } else {
-+
+                if (cbVersions != 0) {
-+
+                    setVersionMessage("You are running the latest version");
-+
+                } else {
-+
+                    setVersionMessage("You are " + cbVersions + " behind");
-+
+                }
-+
+            }
-+
+        } else {
-+
+            setVersionMessage("Unknown version, custom build?");
-+
+        }
-+
+    }
-+
-+
+    private void setVersionMessage(String msg) {
-+
+        lastCheck = System.currentTimeMillis();
-+
+        versionMessage = msg;
-+
+        versionLock.lock();
-+
+        try {
-+
+            hasVersion = true;
-+
+            versionTaskStarted = false;
-+
+            for (CommandSender sender : versionWaiters) {
-+
+                sender.sendMessage(versionMessage);
-+
+            }
-+
+            versionWaiters.clear();
-+
+        } finally {
-+
+            versionLock.unlock();
-+
+        }
-+
+    }
-+
-+
+    private static int getDistance(String repo, String hash) {
-+
+        try {
-+
+            BufferedReader reader = Resources.asCharSource(
-+
+                    new URL("https://hub.spigotmc.org/stash/rest/api/1.0/projects/SPIGOT/repos/" + repo + "/commits?since=" + URLEncoder.encode(hash, "UTF-8") + "&withCounts=true"),
-+
+                    Charsets.UTF_8
-+
+            ).openBufferedStream();
-+
+            try {
-+
+                JSONObject obj = (JSONObject) new JSONParser().parse(reader);
-+
+                return ((Number) obj.get("totalCount")).intValue();
-+
+            } catch (ParseException ex) {
-+
+                ex.printStackTrace();
-+
+                return -1;
-+
+            } finally {
-+
+                reader.close();
-+
+            }
-+
+        } catch (IOException e) {
-+
+            e.printStackTrace();
-+
+            return -1;
-+
+        }
-+
+    }
-254
+}

Bukkit

Commits

Jira Issues

Add shortcut