hearthy.rb

require 'cinch'
require 'csv'

bot = Cinch::Bot.new do
  configure do |c|
    c.nick            = "Hearthy"  
    c.server          = "irc.sandcat.nl"
    c.channels        = ["#hs"]
  end

  helpers do
    def reply_card(m, card, colors)
        # first line: name - type - (race) - class
        reply = Format("%s - #{card["Type"]}" % [Format(colors[card["Rarity"]], card["Name"])])
        reply += " (#{card["Race"]})" if card["Race"] != nil
        if card["Class"] != nil and card["Class"] != "All" then 
          reply += Format(colors[card["Class"]], " - #{card["Class"]}")
        end
        m.reply Format(:bold, reply)
        
        # mana and if available, attack and health
        reply = ""
        reply += "M:#{card["Mana"]} " if card["Mana"] != nil
        reply += "A:#{card["Attack"]} " if card["Attack"] != nil
        reply += "H:#{card["Health"]}" if card["Health"] != nil
        m.reply Format(:bold, reply)
        
        # description if applicable
        m.reply Format(:lime, "#{card["Description"]}") if card["Description"] != nil
    end

    def hs(m, query)
      #  http://rubydoc.info/gems/cinch/Cinch/Formatting     
      colors = Hash.new
      colors["Druid"] = :orange
      colors["Mage"] = :royal
      colors["Warlock"] = :purple
      colors["Shaman"] = :blue
      colors["Warrior"] = :red
      colors["Priest"] = :white
      colors["Rogue"] = :yellow
      colors["Hunter"] = :green
      colors["Paladin"] = :pink
      colors["Epic"] = :purple
      colors["Legendary"] = :orange
      colors["Rare"] = :blue
      colors["Common"] = :white
      colors["Basic"] = :white
      
      # I should perhaps load the list once instead of every query XXX
      # cards obtained from http://hearthstonecardlist.com/
      # open the cards csv file for reading while maintaining column headers and converting numeric values
      cards = CSV.read('cards.csv', :headers => true, :converters=>:numeric, :encoding =>"UTF-8")
    
      # search for all instances of query in the Name column
      found_cards = Array.new
      cards.each { |card|
        if (card["Name"].downcase[query.downcase] != nil) then 
          found_cards.push(card)
        end
      }      

      # act depending on the number of found cards
      case found_cards.length
      when 0
        m.reply "\001ACTION heeft niets kunnen vinden voor [#{query}] :/\001"
      when 1 # one card found
        p found_cards[0]
        card = found_cards[0]

        reply_card(m, card, colors)
      
      else # when multiple cards look like the search query
        # stick all cardnames together
        card_array = Array.new
        found_cards.each { |card| card_array.push("[" + card["Name"] + "]") }
        card_array_string = card_array.join(", ")

        card = found_cards.select{ |c| c["Name"].downcase == query.downcase }[0]

        reply_card(m, card, colors) unless card.nil?

        # and print them
        m.reply "\001ACTION heeft #{found_cards.length} kaarten gevonden: #{card_array_string}"
      end
    end
  end

  on :message, /\[(.+)\]/ do |m, query|
    #m.reply "\001ACTION zoekt naar [#{query}]\001"
    hs(m, query)
  end

  trap "SIGINT" do
    bot.quit
  end
end

bot.start