AcroBot.rb

require 'rubygems'
require 'cinch'
require 'yaml'

#comment goes here


DEFAULT_DICTIONARY = {
  'new' => { 'tbd' => 'To be done.' }
}

module AbbrevBot

  @@home_dir = File.expand_path '~/.config'
  @@install_dir = File.expand_path File.dirname(__FILE__)
  @@abbrevs_file = "#{@@install_dir}/abbrev.yaml"
  @@cfg_file = 'acrobot.yaml'

  def initialize_dictionary
    File.open(@@abbrevs_file, 'w') { |f|
      YAML.dump(DEFAULT_DICTIONARY, f)
    }
    dictionary
  end

  # Try to load the YAML file. If it does not exist initialize the
  # file and populate it with DEFAULT_DICTIONARY.
  # If the file exists but is scrumbled (false), do the same thing.
  #
  def dictionary
    begin
      YAML.load_file(@@abbrevs_file) || initialize_dictionary
    rescue Errno::ENOENT
      initialize_dictionary
    end
  end

  # Open the YAML file with dictionary and look for the abbrev
  def lookup_dictionary(abbrev)
    results = []
    dictionary.each do |key, values|
      next if values.nil?
      matched_keys = values.keys.select { |k| k.to_s.casecmp(abbrev) == 0 }
      # => ['tbd', 'rhel'] ...
      matched_keys.each do |k|
        results << [k, values[k]]
      end
    end
    results.empty? ? false : results
  end

  #Find abbrevs with a certain tag
  def find_values(tag)
    results =[]
    dictionary.each do |key, values|
      next if values.nil?
      matched_keys = values.select { |k,v| v.to_s =~ /@#{tag}( |$)/i }.keys.each do |k|
        results << k
      end
    end
    results
  end

  def save_abbrev(name, description)
    dict = dictionary
    dict['new'] ||= {}
    dict['new'][name.strip] = description.strip
    File.open(@@abbrevs_file, 'w') { |f| YAML.dump(dict, f) }
  end

  def load_settings
    require 'yaml'
    return YAML::load_file("#{@@home_dir}/#{@@cfg_file}") if File.exists? "#{@@home_dir}/#{@@cfg_file}"
    return YAML::load_file("#{@@install_dir}/#{@@cfg_file}")
  end

end

include AbbrevBot

bot = Cinch::Bot.new do

  settings = load_settings

  configure do |c|
   c.nick = settings['nick']         # "acrobot"
   c.realname = settings['realname'] # "IRC Acronym and Abbreviation Expander Bot. '!help' for help"
   c.user = settings['user']         # "acrobot" (user name when connecting)
   c.server = settings['server']     # "irc.freenode.net"
   c.channels = settings['channels'] # ["#acrobot","#katello","#openshift","#satellite6","#zanata","#theforeman"]
#   c.channels = ["#acrobot"]
   c.prefix = settings['prefix']     # /^!/
  end

  on :message, /^!([\w\-\_]+)\=(.+)/ do |m, abbrev, desc|
    nick = m.channel? ? m.user.nick+": " : ""
    save_abbrev(abbrev, desc)
   nick = m.channel? ? m.user.nick+"":""
    m.reply("#{nick} Thanks! [#{abbrev}=#{desc}]")
  end

  on :message, /^!help/i do |m|
    nick = m.channel? ? m.user.nick+": " : ""
    m.reply("To expand an acronym, type (e.g.), !ftp")
    m.reply("To add a new acronym, type (e.g.), !FTP=File Transfer Protocol")
    m.reply("To associate a tag with an acronym, type (e.g.), !IP=Internet Protocol @networking")
    m.reply("To list abbrevs associated with a tag, type eg. !@kernel")
    m.reply("To list all tags, type !@tags")
  end

  on :message, /^!@([\w\-\_]+)$/ do |m, tag|
    tag = tag.strip
    match_abbrevs = find_values(tag)
    nick = m.channel? ? m.user.nick+": " : ""
    if match_abbrevs.empty?
      m.reply("#{nick} Sorry, no such tag. To list all tags, type !@tags")
    else
        m.reply("#{nick}#{match_abbrevs.join(', ')}")
    end
  end

  on :message, /^!([\w\-\_]+)$/ do |m, abbrev|
    abbrev = abbrev.strip
    unless abbrev =~ /^help$/i
      nick_str = m.channel? ? "#{m.user.nick}:" : ""
      if replies = lookup_dictionary(abbrev)
        replies.each do |original_abbrev, value|
          value, *tags = value.split('@')
          reply_str = "%s %s: %s %s" % [
            nick_str,
            Cinch::Formatting.format(:bold, original_abbrev.to_s),
            Cinch::Formatting.format(:bold, value.strip),
            tags.map { |t| "@#{t.strip}" }.join(', ')
          ]
          m.reply(reply_str.strip)
        end
      else
        m.reply("#{nick_str} Sorry, no definition for #{abbrev}")
      end
    end
  end

end

bot.start