This repository has been archived on 2024-05-28. You can view files and clone it, but cannot push or open issues or pull requests.
zhabogram/inc/xmppregister.rb

71 lines
2 KiB
Ruby
Raw Normal View History

2019-05-22 10:41:43 +00:00
module Jabber
module Register
NS_REGISTER = 'jabber:iq:register'
class Responder
attr_accessor :instructions
def initialize(stream)
@stream = stream
@fields = []
@registered_callbacks = []
@stream.add_iq_callback() do |iq|
if iq.query.kind_of?(IqQueryRegister) then
if iq.type == :get then # Registration request
answer = iq.answer(false)
answer.type = :result
query = answer.add(IqQueryRegister.new)
query.add(Field.new(:instructions, @instructions)) unless instructions.nil?
@fields.each do |field| query.add(Field.new(field[0])) end
@stream.send(answer)
elsif iq.type == :set then # Registration response
iq.query.each do |field|
validator = @fields.assoc(field.name.to_sym)[2]
if !validator.call(iq.from, field.text) then
puts "- - n0t acceptable here - -"
answer = iq.answer(true)
answer.type = :error
answer.add(Jabber::ErrorResponse.new('not-acceptable'))
@stream.send(answer)
end
end
# let them know that all looks good!
answer = iq.answer(false)
answer.type = :result
@stream.send(answer)
# Fire off callbacks
@registered_callbacks.each do |cb|
cb.call(iq.from)
end
end
end
end
end
def add_field(name, required, &validator)
@fields << [ name, required, validator ]
end
def add_registered_callback(&cb)
@registered_callbacks << cb
end
end
end
class IqQueryRegister < IqQuery
name_xmlns 'query', Jabber::Register::NS_REGISTER
end
class Field < REXML::Element
def initialize(name, value=nil)
super(name.to_s)
self.text = value
end
end
end