dicebot/dice.js

const diceRegex = () => /\b(\d*)[dD](\d+)\b(?:\s?([+-])\s?(\d+))?/g
const rerollRegex = () => /\b(reroll|re-roll)\s+((?:(?:\S*|\d+s|\d+'s|\d+)\s*)+)/
const numberRegex = () => /\b(one|two|three|four|five|six|seven|eight|nine|ten|ones|twos|threes|fours|fives|sixes|sevens|eights|nines|tens|one's|two's|three's|four's|five's|seven's|eight's|nine's|ten's|\d+)\b/gi
const advRegex = () => /\b(disadvantage|advantage)\b/i
const percentRegex = () => /\b(percentile|percentiles|percent)\b/i
const dicebotRegex = () => /\b(dicebot|dice bot)\b/i
const unicoder = require('./unicoder')

const d6 = {
  stringify: n => unicoder.codesets.dice[n],
  total: (ns, bonus) => ns.reduce((a, b) => a + b, bonus).toString()
}

const coin = {
  stringify: (n, i, total) => `${n === 1 ? unicoder.codesets.circles.red : n === 2 ? unicoder.codesets.circles.blue : unicoder.codesets.circles.black}${total === 1 ? (n === 1 ? ' Heads' : n === 2 ? ' Tails' : ' Sides') : ''}`,
  total: ns => {
    const heads = ns.filter(x => x === 1).length
    const tails = ns.filter(x => x === 2).length

    if (heads === 1 && tails === 0) return 'Heads'
    if (heads === 0 && tails === 1) return 'Tails'
    if (heads === 0 && tails > 0) return `${tails} Tails and no Heads`
    if (heads > 0 && tails === 0) return `${heads} Heads and no Tails`
    if (heads > 0 && tails > 0) return `${heads} Heads and ${tails} Tails`
    return 'The coin disappears'
  }
}

const d10 = {
  stringify: (n, i, total, { percent = false } = {}) => {
    if (total === 2 && percent) {
      if (i === 0) {
        return n === 10 ? '00' : `${n}0`
      } else {
        return n === 10 ? '0' : n.toString()
      }
    } else {
      return unicoder.codesets.negRound[n === 10 ? '0' : n.toString()]
    }
  },
  total: (ns, bonus, { percent = false } = {}) => {
    if (ns.length === 2 & percent) {
      if (ns[0] === 0 && ns[1] === 0) return '100'
      return ((ns[0] === 10 ? 0 : ns[0]) * 10 + (ns[1] % 10)).toString()
    }
    return ns.reduce((a, b) => a + b, bonus).toString()
  },
  bonusText: (x) => {
    const {percent = false, values = []} = x || {}

    if (values.length === 2 && !percent) return '_Did you mean "percentile"?_'
  }
}

const d20 = {
  stringify: n => unicoder.codesets.negRound[n],
  total: (ns, bonus) => ns.reduce((a, b) => a + b, bonus).toString()
}

const ascii = {
  stringify: n => n.toString(),
  total: (ns, bonus) => ns.reduce((a, b) => a + b, bonus).toString()
}

class Roll {
  constructor(roll) {
    if (roll) Object.assign(this, roll)
  }
  value() {
    return this.values.reduce((a, b) => a + b, this.bonus || 0)
  }
  toString() {
    let str = this.values.map((val, i) => this.formatter && this.formatter.stringify ? this.formatter.stringify(val, i, this.values.length, this) : `[${val}]`).join(' + ')
    if (this.bonus < 0) {
      str += ` - ${Math.abs(this.bonus)}`
    } else if (this.bonus > 0) {
      str += ` + ${this.bonus}`
    }
    if (this.values.length > 1 || this.bonus) {
      str += ` = ${this.formatter && this.formatter.total ? this.formatter.total(this.values, this.bonus || 0, this) : this.value()}`
    }
    if (this.rerolls === 1) {
      str += ` with 1 reroll`
    } else if (this.rerolls > 1) {
      str += ` with ${this.rerolls} rerolls`
    }
    return str
  }
}

class BadRoll extends Roll {
  constructor(text) {
    super({text})
  }
  toString() {
    return this.text
  }
}

const random = (rng, n) => Math.floor(rng() * n) + 1

class Dice {
  constructor(dice) {
    if (dice) Object.assign(this, dice)
    if (!this.formatter) {
      if (this.sides === 2) this.formatter = coin
      else if (this.sides === 6) this.formatter = d6
      else if (this.sides === 10) this.formatter = d10
      else if (this.sides <= 20) this.formatter = d20
      else this.formatter = ascii
    }
    if (!this.rng) this.rng = () => Math.random()
  }
  roll() {
    if (this.count > 100) return new BadRoll(`I lost count rolling ${this}`)
    if (this.sides > 256) return new BadRoll(`The faces are too small to read on ${this}`)
    if (this.count === 0) return new BadRoll(`I finished before I started rolling ${this}`)
    if (this.sides === 0) return new BadRoll(`I lost my grip on ${this}`)
    if (this.sides === 1) return new BadRoll(`The mobius ${this} never stopped rolling`)
    if (this.bonusSign === '-' && this.bonus > this.sides) return new BadRoll(`It doesn't seem fair to roll ${this}`)

    let values = []
    for (let i = 0; i < this.count; i++) {
      if (this.modifier === 'advantage') {
        values.push(Math.max(random(this.rng, this.sides), random(this.rng, this.sides)))
      } else if (this.modifier === 'disadvantage') {
        values.push(Math.min(random(this.rng, this.sides), random(this.rng, this.sides)))
      } else {
        values.push(random(this.rng, this.sides))
      }
    }
    let bonus = 0
    switch (this.bonusSign) {
      case '-': bonus = -this.bonus; break
      case '+': bonus = this.bonus; break
    }
    let rerolls = 0
    for (let i = 0; i < 10; i++) {
      let foundAny = false
      for (let d = 0; d < values.length; d++) {
        const v = values[d]
        if (this.rerollNumbers && this.rerollNumbers.includes(v)) {
          values[d] = random(this.rng, this.sides)
          foundAny = true
          rerolls++
        }
      }
      if (!foundAny) break
    }
    return new Roll({
      values,
      bonus,
      rerolls,
      percent: this.percent,
      formatter: this.formatter
    })
  }
  toString() {
    let str =
      this.sides === 2
      ? (
        this.count === 1 ? 'a coin' : `${this.count} coins`
      )
      : (this.count === 2 && this.percent)
      ? 'percentile dice'
      :`${this.count}d${this.sides}`
    if (this.bonusSign) {
      str += this.bonusSign + this.bonus
    }
    if (this.modifier) {
      str += ` with ${this.modifier}`
    }
    if (this.rerollNumbers && this.rerollNumbers.length) {
      if (this.rerollNumbers.length === 1) {
        str += ` and reroll ${this.rerollNumbers[0]}'s`
      } else {
        str += ` and reroll ${this.rerollNumbers.slice(0, this.rerollNumbers.length - 1).join(`'s, `)}'s, and ${this.rerollNumbers[this.rerollNumbers.length - 1]}'s`
      }
    }
    return str
  }
}

Dice.parse = (str, opts) => {
  const rrRegex = rerollRegex()
  const rerolls = []
  const reroll = rrRegex.exec(str)
  const rerollNumbers = []
  if (reroll) {
    const nRegex = numberRegex()
    while (number = nRegex.exec(str)) {
      switch (number[1]) {
        case 'ones':
        case 'one':
          rerollNumbers.push(1)
          break
        case 'twos':
        case 'two':
          rerollNumbers.push(2)
          break
        case 'threes':
        case 'three':
          rerollNumbers.push(3)
          break
        case 'fours':
        case 'four':
          rerollNumbers.push(4)
          break
        case 'fives':
        case 'five':
          rerollNumbers.push(5)
          break
        case 'sixes':
        case 'six':
          rerollNumbers.push(6)
          break
        case 'sevens':
        case 'seven':
          rerollNumbers.push(7)
          break
        case 'eights':
        case 'eight':
          rerollNumbers.push(8)
          break
        case 'nines':
        case 'nine':
          rerollNumbers.push(9)
          break
        case 'tens':
        case 'ten':
          rerollNumbers.push(10)
          break
        default:
          const n = parseFloat(number)
          if (!isNaN(n)) {
            rerollNumbers.push(n)
          }
      }
    }
  }

  const aRegex = advRegex()
  const ad = aRegex.exec(str)
  const dicebot = dicebotRegex().test(str)
  
  const modifier = ad ? (
    ad[1].toLowerCase()[0] === 'a' ? 'advantage' : 'disadvantage'
  ) : null

  const percent = percentRegex().test(str)

  const regex = diceRegex()
  const ret = []
  let dice
  if (percent) {
    ret.push(new Dice({
      count: 2,
      sides: 10,
      percent: true
    }))
  }
  while (dice = regex.exec(str)) {
    ret.push(new Dice({
      match: dice,
      count: dice[1] === '' ? 1 : parseFloat(dice[1]),
      sides: parseFloat(dice[2]),
      bonusSign: dice[3] || null,
      bonus: dice[4] === undefined ? null : parseFloat(dice[4]),
      rerollNumbers,
      modifier,
      percent,
      dicebot,
      rng: opts && opts.rng
    }))
  }
  return ret
}

Dice.chat = (chat, opts) => {
  const dice = Dice.parse(chat, opts)
  if (dice.length) {
    const diceStrings = dice.map(x => x.toString())
    if (diceStrings.length > 1) {
      diceStrings[diceStrings.length - 1] = `and ${diceStrings[diceStrings.length - 1]}`
    }
    const isCoin = dice.length === 1 && dice[0].sides === 2
    const rollingString = `${isCoin ? 'Flipping' : 'Rolling'} ${diceStrings.join(', ')}...`
    const rolls = dice.map(die => die.roll())
    const rollsStrings = rolls.map(roll => roll.toString())
    const bonusString = rolls
      .filter(x => x.formatter && x.formatter.bonusText)
      .map(x => x.formatter.bonusText(x))
      .filter(x => x)
      .map(x => '\n' + x)
      .join('')
    return `${rollingString} ${rollsStrings.join(', ')}.${bonusString}`
  }
}

module.exports = {
  Dice,
  Roll,
  diceRegex,
  rerollRegex,
  advRegex
}