summaryrefslogtreecommitdiff
path: root/data/rbot/plugins/grouphug.rb
blob: 1b08de2c338f92d71788f2ee8d8c522be6d4a606 (plain)
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#-- vim:sw=2:et
#++
#
# :title: Grouphug Plugin for rbot
#
# Author:: Mark Kretschmann <markey@web.de>
# Author:: Casey Link <unnamedrambler@gmail.com>
# Copyright:: (C) 2005 Mark Kretschmann
# Copyright:: (C) 2008 Casey Link
# License:: GPL v2

class GrouphugPlugin < Plugin
  START = '<div id="main"'
  REG  = Regexp.new('<div class="content">\s*<p>(.*?)</p>\s+</div>', Regexp::MULTILINE)
  REGPOST = Regexp.new('title>(.*?) \| Group Hug')
  def initialize
    super
    @confessions = Array.new
    @bot.register_filter(:grouphug, :htmlinfo) { |s| grouphug_filter(s) }
  end

  def help( plugin, topic="" )
    return _("Grouphug plugin. Confess! Usage: 'confess' for random confession, 'confess <number>' for specific one, 'confess <confession>' to share your own confession. Confessions must be at least 10 words.")
  end

  def post_confession(m, params)
    c = params[:confession]
    if c.length < 10
      diff = 10 - c.length
      m.reply _("Confession must be at least 10 words. You need %{m} more.") % {:m => diff}
      return
    end
    uri = "http://beta.grouphug.us/confess"
    form_id = "form_id=confession_node_form"
    op = "op=Submit"
    changed = "changed="
    body = "body=#{c}"
    msg = [form_id,body,changed,op].join("&")

    response = bot.httputil.post(uri, msg)
    debug response.body
    if response.class == Net::HTTPOK
      num = response.body.scan(REGPOST)
      m.reply _("Confession posted: http://beta.grouphug.us/confessions/%{n}") % {:n => num}
    else
      m.reply _("I couldn't share your confession.")
    end
  end

  def get_confessions(html)
    return [] unless html
    start = html.index(START)
    res = html[start, html.length - start].scan(REG)
    return [] unless res
    return res.map { |quote|
      quote[0].ircify_html
    }
  end

  def confess(m, params)
    opts = { :cache => false }
    path = "random"
    begin
      # Fetch a specific question - separate from cache
      if params[:num]
        path = "confessions/#{params[:num]}"
        opts.delete(:cache)
        data = @bot.httputil.get("http://grouphug.us/#{path}", opts)
        confessions = get_confessions(data)
        if confessions.length > 1
          warn "more than one confession found!"
          warn confessions
        end
        confessions << "no confession ##{params[:num]} found" if confessions.empty?
        m.reply confessions.first
      else # Cache random confessions
        if @confessions.empty?
          data = @bot.httputil.get("http://grouphug.us/#{path}", opts)
          @confessions.replace get_confessions(data)
        end
        @confessions << "no confessions found!" if @confessions.empty?
        m.reply @confessions.pop
      end
    rescue Exception => e
      error e
      m.reply "failed to connect to grouphug.us"
    end
  end

  def grouphug_filter(s)
    # check if we like the location of the page
    loc = Utils.check_location(s, %r{http://(?:.*\.)?grouphug\.us})
    return unless loc
    # check if there are any conefssions
    confessions = get_confessions(s[:text])
    return if confessions.empty?
    title = s[:text].ircify_html_title
    # return the first confession
    return {
          :title => title,
          :content => confessions.first,
          :grouphug_confessions => confessions
    }
  end

end


plugin = GrouphugPlugin.new

plugin.default_auth('create', false)

plugin.map "grouphug [:num]",
  :thread => true, :action => :confess, :requirements => { :num => /\d+/ }
plugin.map "confess [:num]",
  :thread => true, :action => :confess, :requirements => { :num => /\d+/ }
plugin.map "confess *confession", :thread => true, :action => :post_confession, :auth_path => 'create'