Merge pull request #1 from acm-uiuc/master

pull in update script

Author: silverdev

.gitignore

@@ -3,3 +3,4 @@
 .nfs.*
 answerback.Linux
 answerback.Darwin
+gitprivate

README.md

@@ -1,2 +1,3 @@
 # Installation
-Install by executing `install.pl` from within the acmdots directory.
+Install by executing `install.py`.
+See `install.py --help` for more information.

install.pl

@@ -0,0 +1,219 @@
+#! /usr/bin/env python
+
+import os
+import os.path
+import optparse
+import shutil
+
+links = {
+	'screenrc':   '.screenrc',
+	'ackrc':      '.ackrc',
+	'toprc':      '.toprc',
+	'dir_colors': '.dir_colors',
+	'lessfilter': '.lessfilter',
+
+	'vim':      '.vim',
+	'vimrc':    '.vimrc',
+	'_vimrc':   '_vimrc',
+	'gvimrc':   '.gvimrc',
+
+	'emacsrc':  '.emacs',
+	'emacs':    '.emacsdir',
+
+	'commonsh': '.commonsh',
+
+	'inputrc':  '.inputrc',
+
+	'bash':         '.bash',
+	'bashrc':       '.bashrc',
+	'bash_profile': '.bash_profile',
+
+	'zsh':      '.zsh',
+	'zshrc':    '.zshrc',
+
+	'ksh':      '.ksh',
+	'kshrc':    '.kshrc',
+	'mkshrc':   '.mkshrc',
+
+	'shinit':   '.shinit',
+
+	'Xdefaults':  '.Xdefaults',
+	'Xresources': '.Xresources',
+
+	'uncrustify.cfg': '.uncrustify.cfg',
+	'indent.pro':     '.indent.pro',
+
+	'xmobarrc':  '.xmobarrc',
+	'xmonad.hs': '.xmonad/xmonad.hs',
+
+	'Wombat.xccolortheme':  'Library/Application Support/Xcode/Color Themes/Wombat.xccolortheme',
+#	'Wombat.dvtcolortheme': 'Library/Developer/Xcode/UserData/FontAndColorThemes/Wombat.dvtcolortheme',
+
+	'gitconfig': '.gitconfig',
+	'gitignore': '.gitignore',
+
+	'tigrc':     '.tigrc',
+
+	'caffeinate': 'bin/caffeinate',
+	'lock':       'bin/lock',
+
+	'git-info':            'bin/git-info',
+	'git-untrack-ignored': 'bin/git-untracked-ignored',
+
+	'gdbinit': '.gdbinit',
+}
+
+scriptdir = os.path.dirname(os.path.realpath(__file__))
+home = os.path.expanduser('~')
+contained = os.path.commonprefix([scriptdir, home]) == home
+
+def read_link_abs(path):
+	"""Read the absolute path to which the symbolic link points.
+
+	Args:
+		path - path to the link
+	Returns:
+		the target absolute path or the empty string if path isn't a link
+	"""
+	if not os.path.islink(path):
+		return ''
+
+	target = os.path.join(os.path.dirname(path), os.readlink(path))
+	return os.path.abspath(target)
+
+def uninstall(file):
+	"""Uninstall a file.
+
+	Args:
+		file - the file to uninstall
+	Returns:
+		True if the file was uninstalled, False otherwise
+	"""
+	src  = os.path.join(scriptdir, file)
+	dest = os.path.join(home, links[file])
+
+	# Only remove if it's a link to a dotfile in the repo
+	if read_link_abs(dest) != src:
+		return False
+
+	os.remove(dest)
+
+	# Remove empty directories that contained the link
+	dir = os.path.dirname(dest)
+	while dir and not os.listdir(dir):
+		os.rmdir(dir)
+		dir = os.path.dirname(dir)
+
+	return True
+
+def uninstall_all():
+	uname = os.uname()[0]
+	links['answerback.' + uname] = 'bin/answerback.' + uname
+	links['gitprivate'] = '.gitprivate'
+
+	i = 0; # Keep track of how many links we remove
+	for file in links:
+		if uninstall(file):
+			i += 1
+
+	print '{0:d} link{1} removed'.format(i, 's' if i != 1 else '')
+
+def install(file, force=False):
+	"""Install a file.
+
+	Args:
+		file - the file to install
+		force - whether to overwrite existing files
+	Returns:
+		True if the file was installed, False otherwise
+	"""
+	src  = os.path.join(scriptdir, file)
+	dest = os.path.join(home, links[file])
+	path = os.path.dirname(dest)
+
+	# If needed, create the directory this file resides in
+	if not os.path.exists(path):
+		os.makedirs(path)
+
+	if os.path.exists(dest):
+		if force:
+			# Remove the destination if it exists
+			if os.path.isdir(dest) and not os.path.islink(dest):
+				shutil.rmtree(dest)
+			else:
+				os.remove(dest)
+		else:
+			# If a link already exists, see if it points to this file
+			# to prevent extra warnings caused by previous runs
+			if read_link_abs(dest) != src:
+				print 'Could not link "{0}" to "{1}": File exists'.format(src, dest)
+			return False
+
+	# Use relative links if the dotfiles are contained in home
+	if contained:
+		os.chdir(home)
+		src  = os.path.relpath(src, path)
+		dest = links[file]
+
+	os.symlink(src, dest)
+	return True
+
+def install_all(force=False):
+	# Compile answerback and add it to links
+	uname = os.uname()[0]
+	answerback_src = os.path.join(scriptdir, 'answerback.c')
+	answerback_bin = os.path.join(scriptdir, 'answerback.' + uname)
+	if os.system('cc {0} -o {1}'.format(answerback_src, answerback_bin)) == 0:
+		links['answerback.' + uname] = 'bin/answerback.' + uname
+	else:
+		print 'Could not compile answerback.'
+
+	# Create private git config and add it to links
+	if not os.path.exists(os.path.join(scriptdir, 'gitprivate')):
+		name = raw_input('Full name for Git config (enter to skip): ')
+		email = raw_input('Email address for Git config (enter to skip): ')
+		if name or email:
+			with open(os.path.join(scriptdir, 'gitprivate'), 'w') as config:
+				config.write('[user]\n')
+				if name:
+					config.write('\tname = ' + name + '\n')
+				if email:
+					config.write('\temail = ' + email + '\n')
+			links['gitprivate'] = '.gitprivate'
+	else:
+		links['gitprivate'] = '.gitprivate'
+
+	i = 0; # Keep track of how many links we added
+	for file in links:
+		if install(file, force):
+			i += 1
+
+	print '{0:d} link{1} created'.format(i, 's' if i != 1 else '')
+
+def main():
+	class RawHelpFormatter(optparse.IndentedHelpFormatter):
+		def format_description(self, description):
+			return description + '\n'
+
+	# Parse arguments
+	parser = optparse.OptionParser(
+		usage='usage: %prog [-h] [-u] [-f]',
+		formatter=RawHelpFormatter(),
+		description= \
+"""Installs symbolic links from dotfile repo into your home directory
+
+Destination directory is "{0}".
+Source files are in "{1}".""".format(home, scriptdir))
+	parser.add_option('-u', '--uninstall', action='store_true',
+		help='unlink dotfiles from your home directory')
+	parser.add_option('-f', '--force', action='store_true',
+		help='force to overwrite existing files')
+	args, _ = parser.parse_args()
+
+	if args.uninstall:
+		uninstall_all()
+	else:
+		install_all(args.force)
+
+if __name__ == '__main__':
+	main()