1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
import py
import os
import click
import errno
from operator import attrgetter
DEFAULT_HOME = os.path.expanduser('~/')
DEFAULT_REPO = os.path.expanduser('~/Dotfiles')
class Repository(object):
"""A repository is a directory that contains dotfiles.
:param repodir: the location of the repository directory
:param homedir: the location of the home directory (primarily for testing)
"""
def __init__(self, repodir, homedir):
self.repodir = repodir
self.homedir = homedir
def __str__(self):
"""Convert repository contents to human readable form."""
return ''.join('%s\n' % item for item in self.contents()).rstrip()
def __repr__(self):
return '<Repository %r>' % self.repodir
def expected_name(self, target):
"""Given a repository target, return the expected symlink name."""
return py.path.local("%s/.%s" % (self.homedir, target.basename))
def contents(self):
"""Given a repository path, discover all existing dotfiles."""
contents = []
self.repodir.ensure(dir=1)
for target in self.repodir.listdir():
target = py.path.local(target)
contents.append(Dotfile(self.expected_name(target), target))
return sorted(contents, key=attrgetter('name'))
class Dotfile(object):
"""An configuration file managed within a repository.
:param name: name of the symlink in the home directory (~/.vimrc)
:param target: where the symlink should point to (~/Dotfiles/vimrc)
"""
def __init__(self, name, target):
self.name = name
self.target = target
def __str__(self):
return self.name.basename
def __repr__(self):
return '<Dotfile %r>' % self.name
@property
def state(self):
if self.target.check(exists=0):
# only for testing, cli should never reach this state
return 'error'
elif self.name.check(exists=0):
# no $HOME symlink
return 'missing'
elif self.name.check(link=0) or not self.name.samefile(self.target):
# if name exists but isn't a link to the target
return 'conflict'
return 'ok'
def add(self):
if self.target.check(exists=1):
raise OSError(errno.EEXIST, self.target)
self.name.move(self.target)
self.link()
def remove(self):
if self.target.check(exists=0):
raise OSError(errno.ENOENT, self.target)
self.name.remove()
self.target.move(self.name)
def link(self):
self.name.mksymlinkto(self.target)
def unlink(self):
self.name.remove()
pass_repo = click.make_pass_decorator(Repository)
@click.group(context_settings=dict(help_option_names=['-h', '--help']))
@click.option('--home-directory', type=click.Path(), default=DEFAULT_HOME,
show_default=True)
@click.option('--repository', type=click.Path(), default=DEFAULT_REPO,
show_default=True)
@click.version_option()
@click.pass_context
def cli(ctx, home_directory, repository):
"""Dotfiles is a tool to make managing your dotfile symlinks in $HOME easy,
allowing you to keep all your dotfiles in a single directory.
"""
ctx.obj = Repository(py.path.local(repository),
py.path.local(home_directory))
@cli.command()
@click.argument('files', nargs=-1, type=click.Path(exists=True))
@pass_repo
def add(repo, files):
"""Add dotfiles to a repository."""
for filename in files:
filename = py.path.local(filename)
click.echo('Dotfile(%s, %s).add()' % (
filename, repo.expected_name(filename)))
@cli.command()
@click.argument('files', nargs=-1, type=click.Path(exists=True))
@pass_repo
def remove(repo, files):
"""Remove dotfiles from a repository."""
for filename in files:
filename = py.path.local(filename)
click.echo('Dotfile(%s, %s).remove()' % (
filename, repo.expected_name(filename)))
@cli.command()
@click.option('-a', '--all', is_flag=True, help='Show all dotfiles.')
@click.option('-c', '--color', is_flag=True, help='Enable color output.')
@pass_repo
def status(repo, all, color):
"""Show all dotfiles in a non-OK state."""
state_info = {
'error': {'char': 'E', 'color': None},
'conflict': {'char': '!', 'color': None},
'missing': {'char': '?', 'color': None},
}
if all:
state_info['ok'] = {'char': ' ', 'color': None}
if color:
state_info['error']['color'] = 'red'
state_info['conflict']['color'] = 'magenta'
state_info['missing']['color'] = 'yellow'
for dotfile in repo.contents():
try:
char = state_info[dotfile.state]['char']
fg = state_info[dotfile.state]['color']
click.secho('%c %s' % (char, dotfile), fg=fg)
except KeyError:
continue
@cli.command()
@click.argument('files', nargs=-1, type=click.Path())
@pass_repo
def link(repo, files):
"""Create any missing symlinks."""
for filename in files:
click.echo('Dotfile(%s).link()' % filename)
@cli.command()
@click.argument('files', nargs=-1, type=click.Path(exists=True))
@pass_repo
def unlink(repo, files):
"""Remove existing symlinks."""
for filename in files:
click.echo('Dotfile(%s).unlink()' % filename)
|
