-- XMPP PING (XEP-0199)

-- library

require 'lm'

require 'iq'

-- public

ping = { }

function ping.send ( conn, to, success, fail )

	iq.send ( conn, to, 'get', success, fail )

end

-- private

local ping_incoming_iq_handler = lm.message_handler.new (

	function ( conn, mess )

		local mtype, smtype = mess:type ()

		if smtype == 'get' then

			local p = mess:child ( 'ping' )

			if p and p:attribute ( 'xmlns' ) == 'urn:xmpp:ping' then

				conn:send ( lm.message.create { mtype = 'iq-result', to = mess:attribute ( 'from' ), id = mess:attribute ( 'id' ) } )

				return true

			end

		end

		return false

	end )

-- mcabber

main.command ( 'ping',

	function ( args )

		local who

		if args[1] then

			who = args[1]

		else

			who = main.full_jid ()

		end

		local time = os.time ()

		ping.send ( lm.connection.bless ( main.connection () ), who,

			function ()

				main.print_info ( ('Pong: %d seconds'):format ( os.time () - time ) )

			end,

			function ( mesg )

				main.print_info ( 'Ping failed: ' .. mesg )

			end )

	end, true, 'jid' )

local ping_handler_registered = false

hooks_d['hook-post-connect'].ping =

	function ( args )

		lm.connection.bless( main.connection () ):handler ( ping_incoming_iq_handler, 'iq', 'normal' )

		ping_handler_registered = true

		hooks_d['hook-post-connect'].ping = nil

		hooks_d['hook-quit'].ping =

			function ( args )

				if ping_handler_registered then

					lm.connection.bless( main.connection () ):handler ( ping_incoming_iq_handler, 'iq' )

				end

			end

	end

main.add_feature ( 'urn:xmpp:ping' )

-- vim: se ts=4: --