root/trunk/jabberbot.rb

require 'RMagick'
require 'cgi'
require 'time'
require 'digest/sha1'
require 'xmpp4r'
require 'xmpp4r/vcard'
require 'xmpp4r/roster'
require 'rubygems'

$: << 'lib'
require 'blog'
Jabber::debug=true

def check_vcard(cl, jid, force_load=false)
  puts "check_vcard(..., #{jid.inspect}, #{force_load.inspect})"
  js = JabberUser.select(:jid=>jid.to_s)
  js.first.delete if js.first and force_load

  if js.first.nil? or force_load
    begin
      vcard = Jabber::Vcard::Helper::get(cl, jid)
      j = JabberUser.new
      j.jid = jid.to_s
      j.jid_sha = Digest::SHA1.new(jid.to_s).hexdigest
      j.nickname = vcard['NICKNAME'] || vcard['FN'] || jid.node.to_s
      j.url = vcard['URL']
      j.photo_mime = vcard['PHOTO/TYPE']

      begin
        binval = Base64::decode64(vcard['PHOTO/BINVAL'].to_s)
        puts "Size of photo of #{jid}: #{binval.size}"
        photo = Magick::Image.from_blob(binval)[0]
        w = photo.columns
        w = 64 if w > 64
        h = photo.rows
        h = 64 if h > 64
        photo.crop_resized!(w, h)
        j.photo = Base64::encode64(photo.to_blob)
        j.photo_w = photo.columns
        j.photo_h = photo.rows
      rescue Exception => e
        puts "Error in PHOTO: #{e.class}: #{e.to_s}\n#{e.backtrace.join("\n")}"
        j.photo = nil
        j.photo_w = 0
        j.photo_h = 0
      end

      j.write

      "Hello #{j.nickname}"
    rescue Exception => e
      e.to_s + " at " + e.backtrace.first
    end
  end
end

blog = Blog.new('config.yaml')
#Jabber::debug = true
cl = Jabber::Client.new(blog.jabberconf['jid'])
cl.connect
cl.auth(blog.jabberconf['password'])
cl.on_exception do
  exit
end
roster = Jabber::Roster::Helper.new(cl)
roster.add_subscription_request_callback { |item,presence|
        roster.accept_subscription(presence.from)
}

cl.add_message_callback do |msg|
  if msg.type != :error
    Thread.new do
      cmd, arg = msg.body.to_s.split(/ /, 2)

      reply = msg.answer(false)
      reply.type = msg.type
      reply.body = case cmd
        when 'refresh-vcard'
          check_vcard(cl, msg.from.strip, true)
        when 'comment'
          url, text = arg.split(/[ \n]/, 2)
          param = {}
          CGI::parse(url.split('?', 2).last.to_s).each { |k,v| param[k] = v.first }
          entry = Entry.by_id(param['entry'].to_s)
          if entry
            comment = Comment.new
            comment.entry_id = entry.entry_id
            comment.date = Time.now.xmlschema
            comment.address = "xmpp:#{msg.from.strip}"
            comment.text = text.to_s.strip
            comment.write

            'Thanks very much!'
          else
            puts "No entry_id #{param['entry'].inspect}"
            "Error: no such entry."
          end
        when 'help'
          "Commands available: help, refresh-vcard"
        else
          "Try \"help\""
      end

      check_vcard(cl, msg.from.strip)
      cl.send(reply)
    end
  end
end

cl.send(Jabber::Presence.new(nil, 'Astroblog2 in public beta now.'))

comment_addresses = []
Comment.select.each { |c|
  comment_addresses << c.address.sub(/^xmpp:/, '').to_s
}
comment_addresses.uniq!
comment_addresses.each { |a|
  check_vcard(cl, a, true)
}

loop do
  cl.send " \t\n"
  sleep 60
end