updates

Author: nomadjimbob

src/main/java/com/stemcraft/SMRegion.java

@@ -135,6 +135,22 @@ public static boolean exists(String name) {
         return regionList.containsKey(name.toLowerCase());
     }
 
+    /**
+     * Get a region name list
+     * @return The region names
+     */
+    public static List<String> getRegionNames() {
+        return new ArrayList<>(regionList.keySet());
+    }
+
+    /**
+     * Get a region list
+     * @return The region list
+     */
+    public static List<SMRegion> getRegions() {
+        return new ArrayList<>(regionList.values());
+    }
+
     /**
      * Get all the regions filtered by World (null for all)
      * @param world Filter by world

src/main/java/com/stemcraft/bridges/SMWorldEdit.java

@@ -1,16 +1,22 @@
 package com.stemcraft.bridges;
 
+import com.sk89q.worldedit.LocalSession;
 import com.sk89q.worldedit.bukkit.BukkitAdapter;
 import com.sk89q.worldedit.bukkit.WorldEditPlugin;
+import com.sk89q.worldedit.math.BlockVector3;
 import com.sk89q.worldedit.regions.Polygonal2DRegion;
 import com.sk89q.worldedit.regions.Region;
+import com.sk89q.worldedit.regions.RegionSelector;
+import com.sk89q.worldedit.regions.selector.CuboidRegionSelector;
+import com.sk89q.worldedit.regions.selector.Polygonal2DRegionSelector;
 import com.stemcraft.STEMCraft;
 import org.bukkit.Location;
 import org.bukkit.entity.Player;
 import org.bukkit.plugin.Plugin;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
 
 public class SMWorldEdit {
     private static final boolean initialized = false;
@@ -71,4 +77,56 @@ public static List<Location> getSelection(Player player) {
 
         return points;
     }
+
+    /**
+     * Set a players world edit selection
+     * @param player The player to set
+     * @param points The selection points
+     */
+    public static void setSelection(Player player, List<Location> points) {
+        WorldEditPlugin worldEdit = getBase();
+        if (worldEdit == null || points.size() < 2) {
+            return;
+        }
+
+        com.sk89q.worldedit.world.World world = BukkitAdapter.adapt(Objects.requireNonNull(points.get(0).getWorld()));
+        LocalSession session = worldEdit.getSession(player);
+        RegionSelector selector;
+
+        if (points.size() == 2) {
+            // Cuboid selection
+            selector = new CuboidRegionSelector(world);
+            BlockVector3 pos1 = toBlockVector3(points.get(0));
+            BlockVector3 pos2 = toBlockVector3(points.get(1));
+
+            try {
+                selector.selectPrimary(pos1, null);
+                selector.selectSecondary(pos2, null);
+            } catch (Exception e) {
+                STEMCraft.error(e);
+            }
+        } else {
+            // Polygon selection
+            selector = new Polygonal2DRegionSelector(world);
+            for (Location point : points) {
+                BlockVector3 pos = toBlockVector3(point);
+                try {
+                    selector.selectPrimary(pos, null);
+                } catch (Exception e) {
+                    STEMCraft.error(e);
+                }
+            }
+        }
+
+        session.setRegionSelector(world, selector);
+    }
+
+    /**
+     * Convert Location to BlockVector3
+     * @param location The location to convert
+     * @return The resulting BlockVector3
+     */
+    private static BlockVector3 toBlockVector3(Location location) {
+        return BlockVector3.at(location.getX(), location.getY(), location.getZ());
+    }
 }

src/main/java/com/stemcraft/commands/SMCommandRegion.java

@@ -17,17 +17,19 @@ public class SMCommandRegion extends SMCommand {
     public SMCommandRegion() {
         super("region");
         description("Region management");
+        alias("rg");
         permission("stemcraft.command.region");
         tabCompletion("create");
         tabCompletion("delete", "{region}");
+        tabCompletion("select", "{region}");
         tabCompletion("list");
         tabCompletion("set", "{region}", "teleport-enter");
         register();
     }
 
     @Override
     public String usage() {
-        return "/region [create|delete|list|load|unload|teleport] [world]";
+        return "/region [create|select|delete|list|load|unload|teleport] [region]";
     }
 
     @Override
@@ -37,7 +39,7 @@ public void execute(SMCommandContext ctx) {
             return;
         }
 
-        String action = ctx.args.shift("create|delete|list|set");
+        String action = ctx.args.shift("create|delete|list|set|select");
         String regionName = "";
         SMRegion region = null;
 
@@ -85,6 +87,16 @@ public void execute(SMCommandContext ctx) {
             assert region != null;
             region.delete();
             ctx.success("Region '{regionName}' deleted", "regionName", regionName);
+        } else if(action.equalsIgnoreCase("select")) {
+            assert region != null;
+
+            if(ctx.fromConsole()) {
+                ctx.error("This command requires to be run by a player");
+                return;
+            }
+
+            SMWorldEdit.setSelection(ctx.player, region.getPoints());
+            ctx.success("Region '{regionName}' selected", "regionName", regionName);
         } else if(action.equalsIgnoreCase("list")) {
             Collection<String> worlds = SMWorld.list();
 

src/main/java/com/stemcraft/tabcompletions/SMTabCompletionRegion.java

@@ -0,0 +1,23 @@
+package com.stemcraft.tabcompletions;
+
+import com.stemcraft.SMRegion;
+import com.stemcraft.SMTabCompletion;
+
+import java.util.List;
+
+public class SMTabCompletionRegion extends SMTabCompletion<SMRegion> {
+
+    public SMTabCompletionRegion() {
+        super("region");
+    }
+
+    @Override
+    public List<String> completions(String arg) {
+        return SMRegion.getRegionNames();
+    }
+
+    @Override
+    public SMRegion resolve(String arg) {
+        return SMRegion.get(arg);
+    }
+}