##################################################################
# Asset Generator (assetGenerator.py)
# Kyle Pulver
# @kylepulver
# http://kpulv.com
# 2013 / 2 / 3
# Version 1.0.0
#
# Put this file in your AS3 Project root.
#
# Modify the ASSET_FOLDER argument to be a path to your assets
# directory from the script.  Usually it's just 'assets'
#
# Modify the OUTPUT_FILE to be the resulting .as file.
#
# Add names of folders to EXCLUDE to skip them.
#
# Dont use spaces in filenames or foldernames if you can help it
#
# The script assumes your dir structure looks something like this
# --
# assets/
#   png/
#     *.png
#   mp3/
#     *.mp3
#   levels/
#     *.oel
# assetGenerator.py
# --
# So your base assets folder has at least one sub directory for
# each major type of asset -- although this is not necessary.
# However, each sub directory beyond the first for each major
# asset type will be added to the asset name in the ouput file.
#
# If you change or improve this script, just let me know!
##################################################################

import os

##################################################################
# Stuff you can change is right below!
##################################################################

# Where to find the assets in relation to the script
ASSET_FOLDER = 'assets'
# The file to output.  Must use / and not \
OUTPUT_FILE = 'src/game/Assets.as'
# What extensions to look for
EXTENSIONS = [
	'png',
	'mp3',
	'oel',
	'ttf'
]
# What prefixes to give to the assets
PREFIXES = [
	'IMG',
	'SOUND',
	'LEVEL',
	'FONT'
]
# Which folders to exclude
EXCLUDE = [
	'psd',
	'exclude'
]

##################################################################
# Magic happens below this line!				
##################################################################

path = os.getcwd() + '\\' + ASSET_FOLDER
output = os.getcwd() + '\\' + OUTPUT_FILE

# Figure out the class name based on the file
className = os.path.splitext(os.path.basename(OUTPUT_FILE))[0]

# Figure out the package name based on the folder
packageName = OUTPUT_FILE[OUTPUT_FILE.find('/') + 1:]
packageSplit = packageName.split('/')
packageSplit.pop()
if (len(packageSplit) > 0):
	packageName = ".".join(packageSplit) + " "
else:
	packageName = ""

# Assemble some data structures!
stuff = {}
prefix = {}
for i in range(len(EXTENSIONS)):
	stuff[EXTENSIONS[i]] = [];
	prefix[EXTENSIONS[i]] = PREFIXES[i] 

# Walk through dem files
for r,d,f in os.walk(path):
	addToFile = True

	for excludes in EXCLUDE:
		if (excludes == os.path.basename(r)):
			addToFile = False

	if (addToFile):		
		for files in f:
			ext = os.path.splitext(files)[1][1:] #grab extension without the '.''
			for key, value in stuff.items():
				if (ext == key):
					stuff[key].append(os.path.relpath(r, output)[3:] + "\\" + files)

# Start writing the file
f = open(output, 'w+')
f.write("""package """ + packageName + """{
	/** Auto generated from assetGenerator.py! :) */
	public class """ + className + """ {

""");

# Crazy writing all the embeds time
for key, value in stuff.items():
	if (len(stuff[key]) > 0):
		f.write ("\t\t/** Generating " + key + " assets! */\n")

		for files in stuff[key]:
			subfolder = files[:files.find(ASSET_FOLDER)]
			subFolders = []
			subFolderCount = len(subfolder.split('\\'))
			folderCount = len(files.split('\\'))

			if (folderCount > subFolderCount + 2):
				start = files.find(ASSET_FOLDER) + len(ASSET_FOLDER) + 1
				poop = files[start:]
				start = poop.find('\\') + 1
				poop = poop[start:]
				poop = poop.split('\\')
				subFolders = poop[:-1]

			fname = os.path.splitext(os.path.basename(files))[0]
			fname = fname.replace(" ", "_") #get rid of spaces, use underscores instead
			fname = fname.upper().replace(prefix[key] + "_", "") #get rid of prefixes already applied in the file name
			assetName = prefix[key] + "_"
			for i in range(len(subFolders)):
				assetName += subFolders[i].upper().replace(" ", "_") + "_"
			assetName += fname

			f.write("\t\t[Embed(source = \"" + files.replace("\\", "/") + "\"")
			
			# Fonts are special
			if (key == 'ttf'):
				f.write(", embedAsCFF = \"false\", fontFamily = \"" + fname + "\")] private static const _" + assetName + ":Class\n")
				f.write("\t\tpublic static const " + assetName + ":String = \"" + fname + "\"\n\n")
			else:
				if (key != 'png' and key != 'mp3'):
					f.write(", mimeType = \"application/octet-stream\"")
				f.write(")] public static const " + assetName + ":Class\n")

		if (key != 'ttf'):
			f.write("\n")

# All done!
f.write("\t}\n}")
f.close()