dotfiles: install.py annotate

author	Ludovic Chabant <ludovic@chabant.com>
date	Wed, 10 Jan 2018 00:00:00 -0800
parents
children	222b477ad678

rev	line source
407 c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	1 import os
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	2 import os.path
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	3 import sys
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 import stat
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 import argparse
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	6 import functools
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 import subprocess
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 import configparser
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11 dotfiles_dir = os.path.abspath(os.path.dirname(__file__))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	13 is_nix = True
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	14 is_windows = False
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	15 if sys.platform == "win32":
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16 is_nix = False
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 is_windows = True
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20 def _p(*paths):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 return os.path.join(dotfiles_dir, *paths).replace('/', os.sep)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24 def nixslash(path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 return path.replace('\\', '/')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	28 def ensure_dir(path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29 full_path = os.path.abspath(os.path.expanduser(path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 if not os.path.isdir(full_path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	31 os.makedirs(full_path, mode=0o700)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34 def mklink(orig_rel_path, link_path, mode=None):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35 orig_full_path = os.path.join(dotfiles_dir, orig_rel_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 link_full_path = os.path.abspath(os.path.expanduser(link_path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 if os.path.islink(link_full_path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	38 print("Unlinking %s" % link_full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39 os.unlink(link_full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	40 elif os.path.exists(link_full_path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41 print("Removing %s" % link_full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42 os.remove(link_full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	44 print("%s -> %s" % (link_full_path, orig_full_path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	45 os.symlink(orig_full_path, link_full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	46 if mode is not None:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 os.chmod(link_full_path, mode)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 def writelines(path, lines):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	51 full_path = os.path.abspath(os.path.expanduser(path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 print("%d lines to %s" % (len(lines), full_path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53 with open(full_path, 'w') as fp:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54 for l in lines:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	55 fp.write(l)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	56 fp.write('\n')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	57
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	58
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	59 def only_on_nix(f):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	60 @functools.wraps(f)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	61 def decorator(args, *kwargs):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	62 if is_nix:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	63 return f(args, *kwargs)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	64 return decorator
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	65
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	67 def only_on_win(f):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	68 @functools.wraps(f)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	69 def decorator(args, *kwargs):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	70 if is_windows:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	71 return f(args, *kwargs)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	72 return decorator
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	73
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	74
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	75 def needs_config(f):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	76 f.__dotfiles_needs_config__ = True
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 return f
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	78
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	80 def run_priority(prio):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	81 def wrapper(f):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	82 f.__dotfiles_priority__ = prio
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	83 return f
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	84 return wrapper
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	85
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	86
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	87 @only_on_nix
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	88 def install_bash():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	89 mklink('bashrc/bashrc', '.bashrc')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	90 mklink('bashrc/bash_profile', '.bash_profile')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	91
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	92
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	93 @only_on_nix
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	94 def install_fish():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	95 ensure_dir('~/.config')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96 mklink('fish', '~/.config/fish')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	97
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	98
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99 def install_vim():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	100 vimrc_path = '~/.vimrc'
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101 if is_windows:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	102 vimrc_path = '~/_vimrc'
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 writelines(vimrc_path, [
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 'set runtimepath+=%s' % nixslash(_p('vim')),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	105 'source %s' % nixslash(_p('vim', 'vimrc'))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 ])
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	107
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	109 @run_priority(2) # Needs to run before `fish`.
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	110 def install_mercurial():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111 hgrc_path = '~/.hgrc'
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112 if is_windows:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	113 hgrc_path = '~/mercurial.ini'
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	114 writelines(hgrc_path, [
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	115 '%%include %s' % _p('hgrc/hgrc'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	116 '[ui]',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	117 'ignore = %s' % _p('hgrc/hgignore'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 '[subrepos]',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119 'git:allowed = true',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120 '[extensions]',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	121 'hggit = %s' % _p('lib/hg/hg-git/hggit/'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	122 'onsub = %s' % _p('lib/hg/onsub/onsub.py'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123 'allpaths = %s' % _p('lib/hg/allpaths/mercurial_all_paths.py'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124 'prompt = %s' % _p('lib/hg/hg-prompt/prompt.py'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125 'evolve = %s' % _p('lib/hg/mutable-history/hgext3rd/evolve'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	126 'terse-status = %s' % _p('lib/hg/terse-status/terse-status.py')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 ])
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	128 if is_nix:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	129 print("Building fast-hg-prompt...")
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	130 compile_ok = True
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	131 try:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	132 subprocess.check_call(['make'], cwd=_p('lib/hg/fast-hg-prompt'))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	133 except subprocess.CalledProcessError:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	134 compile_ok = False
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136 for n in ['bookmark', 'remote', 'status']:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	137 link_path = os.path.expanduser('~/.local/bin/fast-hg-%s' % n)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	138 if compile_ok:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139 mklink('lib/hg/fast-hg-prompt/fast-hg-%s' % n, link_path,
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	140 mode=(stat.S_IRUSR \| stat.S_IWUSR \| stat.S_IXUSR))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	141 elif os.path.islink(link_path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	142 os.unlink(link_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	143 elif os.path.exists(link_path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144 os.remove(link_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147 def install_git():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	148 writelines('~/.gitconfig', [
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	149 '[include]',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	150 'path = %s' % _p('git/gitconfig')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	151 ])
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152 if is_windows:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	153 subprocess.check_call(
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154 ['setx', 'GIT_SSH',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 '%USERPROFILE%\\Dropbox\\Utilities\\plink.exe'],
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156 shell=True)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	159 @only_on_nix
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	160 def install_tmux():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161 mklink('tmux/tmux.conf', '~/.tmux.conf')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	162
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	163
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	164 @only_on_nix
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165 def install_weechat():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	166 mklink('weechat', '~/.weechat')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	169 @only_on_nix
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	170 def install_mutt():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	171 writelines('~/.muttrc', [
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	172 'source "gpg2 -dq %s \|"' % _p('mutt/variables.gpg'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 'source "%s"' % _p('mutt/muttrc'),
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	174 'source "%s"' % _p('lib/mutt/mutt-colors-solarized/'
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	175 'mutt-colors-solarized-dark-256.muttrc')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 ])
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	178
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	179 def clone_git(url, path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	180 if os.path.isdir(path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	181 print("Skipping git clone of %s -- directory exists" % path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 print("git clone %s %s" % (url, path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 ensure_dir(os.path.dirname(path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184 subprocess.check_call(['git', 'clone', url, path])
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	187 def clone_hg(url, path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188 if os.path.isdir(path):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	189 print("Skipping hg clone of %s -- directory exists" % path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	190 print("hg clone %s %s" % (url, path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191 ensure_dir(os.path.dirname(path))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192 env = dict(os.environ)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	193 env.update({'HGPLAIN': '1'})
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	194 subprocess.check_call(['hg', 'clone', url, path], env=env)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	195
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	196
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	197 @needs_config
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	198 @run_priority(100)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	199 def install_subrepos(cfg):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	200 if not cfg.has_section('subrepos'):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	201 return
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	202
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	203 for path, url in cfg.items('subrepos'):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	204 full_path = _p(path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	205 if url.startswith('[git]'):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	206 clone_git(url[len('[git]'):], full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	207 else:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	208 clone_hg(url, full_path)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	209
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	210
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	211 def main():
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	212 print("dotfiles installer")
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	213 print("python %s" % sys.version)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	214 print("on %s" % sys.platform)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	215 print('')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	216
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	217 cfg = configparser.ConfigParser()
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	218 cfg.read(_p('install.cfg'))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	219
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	220 mod_names = ['all']
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	221 this_mod = sys.modules[__name__]
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	222 for an in dir(this_mod):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	223 if not an.startswith('install_'):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	224 continue
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	225
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	226 name = an[len('install_'):]
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	227 mod_names.append(name)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	228
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	229 parser = argparse.ArgumentParser()
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	230 parser.add_argument(
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	231 'module', nargs='*',
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	232 choices=mod_names,
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	233 help="Which module(s) to install. Defaults to all modules.")
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	234 args = parser.parse_args()
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	235
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	236 funcs = []
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	237 selected_mods = set(args.module)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	238 if 'all' in selected_mods:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	239 selected_mods = set(mod_names)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	240 selected_mods.remove('all')
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	241 for mn in selected_mods:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	242 func = getattr(this_mod, 'install_%s' % mn)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	243 funcs.append((mn, func))
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	244
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	245 funcs = sorted(funcs, key=_get_install_func_priority, reverse=True)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	246 for name, func in funcs:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	247 print("Installing %s" % name)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	248 if hasattr(func, '__dotfiles_needs_config__'):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	249 func(cfg)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	250 else:
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	251 func()
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	252
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	253
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	254 def _get_install_func_priority(func_info):
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	255 func = func_info[1]
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	256 return getattr(func, '__dotfiles_priority__', 0)
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	257
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	258
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	259 if __name__ == '__main__':
c6da0c9f40ae Replace subrepos with an install script. Finally. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	260 main()

407

c6da0c9f40ae Replace subrepos with an install script. Finally.