tools.devnull.boteco.plugins.diceroll.SimpleDiceRoll Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of boteco-plugin-diceroll Show documentation
The newest version!
/*
 * The MIT License
 *
 * Copyright (c) 2016 Marcelo "Ataxexe" Guimarães 
 *
 * 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 tools.devnull.boteco.plugins.diceroll;

import java.util.Arrays;
import java.util.function.Function;

public class SimpleDiceRoll implements DiceRoll {

  private final Function diceFunction;

  public SimpleDiceRoll(Function diceFunction) {
    this.diceFunction = diceFunction;
  }

  public SimpleDiceRoll() {
    this(SimpleDice::new);
  }

  @Override
  public int roll(String expression) {
    String[] rolls = expression.split("\\s*[+]\\s*");
    if (rolls.length > 1) {
      return Arrays.stream(rolls)
          .mapToInt(this::roll)
          .reduce(0, (left, right) -> left + right);
    } else {
      return _roll(expression);
    }
  }

  private int _roll(String expression) {
    int total = 0;
    int dices = 0;
    int sides = 0;

    if (expression.startsWith("d")) {
      dices = 1;
      sides = Integer.parseInt(expression.substring(1));
    } else if(expression.matches("\\d+")) {
      total = Integer.parseInt(expression);
    } else {
      String[] split = expression.split("d");
      dices = Integer.parseInt(split[0]);
      sides = Integer.parseInt(split[1]);
    }

    for (int i = 0; i < dices; i++) {
      total += diceFunction.apply(sides).roll();
    }

    return total;
  }

}