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
|
import click
from .exceptions import DotfileException
from .repository import Repositories
def single(repos):
"""Raise an exception if multiple repositories are provided.
Certain operations (add, remove, etc...) can only be applied to a
single repository while other operations (list) can be applied
across multiple repositories.
"""
if len(repos) > 1:
raise click.BadParameter('Must specify exactly one repository.',
param_hint=['-r', '--repo'])
return repos[0]
def confirm(method, files, repo):
"""Return a list of files, or all files if none were specified.
When no files are specified, all files are assumed. But before we
go ahead, confirm to make sure this is the intended operation.
"""
if files:
# user has specified specific files, so we are not assuming all
return files
# no files provided, so we assume all files after confirmation
message = 'Are you sure you want to %s all dotfiles?' % method
click.confirm(message, abort=True)
return str(repo).split()
def show(repo, state):
"""TODO"""
for dotfile in repo.contents():
try:
display = state[dotfile.state]
except KeyError:
continue
char = display['char']
name = dotfile.short_name(repo.home)
fg = display.get('color', None)
bold = display.get('bold', False)
click.secho('%c %s' % (char, name), fg=fg, bold=bold)
def perform(method, files, repo, copy, debug):
"""Perform an operation on one or more dotfiles."""
for dotfile in repo.dotfiles(files):
try:
getattr(dotfile, method)(copy, debug)
if not debug:
msg = '%s%s' % (method, 'd' if method[-1] == 'e' else 'ed')
click.echo('%s %s' % (msg, dotfile.short_name(repo.home)))
except DotfileException as err:
click.echo(err)
pass_repos = click.make_pass_decorator(Repositories)
CONTEXT_SETTINGS = dict(auto_envvar_prefix='DOTFILES',
help_option_names=['-h', '--help'])
@click.group(context_settings=CONTEXT_SETTINGS)
@click.option('--repo', '-r', type=click.Path(), multiple=True,
help='A repository path.', default=['~/Dotfiles'],
show_default=True)
@click.version_option(None, '-v', '--version')
@click.pass_context
def cli(ctx, repo):
"""Dotfiles is a tool to make managing your dotfile symlinks in $HOME easy,
allowing you to keep all your dotfiles in a single directory.
"""
try:
ctx.obj = Repositories(repo)
except FileNotFoundError as e:
raise click.ClickException('Directory not found: %s' % e)
@cli.command()
@click.option('-c', '--copy', is_flag=True,
help='Copy files instead of creating symlinks.')
@click.option('-d', '--debug', is_flag=True,
help='Show what would be executed.')
@click.argument('files', nargs=-1, type=click.Path())
@pass_repos
def add(repos, copy, debug, files):
"""Add dotfiles to a repository."""
repo = single(repos)
perform('add', files, repo, copy, debug)
@cli.command()
@click.option('-d', '--debug', is_flag=True,
help='Show what would be executed.')
@click.argument('files', nargs=-1, type=click.Path(exists=True))
@pass_repos
def remove(repos, debug, files):
"""Remove dotfiles from a repository."""
repo = single(repos)
files = confirm('remove', files, repo)
perform('remove', files, repo, False, debug)
if not debug:
# pruning will remove any remaining empty directories
repo.prune()
@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_repos
def status(repos, all, color):
"""Show current status of dotfiles.
By default only non-OK dotfiles are shown. This can be overridden
with the '-a, --all' flag.
Legend:
l: symlink c: copy e: external symlink
?: missing !: conflict
Meaning:
* Missing: Not found in your home directory.
* Conflict: Different from the file in your home directory.
"""
bold = True if all and not color else False
state = {
'missing': {'char': '?', 'bold': bold},
'conflict': {'char': '!', 'bold': bold},
}
if all:
state['link'] = {'char': 'l'}
state['copy'] = {'char': 'c'}
state['external'] = {'char': 'e'}
if color:
state['missing'].update( {'color': 'yellow'})
state['conflict'].update({'color': 'magenta'})
for repo in repos:
show(repo, state)
@cli.command()
@click.option('-c', '--copy', is_flag=True,
help='Copy files instead of creating symlinks.')
@click.option('-d', '--debug', is_flag=True,
help='Show what would be executed.')
@click.argument('files', nargs=-1, type=click.Path())
@pass_repos
def enable(repos, copy, debug, files):
"""Link dotfiles into your home directory."""
repo = single(repos)
files = confirm('enable', files, repo)
perform('enable', files, repo, copy, debug)
@cli.command()
@click.option('-d', '--debug', is_flag=True,
help='Show what would be executed.')
@click.argument('files', nargs=-1, type=click.Path())
@pass_repos
def disable(repos, debug, files):
"""Unlink dotfiles from your home directory."""
repo = single(repos)
files = confirm('disable', files, repo)
perform('disable', files, repo, False, debug)
|
