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

import datetime

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 today's date in ISO format YYYY-MM-DD

today = datetime.date.today().isoformat()

print "DATE", today

# insert the date and changeset into the raptor_version.py file

filename = "../../python/raptor_version.py"

lines = []

try:

	file = open(filename, "r")

	for line in file.readlines():

		if "ISODATE" in line and "CHANGESET" in line:

			line = line.replace("ISODATE", today)

			line = line.replace("CHANGESET", changeset)

			if prototype:

				line = line.replace("system", "system PROTOTYPE")

			lines.append(line)

		else:

			lines.append(line)

except IOError, ex:

	sys.stderr.write("error: failed to read file '%s'\n%s" % (filename, str(ex)))

	sys.exit(1)

finally:

	file.close()

# ... and write the modified raptor_version.py file

try:

	file = open(filename, "w")

	for line in lines:

		file.write(line)

except IOError, ex:

	sys.stderr.write("error: failed to write file '%s'\n%s" % (filename, str(ex)))

	sys.exit(1)

finally:

	file.close()

# check that we really did change 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 today in version or not changeset in version:

		sys.stderr.write("error: date or 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)

# run the Linux installer maker script

package_sbs = subprocess.Popen(["./package_sbs.sh", "-s"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)

(stdout, stderr) = package_sbs.communicate()

if package_sbs.returncode != 0:

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

	sys.exit(1)

# find the name of the archive in /tmp

match = re.search('archive "([^"]+)" successfully created', stdout)

if match:

	tmp_archive = "/tmp/" + match.group(1)

	print "TMP ARCHIVE", tmp_archive

else:

	sys.stderr.write("error: failed to find linux archive file.\n")

	sys.exit(1)

# move it to the WORKSPACE root

if 'WORKSPACE' in os.environ:

	name = re.sub(r'/tmp/(sbs-\d+\.\d+\.\d+-).*', r'\1', tmp_archive)

	if prototype:

		fullname = name + "PROTOTYPE-" + changeset + ".run"

	else:

		fullname = name + changeset + ".run"

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

	print "WORKSPACE ARCHIVE", final_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)

# the end