# vim: set ts=2 sw=2 tw=99 noet ft=python: 
import os
import re
import subprocess
from ambuild.cache import Cache
import ambuild.command as command

#Quickly try to ascertain the current repository revision
def GetVersion():
	rev = command.RunDirectCommand(AMBuild, ['git', 'rev-list', '--count', 'HEAD']).stdoutText.strip()
	cset = command.RunDirectCommand(AMBuild, ['git', 'log', '--pretty=format:%h', '-n', '1']).stdoutText.strip()

	if not rev or not cset:
		raise Exception('Could not determine repository version')

	return (rev, cset)

def PerformReversioning():
	rev, cset = GetVersion()
	cacheFile = os.path.join(AMBuild.outputFolder, '.ambuild', 'hgcache')
	cache = Cache(cacheFile)
	if os.path.isfile(cacheFile):
		cache.LoadCache()
		if cache.HasVariable('cset') and cache['cset'] == cset:
			return False
	cache.CacheVariable('cset', cset)

	productFile = open(os.path.join(AMBuild.sourceFolder, 'product.version'), 'r')
	productContents = productFile.read()
	productFile.close()
	m = re.match('(\d+)\.(\d+)\.(\d+)(.*)', productContents)
	if m == None:
		raise Exception('Could not detremine product version')
	major, minor, release, tag = m.groups()

	incFolder = os.path.join(AMBuild.sourceFolder, 'extension')
	incFile = open(os.path.join(incFolder, 'version_auto.h'), 'w')
	incFile.write("""
#ifndef _AUTO_VERSION_INFORMATION_H_
#define _AUTO_VERSION_INFORMATION_H_

#define SM_BUILD_TAG	 	\"{0}\"
#define SM_BUILD_UNIQUEID	\"{1}:{2}\" SM_BUILD_TAG
#define SM_VERSION			\"{3}.{4}.{5}\"
#define SM_FULL_VERSION		SM_VERSION SM_BUILD_TAG
#define SM_FILE_VERSION		{6},{7},{8},0

#endif /* _AUTO_VERSION_INFORMATION_H_ */

""".format(tag, rev, cset, major, minor, release, major, minor, release))
	incFile.close()
	cache.WriteCache()

PerformReversioning()


