ZapRequest.java

package net.sourceforge.kolmafia.request;

import java.io.BufferedReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import net.java.dev.spellcast.utilities.LockableListModel;
import net.java.dev.spellcast.utilities.SortedListModel;
import net.sourceforge.kolmafia.AdventureResult;
import net.sourceforge.kolmafia.KoLCharacter;
import net.sourceforge.kolmafia.KoLConstants;
import net.sourceforge.kolmafia.KoLConstants.MafiaState;
import net.sourceforge.kolmafia.KoLmafia;
import net.sourceforge.kolmafia.RequestLogger;
import net.sourceforge.kolmafia.objectpool.ItemPool;
import net.sourceforge.kolmafia.persistence.ItemDatabase;
import net.sourceforge.kolmafia.preferences.Preferences;
import net.sourceforge.kolmafia.session.InventoryManager;
import net.sourceforge.kolmafia.session.ResultProcessor;
import net.sourceforge.kolmafia.utilities.FileUtilities;
import net.sourceforge.kolmafia.utilities.StringUtilities;

public class ZapRequest extends GenericRequest {
  private static final Pattern ZAP_PATTERN = Pattern.compile("whichitem=(\\d+)");
  private static final Pattern ACQUIRE_PATTERN =
      Pattern.compile("You acquire an item: <b>(.*?)</b>");
  private static final Pattern OPTION_PATTERN =
      Pattern.compile("<option value=(\\d+) descid='.*?'>.*?</option>");

  private static final Map<Integer, List<String>> zapGroups = new HashMap<>();

  private AdventureResult item;
  private AdventureResult acquired;

  public ZapRequest(final AdventureResult item) {
    super("wand.php");

    this.item = null;

    if (KoLCharacter.getZapper() == null) {
      return;
    }

    this.item = item;

    this.addFormField("action", "zap");
    this.addFormField("whichwand", String.valueOf(KoLCharacter.getZapper().getItemId()));
    this.addFormField("whichitem", String.valueOf(item.getItemId()));
  }

  private static void initializeList() {
    if (!ZapRequest.zapGroups.isEmpty()) {
      return;
    }

    try (BufferedReader reader =
        FileUtilities.getVersionedReader("zapgroups.txt", KoLConstants.ZAPGROUPS_VERSION)) {
      String line;

      while ((line = FileUtilities.readLine(reader)) != null) {
        List<String> list = StringUtilities.tokenizeString(line);
        for (String name : list) {
          int itemId = ItemDatabase.getItemId(name, 1, false);
          if (itemId < 0) {
            RequestLogger.printLine("Unknown item in zap group: " + name);
            continue;
          }
          ZapRequest.zapGroups.put(itemId, list);
        }
      }
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  public static LockableListModel<AdventureResult> getZappableItems() {
    ZapRequest.initializeList();

    SortedListModel<AdventureResult> matchingItems = new SortedListModel<>();
    if (Preferences.getBoolean("relayTrimsZapList")) {
      matchingItems.addAll(
          KoLConstants.inventory.stream()
              .filter(i -> ZapRequest.zapGroups.containsKey(i.getItemId()))
              .collect(Collectors.toList()));
    } else {
      matchingItems.addAll(KoLConstants.inventory);
    }
    return matchingItems;
  }

  public static List<String> getZapGroup(int itemId) {
    ZapRequest.initializeList();

    return ZapRequest.zapGroups.getOrDefault(itemId, new ArrayList<>());
  }

  @Override
  public void run() {
    if (this.item == null) {
      return;
    }

    if (KoLCharacter.getZapper() == null) {
      KoLmafia.updateDisplay(MafiaState.ERROR, "You don't have a wand.");
      return;
    }

    if (InventoryManager.hasItem(this.item, true)) {
      InventoryManager.retrieveItem(this.item);
    }

    if (!KoLConstants.inventory.contains(this.item)) {
      KoLmafia.updateDisplay(MafiaState.ERROR, "You don't have a " + this.item.getName() + ".");
      return;
    }

    KoLmafia.updateDisplay("Zapping " + this.item.getName() + "...");
    super.run();
  }

  @Override
  public void processResults() {
    // Remove item if zap succeeded. Remove wand if it blew up.
    ZapRequest.parseResponse(this.getURLString(), this.responseText);

    // "The Crown of the Goblin King shudders for a moment, but
    // nothing happens."
    if (this.responseText.contains("nothing happens")) {
      KoLmafia.updateDisplay(MafiaState.ERROR, "The " + this.item.getName() + " is not zappable.");
      return;
    }

    Matcher acquiresMatcher = ZapRequest.ACQUIRE_PATTERN.matcher(responseText);
    String acquired = acquiresMatcher.find() ? acquiresMatcher.group(1) : null;
    if (acquired != null) {
      this.acquired = ItemPool.get(acquired, 1);
    }

    // Notify the user of success.
    KoLmafia.updateDisplay(
        this.item.getName()
            + " has been transformed into "
            + (acquired != null ? acquired : "an unknown item")
            + ".");
  }

  public static void parseResponse(final String urlString, final String responseText) {
    if (!urlString.startsWith("wand.php")) {
      return;
    }

    if (responseText.contains("nothing happens")) {
      return;
    }

    // If it blew up, remove wand and zero usages
    if (responseText.contains("abruptly explodes")) {
      ResultProcessor.processResult(KoLCharacter.getZapper().getNegation());
      // set to -1 because will be incremented below
      Preferences.setInteger("_zapCount", -1);
      // a new wand can be made in 3 days
      Preferences.setInteger("lastZapperWandExplosionDay", KoLCharacter.getCurrentDays());
    }

    Matcher itemMatcher = ZapRequest.ZAP_PATTERN.matcher(urlString);
    if (!itemMatcher.find()) {
      return;
    }

    // Remove the item which was transformed.
    int itemId = StringUtilities.parseInt(itemMatcher.group(1));
    ResultProcessor.removeItem(itemId);

    // increment zap count
    Preferences.increment("_zapCount");
  }

  public static void decorate(final StringBuffer buffer) {
    // Don't trim the list if user wants to see all items
    if (!Preferences.getBoolean("relayTrimsZapList")) return;

    ZapRequest.initializeList();

    int selectIndex = buffer.indexOf("<select");
    if (selectIndex == -1) return;
    selectIndex = buffer.indexOf(">", selectIndex) + 1;
    int endSelectIndex = buffer.indexOf("</select>", selectIndex);
    Matcher optionMatcher =
        ZapRequest.OPTION_PATTERN.matcher(buffer.substring(selectIndex, endSelectIndex));
    buffer.delete(selectIndex, endSelectIndex);

    int itemId;
    StringBuilder zappableOptions = new StringBuilder();
    while (optionMatcher.find()) {
      itemId = Integer.parseInt(optionMatcher.group(1));
      if (itemId == 0 || ZapRequest.zapGroups.containsKey(itemId)) {
        zappableOptions.append(optionMatcher.group());
      }
    }

    buffer.insert(selectIndex, zappableOptions);
    int pos = buffer.lastIndexOf("</center>");
    buffer.insert(
        pos,
        "KoLmafia trimmed this list to the items it knows to be zappable, which may not include recently discovered or modified items.  <a href=\"wand.php?whichwand="
            + KoLCharacter.getZapper().getItemId()
            + "&notrim=1\">Click here</a> for the full list.");
  }

  public static boolean registerRequest(final String urlString) {
    if (!urlString.startsWith("wand.php")) {
      return false;
    }

    Matcher itemMatcher = ZapRequest.ZAP_PATTERN.matcher(urlString);
    if (!itemMatcher.find()) {
      return true;
    }

    int itemId = StringUtilities.parseInt(itemMatcher.group(1));
    AdventureResult item = ItemPool.get(itemId);

    RequestLogger.updateSessionLog();
    RequestLogger.updateSessionLog("zap " + item.getName());

    return true;
  }

  public AdventureResult getAcquired() {
    return acquired;
  }
}