import untidy
import sys

OUTPUT_DIRECTORY = "fuzzedXML"

def create_xml_files(xml_sample_file):
	xmlInput = open(xml_sample_file, "r").read()
	oXMLFuzzer = untidy.xmlFuzzer()
	oXMLFuzzer.setRepetitions([3,30,60])
	fuzzedXMLs = oXMLFuzzer.fuzz(xmlInput)

	counter = 0
	for fuzzedXML in fuzzedXMLs:
			xml_output_file = xml_sample_file + str(counter)
			counter += 1	
			full_path = OUTPUT_DIRECTORY + "/" + xml_output_file
			open(full_path, "w").write(fuzzedXML)

if __name__ == '__main__':
		try:
				input_xml_file = sys.argv[1]
		except IndexError:
				print "Enter sample XML file"
				sys.exit(-1)

		print "Mixing XML file %s...." % input_xml_file
		create_xml_files(input_xml_file)