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