@ -9,9 +9,8 @@
/ disconnect — Disconnect from Telegram network
/ disconnect — Disconnect from Telegram network
/ logout — Disconnect from Telegram network and forget session
/ logout — Disconnect from Telegram network and forget session
/ sessions — Shows current active sessions (available for admins)
/ info — Show information and usage statistics of this instance (only for JIDs specified as administrators)
/ debug — Shows some debug information (available for admins)
/ restart — Restart this instance (only for JIDs specified as administrators)
/ restart — Reset Zhabogram (available for admins)
'
'
#############################
#############################
@ -19,21 +18,27 @@
#############################
#############################
## XMPP Transport Class #####
## XMPP Transport Class #####
#############################
#############################
include Jabber :: Discovery
include Jabber :: Dataforms
class XMPPComponent
class XMPPComponent
# init class and set logger #
# init class and set logger #
def initialize ( params )
def initialize ( params )
@@loglevel = params [ 'loglevel' ] || Logger :: DEBUG
@@loglevel = params [ 'loglevel' ] || Logger :: DEBUG
@logger = Logger . new ( STDOUT ) ; @logger . level = @@loglevel ; @logger . progname = '[XMPPComponent]'
@logger = Logger . new ( STDOUT ) ; @logger . level = @@loglevel ; @logger . progname = '[XMPPComponent]'
@config = { host : params [ " host " ] || 'localhost' , port : params [ " port " ] || 8899 , jid : params [ " jid " ] || 'tlgrm.localhost' , secret : params [ 'password' ] || '' , admins : params [ 'admins' ] || [ ] , debug : params [ 'debug' ] || false } # default config
@config = { host : params [ " host " ] || 'localhost' , port : params [ " port " ] || 8899 , jid : params [ " jid " ] || 'tlgrm.localhost' , secret : params [ 'password' ] || '' , admins : params [ 'admins' ] || [ ] , debug : params [ 'debug' ] } # default config
@sessions = { }
@sessions = { }
@presence_que = { }
@db = SQLite3 :: Database . new ( params [ 'db_path' ] || 'users.db' )
@db = SQLite3 :: Database . new ( params [ 'db_path' ] || 'users.db' )
@db . execute ( " CREATE TABLE IF NOT EXISTS users(jid varchar(256), login varchar(256), PRIMARY KEY(jid) ); " )
@db . execute ( " CREATE TABLE IF NOT EXISTS users(jid varchar(256), login varchar(256), PRIMARY KEY(jid) ); " )
@db . results_as_hash = true
@db . results_as_hash = true
self . load_db ( )
end
end
# load sessions from db #
# load sessions from db #
def load_db ( jid = nil ) # load
def load_db ( jid = nil )
@logger . info " Initializing database.. "
@logger . info " Initializing database.. "
query = ( jid . nil? ) ? " SELECT * FROM users " : " SELECT * FROM users where jid = '%s'; " % jid
query = ( jid . nil? ) ? " SELECT * FROM users " : " SELECT * FROM users where jid = '%s'; " % jid
@logger . debug ( query )
@logger . debug ( query )
@ -41,34 +46,52 @@ class XMPPComponent
end
end
# store session to db #
# store session to db #
def update_db ( jid , delete = false ) # write
def update_db ( jid , delete = false , register = false )
return if not @sessions . key? jid
login = ( not register and @sessions . key? jid ) ? @sessions [ jid ] . login . to_s : register
return if not login
@logger . info " Writing database [%s].. " % jid . to_s
@logger . info " Writing database [%s].. " % jid . to_s
query = ( delete ) ? " DELETE FROM users where jid = '%s'; " % jid . to_s : " INSERT OR REPLACE INTO users(jid, login) VALUES('%s', '%s'); " % [ jid . to_s , @sessions [ jid ] . login . to_s ]
query = ( delete ) ? " DELETE FROM users where jid = '%s'; " % jid . to_s : " INSERT OR REPLACE INTO users(jid, login) VALUES('%s', '%s'); " % [ jid . to_s , login ]
@logger . debug query
@logger . debug query
@db . execute ( query )
@db . execute ( query )
end
end
# connecting to XMPP server #
# connecting to XMPP server #
def connect ( ) # :jid => transport_jid, :host => xmpp_server, :port => xmpp_component_port, :secret => xmpp_component_secret
def connect ( ) # :jid => transport_jid, :host => xmpp_server, :port => xmpp_component_port, :secret => xmpp_component_secret
@logger . info " Connecting.. "
begin
begin
Jabber :: debug = @config [ :debug ]
Jabber :: debug = @config [ :debug ]
@@transport = Jabber :: Component . new ( @config [ :jid ] )
@@transport . connect ( @config [ :host ] , @config [ :port ] )
# component
@@transport . auth ( @config [ :secret ] )
@component = Jabber :: Component . new ( @config [ :jid ] )
@@transport . add_message_callback do | msg | msg . first_element_text ( 'body' ) ? self . message_handler ( msg ) : nil end
@component . connect ( @config [ :host ] , @config [ :port ] )
@@transport . add_presence_callback do | presence | self . presence_handler ( presence ) end
@component . auth ( @config [ :secret ] )
@@transport . add_iq_callback do | iq | self . iq_handler ( iq ) end
@component . add_message_callback do | msg | msg . first_element_text ( 'body' ) ? self . message_handler ( msg ) : nil end
@@transport . on_exception do | exception , stream , state | self . survive ( exception , stream , state ) end
@component . add_presence_callback do | presence | self . presence_handler ( presence ) end
@logger . info " Connection established "
@component . add_iq_callback do | iq | self . iq_handler ( iq ) end
self . load_db ( )
@component . on_exception do | exception , stream , state | self . survive ( exception , stream , state ) end
@logger . info 'Found %s sessions in database.' % @sessions . count
@logger . info " Connection to XMPP server established! "
# disco
@disco = Jabber :: Discovery :: Responder . new ( @component )
@disco . identities = [ Identity . new ( 'gateway' , 'Telegram Gateway' , 'telegram' ) ]
@disco . add_features ( [ 'http://jabber.org/protocol/disco' , 'jabber:iq:register' ] )
# janbber::iq::register
@iq_register = Jabber :: Register :: Responder . new ( @component )
@iq_register . instructions = 'Please enter your Telegram login'
@iq_register . add_field ( :login , true ) do | jid , login | self . process_command ( jid , '/login %s' % login ) end
# jabber::iq::gateway
@iq_gateway = Jabber :: Gateway :: Responder . new ( @component ) do | iq , query | ( @sessions . key? iq . from . bare . to_s and @sessions [ iq . from . bare . to_s ] . online? ) ? @sessions [ iq . from . bare . to_s ] . resolve_username ( query ) . to_s + '@' + @component . jid . to_s : '' end
@iq_gateway . description = " Specify @username / ID / https://t.me/link "
@iq_gateway . prompt = " Telegram contact "
@logger . info 'Loaded %s sessions from database.' % @sessions . count
@sessions . each do | jid , session | self . presence ( jid , nil , :subscribe ) end
@sessions . each do | jid , session | self . presence ( jid , nil , :subscribe ) end
Thread . stop ( )
Thread . new { while @component . is_connected? do @presence_que . each_value { | p | @component . send ( p ) } ; @presence_que . clear ; sleep ( 60 ) ; end } # presence updater thread
Thread . stop ( )
rescue Interrupt
rescue Interrupt
@logger . error 'Interrupted!'
@logger . error 'Interrupted!'
@@transport . on_exception do | exception , | end
@componen t. on_exception do | exception , | end
self . disconnect ( )
self . disconnect ( )
return - 11
return - 11
rescue Exception = > e
rescue Exception = > e
@ -80,35 +103,35 @@ class XMPPComponent
# transport shutdown #
# transport shutdown #
def disconnect ( )
def disconnect ( )
@logger . info " Closing all connections..."
@logger . info " Closing connections..."
@sessions . each do | jid , session | @sessions [ jid ] . disconnect ( ) end
@sessions . each do | jid , session | @sessions [ jid ] . disconnect ( ) end
@@transpor t. close ( )
@componen t. close ( )
end
end
# vse umrut a ya ostanus'... #
# vse umrut a ya ostanus'... #
def survive ( exception , stream , state )
def survive ( exception , stream , state )
@logger . error " Stream error on :%s (%s) " % [ state . to_s , exception . to_s ]
@logger . error " Stream error on :%s (%s) " % [ state . to_s , exception . to_s ]
@logger . info " Trying to re ssurect XMPP stream.."
@logger . info " Trying to re vive stream.."
self . connect ( )
self . connect ( )
end
end
# message to users #
# message to users #
def message ( to , from = nil , body = '' )
def message ( to , from = nil , body = '' )
@logger . info " Sending message from <%s> to <%s> " % [ from || @@transpor t. jid , to ]
@logger . info " Sending message from <%s> to <%s> " % [ from || @componen t. jid , to ]
msg = Jabber :: Message . new
msg = Jabber :: Message . new
msg . from = ( from ) ? " %s@%s " % [ from , @@transport . jid . to_s ] : @@transpor t. jid
msg . from = ( from ) ? " %s@%s " % [ from , @component . jid . to_s ] : @componen t. jid
msg . to = to
msg . to = to
msg . body = body
msg . body = body
msg . type = :chat
msg . type = :chat
@logger . debug msg . to_s
@logger . debug msg . to_s
@@transpor t. send ( msg )
@componen t. send ( msg )
end
end
# presence update #
# presence update #
def presence ( to , from = nil , type = nil , show = nil , status = nil , nickname = nil , photo = nil )
def presence ( to , from = nil , type = nil , show = nil , status = nil , nickname = nil , photo = nil , immediately = true )
@logger . debug " Presence update request from %s .." % from . to_s
@logger . debug " Presence update request from %s (immed = %s) .." % [ from . to_s , immediately ]
req = Jabber :: Presence . new ( )
req = Jabber :: Presence . new ( )
req . from = from . nil? ? @@transpor t. jid : " %s@%s " % [ from , @@transpor t. jid ] # presence <from>
req . from = from . nil? ? @componen t. jid : " %s@%s " % [ from , @componen t. jid ] # presence <from>
req . to = to # presence <to>
req . to = to # presence <to>
req . type = type unless type . nil? # pres. type
req . type = type unless type . nil? # pres. type
req . show = show unless show . nil? # presence <show>
req . show = show unless show . nil? # presence <show>
@ -116,7 +139,7 @@ class XMPPComponent
req . add_element ( 'nick' , { 'xmlns' = > 'http://jabber.org/protocol/nick' } ) . add_text ( nickname ) unless nickname . nil? # nickname
req . add_element ( 'nick' , { 'xmlns' = > 'http://jabber.org/protocol/nick' } ) . add_text ( nickname ) unless nickname . nil? # nickname
req . add_element ( 'x' , { 'xmlns' = > 'vcard-temp:x:update' } ) . add_element ( " photo " ) . add_text ( photo ) unless photo . nil? # nickname
req . add_element ( 'x' , { 'xmlns' = > 'vcard-temp:x:update' } ) . add_element ( " photo " ) . add_text ( photo ) unless photo . nil? # nickname
@logger . debug req . to_s
@logger . debug req . to_s
@@transpor t. send ( req )
( immediately ) ? @componen t. send ( req ) : @presence_que . store ( to , req )
end
end
# request timezone information #
# request timezone information #
@ -125,11 +148,11 @@ class XMPPComponent
iq = Jabber :: Iq . new
iq = Jabber :: Iq . new
iq . type = :get
iq . type = :get
iq . to = jid
iq . to = jid
iq . from = @@transpor t. jid
iq . from = @componen t. jid
iq . id = 'time_req_1'
iq . id = 'time_req_1'
iq . add_element ( " time " , { " xmlns " = > " urn:xmpp:time " } )
iq . add_element ( " time " , { " xmlns " = > " urn:xmpp:time " } )
@logger . debug iq . to_s
@logger . debug iq . to_s
@@transpor t. send ( iq )
@componen t. send ( iq )
end
end
#############################
#############################
@ -141,7 +164,7 @@ class XMPPComponent
return if msg . type == :error
return if msg . type == :error
@logger . info 'Received message from <%s> to <%s>' % [ msg . from . to_s , msg . to . to_s ]
@logger . info 'Received message from <%s> to <%s>' % [ msg . from . to_s , msg . to . to_s ]
@logger . debug msg . to_s
@logger . debug msg . to_s
if msg . to == @@transpor t. jid then self . process_command ( msg . from , msg . first_element_text ( 'body' ) ) ; return ; end # treat message as internal command if received as transport jid
if msg . to == @componen t. jid then self . process_command ( msg . from , msg . first_element_text ( 'body' ) ) ; return ; end # treat message as internal command if received as transport jid
if @sessions . key? msg . from . bare . to_s then self . request_tz ( msg . from ) if not @sessions [ msg . from . bare . to_s ] . tz_set? ; @sessions [ msg . from . bare . to_s ] . process_outgoing_msg ( msg . to . to_s . split ( '@' ) [ 0 ] . to_i , msg . first_element_text ( 'body' ) ) ; return ; end #if @sessions.key? msg.from.bare.to_s and @sessions[msg.from.bare.to_s].online? # queue message for processing session is active for jid from
if @sessions . key? msg . from . bare . to_s then self . request_tz ( msg . from ) if not @sessions [ msg . from . bare . to_s ] . tz_set? ; @sessions [ msg . from . bare . to_s ] . process_outgoing_msg ( msg . to . to_s . split ( '@' ) [ 0 ] . to_i , msg . first_element_text ( 'body' ) ) ; return ; end #if @sessions.key? msg.from.bare.to_s and @sessions[msg.from.bare.to_s].online? # queue message for processing session is active for jid from
end
end
@ -149,9 +172,9 @@ class XMPPComponent
def presence_handler ( prsnc )
def presence_handler ( prsnc )
@logger . debug " Received presence :%s from <%s> to <%s> " % [ prsnc . type . to_s , prsnc . from . to_s , prsnc . to . to_s ]
@logger . debug " Received presence :%s from <%s> to <%s> " % [ prsnc . type . to_s , prsnc . from . to_s , prsnc . to . to_s ]
@logger . debug ( prsnc . to_s )
@logger . debug ( prsnc . to_s )
if prsnc . type == :subscribe then reply = prsnc . answer ( false ) ; reply . type = :subscribed ; @@transpor t. send ( reply ) ; end # send "subscribed" reply to "subscribe" presence
if prsnc . type == :subscribe then reply = prsnc . answer ( false ) ; reply . type = :subscribed ; @componen t. send ( reply ) ; end # send "subscribed" reply to "subscribe" presence
if prsnc . to == @@transpor t. jid and @sessions . key? prsnc . from . bare . to_s and prsnc . type == :unavailable then @sessions [ prsnc . from . bare . to_s ] . disconnect ( ) ; return ; end # go offline when received offline presence from jabber user
if prsnc . to == @componen t. jid and @sessions . key? prsnc . from . bare . to_s and prsnc . type == :unavailable then @sessions [ prsnc . from . bare . to_s ] . disconnect ( ) ; return ; end # go offline when received offline presence from jabber user
if prsnc . to == @@transpor t. jid and @sessions . key? prsnc . from . bare . to_s then self . request_tz ( prsnc . from ) ; @sessions [ prsnc . from . bare . to_s ] . connect ( ) ; return ; end # connect if we have session
if prsnc . to == @componen t. jid and @sessions . key? prsnc . from . bare . to_s then self . request_tz ( prsnc . from ) ; @sessions [ prsnc . from . bare . to_s ] . connect ( ) ; return ; end # connect if we have session
end
end
# new iq (vcard/tz) request to XMPP component #
# new iq (vcard/tz) request to XMPP component #
@ -177,7 +200,7 @@ class XMPPComponent
reply . type = :result
reply . type = :result
reply . elements [ " vCard " ] = vcard
reply . elements [ " vCard " ] = vcard
@logger . debug reply . to_s
@logger . debug reply . to_s
@@transpor t. send ( reply )
@componen t. send ( reply )
# time response #
# time response #
elsif iq . type == :result and iq . elements [ " time " ] and @sessions . key? iq . from . bare . to_s then
elsif iq . type == :result and iq . elements [ " time " ] and @sessions . key? iq . from . bare . to_s then
@logger . debug " Timezone response from <%s> " % iq . from . to_s
@logger . debug " Timezone response from <%s> " % iq . from . to_s
@ -188,7 +211,7 @@ class XMPPComponent
reply = iq . answer
reply = iq . answer
reply . type = :error
reply . type = :error
end
end
@@transpor t. send ( reply )
@componen t. send ( reply )
end
end
#############################
#############################
@ -213,34 +236,24 @@ class XMPPComponent
@sessions [ from . bare . to_s ] . disconnect ( true ) if @sessions . key? from . bare . to_s
@sessions [ from . bare . to_s ] . disconnect ( true ) if @sessions . key? from . bare . to_s
self . update_db ( from . bare . to_s , true )
self . update_db ( from . bare . to_s , true )
@sessions . delete ( from . bare . to_s )
@sessions . delete ( from . bare . to_s )
when '/ debug ' # show some debug information
when '/ info ' # show some debug information
return if not @config [ :admins ] . include? from . bare . to_s
return if not @config [ :admins ] . include? from . bare . to_s
GC . start
response = " Information about this instance: \n \n "
dump = ( defined? Memprof2 ) ? " /tmp/zhabogram.%s.dump " % Time . now . to_i : nil
Memprof2 . report ( out : dump ) if dump
response = " Debug information: \n \n "
response += " Running from: %s \n " % ` ps -p #{ $$ } -o lstart ` . lines . last . strip
response += " Running from: %s \n " % ` ps -p #{ $$ } -o lstart ` . lines . last . strip
response += " Sessions: %d online | %d total \n " % [ @sessions . inject ( 0 ) { | cnt , ( jid , sess ) | cnt = ( sess . online? ) ? cnt + 1 : cnt } , @sessions . count ]
response += " System memory used: %d KB \n " % ` ps -o rss -p #{ $$ } ` . lines . last . strip . to_i
response += " System memory used: %d KB \n " % ` ps -o rss -p #{ $$ } ` . lines . last . strip . to_i
response += " Objects memory allocated: %d bytes \n " % ` cut -d' ' -f1 #{ dump } ` . lines . map ( & :to_i ) . reduce ( 0 , :+ ) if dump
response += " \n \n Sessions: %d online | %d total \n " % [ @sessions . inject ( 0 ) { | cnt , ( jid , sess ) | cnt = ( sess . online? ) ? cnt + 1 : cnt } , @sessions . count ]
response += " \n Detailed memory info saved to %s \n " % dump if dump
@sessions . each do | jid , session | response += " JID: %s | Login: %s | Status: %s (%s) | %s \n " % [ jid , session . login , ( session . online == true ) ? 'Online' : 'Offline' , session . auth_state , ( session . me ) ? session . format_username ( session . me . id ) : 'Unknown' ] end
response += " \n Run this transport with --profiler (depends on gem memprof2) to get detailed memory infnormation. \n " if not dump
self . message ( from . bare , nil , response )
when '/sessions' # show active sessions
return if not @config [ :admins ] . include? from . bare . to_s
response = " Active sessions list: \n \n "
@sessions . each do | jid , session | response += " JID: %s | Login: %s | Status: %s (%s) | Telegram profile: %s \n " % [ jid , session . login , ( session . online == true ) ? 'Online' : 'Offline' , session . auth_state , ( session . me ) ? session . format_username ( session . me . id ) : 'Unknown' ] end
self . message ( from . bare , nil , response )
self . message ( from . bare , nil , response )
when '/restart' # reset transport
when '/restart' # reset transport
return if not @config [ :admins ] . include? from . bare . to_s
return if not @config [ :admins ] . include? from . bare . to_s
self . message ( from . bare , nil , 'Trying to restart all active sessions and reconnect to XMPP server..' )
self . message ( from . bare , nil , 'Trying to restart all active sessions and reconnect to XMPP server..' )
sleep ( 0. 5 )
sleep ( 1 )
Process . kill ( " INT " , Process . pid )
Process . kill ( " INT " , Process . pid )
else # unknown command -- display help #
else # unknown command -- display help #
self . message ( from . bare , nil , :: HELP_MESSAGE )
self . message ( from . bare , nil , :: HELP_MESSAGE )
end
end
return
return true
end
end
end
end