KitList.java

/*
 * Copyright (c) 2010 SimpleServer authors (see CONTRIBUTORS)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package simpleserver.config;

import java.util.Map.Entry;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.xml.sax.SAXException;

import simpleserver.Color;
import simpleserver.Player;
import simpleserver.bot.BotController.ConnectException;
import simpleserver.config.xml.Permission;

import com.google.common.collect.ImmutableList;

public class KitList extends PropertiesConfig {
  public static final class Kit {
    public final Permission perm;
    public final ImmutableList<Entry> items;

    public static final class Entry {
      private final int item;
      private final short damage;
      private final int amount;

      public Entry(int item, short damage, int amount) {
        this.item = item;
        this.damage = damage;
        this.amount = amount;
      }

      public Entry(int item, int amount) {
        this(item, (short) 0, amount);
      }

      public int item() {
        return item;
      }

      public short damage() {
        return damage;
      }

      public int amount() {
        return amount;
      }
    }

    public Kit(String perm, ImmutableList<Entry> items) throws SAXException {
      this.perm = new Permission(perm);
      this.items = items;
    }
  }

  private ConcurrentMap<String, Kit> kits;

  public KitList() {
    super("kit-list.txt");

    kits = new ConcurrentHashMap<String, Kit>();
  }

  public boolean giveKit(Player player, String kitName) {
    Kit kit = kits.get(kitName.toLowerCase());
    if ((kit != null) && (kit.perm.contains(player))) {
      try {
        player.give(kit);
      } catch (ConnectException e) {
        player.addTMessage(Color.RED, "Giving Kit %s failed!", kitName);
      }
      return true;
    }
    return false;
  }

  public void listKits(Player player) {
    StringBuilder kitList = new StringBuilder();
    for (String name : kits.keySet()) {
      Kit kit = kits.get(name);
      if (kit.perm.contains(player)) {
        kitList.append(name);
        kitList.append(", ");
      }
    }
    if (kitList.length() == 0) {
      player.addTMessage(Color.RED, "You can't use any kits");
    } else {
      player.addTCaptionedMessage("Allowed kits", kitList.substring(0, kitList.length() - 2));
    }
  }

  @Override
  public void load() {
    super.load();

    kits.clear();
    for (Entry<Object, Object> entry : properties.entrySet()) {
      String[] options = entry.getValue().toString().split("\\|");
      if (options.length < 2) {
        System.out.println("Skipping bad kit list entry " + entry.getValue());
        continue;
      }

      boolean legacy = false;

      ImmutableList.Builder<Kit.Entry> items = ImmutableList.builder();
      for (int c = 1; c < options.length; ++c) {

        if (options[c].contains(":")) {
          // legacy
          Kit.Entry item = loadLegacyEntry(options[c]);
          if (item != null) {
            items.add(item);
            legacy = true;
          }
          continue;
        }

        String[] item = options[c].split("\\*");
        String[] data = item[0].split("\\.");
        if (item.length < 1 || item.length > 2 || data.length > 2 || data.length < 1) {
          System.out.println("Skipping bad kit item " + options[c]);
          continue;
        }

        Integer block;
        Short damage = 0;
        Integer amount = 1;

        try {
          if (data.length == 2) {
            damage = Short.parseShort(data[1]);
          }
          block = Integer.parseInt(data[0]);
          if (item.length == 2) {
            amount = Integer.parseInt(item[1]);
          }
        } catch (NumberFormatException e) {
          System.out.println("Skipping bad kit item " + options[c]);
          continue;
        }

        items.add(new Kit.Entry(block, damage, amount));
      }

      Kit kit;
      try {
        kit = new Kit(options[0], items.build());
      } catch (SAXException e) {
        System.out.println("Invalid permission for kit " + options[0] + ": " + e.getMessage());
        continue;
      }
      kits.put(entry.getKey().toString().toLowerCase(), kit);

      if (legacy) {
        StringBuilder convertedEntry = new StringBuilder(kit.perm.toString());
        for (Kit.Entry item : kit.items) {
          convertedEntry.append("|" + item.item);
          if (item.damage != 0) {
            convertedEntry.append("." + item.damage);
          }
          if (item.amount != 1) {
            convertedEntry.append("*" + item.amount);
          }
        }
        properties.setProperty(entry.getKey().toString(), convertedEntry.toString());
        System.out.println("Converting Kit " + entry.getKey().toString() + " to new format.");
      }
    }
  }

  private Kit.Entry loadLegacyEntry(String line) {
    String[] item = line.split(":");
    if (item.length != 2) {
      System.out.println("Skipping bad kit item " + line);
      return null;
    }

    Integer block;
    Integer amount;
    try {
      block = Integer.parseInt(item[0]);
      amount = Integer.parseInt(item[1]);
    } catch (NumberFormatException e) {
      System.out.println("Skipping bad kit item " + line);
      return null;
    }

    return new Kit.Entry(block, amount);
  }
}