local lm   = require 'lm'

local evil = require 'lm.evil'

evil.handler (

	function ( mess )

		local evillevel = tonumber(main.option ( 'lua_evil_sensibility' ))

		local mtype, smtype = mess:type ()

		if evillevel > 1 then

			main.print_info ( mess:attribute ( 'from' ), 'Evil stanza of type ' .. mtype .. '.' .. smtype .. ' detected!' )

		elseif evillevel > 0 then

			print ( 'Tainted by evil stanza of type ' .. mtype .. '.' .. smtype .. ' from ' .. ( mess:attribute ( 'from' ) or '... unknown in black' ) )

		end

		return main.yesno ( main.option ( 'lua_filter_evil' ) )

	end )

local stat2xmpp = {

	free     = 'chat',

	online   = '',

	away     = 'away',

	dnd      = 'dnd',

	notavail = 'xa',

	offline  = 'unavailable',

}

-- TODO improve interface, check if we sending right thing for offline

main.command ( 'evil',

	function ( args )

		local connection = main.connection ()

		if not connection then

			print "You are not online!"

			return

		end

		local conn = lm.connection.bless ( connection )

		if args[1] == 'status' then

			local text = ''

			for i, mesg in ipairs ( args ) do

				if i > 2 then

					text = text .. ' ' .. mesg

				end

			end

			local st = stat2xmpp[args[2]]

			if not st then

				st = ''

			end

			evil.presence ( conn, args.t, st, text:sub ( 2 ) )

		else

			local text = ''

			if args[1] == 'message' then

				for i, mesg in ipairs ( args ) do

					if i > 1 then

						text = text .. ' ' .. mesg

					end

				end

			else

				for i, mesg in ipairs ( args ) do

					text = text .. ' ' .. mesg

				end

			end

			local mtype = 'chat'

			if args.k then

				mtype = args.k

			end

			local who

			if args.t then

				who = args.t

			else

				who = main.current_buddy ()

			end

			evil.message ( conn, who, mtype, text:sub ( 2 ) )

		end

	end, true )

commands_help['evil'] = "[-t jid] [status stat [message] | [-k message_type] [message] message]\n\nSends evil message or presence.\nmessage_type may be chat, normal, headline.\nNote, that for now it will not change mcabber's status."

local evil_handler = lm.message_handler.new ( evil.stanza_handler )

local evil_handler_registered = false

local evil_pc_handler =

	function ( args )

		local connection = main.connection ()

		if connection then

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

			lm.connection.bless( main.connection () ):handler ( evil_handler, 'message',  'first' )

			lm.connection.bless( main.connection () ):handler ( evil_handler, 'presence', 'first' )

			evil_handler_registered = true

		end

	end

local evil_pd_handler = 

	function ( args )

		if evil_handler_registered then

			local connection = main.connection ()

			if connection then

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

				lm.connection.bless( main.connection () ):handler ( evil_handler, 'message'  )

				lm.connection.bless( main.connection () ):handler ( evil_handler, 'presence' )

			end

			evil_handler_registered = false

		end

	end

main.hook ( 'hook-post-connect',   evil_pc_handler )

main.hook ( 'hook-pre-disconnect', evil_pd_handler )

main.hook ( 'hook-lua-start',

	function ( args )

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

		evil_pc_handler ()

	end )

main.hook ( 'hook-lua-quit',

	function ( args )

		main.del_feature ( 'http://jabber.org/protocol/evil' )

		evil_pd_handler ()

	end )

local char2xmpp = {

	f = 'chat',

	o = '',

	a = 'away',

	d = 'dnd',

	n = 'xa',

	_ = 'unavailable',

}

-- hack, but working ;)

main.hook ( 'hook-my-status-change',

	function ( args )

		if main.yesno ( main.option ( 'lua_evil_mode' ) ) then

			local connection = main.connection ()

			if connection then

				evil.presence ( lm.connection.bless ( connection ), nil, char2xmpp[args.new_status], args.message )

			end

		end

	end )

-- vim: se ts=4 sw=4: --