local lm  = require 'lm'

local ibb = require 'lm.ibb'

local mc_incoming_files = { }

ibb.handler (

	function ( from, accept, reject )

		local fid = #mc_incoming_files + 1

		mc_incoming_files[fid] = {

				from   = from,

				accept =

					function ( name )

						mc_incoming_files[fid].name = name

						accept (

							function ( data )

								local h = io.open ( mc_incoming_files[fid].name, 'w' )

								if not h then

									print ( 'Cannot open output file: ' .. mc_incoming_files[fid].name )

									return

								end

								h:write ( data )

								h:close ()

								print ( 'Stream ' .. fid .. ' successfully saved to ' .. mc_incoming_files[fid].name )

								mc_incoming_files[fid] = nil

							end,

							function ( mesg )

								main.print_info ( from, 'Stream error: ' .. mesg )

								mc_incoming_files[fid] = nil -- XXX

							end )

					end,

				reject =

					function ()

						reject ()

						print ( 'Stream ' .. fid .. ' rejected' )

						mc_incoming_files[fid] = nil

					end,

		}

		main.print_info ( from, from .. ' wants you to receive stream. Use /ibb [accept|reject] ' .. fid .. ' to process his request.' )

	end )

local ibb_sid = 0

main.command ( 'ibb',

	function ( args )

		local action = args[1]

		if action == 'send' then

			local who

			if args.t then

				who = args.t

			else

				who = main.full_jid ()

			end

			local fname = args[2]

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

			local sid   = ibb_sid

			ibb_sid     = ibb_sid + 1

			local stream = ibb.new ( conn, who, 4096, sid )

			stream:open (

				function ()

					main.print_info ( who, 'Stream accepted' )

					local noerr = true

					local h     = io.open ( fname, 'r' )

					if not h then

						print ( 'Cannot open file ' .. fname )

						return

					end

					local data = h:read ( '*a' ) -- In fact, it is better to read it in chunks :/

					h:close ()

					local fail =

						function ( mesg )

							noerr = false

							main.print_info ( who, 'Stream error: ' .. mesg )

						end

					stream:send ( data,

						function ( seq )

							main.print_info ( who, 'Delivery notification of chunk #' .. seq )

						end, fail )

					if noerr then

						stream:close (

							function ()

								main.print_info ( who, 'Stream finalizing notification' )

							end, fail )

					end

					if noerr then

						main.print_info ( who, 'Stream sent' )

					else

						main.print_info ( who, 'Stream error occured' )

					end

				end,

				function ( mesg )

					main.print_info ( who, 'Stream initiation error: ' .. mesg )

				end )

		elseif action == 'accept' then

			local id = tonumber(args[2])

			if mc_incoming_files[id] then

				mc_incoming_files[id].accept ( args[3] )

			end

		elseif action == 'reject' then

			local id = tonumber(args[2])

			if mc_incoming_files[id] then

				mc_incoming_files[id].reject ()

			end

		else

			local text = ''

			for sid, data in pairs ( mc_incoming_files ) do

				text = text .. '\n' ..  sid .. ': ' .. data.from .. ' --> ' .. ( data.name or '?' )

			end

			if text ~= '' then

				print ( 'List of incoming streams:' .. text )

			else

				print ( 'No streams' )

			end

		end

	end, true, { "send", "accept", "reject" } )

commands_help['ibb'] = "[[-t target_jid] send filename | accept sid filename | reject sid]\n\nRequests, accepts or rejects sending file via in-band bytestream."

local ibb_handler            = lm.message_handler.new ( ibb.iq_handler )

local ibb_handler_registered = false

hooks_d['hook-post-connect'].ibb =

	function ( args )

		lm.connection.bless( main.connection () ):handler ( ibb_handler, 'iq', 'normal' )

		ibb_handler_registered = true

		hooks_d['hook-post-connect'].ibb = nil

		hooks_d['hook-quit'].ibb =

			function ( args )

				if ibb_handler_registered then

					lm.connection.bless( main.connection () ):handler ( ibb_handler, 'iq' )

				end

			end

	end

main.add_feature ( 'http://jabber.org/protocol/ibb' )

-- vim: se ts=4: --