import tempfile
import zipfile
import shutil
import os
import sys
from subprocess import Popen
 
zipfname=sys.argv[1]
 
tempdir = tempfile.mkdtemp()
try:
	tempname = os.path.join(tempdir, 'new.zip')
	with zipfile.ZipFile(zipfname, 'r') as zipread:
		with zipfile.ZipFile(tempname, 'w') as zipwrite:
			for item in zipread.infolist():
				data = zipread.read(item.filename)
				if item.filename [-3:]=="bsp":
					data=data[:4]+chr(0x2e)+data[5:]
					print item.filename + " version fixed to 46"
				if item.filename [-3:]=="png" and False:
					tempimg= os.path.join(tempdir, item.filename)
					zipread.extract(item.filename,tempdir)
					with open(os.devnull, 'w') as fp:
						cmd = Popen(("png2tga -c %s %s" % (tempimg,os.path.dirname(tempimg))), stdout=fp, shell=False)
					cmd.communicate()
					item.filename=item.filename[:-3]+"tga"
					img=open(tempimg[:-3]+"tga")
					data=img.read()
					img.close()
					print item.filename + " converted from png format"
				if item.filename [-6:]=="shader":
					zipread.extract(item.filename,tempdir)
					data=""
					with open(os.path.join(tempdir, item.filename)) as f:
						for line in f:
							if "novlcollapse" in line:
								print "Shader: " + item.filename + " fixed in line: " + line
								continue
							data+=line+"\n"
					os.remove(os.path.join(tempdir, item.filename))
				zipwrite.writestr(item, data )
	shutil.move(tempname, zipfname)
finally:
	shutil.rmtree(tempdir)