Merge branch 'master' into gregoa
[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
180     GIT_PS1_SHOWUNTRACKEDFILES= PSMOVEIN="movein:${1}" $SHELL -i || :
181 }
182
183 execin() {
184     local REPO
185     if [ $# -lt 2 ]; then
186        usage
187     fi
188
189     REPO=$1;shift
190
191     export GIT_DIR="$LOCAL_REPOS/${REPO}.git"
192
193     "$@"
194 }
195
196 add() {
197     if [ $# -lt 1 ]; then
198        usage
199     fi
200
201     for REPO in "$@"; do
202         REPO_NAME=$REPO.git
203         LOCAL_REPO=$LOCAL_REPOS/$REPO_NAME
204         REPO_URL=ssh://$GIT_HOST/$REMOTE_REPOS/$REPO_NAME
205
206         if [ -e "$LOCAL_REPO" ]; then
207             echo $LOCAL_REPO already exists
208             exit 1
209         else
210             trap "unset GIT_DIR; unset GIT_WORK_TREE; rm -rf $LOCAL_REPO" 0
211             mkdir -p "$LOCAL_REPO"
212             export GIT_DIR="$LOCAL_REPO"
213             git_work_tree "$REPO_NAME"
214             git init --bare
215             git remote add origin $REPO_URL
216             git config branch.master.remote origin
217             git config branch.master.merge refs/heads/master
218             git config core.bare false
219             git config core.worktree "$GIT_WORK_TREE"
220             git config status.showUntrackedFiles no
221             git pull
222             trap - 0
223
224             mr -c "$MRCONFIG" config "$LOCAL_REPO" checkout="git_fake_bare_checkout '$REPO_URL' '$REPO_NAME' '$GIT_WORK_TREE'"
225         fi
226     done
227 }
228
229 list() {
230     find "${LOCAL_REPOS}" -mindepth 1 -type d -name '*.git' | sed "s,${LOCAL_REPOS}/,,"
231 }
232
233 listremote() {
234     ssh $GIT_HOST "
235         find '${REMOTE_REPOS}' -mindepth 1 -type d -name '*.git' | sed 's,${REMOTE_REPOS}/,,'
236     " </dev/null
237 }
238
239 locate() {
240     local REPO
241     if [ $# -ne 1 ]; then
242         usage
243     fi
244     for REPO in $($0 list); do
245         (cd /; $0 exec "$REPO" git ls-files | sed -nr "/$1/{s/^/$REPO:/p}")
246     done
247 }
248
249 new() {
250     if [ $# -lt 2 ]; then
251         usage
252     fi
253     REPO_NAME=$1.git ; shift
254     LOCAL_REPO="$LOCAL_REPOS/$REPO_NAME"
255     REPO_URL="ssh://$GIT_HOST/$REMOTE_REPOS/$REPO_NAME"
256
257     if [ ! -e "$1" ]; then
258         echo $1 not found
259         exit 1
260     fi
261
262     if [ -e "$LOCAL_REPO" ]; then
263         echo $LOCAL_REPO already exists
264         exit 1
265     else
266         trap "unset GIT_DIR; unset GIT_WORK_TREE; rm -rf $LOCAL_REPO" 0
267         mkdir -p "$LOCAL_REPO"
268
269         if [ $CREATE_REMOTE -ne 0 ]; then
270             ssh $GIT_HOST "
271                 GIT_DIR=$REMOTE_REPOS/$REPO_NAME git --bare init
272             " </dev/null
273         fi
274
275         export GIT_DIR="$LOCAL_REPO"
276         git_work_tree "$REPO_NAME"
277         git init --bare
278         git remote add origin $REPO_URL
279         git config branch.master.remote origin
280         git config branch.master.merge refs/heads/master
281         git config core.bare false
282         git config core.worktree "$GIT_WORK_TREE"
283         git config status.showUntrackedFiles no
284         git add "$@"
285         git commit -m "initial checkin"
286         git push --all
287
288         trap - 0
289
290         mr -c "$MRCONFIG" config "$LOCAL_REPO" checkout="git_fake_bare_checkout '$REPO_URL' '$REPO_NAME' '$GIT_WORK_TREE'"
291     fi
292
293 }
294
295 preflight() {
296     # Check a few requirements before doing any work
297     errors=0
298     set +e
299     for binary in mr git; do
300         bin=$(which ${binary})
301         if [ -z "${bin}" ]; then
302             echo "Missing required program: ${binary}"
303             errors=$(( errors + 1 ))
304         fi
305     done
306     set -e
307     if [ $errors -ne 0 ]; then
308         echo "Errors found, exiting"
309         exit 2
310     fi
311 }
312
313 if [ $# -lt 1 ]; then
314     usage
315     exit 1
316 fi
317
318 command=$1 ; shift
319 case "$command" in
320     init)
321        preflight
322        init "$@"
323        ;;
324     add)
325        preflight
326        add "$@"
327        ;;
328     new)
329        preflight
330        new "$@"
331        ;;
332     login)
333        login "$@"
334        ;;
335     exec)
336        execin "$@"
337        ;;
338     ls)
339        list
340        ;;
341     list)
342        list
343        ;;
344     ls-r)
345        listremote
346        ;;
347     list-remote)
348        listremote
349        ;;
350     locate)
351        locate "$@"
352        ;;
353
354     *)
355        usage
356        exit 1
357        ;;
358 esac