Allow storage of repos inside .zit
[zit] / zit
1 #!/bin/bash
2
3 abort() {
4         echo $1
5         exit 1
6 }
7
8 cmd="$1"
9 shift
10
11 USAGE="usage: zit COMMAND FILE [ARGS...]"
12
13 zit_help() {
14         cmd="$1"
15         shift
16         case $cmd in
17                 git)
18                 git help "$@"
19                 ;;
20                 *)
21                 echo $USAGE
22                 echo ""
23                 echo "Set up a git repository under .zit.FILE to track changes for FILE."
24                 echo "File must be a regular file and in the current directory."
25                 echo ""
26                 echo "Zit commands:"
27                 echo "   init           Synonym for track"
28                 echo "   list           Synonym for tracked"
29                 echo "   track  Start tracking changes to FILE"
30                 echo "   tracked        List tracked files in current directory"
31                 echo ""
32                 echo "See 'zit help git' or 'git help' for git commands."
33                 ;;
34         esac
35 }
36
37 ZIT_DIR=.zit
38
39 zit_setup() {
40         ZIT_FILE="$1"
41         test $ZIT_FILE || abort "Please specify a file"
42         test -f $ZIT_FILE || abort "No such file $ZIT_FILE"
43         test $ZIT_FILE = "`basename $ZIT_FILE`" || abort "Sorry, Zit only works on files in the current directory"
44
45         export GIT_WORK_TREE="`pwd`"
46
47         # first, check if a repo exists already, looking for
48         # .zit/file.git or .file.git, in that order
49         # if neither is found, and .zit exists, set the repo dir
50         # to .zit/file.git, otherwise set it to .file.git
51         GIT_DIR="$ZIT_DIR/$ZIT_FILE.git"
52         if ! test -d "$GIT_DIR"; then
53                 GIT_DIR=".$ZIT_FILE.git"
54                 if ! test -d "$GIT_DIR"; then
55                         test -d "$ZIT_DIR" && GIT_DIR="$ZIT_DIR/$ZIT_FILE.git"
56                 fi
57         fi
58         export GIT_DIR
59 }
60
61 zit_init() {
62         if test "$1"; then
63                 zit_setup $1
64                 test -e $GIT_DIR && abort "$GIT_DIR exists, is $ZIT_FILE tracked already?"
65                 mkdir $GIT_DIR && echo "Initializing Zit repository in $GIT_DIR"
66                 test -d $GIT_DIR || abort "Failed to create $GIT_DIR"
67                 git init || abort "Failed to initialize Git repository in $GIT_DIR" 
68                 echo "*" > $GIT_DIR/info/exclude
69                 git add -f $ZIT_FILE || abort "Failed to add $ZIT_FILE"
70                 git commit "$@" || abort "Failed to make first commit for $ZIT_FILE"
71         else
72                 if test -d "$ZIT_DIR"; then
73                         echo "$ZIT_DIR exists already"
74                         exit
75                 fi
76                 test -e $ZIT_DIR && abort "$ZIT_DIR exists but it's not a directory, cannot continue"
77                 mkdir $ZIT_DIR
78         fi
79 }
80
81 zit_list() {
82         export GIT_WORK_TREE="`pwd`"
83         for file in .*.git; do
84                 if ! test -e $file; then
85                         echo "(no files tracked by zit)"
86                         break
87                 fi
88                 export GIT_DIR="$file"
89                 file="${file#.}"
90                 file="${file%.git}"
91                 # TODO show actual file status
92                 git ls-files -t $file
93         done;
94 }
95
96 case $cmd in
97         "")
98         echo $USAGE
99         ;;
100         help)
101         zit_help "$@"
102         ;;
103         init|track)
104         zit_init $1
105         ;;
106         list|tracked)
107         zit_list
108         ;;
109         # Most commands will work with the generic catch-all mechanism used
110         # below, but some of them require a more thorough analysis of the
111         # parameters to decide whether $ZIT_FILE should be put back into the
112         # parameter list or not. For example,
113         # $ zit commit somefile
114         # wouldn't do what one expects it to do, unless 'add' is run first,
115         # (except that
116         # $ zit add somefile
117         # wouldn't work either), however
118         # $ zit commit somefile -a
119         # would work correctly. So we handle some commands separately (for the
120         # moment just add and commit)
121         add|commit)
122         zit_setup $1
123         shift
124         git $cmd "$@" "$ZIT_FILE"
125         ;;
126         # the raw<command> method can be used to not replicate $ZIT_FILE in the
127         # parameter list
128         raw*)
129         zit_setup $1
130         shift
131         git ${cmd#raw} "$@"
132         ;;
133         *)
134         zit_setup $1
135         shift
136         git $cmd "$@"
137         ;;
138 esac