#!/usr/bin/python
# this script is a total hack it works and I'll clean it up later
import sys,getopt, optparse, pdb, re
def raw_lines(file):
	try:
		raw_lines = open(file, "r")
	except Exception:
		print("Failed to open ",file,". Do you have the file name correct?")
		sys.exit(1)
	Rlines = raw_lines.readlines()
	return Rlines

def parse_file(file,file_name):
	cleanup = []
	for line in file:
               # match=re.search("\n", line) # the next few lines are notes and can be ignored
               # if match:
               #   line=line.replace("\n","")
		#for x in line:
		#	clean = filter(lambda y: y != '\n', x)
		clean = line.rstrip()
		cleanup.append(clean)
	try:
		header = cleanup.index('BSSID, First time seen, Last time seen, channel, Speed, Privacy, Cipher, Authentication, Power, # beacons, # IV, LAN IP, ID-length, ESSID, Key')
		stationStart = cleanup.index('Station MAC, First time seen, Last time seen, Power, # packets, BSSID, Probed ESSIDs')
		del cleanup[header]
	except Exception:
		print("You seem to have provided an improper input file"" '",file_name,"' ""Please make sure you are loading an airodump csv file and not a Pcap")
		sys.exit(1)
	Clients = cleanup[stationStart:] #splits off the clients into their own list
	stationStart = stationStart - 1 #ulgy hack to make sure the heading gets deleted from end of the APs List
	del cleanup[stationStart:]#removed all of the client info leaving only the info on available target AP's in ardump maybe I should create a new list for APs?
	lines = [cleanup,Clients]
	return lines
def join_write(data,name):
	file = open(name,'a')
	for line in data[0]:
		line=line.rstrip()
		if len(line)>1:
			file.write(line+'\n')
	for line in data [1]:
		if len(line)>1:
			file.write(line+'\n')
	file.close()
def showBanner():
	print("Airodump Joiner\nJoin Two Airodump CSV Files\n\n\t-i\tInput Files [ foo_name_1 foo_name_2 foo_name_3 .....] \n\t-o\tOutput File\n")

def file_pool(files):
	AP = []
	Clients = []
	for file in files:
		ret = raw_lines(file)
		ret = parse_file(ret,file)
		AP.extend(ret[1])
		Clients.extend(ret[0])
	lines = [AP,Clients]
	output = sort_file(lines)
	return output

def sort_file(input):
	AP = ['BSSID, First time seen, Last time seen, channel, Speed, Privacy, Cipher, Authentication, Power, # beacons, # IV, LAN IP, ID-length, ESSID, Key']
	Clients = ['\nStation MAC, First time seen, Last time seen, Power, # packets, BSSID, Probed ESSIDs']
	Clients.extend(input[0])
	AP.extend(input[1])
	output = [AP,Clients]
	return output

	

if  __name__ == "__main__":
	if len(sys.argv) <= 1:
		showBanner()
		sys.exit(1)

	parser = optparse.OptionParser("usage: %prog [options] arg1 arg2 arg3 .....")
	parser.add_option("-o", "--output",  dest="output",nargs=1, help="output file to write to")
	parser.add_option("-i", "--file", dest="filename", nargs=2 ,help="Input files to read data from requires at least two arguments")
	
	(options, args) = parser.parse_args()
	filenames = options.filename
	outfile = options.output
	if outfile == None:
		print("You must provide a file name to write out to. IE... -o foo.csv\n")
		showBanner()
		sys.exit(1)
	elif filenames == None:
		print("You must provide at least two file names to join. IE... -i foo1.csv foo2.csv\n")
		showBanner()
		sys.exit(1)
	for file_name in args:
		filenames += (file_name,)	
	return_var = file_pool(filenames)
	return_var = join_write(return_var,outfile)

	
