Support repository names with / in them (e.g., "etc/vim")
[gregoa/movein.git] / movein
1 #!/bin/bash
2 #
3 # Copyright © 2008 Mike O'Connor <stew@vireo.org>
4 #
5 # This program is free software; you can redistribute it and/or modify it
6 # under the terms of the GNU General Public License as published by the
7 # Free Software Foundation; either version 2, or (at your option) any
8 # later version.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program; if not, write to the Free Software
17 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
18 # USA.
19
20 set -e
21 set -u
22
23 usage() {
24     bn=$(basename $0)
25     cat <<EOF
26 $0: manage home directory using mr
27
28 NAME
29   $bn - a command to manage detached git repositories using mr
30
31 SYNOPSIS
32   $bn command [options]
33
34   $bn init
35   $bn ls
36   $bn list
37   $bn ls-r
38   $bn list-remote
39   $bn add repository_name
40   $bn new repository_name file1 [file2 file3...]
41   $bn login repository_name
42   $bn exec repository_name command [arg1 arg2...]
43
44 COMMANDS
45   init
46       create ~/.moveinrc, create/update  ~/.mrconfig
47
48   ls
49   list
50       show a list of local repositories
51
52   ls-r
53   list-remote
54       show a list of remote repositories
55
56   locate pattern
57       locate which repositories contain files matching pattern
58
59   add repository_name
60       checkout one or more repositories from the remote host and add it to
61       mr's configuration
62
63   new repository_name file1 [file2 file3...]
64       create a new repository on the remote host, and checkin the
65       listed files to the new repository.   Add the new repository
66       to mr's configuration
67
68   login repository_name
69       start a subshell in repository_name
70
71   exec repository_name command [arg1 arg2...]
72       execute a command in the context of repository_name, for example:
73         $bn exec myrepo git status
74         $bn exec myrepo git ls-files
75
76 EOF
77
78     exit 1
79 }
80
81 [ $# -ge 1 ] || usage
82
83 GIT_HOST=git.$(hostname -d || echo "example.com")
84 REMOTE_REPOS=~/git
85 LOCAL_REPOS=~/.movein
86 MRCONFIG=~/.mrconfig
87 MOVEINRC=~/.moveinrc
88 CREATE_REMOTE=1
89
90 [ -e "$MOVEINRC" ] && . "$MOVEINRC"
91
92 init() {
93
94     if [ $# -ne 0 ]; then
95         usage
96     fi
97
98     if [ -e $MOVEINRC ]; then
99         echo $MOVEINRC already exists
100         exit 1
101     fi
102
103     echo -n "git server hostname? [${GIT_HOST}] "
104     read GIT_HOST
105     if [ -z "$GIT_HOST" ]; then
106         GIT_HOST=git.$(hostname -d)
107     fi
108
109     echo -n "path to remote repositories? [~/git] "
110     read REMOTE_REPOS
111     if [ -z "$REMOTE_REPOS" ]; then
112         REMOTE_REPOS=~/git
113     fi
114
115     case $GIT_HOST in
116         *github*)
117             DEFAULT_CREATE_REMOTE=0
118             CREATE_REMOTE_OPTIONS="[yN]"
119             ;;
120         *)
121             DEFAULT_CREATE_REMOTE=1
122             CREATE_REMOTE_OPTIONS="[Yn]"
123             ;;
124     esac
125
126     echo -n "should \"movein new\" run \"git init\" on the remote host? (github users should say \"no\") $CREATE_REMOTE_OPTIONS"
127     read CREATE_REMOTE_SELECTION
128     case "$CREATE_REMOTE_SELECTION" in
129         [yY]) CREATE_REMOTE=1 ;;
130         [nN]) CREATE_REMOTE=0 ;;
131         *) CREATE_REMOTE=$DEFAULT_CREATE_REMOTE
132     esac
133
134     echo -n "Local repository directory? [~/.movein] "
135     read LOCAL_REPOS
136     if [ -z "$LOCAL_REPOS" ]; then
137         LOCAL_REPOS=~/.movein
138     fi
139
140     echo -n "Location of .mrconfig file? [~/.mrconfig] "
141     read MRCONFIG
142     if [ -z "$MRCONFIG" ]; then
143         MRCONFIG=~/.mrconfig
144     fi
145
146     cat <<EOF > $MOVEINRC
147 GIT_HOST=$GIT_HOST
148 REMOTE_REPOS=$REMOTE_REPOS
149 LOCAL_REPOS=$LOCAL_REPOS
150 MRCONFIG=$MRCONFIG
151 CREATE_REMOTE=$CREATE_REMOTE
152 EOF
153
154     if [ ! -d "$LOCAL_REPOS" ]; then
155        mkdir -p "$LOCAL_REPOS"
156     fi
157
158     mr -c "$MRCONFIG" config DEFAULT include="cat /usr/share/mr/git-fake-bare"
159
160 }
161
162 git_work_tree() {
163     local TEMP_REPO
164     TEMP_REPO=$1
165     GIT_WORK_TREE=../../
166
167     while [ "${TEMP_REPO#*/}" != "$TEMP_REPO" ]; do
168         TEMP_REPO="${TEMP_REPO#*/}"
169         GIT_WORK_TREE="../$GIT_WORK_TREE"
170     done
171 }
172
173 login() {
174     if [ $# -ne 1 ]; then
175        usage
176     fi
177
178     export GIT_DIR="$LOCAL_REPOS/${1}.git"
179     export GIT_WORK_TREE="$GIT_DIR/$(git config --get core.worktree)"
180
181     GIT_PS1_SHOWUNTRACKEDFILES= PSMOVEIN="movein:${1}" $SHELL -i || :
182 }
183
184 execin() {
185     local REPO
186     if [ $# -lt 2 ]; then
187        usage
188     fi
189
190     REPO=$1;shift
191
192     export GIT_DIR="$LOCAL_REPOS/${REPO}.git"
193     export GIT_WORK_TREE="$GIT_DIR/$(git config --get core.worktree)"
194
195     "$@"
196 }
197
198 add() {
199     if [ $# -lt 1 ]; then
200        usage
201     fi
202
203     for REPO in "$@"; do
204         REPO_NAME=$REPO.git
205         LOCAL_REPO=$LOCAL_REPOS/$REPO_NAME
206         REPO_URL=ssh://$GIT_HOST/$REMOTE_REPOS/$REPO_NAME
207
208         if [ -e "$LOCAL_REPO" ]; then
209             echo $LOCAL_REPO already exists
210             exit 1
211         else
212             trap "unset GIT_DIR; unset GIT_WORK_TREE; rm -rf $LOCAL_REPO" 0
213             mkdir -p "$LOCAL_REPO"
214             export GIT_DIR="$LOCAL_REPO"
215             git_work_tree "$REPO_NAME"
216             git init --bare
217             git remote add origin $REPO_URL
218             git config branch.master.remote origin
219             git config branch.master.merge refs/heads/master
220             git config core.bare false
221             git config core.worktree "$GIT_WORK_TREE"
222             git config status.showUntrackedFiles no
223             git pull
224             trap - 0
225
226             mr -c "$MRCONFIG" config "$LOCAL_REPO" checkout="git_fake_bare_checkout '$REPO_URL' '$REPO_NAME' '$GIT_WORK_TREE'"
227         fi
228     done
229 }
230
231 list() {
232     find "${LOCAL_REPOS}" -mindepth 1 -type d -name '*.git' | sed "s,${LOCAL_REPOS}/,,"
233 }
234
235 listremote() {
236     ssh $GIT_HOST "
237         find '${REMOTE_REPOS}' -mindepth 1 -type d -name '*.git' | sed 's,${REMOTE_REPOS}/,,'
238     " </dev/null
239 }
240
241 locate() {
242     local REPO
243     if [ $# -ne 1 ]; then
244         usage
245     fi
246     for REPO in $($0 list); do
247         (cd /; $0 exec "$REPO" git ls-files | sed -nr "/$1/{s/^/$REPO:/p}")
248     done
249 }
250
251 new() {
252     if [ $# -lt 2 ]; then
253         usage
254     fi
255     REPO_NAME=$1.git ; shift
256     LOCAL_REPO="$LOCAL_REPOS/$REPO_NAME"
257     REPO_URL="ssh://$GIT_HOST/$REMOTE_REPOS/$REPO_NAME"
258
259     if [ ! -e "$1" ]; then
260         echo $1 not found
261         exit 1
262     fi
263
264     if [ -e "$LOCAL_REPO" ]; then
265         echo $LOCAL_REPO already exists
266         exit 1
267     else
268         trap "unset GIT_DIR; unset GIT_WORK_TREE; rm -rf $LOCAL_REPO" 0
269         mkdir -p "$LOCAL_REPO"
270
271         if [ $CREATE_REMOTE -ne 0 ]; then
272             ssh $GIT_HOST "
273                 GIT_DIR=$REMOTE_REPOS/$REPO_NAME git --bare init
274             " </dev/null
275         fi
276
277         export GIT_DIR="$LOCAL_REPO"
278         git_work_tree "$REPO_NAME"
279         git init --bare
280         git remote add origin $REPO_URL
281         git config branch.master.remote origin
282         git config branch.master.merge refs/heads/master
283         git config core.bare false
284         git config core.worktree "$GIT_WORK_TREE"
285         git config status.showUntrackedFiles no
286         git add "$@"
287         git commit -m "initial checkin"
288         git push --all
289
290         trap - 0
291
292         mr -c "$MRCONFIG" config "$LOCAL_REPO" checkout="git_fake_bare_checkout '$REPO_URL' '$REPO_NAME' '$GIT_WORK_TREE'"
293     fi
294
295 }
296
297 preflight() {
298     # Check a few requirements before doing any work
299     errors=0
300     set +e
301     for binary in mr git; do
302         bin=$(which ${binary})
303         if [ -z "${bin}" ]; then
304             echo "Missing required program: ${binary}"
305             errors=$(( errors + 1 ))
306         fi
307     done
308     set -e
309     if [ $errors -ne 0 ]; then
310         echo "Errors found, exiting"
311         exit 2
312     fi
313 }
314
315 if [ $# -lt 1 ]; then
316     usage
317     exit 1
318 fi
319
320 command=$1 ; shift
321 case "$command" in
322     init)
323        preflight
324        init "$@"
325        ;;
326     add)
327        preflight
328        add "$@"
329        ;;
330     new)
331        preflight
332        new "$@"
333        ;;
334     login)
335        login "$@"
336        ;;
337     exec)
338        execin "$@"
339        ;;
340     ls)
341        list
342        ;;
343     list)
344        list
345        ;;
346     ls-r)
347        listremote
348        ;;
349     list-remote)
350        listremote
351        ;;
352     locate)
353        locate "$@"
354        ;;
355
356     *)
357        usage
358        exit 1
359        ;;
360 esac