-- IN-BAND REGISTRATION (XEP-0077)

-- library

require 'lm'

require 'iq'

require 'x_data'

-- public

iq_register = { }

function iq_register.parse ( query )

	local form  = { xmlns = 'jabber:iq:register', type = 'form' }

	local instructions = query:child ( 'instructions' )

	if instructions then

		form.instructions = instructions:value ()

	end

	-- XXX how it can be mapped to common form?

	--     and needs it be supplied?

	if query:child ( 'registered' ) then

		form.registered = true

	end

	local x = query:child ( 'x' )

	if x and x:attribute ( 'xmlns' ) == 'jabber:x:data' then

		form = x_data.parse ( x )

		local format = form.format

		form.format =

			function ( form, root )

				root.query = format ( form, { xmlns = 'jabber:iq:register' } )

				return root

			end

		return form

	end

	local field = query:children ()

	while field do

		local name  = field:name ()

		if name ~= 'instructions' and name ~= 'registered' then

			table.insert ( form, { type = 'text-single', var = name, value = field:value () or '' } )

		end

		field = field:next ()

	end

	form.format =

		function ( form, root )

			root.query = { xmlns = 'jabber:iq:register' }

			for index, field in ipairs ( form ) do

				root.query[field.var] = field.value

			end

			return root

		end

	return form

end

function iq_register.register ( conn, to, success, fail )

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

		{

			query = { xmlns = 'jabber:iq:register' },

		},

		function ( mess )

			local query = mess:child ( 'query' )

			if query and query:attribute ( 'xmlns' ) == 'jabber:iq:register' then

				success ( iq_register.parse ( query ),

					function ( form, success, fail )

						form.type = 'submit' -- XXX

						iq.send ( conn, to, 'set', form.format ( form, { } ), success , fail )

					end,

					function ( form, success, fail )

						success ()

					end )

			end

		end, fail )

end

function iq_register.unregister ( conn, to, success, fail )

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

		{

			query = { xmlns = 'jabber:iq:register',

				remove = { },

			},

		},

		function ( mess )

			success ()

		end,

		function ( mesg, mess )

			local query = mess:child ( 'query' )

			if query and query:attribute ( 'xmlns' ) == 'jabber:iq:register' then

				success ( iq_register.parse ( query ),

					function ( form, success, fail )

						form.type = 'submit' -- XXX

						iq.send ( conn, to, 'set', form.format ( form, { } ), success, fail )

					end,

					function ( form, success, fail )

						success ()

					end )

			else

				fail ( mesg )

			end

		end )

end

-- mcabber

main.command ( 'register',

	function ( args )

		local who

		if args and args ~= '' then

			who = args

		else

			who = main.full_jid ()

		end

		iq_register.register ( lm.connection.bless ( main.connection () ), who,

			function ( form, submit, reject )

				local id = #forms + 1

				forms[id] = {

					form   = form,

					submit =

						function ( form )

							submit ( form,

								function ()

									main.print_info ( who, 'Successfully registered' )

								end,

								function ( mesg )

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

								end )

						end,

					reject =

						function ( form )

							reject ( form,

								function ()

									main.print_info ( who, 'Registration cancelled' )

								end,

								function ( mesg )

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

								end )

						end,

				}

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

			end,

			function ( mesg )

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

			end )

	end, false, 'jid' )

main.command ( 'cancel',

	function ( args )

		local who

		if args and args ~= '' then

			who = args

		else

			who = main.full_jid ()

		end

		iq_register.unregister ( lm.connection.bless ( main.connection () ), who,

			function ( form, submit, reject )

				if not form then

					main.print_info ( who, 'Successfully unregistered' )

				else

					local id = #forms + 1

					forms[id] = {

						form   = form,

						submit =

							function ( form )

								submit ( form,

									function ()

										main.print_info ( who, 'Successfully unregistered' )

									end,

									function ( mesg )

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

									end )

							end,

						reject =

							function ( form )

								reject ( form,

									function ()

										main.print_info ( who, 'Unregistration cancelled' )

									end,

									function ( mesg )

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

									end )

							end,

					}

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

				end

			end,

			function ( mesg )

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

			end )

	end, false, 'jid' )

commands_help['register'] = "[jid]\n\nSends registration request to jid (or current buddy). You, probably, then will need to fill and send some form."

commands_help['cancel'] = "[jid]\n\nSends registration cancellation request to jid (or current buddy). May require a form filling."

-- vim: se ts=4: --