/
wiki.rb
57 lines (43 loc) · 1.1 KB
/
wiki.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
module Selbot2
class Wiki
include Cinch::Plugin
HELPS << [":wiki", "search the wiki"]
self.prefix = Selbot2::PREFIX
match /wiki ((?:\w*\s?){0,3})[\\|$]?/
def execute(message, query)
query = query.strip.gsub(' ', '-')
replies = replies_for("pagename:#{query}")
if replies.empty?
replies = replies_for(query)
end
if replies.empty?
message.reply "No results."
return
end
replies.each_with_index do |resp, idx|
message.reply "#{idx + 1}: #{resp}"
end
end
def escaper
@escaper ||= URI::Parser.new
end
def replies_for(query)
resp = RestClient.get "#{Selbot2::REPO}/wiki"
doc = Nokogiri.HTML(resp.downcase)
rows = doc.css("#wiki-content a[href*='#{query.downcase}']")
rows[0..2].map { |d| Page.new(d).url }.compact
end
class Page
def initialize(doc)
@doc = doc
end
def name
@doc.text.strip
end
def url
path = @doc['href']
"https://github.com#{path}"
end
end # Page
end # Wiki
end # Selbot2