#!/usr/bin/python
# ----------------------------------------------------------
# DESCRIPTION:
# 
# A Script that switches application configuration files
# based on the current system network location (I myself
# use this for changing proxy server configurations)
# 
# Copyright (c) 2007-2008 Ali Rantakari
# http://hasseg.org
# 
# ----------------------------------------------------------
# REQUIRES:
# 
# - OS X Leopard (10.5)
# - Growl (and the growlnotify script (distributed with
#          growl))
# 
# ----------------------------------------------------------
# USAGE:
# 
# - Go through the 'settings' area below and change the
#   values to your liking
# 
# - Create configuration files for each location and each
#   application you would like to change configs for, using
#   the following naming pattern:
# 
#      <appname>.<config_file_name>.<location_name>
# 
#      ( for example: firefox.user.js.Work )
# 
#   These files must be located in the same folder as this
#   script. If you wish to add a common, location-agnostic
#   prefix for an app's configuration file, create a file
#   with the following naming pattern, and it will be
#   prepended to the configuration file before it is set:
# 
#      <appname>.<config_file_name>
#   
#   You can also create a file called "defaultconfig-prefix",
#   which will be prepended to all configuration files as
#   they are set, regardless of application and location.
# 
# - Call the script to switch the configuration files.
#   I Suggest using some automated system for this - I
#   myself use MarcoPolo. Since this is a Python script and
#   MarcoPolo doesn't want to run it (using the "ShellScript"
#   action,) make it call the included switchProxiesCaller
#   bash script (which will then call this one) each time
#   the network location changes.
# 
# ----------------------------------------------------------
# 


import string, sys, os, shutil



def getMyBaseName():
	return os.path.basename(sys.argv[0])

def getMyDirName():
	return os.path.dirname(sys.argv[0])





# ----------------------------------------------------------------
# 
# SETTINGS:
# 
# 

# The applications you want to switch configuration files for.
# 
# A dictionary where each key is the app name and corresponding
# values are dictionaries with the following two keys:
#   file : the app's config file name
#   dir  : the directory where the config file is located
#          (without trailing slash)
# 
# examples:
#   apps['firefox'] = {'file':'user.js', 'dir':'/Users/me/Library/Application Support/Firefox/Profiles/wn7lxarp.default'}
#   apps['subversion'] = {'file':'servers', 'dir':'/Users/me/.subversion'}
#   apps['bash'] = {'file':'.bash_profile', 'dir':'/Users/me'}
# 
apps = {}







# locations of CLI apps used
# 
# - system: (these should be here)
# 
osascript = "/usr/bin/osascript"
# 
# - custom: (these you need to get yourself)
# 
growlnotify = "/usr/local/bin/growlnotify"





# GROWL SETTINGS:

# whether or not to use growl for notifications
# (if this is False, notifications will be
#  printed to stdout)
# 
growl_use = True

# the name of this application 
# (to send to growlnotify)
# 
growl_appname = "Config switch script"

# the file to use as an icon in growl
# notifications
# 
growl_icon = getMyDirName()+"/icon.png"



# - - - - - - - - - - - - - - - - - - - - - -
# settings end here.
# ----------------------------------------------------------------
# 









# BEGIN: functions -----------------------------------------


# displays a notification/message to the user
def showMessage(title, msg):
	title = title.replace('"', '\'')
	msg = msg.replace('"', '\'')
	if (growl_use):
		if (os.path.exists(growlnotify)):
			os.system(growlnotify+' -a "'+growl_appname+'" -t "'+title+'" -m "'+msg+'" --image "'+growl_icon+'"')
		else:
			print " "
			print "** Error:"
			print "   growlnotify not found in path: \""+growlnotify+"\""
	else:
		print " "
		print "** "+title+":"
		print "   "+msg



# returns the name of the current OS X network location
def getCurrLocation():
	cmd = osascript+"""<<END
	tell application "System Events"
		tell network preferences
			return the name of current location
		end tell
	end tell
	END"""
	fin,fout = os.popen2(cmd)
	return fout.read().rstrip('\n')



# concatenates two files (first & second) into a
# third file (output)
def concatenateFiles(first, second, output):
	file = open(first, 'r')
	data = file.read()
	file.close()
	
	file = open(second, 'r')
	data += file.read()
	file.close()
	
	outputfile = open(output, 'w')
	outputfile.write(data)
	outputfile.close()


def getFileContents(filepath):
	file = open(filepath, 'r')
	data = file.read()
	file.close()
	return data
	


# --end--: functions - - - - - - - - - - - - - - - - - - - -






currloc = getCurrLocation()

if (currloc != None) and (currloc != ""):
	
	for appkey in apps:
		app = apps[appkey]
		origcfg = app['dir']+"/"+app['file']
		thiscfg = getMyDirName()+"/"+appkey+"."+app['file']+"."+currloc
		
		if os.path.exists(thiscfg):
			
			newcontents = ""
			
			if os.path.exists(getMyDirName()+"/defaultconfig-prefix"):
				newcontents = newcontents + getFileContents(getMyDirName()+"/defaultconfig-prefix")
			
			if os.path.exists(getMyDirName()+"/"+appkey+"."+app['file']):
				newcontents = newcontents + getFileContents(getMyDirName()+"/"+appkey+"."+app['file'])
			
			newcontents = newcontents + getFileContents(thiscfg)
			
			try:
				os.remove(origcfg)
				outputfile = open(origcfg, 'w')
				outputfile.write(newcontents)
				outputfile.close()
				showMessage("Settings for "+appkey+" OK", 'When you run '+appkey+' the next time, settings for "'+currloc+'" will be used.')
			except (IOError, os.error), why:
				showMessage("Error", 'Can not write new '+appkey+' config file: "'+str(why)+'"')
			
		else:
			showMessage("Error", 'Can not find '+appkey+' configuration for location "'+currloc+'"')
	
else:
	showMessage("Error", "Can not determine current network location")