dice-n-easy.rb

require 'discordrb'
require 'dicebag'
require 'dotenv'

Dotenv.load
PREFIX = ENV['DICENEASY_PREFIX'] || '/'
ROLLCOMMAND = ENV['DICENEASY_ROLLCOMMAND'] || 'r'
REDIS_HOST = ENV['DICENEASY_REDISHOST']
REDIS_PASSWORD = ENV['DICENEASY_REDISPASSWORD']
REDIS_URL = ENV['REDIS_URL']
ADD_MACRO_COMMAND = ENV['DICENEASY_ADDMACROCOMMAND'] || 'am'
USE_MACRO_COMMAND = ENV['DICENEASY_USEMACROCOMMAND'] || 'm'
TOKEN = ENV['DICENEASY_TOKEN']
DICE_PREFIX = "#{PREFIX}#{ROLLCOMMAND}"
puts "Using redis #{REDIS_HOST if REDIS_HOST} for macros" if REDIS_HOST || REDIS_URL
puts "Using a password for redis" if REDIS_HOST and REDIS_PASSWORD
puts "Using token: %s" % TOKEN.slice(-5,5)

def get_user_or_nick(event)
  user = event.user.nick != nil ? event.user.nick : event.user.name
end
def roll_dice(dice_string)
  puts "Rolling #{dice_string}"
  DiceBag::Roll.new(dice_string).result()
end
def format_embed(embed, result, user, macro = nil, macro_string = nil)
  embed.title = "#{user} rolled #{result}"
  reason = ""
  result.each do |section|
    reason = "%s %sd%s:%s".% [reason, section.count, section.sides, section.tally] if defined? section.count
  end
  embed.description = reason
  embed.color = 4289797
  if macro && macro_string
    embed.fields = [
      Discordrb::Webhooks::EmbedField.new(name: macro, value: macro_string),
    ]
  end
  embed
end
def embed_help_message(embed)
  embed.title = "I'm your friendly neighborhood dice bot here to help!"
  embed.description = "The following are some common dice rolls"
  embed.fields = [
    Discordrb::Webhooks::EmbedField.new(name: "Advantage", value: "#{DICE_PREFIX} 2d20k1+5"),
    Discordrb::Webhooks::EmbedField.new(name: "Disadvantage", value: "#{DICE_PREFIX} 2d20kl1+5"),
    Discordrb::Webhooks::EmbedField.new(name: "Reroll 1s", value: "#{DICE_PREFIX} 1d6 r1"),
    Discordrb::Webhooks::EmbedField.new(name: "Explode 5s and 6s", value: "#{DICE_PREFIX} 1d6 e5"),
    Discordrb::Webhooks::EmbedField.new(name: "Adding a macro", value: "#{PREFIX}#{ADD_MACRO_COMMAND} attack 1d20+4"),
    Discordrb::Webhooks::EmbedField.new(name: "Use a macro", value: "#{PREFIX}#{USE_MACRO_COMMAND} attack"),
  ]
  embed.color = 4289797
end

@bot = Discordrb::Commands::CommandBot.new token: ENV['DICENEASY_TOKEN'], compress_mode: :large, ignore_bots: true, fancy_log: true, prefix: PREFIX
@bot.gateway.check_heartbeat_acks = false

@bot.command ROLLCOMMAND.to_sym do |event|
    puts "ROLLING: #{event.content}"
    input = event.content.strip
    user = get_user_or_nick(event)
    
    dice_string = input.delete_prefix(DICE_PREFIX).strip
    begin
      result = roll_dice(dice_string)
      event.send_embed do |embed| format_embed(embed, result, user) end
    rescue
      puts "Dice roll #{dice_string} failed"
      event.respond "#{user} oops! #{dice_string} doesn't compute. Please try again. PM me `help` for help."
    end
end

@bot.pm do |event|
  puts "PM: #{event.content}"
  next if !event.content.downcase.include? 'help'
  event.send_embed do |embed| embed_help_message embed end
end

if REDIS_HOST || REDIS_URL
  require 'redis'
  redis = REDIS_HOST ? Redis.new(host: REDIS_HOST, password: REDIS_PASSWORD) : Redis.new(url: REDIS_URL)

  @bot.command ADD_MACRO_COMMAND.to_sym do |event|
    puts "ADD MACRO: #{event.content}"
    input = event.content.strip
    macro_name = input.split[1].upcase
    dice_string = input.split.drop(2).join(' ')
    user = get_user_or_nick(event)
    begin
      roll_dice(dice_string)
      key = "#{user}:#{macro_name}"
      redis.set(key, dice_string)
      event.respond "#{user} added macro #{input.split[1]} as #{dice_string}. Use with #{PREFIX}#{USE_MACRO_COMMAND} #{input.split[1]}"
    rescue
      puts "Add macro failed #{input}"
      event.respond "#{user} oops! #{input} isn't a macro. Add macros look like: #{PREFIX}#{ADD_MACRO_COMMAND} <macro name> <dice to roll>. Please make sure the dice command is valid and there are no spaces in the macro name."
    end
  end

  @bot.command USE_MACRO_COMMAND.to_sym do |event|
    puts "USE MACRO:#{event.content}"
    input = event.content.strip
    macro_name = input.split[1].upcase
    user = get_user_or_nick(event)
    begin
      key = "#{user}:#{macro_name}"
      dice_string = redis.get(key)
      result = roll_dice(dice_string)
      event.send_embed do |embed| format_embed(embed, result, user, macro_name, dice_string) end
    rescue
      puts "Use macro failed #{input}"
      event.respond "#{user} oops! #{input} isn't a saved macro. Remember macros are user and nickname specific."
    end
  end
end
@bot.run