wordlist provider

1 files changed, 44 insertions, 0 deletions

diff --git a/lib/rbot/core/utils/wordlist.rb b/lib/rbot/core/utils/wordlist.rb
new file mode 100755
index 00000000..23e141ac
--- /dev/null
+++ b/lib/rbot/core/utils/wordlist.rb
@@ -0,0 +1,44 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: rbot wordlist provider
+#
+# Author:: Raine Virta <rane@kapsi.fi>
+
+require "find"
+
+module ::Irc
+class Bot
+class Wordlist
+  def self.wordlist_base
+    @@wordlist_base ||= File.join(Utils.bot.botclass, 'wordlists')
+  end
+
+  def self.get(path, options={})
+    opts = { :spaces => false }.merge(options)
+
+    wordlist_path = File.join(wordlist_base, path)
+    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
+
+    wordlist.map! { |l| l.strip }
+    wordlist.reject do |word|
+      word =~ /\s/ && !opts[:spaces] ||
+      word.empty?
+    end
+  end
+end
+end
+end
\ No newline at end of file