-- 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: --