require 'rex'
require 'rex/logging'
require 'msf/base'

module Msf

###
#
# This module provides an initialization interface for logging.
#
###
class Logging

	@@initialized = false
	@@session_logging = false

	#
	# Initialize logging.
	#
	def self.init
		if (! @@initialized)
			@@initialized = true

			f = Rex::Logging::Sinks::Flatfile.new(
				Msf::Config.log_directory + File::SEPARATOR + "framework.log")

			# Register each known log source
			[
				Rex::LogSource,
				Msf::LogSource,
				'base',
			].each { |src|
				register_log_source(src, f)
			}
		end
	end

	#
	# Enables a log source.
	#
	def self.enable_log_source(src, level = 0)
		if (log_source_registered?(src) == false)
			f = Rex::Logging::Sinks::Flatfile.new(
				Msf::Config.log_directory + File::SEPARATOR + "#{src}.log")
	
			register_log_source(src, f, level)
		end
	end

	#
	# Stops logging for a given log source.
	#
	def self.disable_log_source(src)
		deregister_log_source(src)
	end

	#
	# Sets whether or not session logging is to be enabled.
	#
	def self.enable_session_logging(tf)
		@@session_logging = tf
	end

	#
	# Returns whether or not session logging is enabled.
	#
	def self.session_logging_enabled?
		@@session_logging || false
	end

	#
	# Starts logging for a given session.
	#
	def self.start_session_log(session)
		if (log_source_registered?(session.log_source) == false)
			f = Rex::Logging::Sinks::Flatfile.new(
			Msf::Config.session_log_directory + File::SEPARATOR + "#{session.log_file_name}.log")
	
			register_log_source(session.log_source, f)
			
			rlog("\n[*] Logging started: #{Time.now}\n\n", session.log_source)
		end
	end

	#
	# Stops logging for a given session.
	#
	def self.stop_session_log(session)
		rlog("\n[*] Logging stopped: #{Time.now}\n\n", session.log_source)

		deregister_log_source(session.log_source)
	end

end

end