SearchContainerCommand.java

/*
 * Copyright (C) 2011-2022 lishid. All rights reserved.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package com.lishid.openinv.command;

import com.lishid.openinv.util.TabCompleter;
import com.lishid.openinv.util.lang.LanguageManager;
import com.lishid.openinv.util.lang.Replacement;
import com.lishid.openinv.util.SearchHelper;
import org.bukkit.Chunk;
import org.bukkit.Material;
import org.bukkit.World;
import org.bukkit.block.BlockState;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.command.TabExecutor;
import org.bukkit.entity.Player;
import org.bukkit.Nameable;
import org.bukkit.inventory.InventoryHolder;
import org.bukkit.plugin.Plugin;
import org.jetbrains.annotations.NotNull;

import java.util.Collections;
import java.util.List;
import java.util.Locale;

/**
 * Command for searching containers in a radius of chunks.
 */
public class SearchContainerCommand implements TabExecutor {

  private final @NotNull Plugin plugin;
  private final @NotNull LanguageManager lang;

  public SearchContainerCommand(@NotNull Plugin plugin, @NotNull LanguageManager lang) {
    this.plugin = plugin;
    this.lang = lang;
  }

  @Override
  public boolean onCommand(
      @NotNull CommandSender sender,
      @NotNull Command command,
      @NotNull String label,
      @NotNull String[] args) {
    if (!(sender instanceof Player senderPlayer)) {
      lang.sendMessage(sender, "messages.error.consoleUnsupported");
      return true;
    }

    if (args.length < 1) {
      // Must supply material
      return false;
    }

    Material material = Material.matchMaterial(args[0]);

    if (material == null) {
      lang.sendMessage(
          sender,
          "messages.error.invalidMaterial",
          new Replacement("%target%", args[0]));
      return false;
    }

    int radius = 5;

    if (args.length > 1) {
      try {
        radius = Integer.parseInt(args[1]);
      } catch (NumberFormatException e) {
        // Invalid radius supplied
        return false;
      }
    }

    // Clamp radius.
    int configMax = plugin.getConfig().getInt("settings.command.searchcontainer.max-radius", 10);
    radius = Math.max(0, Math.min(radius, configMax));

    World world = senderPlayer.getWorld();
    Chunk centerChunk = senderPlayer.getLocation().getChunk();
    StringBuilder locations = new StringBuilder();
    locations.append("\n");

    for (int dX = -radius; dX <= radius; ++dX) {
      for (int dZ = -radius; dZ <= radius; ++dZ) {
        if (!world.loadChunk(centerChunk.getX() + dX, centerChunk.getZ() + dZ, false)) {
          continue;
        }
        Chunk chunk = world.getChunkAt(centerChunk.getX() + dX, centerChunk.getZ() + dZ);
        for (BlockState tileEntity : chunk.getTileEntities()) {
          if (!(tileEntity instanceof InventoryHolder holder)) {
            continue;
          }
          if (!SearchHelper.findMatch(holder.getInventory(), itemStack -> itemStack.getType() == material)) {
            continue;
          }
          locations.append(holder.getInventory().getType().name().toLowerCase(Locale.ENGLISH)).append(": ");
          if (holder instanceof Nameable named) {
            String customName = named.getCustomName();
            if (customName != null) {
              locations.append(customName.toLowerCase(Locale.ENGLISH).strip()).append(" - ");
            }
          }
          locations.append(" (").append(tileEntity.getX()).append(',').append(tileEntity.getY()).append(',')
              .append(tileEntity.getZ()).append(")\n");
        }
      }
    }

    if (locations.length() < 2) {
      lang.sendMessage(
          sender,
          "messages.info.container.noMatches",
          new Replacement("%target%", material.name()));
      return true;
    }

    lang.sendMessage(
        sender,
        "messages.info.container.matches",
        new Replacement("%target%", material.name()),
        new Replacement("%detail%", locations.toString()));
    return true;
  }

  @Override
  public List<String> onTabComplete(
      @NotNull CommandSender sender,
      @NotNull Command command,
      @NotNull String label,
      String[] args) {
    if (args.length < 1 || args.length > 2 || !command.testPermissionSilent(sender)) {
      return Collections.emptyList();
    }

    String argument = args[args.length - 1];
    if (args.length == 1) {
      return TabCompleter.completeEnum(argument, Material.class);
    } else {
      return TabCompleter.completeInteger(argument);
    }
  }

}