-- PERSONAL EVENTING PROTOCOL (XEP-0163)

-- library

require 'lm'
require 'iq'
require 'mpd'

-- public

pep = {
	handlers = {},
}

function pep.publish ( conn, node, data, success, fail )
--	local bjid = conn:jid():gsub ( '/.*', '' )
	data.xmlns = 'http://jabber.org/protocol/' .. node
	iq.send ( conn, nil, 'set',
		{
			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub',
				publish = { node = 'http://jabber.org/protocol/' .. node,
					item = { -- id = "current",
						[node] = data,
					},
				},
			},
--[[
			configure = {
				x = {
					field = {{ type = "hidden", var = 'FORM_TYPE',
						value = { 'http://jabber.org/protocol/pubsub#node_config' },
					},{ var = "pubsub#access_model",
						value = { 'presence' },
					}},
				},
			},
--]]
		}, success, fail )
end

-- private

-- XXX in fact, it is not a pep handler, it is pubsub handler.
--     should it go there?
local pep_handler_registered       = false
local pep_incoming_message_handler = lm.message_handler.new (
	function ( conn, mess )
		local e = mess:child ( 'event' )
		if e and e:attribute ( 'xmlns' ) == 'http://jabber.org/protocol/pubsub#event' then
			local is = e:child ( 'items' )
			if is then
				local from = mess:attribute ( 'from' )
				local node = is:attribute ( 'node' )
				local item = is:child ( 'item' )
				if item then
					local handled = true -- XXX should we do this? well, if it becomes general pubsub handler - then no.
					local n       = item:children ()
					while n do
						local xmlns = n:attribute ( 'xmlns' )
						if pep.handlers[xmlns] then
							if not pep.handlers[xmlns] ( from, node, n ) then
								handled = false
							end
						else
							handled = false
						end
						n = n:next ()
					end
					return handled
				end
			end
		end
		return false
	end )

-- mcabber

local tune_enabled = false
local mpd_pub_song = { }

-- XXX in fact, we should separate library code from client code here, but it is silly.
--     seting up four registrators, four default handlers, four handler storage variables...
--     and, in fact, we cannot do much about complex datatypes of mood and activity.
pep.handlers['http://jabber.org/protocol/tune'] =
		function ( from, node, data )
			local self = false
			if from == lm.connection.bless ( main.connection () ):jid():gsub ( '/.*', '' ) then -- o_O
				self         = true
				mpd_pub_song = { }
			end
			local item = data:children ()
			local text = ''
			while item do
				local name  = item:name ()
				local value = item:value ()
				if self then
					mpd_pub_song[name] = { value or '' }
				end
				text = ("%s\n- %s: %s"):format ( text, item:name (), item:value () or '' )
				item = item:next ()
			end
			if main.yesno ( main.option ( 'lua_pep_notification' ) ) then
				if text ~= '' then
					text = 'Now listening to:' .. text
				else
					text = 'Now not listening to anything'
				end
			end
			main.print_info ( from, text )
			return true
		end
pep.handlers['http://jabber.org/protocol/mood'] =
		function ( from, node, data )
			if not main.yesno ( main.option ( 'lua_pep_notification' ) ) then
				return true
			end
			local item = data:children ()
			local mood, desc
			while item do
				if item:name () == 'text' then
					desc = item:value ()
				else
					mood = item:name ()
					-- here we can add child elements handling (by namespace)
				end
				item = item:next ()
			end
			if mood then
				main.print_info ( from, ("Buddy's mood now %s %s"):format ( mood, desc or '' ) )
			else
				main.print_info ( from, "Buddy hides his mood" )
			end
		end
pep.handlers['http://jabber.org/protocol/activity'] =
		function ( from, node, data )
			if not main.yesno ( main.option ( 'lua_pep_notification' ) ) then
				return true
			end
			local item = data:children ()
			local activity, desc
			while item do
				if item:name () == 'text' then
					desc = item:value ()
				else
					activity = item:name ()
					local subitem = item:children ()
					if subitem then
						-- here we can check for non-standard subactivity elements,
						-- add subactivity child elements handling
						activity = ("%s: %s"):format ( activity, subitem:name () )
					end
				end
				item = item:next ()
			end
			if activity then
				main.print_info ( from, ("Now %s %s"):format ( activity, desc or '' ) )
			else
				main.print_info ( from, "Buddy hides his activity" )
			end
			return true
		end
pep.handlers['http://jabber.org/protocol/geoloc'] =
		function ( from, node, data )
			if not main.yesno ( main.option ( 'lua_pep_notification' ) ) then
				return true
			end
			local item = data:children ()
			local text = ''
			while item do
				text = ("%s\n- %s: %s"):format ( text, item:name (), item:value () or '' )
				item = item:next ()
			end
			if text ~= '' then
				text = 'Now at:' .. text
			else
				text = 'Now in unknown location'
			end
			main.print_info ( from, text )
			return true
		end

local function mpd_getstatus ()
	local status = mpd.call_command { 'status' }
	if not tune_enabled or ( status.state ~= 'play' and status.state ~= 'pause' ) then
		for k, v in pairs ( mpd_pub_song ) do -- if there is anything published, publish nothing
			return { }
		end
		return nil
	end
	
	local song      = mpd.call_command { 'currentsong' }
	local dir, file = song.file:match ( '(.+)/(.-)' )
	-- populate according to currentsong fields: artist - artist, length - time, source - album, title - title, track - id, rating - ?, uri - ?
	local ret = {
		artist = { song.artist or 'Unknown' },
		length = { song.time                },
		source = { song.album  or dir       },
		title  = { song.title  or file      },
		track  = { song.id                  },
	}

	if not song.time or song.time == '0' then -- XXX
		ret.length = nil
	end

	local modified = false
	for k, v in pairs ( ret ) do
		if not mpd_pub_song[k] or mpd_pub_song[k][1] ~= v[1] then
			modified = true
			break
		end
	end
	if not modified then
		for k, v in pairs ( mpd_pub_song ) do
			if not ret[k] or ret[k][1] ~= v[1] then
				modified = true
				break
			end
		end
	end

	if modified then
		return ret
	else
		return nil
	end
end

local function dummy ()
end

local function mpd_callback ()
	local sdata = mpd_getstatus ()
	if sdata then
		pep.publish ( lm.connection.bless ( main.connection () ), 'tune', sdata, dummy, dummy )
	end
	if tune_enabled then
		return true
	else
		return false
	end
end

-- do not call it too fast, or you end up with many daemons at once
local function enable_tune ( yn )
	if yn == nil then
		yn = true
	end
	if yn then
		if not tune_enabled then
			main.timer ( 15, mpd_callback )
			tune_enabled = true
			-- update status
		end
	else
		if tune_enabled then
			tune_enabled = false
			-- update status
		end
	end
end

main.command ( 'tune',
	function ( args )
		local enable = main.yesno ( args )
		if enable == nil then
			if tune_enabled then
				print ( "Tune notifications enabled" )
			else
				print ( "Tune notifications disabled" )
			end
		else
			enable_tune ( enable )
		end
	end, false, 'yesno' )
main.command ( 'mood',
	function ( args )
		local data = { }
		local mood, text = args[1], args[2]
		if text then
			data.text  = { text }
		end
		if mood then
			data[mood] = { }
		end
		pep.publish ( lm.connection.bless ( main.connection () ), 'mood', data, dummy, dummy )
	end, true )
main.command ( 'activity',
	function ( args )
		local data = { }
		local activity, text = args[1], args[2]
		if text then
			data.text = { text }
		end
		local act, subact = activity:match ( "(.-)%-(.+)" )
		if not act then
			act = activity
		end
		if act ~= '' then
			data[act] = { }
			if subact then
				data[act][subact] = { }
			end
		end
		pep.publish ( lm.connection.bless ( main.connection () ), 'activity', data, dummy, dummy )
	end, true )
main.command ( 'location',
	function ( args )
		local data = { }
		for key, val in pairs ( args ) do
			data[key] = { val }
		end
		pep.publish ( lm.connection.bless ( main.connection () ), 'geoloc', data, dummy, dummy )
	end, true )

commands_help['tune']     = "[enable|disable|on|off|yes|no|true|false]\n\nEnables or disables publishing of notifications about playing music in your player (currently only mpd is supported)."
commands_help['mood']     = "[mood [message]]\n\nPublishes your mood.\nNote, that for now it does not checks for mood validity, so, see xep0107 for valid moods."
commands_help['activity'] = "[activity[-specific_activity] [text]]\n\nPublishes your activity.\nNote, that for now it does not checks for activity validity, so, see xep0108 for valid activity values."
commands_help['location'] = "[-key value [-key value ...]]\n\nPublishes your current geolocation.\nValid keys are accuracy, alt, area, bearing, building, country, datum, description, error, floor, lat, locality, lon, postalcode, region, room, speed, street, text, timestamp and uri, according to xep0080."

hooks_d['hook-post-connect'].pep =
	function ( args )
		lm.connection.bless( main.connection () ):handler ( pep_incoming_message_handler, 'message', 'normal' )
		pep_handler_registered = true
		if tune_enabled then
			mpd_callback ()
		end
		-- XXX may it confuse pairs()?
		hooks_d['hook-post-connect'].pep =
			function ( args )
				if tune_enabled then
					mpd_callback ()
				end
			end
		hooks_d['hook-quit'].pep =
			function ( args )
				if pep_handler_registered then
					lm.connection.bless( main.connection () ):handler ( pep_incoming_message_handler, 'message' )
					pep_handler_registered = false
				end
			end
	end

-- XXX this really should be initialized after connection establishment (?)
--     but as this thing is implemented by now, it will be cached by server,
--     and, thus, we will be unable to get notifications.
main.add_feature ( 'http://jabber.org/protocol/tune+notify' )
main.add_feature ( 'http://jabber.org/protocol/tune' )
main.add_feature ( 'http://jabber.org/protocol/mood+notify' )
main.add_feature ( 'http://jabber.org/protocol/mood' )
main.add_feature ( 'http://jabber.org/protocol/activity+notify' )
main.add_feature ( 'http://jabber.org/protocol/activity' )
main.add_feature ( 'http://jabber.org/protocol/geoloc+notify' )
main.add_feature ( 'http://jabber.org/protocol/geoloc' )

-- vim: se ts=4: --