#!/usr/bin/env python

# Set BIRCH_PLATFORM environment variable in platform.source
# and platform.profile.source by uncommenting one of the 
# lines in each file.

#Synopsis: setplatform.sh platform
#

import os, os.path, sys, shutil

def replaceAll(file, searchExp, replaceExp):
	shutil.move( file, file + "~" )
	destination = open( file, "w" )
	source= open( file + "~", "r" )
	for line in source:
		if searchExp in line:
			line = line.replace(searchExp,replaceExp)
		destination.write( line )
	source.close()
	destination.close()	

def setplatform(BASEDIR = os.path.abspath(os.path.join(os.getcwd(), os.pardir))):
	PLAT_SRC = os.path.join(BASEDIR, "local", "admin", "platform.source")
	PROF_SRC = os.path.join(BASEDIR, "local", "admin", "platform.profile.source")
	
	if len(sys.argv) < 1:
		print ("setplatform.sh: ERROR! YOU NEED TO SPECIFY A PLAFORM!")
	platform = sys.argv[1]
	
	if platform in ["solaris-sparc", "solaris-amd64", "linux-intel", "linux-x86_64", "osx-x86_64", "winxp-32", "win7-64" ]:
	
		replaceAll(PLAT_SRC, "#setenv BIRCH_PLATFORM " + platform, "setenv BIRCH_PLATFORM " + platform)
		replaceAll(PROF_SRC, "#BIRCH_PLATFORM=" + platform, "BIRCH_PLATFORM=" + platform)
	else:
		print ('setplatform.sh: invalid platform (' + platform + ')')
		sys.exit(1)
		
if __name__=="__main__":
	setplatform(sys.argv[1])