forked from colonelpanic/dotfiles
125 lines
3.2 KiB
Python
125 lines
3.2 KiB
Python
import os
|
|
import sys
|
|
|
|
from invoke import Collection, ctask
|
|
|
|
from . import osx
|
|
from . import linux
|
|
from .util import DOTFILES_DIRECTORY, RESOURCES_DIRECTORY
|
|
|
|
|
|
ns = Collection()
|
|
ns.add_collection(osx)
|
|
ns.add_collection(linux)
|
|
|
|
|
|
@ctask(default=True)
|
|
def setup(ctx):
|
|
ctx.config['run']['pty'] = False
|
|
ctx.config['run']['warn'] = True
|
|
dotfiles(ctx, 'f')
|
|
if 'darwin' in sys.platform:
|
|
osx.setup(ctx)
|
|
else:
|
|
linux.setup(ctx)
|
|
install_python_libraries(ctx)
|
|
powerline(ctx)
|
|
install_npm_libraries(ctx)
|
|
change_shell(ctx)
|
|
|
|
|
|
@ctask
|
|
def dotfiles(ctx, flags=''):
|
|
ctx.run('hash dotfiles || sudo pip install dotfiles')
|
|
return ctx.run('dotfiles -s{1} -R {0}'.format(DOTFILES_DIRECTORY, flags))
|
|
|
|
|
|
@ctask
|
|
def dropbox_dotfiles(ctx, flags='f'):
|
|
ctx.run('hash dotfiles || sudo pip install dotfiles')
|
|
ctx.run('dotfiles -s{1} -R {0}'.format(
|
|
os.path.join(
|
|
os.path.expanduser('~'), 'Dropbox', 'configs', 'dotfiles'
|
|
),
|
|
flags
|
|
))
|
|
link_filenames(ctx, [('~/Dropbox/configs/custom.el', '~/.emacs.d/custom.el')], force=True)
|
|
|
|
|
|
@ctask
|
|
def powerline(ctx):
|
|
ctx.run('sudo pip install psutil')
|
|
ctx.run('sudo pip install git+git://github.com/Lokaltog/powerline')
|
|
|
|
|
|
@ctask
|
|
def install_python_libraries(ctx):
|
|
ctx.run('sudo pip install setuptools --upgrade')
|
|
ctx.run('sudo pip install -r {0}'.format(
|
|
os.path.join(RESOURCES_DIRECTORY, 'requirements.txt')
|
|
))
|
|
|
|
|
|
@ctask
|
|
def install_npm_libraries(ctx):
|
|
ctx.run('source {0}'.format(
|
|
os.path.join(RESOURCES_DIRECTORY, 'npm.sh')
|
|
))
|
|
|
|
|
|
@ctask
|
|
def vimstall(ctx):
|
|
ctx.run('vim +BundleInstall! +q +q')
|
|
|
|
|
|
@ctask
|
|
def change_shell(ctx):
|
|
ctx.run('sudo chsh -s $(which zsh) $(whoami)')
|
|
|
|
|
|
@ctask
|
|
def link_dropbox_other(ctx, force=False):
|
|
link_pairs = (
|
|
('~/Dropbox/Documents', '~/Documents'),
|
|
('~/Dropbox/Pictures', '~/Pictures'),
|
|
('~/Dropbox/org', '~/org')
|
|
('~/Dropbox/Desktop', '~/Desktop')
|
|
)
|
|
link_filenames(ctx, link_pairs, force)
|
|
|
|
|
|
def link_filenames(ctx, link_pairs, force=False):
|
|
for source, destination in link_pairs:
|
|
destination = os.path.expanduser(destination)
|
|
source = os.path.expanduser(source)
|
|
if force:
|
|
ctx.run("sudo rm -rf {0}".format(destination))
|
|
if os.path.exists(destination):
|
|
print("Skipping {0} because path already exists".format(destination))
|
|
else:
|
|
print("Linking {0} to {1}".format(destination, source))
|
|
ctx.run('ln -s {0} {1}'.format(source, destination))
|
|
|
|
|
|
@ctask
|
|
def customize_user_settings(ctx):
|
|
input_function = input if sys.version_info.major == 3 else raw_input
|
|
username = input_function("Enter the user's full name: ")
|
|
email = input_function("Enter the user's email address: ")
|
|
with open(os.path.expanduser('~/.gitconfig.custom'), 'w') as custom_file:
|
|
custom_file.write("""[user]
|
|
name = {0}
|
|
email = {1}""".format(username, email))
|
|
|
|
|
|
ns.add_task(change_shell)
|
|
ns.add_task(customize_user_settings)
|
|
ns.add_task(dotfiles)
|
|
ns.add_task(dropbox_dotfiles)
|
|
ns.add_task(install_npm_libraries)
|
|
ns.add_task(install_python_libraries)
|
|
ns.add_task(link_dropbox_other)
|
|
ns.add_task(powerline)
|
|
ns.add_task(setup)
|
|
ns.add_task(vimstall)
|