#!/usr/local/bin/ruby -w
require 'find'

@results = Array.new

# Iterate through the child directories & call the parse file method
def scan_dirs
	root = "/var/qmail/queue/mess"
	Find.find(root) do |file|
		parse_file(file)
	end
	# Sort on the second element in our array
	@results.sort! {|x, y| y[1] <=> x[1]}
	print_results
end

# Parse each file for the information we want
def parse_file(path)
	
	file =	path[(path.length-7), path.length]
	sourceip = ""
	email = ""
	subject = ""
	line_no = 0

	File.open(path, 'r').each do |line|
		
		line = line.strip # Remove any \n\r nil, etc
		line_no += 1
		
		if line_no == 1
			if line.match("invoked for bounce")
				# Internal Bounce Msg
				sourceip = "SMTP"
			end
		end
		
		if (line_no == 2 and sourceip.empty?)
			if line.match("webmail.internet.net")
				sourceip = "Webmail"
			else
				sourceip = line.scan(/\b(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\b/)
				if sourceip.empty?
					sourceip = "No Source IP**"
				end
			end
		end

		if (line.match("SquirrelMail") and sourceip == "Webmail") or
			 (line.match("From:") and sourceip != "Webmail")
			 if email.empty?
			 	  email = get_email(line)
			 end
		end

		if line.match("Subject:") and subject.empty? 
			subject = truncate(line,50)
		end

		if line_no == 20 #Nothing more we want to read in the file
		@results << ["#{file}", "#{sourceip}", "#{email}", "#{subject}"]
			line_no = 0
			return
		end
	end
end

# Truncate subject line
def truncate(string, width)
  if string.length <= width
    string
  else
    string[0, width-3] + "..."
  end
end

# Print out results
def print_results
	print "\e[2J\e[f"
	
	print "Mess#".ljust(10," ")
	print "Source".ljust(18," ")
	print "Email Addrress".ljust(30, " ")
	print "Subject".ljust(50, " ")
	1.times { print "\n" }
	111.times { print "-" }
	1.times { print "\n" }
	
	@results.each do |line|
		print line[0].ljust(10," ")
		print line[1].ljust(18," ")
		print line[2].ljust(30, " ")
		print line[3].ljust(50, " ")
	
		1.times { print "\n" }
	end
end

# Get email address from line/string
def get_email(line_to_parse)
	# Pull the email address from the line
	line_to_parse.scan(/\b[A-Z0-9._%+-]+@[A-Z0-9.-]+\.[A-Z]{2,4}\b/i).flatten
end

# Ok, begin our scan
scan_dirs
exit