class Irc::Bot::Wordlist
Public Class Methods
exist?(bot, path)
click to toggle source
# File lib/rbot/core/utils/wordlist.rb, line 55 def self.exist?(bot, path) wordlist_base = bot.path('wordlists') fn = path.to_s # refuse to check outside of the wordlist base directory return false if fn =~ /\.\.\// File.exist?(File.join(wordlist_base, fn)) end
get(bot, where, options={})
click to toggle source
# File lib/rbot/core/utils/wordlist.rb, line 13 def self.get(bot, where, options={}) wordlist_base = bot.path('wordlists') opts = { :spaces => false }.merge(options) wordlist_path = File.join(wordlist_base, where) raise "wordlist not found: #{wordlist_path}" unless File.exist?(wordlist_path) # Location is a directory -> combine all lists beneath it wordlist = if File.directory?(wordlist_path) wordlists = [] Find.find(wordlist_path) do |path| next if path == wordlist_path wordlists << path unless File.directory?(path) end wordlists.map { |list| File.readlines(list) }.flatten else File.readlines(wordlist_path) end # wordlists are assumed to be UTF-8, but we need to strip the BOM, if present wordlist.map! { |l| l.sub("\xef\xbb\xbf",'').strip } wordlist.reject do |word| word =~ /\s/ && !opts[:spaces] || word.empty? end end
list(bot, options={})
click to toggle source
Return an array with the list of available wordlists. Available options:
- pattern
-
pattern that should be matched by the wordlist filename
# File lib/rbot/core/utils/wordlist.rb, line 44 def self.list(bot, options={}) wordlist_base = bot.path('wordlists') pattern = options[:pattern] || "**" # refuse patterns that contain ../ return [] if pattern =~ /\.\.\// striplen = wordlist_base.length+1 Dir.glob(File.join(wordlist_base, pattern)).map { |name| name[striplen..-1] } end