-- PUBLISH-SUBSCRIBE (BEP-0060)

-- library

require 'lm'

require 'iq'

require 'x_data'

-- public

pubsub = { }

-- SUBSCRIBER USE CASES

function pubsub.subscribe ( conn, to, node, success, fail )

	local jid = conn:jid():gsub ( '/.*', '' )

	iq.send ( conn, to, 'set',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub',

				subscribe = { node = node, jid = jid },

			},

		},

		function ( mess )

			local s = mess:path ( 'pubsub', 'subscription' )

			if s then

				success ( s:attribute ( 'subid' ) )

			else

				success ()

			end

		end,

		fail )

end

function pubsub.unsubscribe ( conn, to, node, success, fail )

	local jid = conn:jid():gsub ( '/.*', '' )

	iq.send ( conn, to, 'set',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub',

				unsubscribe = { node = node, jid = jid },

			},

		}, success, fail )

end

-- I found no servers with subscription options support thus it is not implemented.

-- untested :(

function pubsub.retrieve ( conn, to, node, success, fail, max, ids )

	local items = { node = node, max_items = max }

	if ids then

		items.item = { }

		for k, id in pairs ( ids ) do

			table.insert ( items.item, { id = id } )

		end

	end

	iq.send ( conn, to, 'get',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub',

				items = items,

			},

		},

		function ( mess )

			local items = mess:path ( 'pubsub', 'items' )

			if items then

				local from = mess:attribute ( 'from' )

				local node = items:attribute ( 'node' )

				local item = items:children ()

				while item do

					success ( from, node, item ) -- XXX use registered xmlns handlers for that?

					item = item:next ()

				end

			else

				-- XXX

			end

		end, fail )

end

-- OWNER USE CASES

-- node may be nil

function pubsub.create_node ( conn, to, node, success, fail )

	iq.send ( conn, to, 'set',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub',

				create = { node = node },

			},

		},

		function ( mess )

			if node then

				success ()

			else

				local create = mess:path ( 'pubsub', 'create' )

				if create then

					success ( create:attribute ( 'node' ) )

				else

					success ()

				end

			end

		end, fail )

end

function pubsub.delete_node ( conn, to, node, success, fail )

	iq.send ( conn, to, 'set',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

				delete = { node = node },

			},

		}, success, fail )

end

function pubsub.purge_node ( conn, to, node, success, fail )

	iq.send ( conn, to, 'set',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

				purge = { node = node },

			},

		}, success, fail )

end

function pubsub.configure_node ( conn, to, node, success, fail )

	iq.send ( conn, to, 'get',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

				configure = { node = node },

			},

		},

		function ( mess )

			local x = mess:path ( 'pubsub', 'configure', 'x' )

			if x then

				success ( x_data.parse ( x ),

					function ( form, success, fail )

						iq.send ( conn, to, 'set',

							{

								pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

									configure = form.format ( form, { node = node } ),

								},

							}, success, fail )

					end,

					function ( form, success, fail )

						iq.send ( conn, to, 'set',

							{

								pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

									configure = { node = node,

										x = { xmlns = 'jabber:x:data', type = 'cancel' }, -- FIXME

									},

								},

							}, success, fail )

					end )

			else

				fail ( mess:xml () ) -- XXX

			end

		end, fail )

end

function pubsub.list_subscriptions ( conn, to, node, success, fail )

	iq.send ( conn, to, 'get',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

				subscriptions = { node = node },

			},

		},

		function ( mess )

			local s = mess:path ( 'pubsub', 'subscriptions' )

			if s then

				local sub = s:children ()

				local ret = { }

				while sub do

					table.insert ( ret, { jid = sub:attribute ( 'jid' ), subscription = sub:attribute ( 'subscription' ), subid = sub:attribute ( 'subid' ) } )

					sub = sub:next ()

				end

				success ( ret )

			else

				fail ( mess:xml () ) -- XXX

			end

		end, fail )

end

function pubsub.modify_subscription ( conn, to, node, jid, state, success, fail, id )

	iq.send ( conn, to, 'set',

		{

			pubsub = { xmlns = 'http://jabber.org/protocol/pubsub#owner',

				subscriptions = { node = node,

					subscription = { jid = jid, subscription = state, id = id },

				},

			},

		}, success, fail )

end

-- mcabber

main.command ( 'node',

	function ( args )

		local who, action, node = args.t, args[1], args[2]

		local conn = lm.connection.bless ( main.connection () )

		if not who then

			who = main.current_buddy ()

		end

		if action == 'subscribe' then

			pubsub.subscribe ( conn, who, node,

				function ( id )

					if id then

						main.print_info ( who, 'Subscription succeeds with id ' .. id )

					else

						main.print_info ( who, 'Subscription successful' )

					end

				end,

				function ( mesg )

					main.print_info ( who, 'Subscription unsuccessful: ' .. mesg )

				end )

		elseif action == 'unsubscribe' then

			pubsub.unsubscribe ( conn, who, node,

				function ()

					main.print_info ( who, 'Unubscription successful' )

				end,

				function ( mesg )

					main.print_info ( who, 'Unsubscription unsuccessful: ' .. mesg )

				end )

		elseif action == 'retrieve' or action == 'items' or action == 'get' then

			pubsub.retrieve ( conn, who, node,

				function ( from, node, item )

					main.print_info ( who, 'Item from ' .. from .. ', node ' .. node .. ':\n' .. item:xml () ) 

				end,

				function ( mesg )

					main.print_info ( who, 'Retrieval failed: ' .. mesg )

				end, args.m )

		elseif action == 'create' or action == 'new' then

			pubsub.create_node ( conn, who, node,

				function ( node )

					if node then

						main.print_info ( who, 'Node ' .. node .. ' successfully created' )

					else

						main.print_info ( who, 'Node successfully created' )

					end

				end,

				function ( mesg )

					main.print_info ( who, 'Creation failed: ' .. mesg )

				end )

		elseif action == 'delete' or action == 'del' then

			pubsub.delete_node ( conn, who, node,

				function ()

					main.print_info ( who, 'Node deleted' )

				end,

				function ( mesg )

					main.print_info ( who, 'Node deletion failed: ' .. mesg )

				end )

		elseif action == 'purge' or action == 'del_items' then

			pubsub.purge_node ( conn, who, node,

				function ()

					main.print_info ( who, 'Node purged' )

				end,

				function ( mesg )

					main.print_info ( who, 'Node purge failed: ' .. mesg )

				end )

		elseif action:sub ( 1, 4 ) == 'conf' then

			pubsub.configure_node ( conn, who, node,

				function ( form, submit, reject )

					local id = #forms + 1

					forms[id] = {

						form = form,

						submit =

							function ( form )

								submit ( form,

									function ()

										main.print_info ( who, 'Node configuration completed' )

									end,

									function ( mesg )

										main.print_info ( who, 'Node configuration failed: ' .. mesg )

									end )

							end,

						reject =

							function ( form )

								reject ( form,

									function ()

										main.print_info ( who, 'Node configuration cancelled' )

									end,

									function ( mesg )

										main.print_info ( who, 'Node configuration cancellation failed: ' .. mesg )

									end )

							end,

					}

					print ( 'You have new form ' .. id )

				end,

				function ( mesg )

					main.print_info ( who, 'Node configuration failed: ' .. mesg )

				end )

		elseif action == 'subscriptions' or action == 'subscribers' then

			pubsub.list_subscriptions ( conn, who, node,

				function ( s )

					local text = ''

					for i, v in ipairs ( s ) do

						local subid = v.subid

						if subid then

							subid = '(id ' .. subid .. ')'

						else

							subid = ''

						end

						text = text .. ('\n- [%s] %s %s'):format ( v.subscription, v.jid, subid )

					end

					if text ~= '' then

						main.print_info ( who, 'Node subscriptions:' .. text )

					else

						main.print_info ( who, 'No subscriptions' )

					end

				end,

				function ( mesg )

					main.print_info ( who, 'Node subscriptions retrieval failed: ' .. mesg )

				end )

		elseif action == 'subscription' or action == 'modify' then -- XXX

			pubsub.modify_subscription ( conn, args.t or main.current_buddy (), node, args[3], args[4],

				function ()

					main.print_info ( who, 'Subscription modified' )

				end,

				function ( mesg )

					main.print_info ( who, 'Subsrciption modification failed: ' .. mesg )

				end, args[5] )

		else

			print ( 'Error: unknown action' )

		end

	end, true )

-- FIXME

commands_help['node']           = "[-t jid] [-m max_items] action [node_name]\n\nAction can be subscribe, unsubscribe, retrieve (items, get), create (new), delete (del), purge (del_items), configure (conf*), subscriptions (subscribers), subscription (modify?)"

--[[

commands_help['subscribe']      = "[-t jid] node_name\n\nSends pubsub subscription request to specified node of jid or current buddy."

commands_help['unsubscribe']    = "[-t jid] node_name\n\nSends pubsub unsubscription request to specified node of jid or current buddy."

commands_help['retrieve']       = "[-t jid] [-m max_items] node_name\n\nSends pubsub items retrieval request to specified node of jid or current buddy.\nNote, that we cannot know, how to deal with these itemss, so, raw xml will be printed as a result."

commands_help['create_node']    = "[-t jid] [node_name]\n\nSends pubsub node creation request to specified node of jid or current buddy. If node name is not specified, server can generate unique id for it, if supported."

commands_help['configure_node'] = "[-t jid] node_name\n\nSends pubsub node configuration request to specified node of jid or current buddy."

commands_help['delete_node']    = "[-t jid] node_name\n\nSends pubsub node deletion request to specified node of jid or current buddy."

commands_help['purge_node']     = "[-t jid] node_name\n\nSends pubsub node items purging request to specified node of jid or current buddy."

commands_help['subscriptions']  = "[-t jid] node_name\n\nSends pubsub subscription list request to specified node of jid or current buddy."

commands_help['subscription']   = "[-t jid] node_name subscriber_jid state [subscription_id]\n\nSends pubsub subscription modification request to change subscription state of 'subscriber_jid' to 'state'. Optional id is used when multiple subscriptions for one jid are available."

--]]

-- vim: se ts=4: --