# hudson runs this from the raptor/util/install-windows directory

import os

import re

import shutil

import subprocess

import sys

# run "hg id" to get the current branch name and tip changeset

hgid = subprocess.Popen(["hg", "id"], stdout=subprocess.PIPE)

stdout = hgid.communicate()[0]

if hgid.returncode == 0 and len(stdout) >= 12:

	changeset = stdout[0:12]

	print "CHANGESET", changeset

	prototype = ("wip" in stdout or "fix" in stdout)

	print "PROTOTYPE", prototype

else:

	sys.stderr.write("error: failed to get tip mercurial changeset.\n")

	sys.exit(1)

# get the raptor version string

sbs_v = subprocess.Popen(["../../bin/sbs", "-v"], stdout=subprocess.PIPE)

version = sbs_v.communicate()[0]

if sbs_v.returncode == 0:

	print "VERSION", version

	if not changeset in version:

		sys.stderr.write("error: changeset does not match the sbs version.\n")

		sys.exit(1)

        if prototype and not "PROTOTYPE" in version:

		sys.stderr.write("error: the sbs version should be marked PROTOTYPE.\n")

		sys.exit(1)

else:

	sys.stderr.write("error: failed to get sbs version.\n")

	sys.exit(1)

# find the SBS_HOME and WIN32_SUPPORT

if 'SBS_HOME' in os.environ:

	sbs_home = os.environ['SBS_HOME']

else:

	sys.stderr.write("error: no SBS_HOME is set.\n")

	sys.exit(1)

if 'WIN32_SUPPORT' in os.environ:

	win32_support = os.environ['WIN32_SUPPORT']

else:

	sys.stderr.write("error: no WIN32_SUPPORT is set.\n")

	sys.exit(1)

# run the Windows installer maker script

if prototype:

	postfix = "-PROTOTYPE-" + changeset

else:

	postfix = "-" + changeset

package_sbs = subprocess.Popen(["python", "raptorinstallermaker.py",

                                "-s", sbs_home, "-w", win32_support,

                                "--postfix=" + postfix],

                                stdout=subprocess.PIPE) #, stderr=subprocess.PIPE)

(stdout, stderr) = package_sbs.communicate()

if package_sbs.returncode == 0:

	match = re.search('Output: "([^"]+)"', stdout)

	zip_match = re.search('Zipoutput: "([^"]+)"', stdout)

	if match:

		tmp_archive = match.group(1)

		print "TMP ARCHIVE", tmp_archive

	else:

		sys.stderr.write("error: failed to find packaged filename.\n")

		sys.exit(1)

	if zip_match:

		tmp_zip_archive = zip_match.group(1)

		print "TMP ZIP ARCHIVE", tmp_zip_archive

	else:

		sys.stderr.write("error: failed to find zip filename.\n")

		sys.exit(1)

else:

	sys.stderr.write("error: failed to create windows package of sbs.\n")

	sys.exit(1)

# move the results to WORKSPACE

if 'WORKSPACE' in os.environ:

	final_archive = os.path.join(os.environ['WORKSPACE'], os.path.basename(tmp_archive))

	final_zip_archive = os.path.join(os.environ['WORKSPACE'], os.path.basename(tmp_zip_archive))

	print "WORKSPACE ARCHIVE", final_archive

	print "WORKSPACE ZIP ARCHIVE", final_zip_archive

else:

	sys.stderr.write("error: no WORKSPACE is set.\n")

	sys.exit(1)

try:

	shutil.move(tmp_archive, final_archive)

except Error, err:

	sys.stderr.write("error: could not rename '%s' as '%s'.\n" % (tmp_archive, final_archive))

	sys.exit(1)

try:

	shutil.move(tmp_zip_archive, final_zip_archive)

except Error, err:

	sys.stderr.write("error: could not rename '%s' as '%s'.\n" % (tmp_zip_archive, final_zip_archive))

	sys.exit(1)

# the end