local lm     = require 'lm'
local pubsub = require 'lm.pubsub'
local avatar = require 'lm.avatar'

pubsub.handler ( 'urn:xmpp:avatar:metadata',
	function ( from, node, data, id )
		if not main.yesno ( main.option ( 'lua_pep_notification' ) ) then
			return true
		end
		local item = data:child ()
		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: --