-- USER AVATAR (XEP-0084)
-- library
require 'lm'
require 'sha1'
require 'base64'
require 'pubsub'
require 'pep'
-- public
avatar = { }
-- TODO 'temporary disabling'
-- however I cannot see a method to enable it back without republishing avatar :(
-- this requires client to know, what is published on the server now
-- maybe we can do that by requesting item without payload from server
function avatar.publish ( conn, data, success, fail, height, width )
local id = sha1.digest ( data )
pep.publish ( conn, 'urn:xmpp:avatar:data',
{ id = id,
data = { xmlns = 'urn:xmpp:avatar:data',
base64.encode ( data ),
},
},
function ()
pep.publish ( conn, 'urn:xmpp:avatar:metadata',
{ id = id,
info = { bytes = data:len (), id = id, type = 'image/png', height = height, width = width },
}, success, fail )
end, fail )
end
function avatar.publish_url ( conn, url, iid, size, id, mtype, success, fail, height, width )
pep.publish ( conn, 'urn:xmpp:avatar:metadata',
{ id = iid,
info = { bytes = size, id = id, type = mtype, url = url, height = height, width = width },
}, success, fail )
end
function avatar.get ( conn, from, id, success, fail )
pubsub.retrieve ( conn, from, 'urn:xmpp:avatar:data',
function ( from, node, item )
local data = item:child ( 'data' )
if data then
success ( data:value () )
else
-- XXX
end
end, fail, nil, { id } )
end
-- mcabber
pep.handlers['urn:xmpp:avatar:metadata'] =
function ( from, node, data )
if not main.yesno ( main.option ( 'lua_pep_notification' ) ) then
return true
end
local item = data:children ()
while item do
main.print_info ( from, ('Avatar: %s [%s] %s bytes, %sx%s %s'):format (
item:attribute ( 'id' ),
item:attribute ( 'type' ),
item:attribute ( 'bytes' ),
item:attribute ( 'width' ) or '?',
item:attribute ( 'height' ) or '?',
item:attribute ( 'url' ) or '' ) )
item = item:next ()
end
end
main.command ( 'avatar',
function ( args )
local action = args[1]
if action == 'get' then
local who
if args.t then
who = args.t
else
who = main.current_buddy ()
end
avatar.get ( lm.connection.bless ( main.connection () ), who, args[2],
function ( data )
local h = io.open ( args[3], 'w' )
if h then
h:write ( data )
h:close ()
main.print_info ( who, 'Avatar saved to ' .. args[3] )
else
print ( 'Cannot open file for writing ' .. args[3] )
end
end,
function ( mesg )
main.print_info ( who, 'Error obtaining avatar: ' .. mesg )
end )
else
local file = action
if action == 'set' then
file = args[2]
end
local h = io.open ( file )
if h then
data = h:read ( '*a' )
h:close ()
avatar.publish ( lm.connection.bless ( main.connection () ), data,
function ()
print ( 'Avatar published' )
end,
function ( mesg )
print ( 'Avatar publishing error: ' .. mesg )
end )
else
print ( 'Cannot open file ' .. file )
end
end
end, true, 'file' )
commands_help['avatar'] = '[-t jid] get id filename | [set] filename\n\nGet action tries to get from server avatar with specified id and save it to \'filename\'.\nSet action publishes avatar to server. File must be a PNG image.'
main.add_feature ( 'urn:xmpp:avatar:metadata+notify' )
-- vim: se ts=4: --