updated to add xsede tests back in for when it's an isolated checkout.
[feisty_meow.git] / scripts / rev_control / version_control.sh
1 #!/bin/bash
2
3 # these are helper functions for doing localized revision control.
4 # this script should be sourced into other scripts that use it.
5
6 # one unpleasantry to take care of first; cygwin barfs aggressively if the TMP directory
7 # is a DOS path, but we need it to be a DOS path for our XSEDE testing, so that blows.
8 # to get past this, TMP gets changed below to a hopefully generic and safe place.
9
10 if [[ "$TMP" =~ .:.* ]]; then
11   echo making weirdo temporary directory for DOS path.
12   export TMP=/tmp/rev_control_$USER
13 fi
14 if [ ! -d "$TMP" ]; then
15   mkdir $TMP
16 fi
17 if [ ! -d "$TMP" ]; then
18   echo "Could not create the temporary directory TMP in: $TMP"
19   echo "This script will not work properly without an existing TMP directory."
20 fi
21
22 this_host=
23 # gets the machine's hostname and stores it in the variable "this_host".
24 function get_our_hostname()
25 {
26   if [ "$OS" == "Windows_NT" ]; then
27     this_host=$(hostname)
28   elif [ ! -z "$(echo $MACHTYPE | grep apple)" ]; then
29     this_host=$(hostname)
30   elif [ ! -z "$(echo $MACHTYPE | grep suse)" ]; then
31     this_host=$(hostname --long)
32   else
33     this_host=$(hostname)
34   fi
35   #echo "hostname is $this_host"
36 }
37
38 # this function sets a variable called "home_system" to "true" if the
39 # machine is considered one of fred's home machines.  if you are not
40 # fred, you may want to change the machine choices.
41 export home_system=
42 function is_home_system()
43 {
44   # load up the name of the host.
45   get_our_hostname
46   # reset the variable that we'll be setting.
47   home_system=
48   if [[ $this_host == *.gruntose.blurgh ]]; then
49     home_system=true
50 #temp code
51 elif [[ $this_host == buildy ]]; then
52 home_system=true
53 elif [[ $this_host == simmy ]]; then
54 home_system=true
55 #temp code
56   fi
57 }
58
59 # we only want to totally personalize this script if the user is right.
60 function check_user()
61 {
62   if [ "$USER" == "fred" ]; then
63     export SVNUSER=fred_t_hamster@
64     export EXTRA_PROTOCOL=+ssh
65   else
66     export SVNUSER=
67     export EXTRA_PROTOCOL=
68   fi
69 }
70
71 # calculates the right modifier for hostnames / repositories.
72 modifier=
73 function compute_modifier()
74 {
75   modifier=
76   directory="$1"; shift
77   in_or_out="$1"; shift
78   check_user
79   # some project specific overrides.
80   if [[ "$directory" == hoople* ]]; then
81     modifier="svn${EXTRA_PROTOCOL}://${SVNUSER}svn.code.sf.net/p/hoople2/svn/"
82   fi
83   if [[ "$directory" == yeti* ]]; then
84     modifier="svn${EXTRA_PROTOCOL}://${SVNUSER}svn.code.sf.net/p/yeti/svn/"
85   fi
86   # see if we're on one of fred's home machines.
87   is_home_system
88   # special override to pick local servers when at home.
89   if [ "$home_system" == "true" ]; then
90     if [ "$in_or_out" == "out" ]; then
91       # need the right home machine for modifier when checking out.
92 #huhhh?      modifier="svn://shaggy/"
93       modifier=
94     else 
95       # no modifier for checkin.
96       modifier=
97     fi
98   fi
99 }
100
101 # selects the method for check-in based on where we are.
102 function do_checkin()
103 {
104   local directory="$1"; shift
105   pushd "$directory" &>/dev/null
106   retval=0  # normally successful.
107   if [ -d "CVS" ]; then
108     cvs ci .
109     retval=$?
110   elif [ -d ".svn" ]; then
111     svn ci .
112     retval=$?
113   elif [ -d ".git" ]; then
114     # snag all new files.  not to everyone's liking.
115     git add --all .
116     retval=$?
117     # tell git about all the files and get a check-in comment.
118     git commit .
119     retval+=$?
120     # upload the files to the server so others can see them.
121     git push 2>&1 | grep -v "X11 forwarding request failed"
122     retval+=$?
123   else
124     echo no repository in $directory
125     retval=1
126   fi
127   popd &>/dev/null
128   return $retval
129 }
130
131 function do_diff
132 {
133   local directory="$1"; shift
134   pushd "$directory" &>/dev/null
135   retval=0  # normally successful.
136
137   # only update if we see a repository living there.
138   if [ -d ".svn" ]; then
139     svn diff .
140   elif [ -d ".git" ]; then
141     git diff 
142   elif [ -d "CVS" ]; then
143     cvs diff .
144   fi
145
146   popd &>/dev/null
147   return $retval
148 }
149
150 function do_report_new
151 {
152   local directory="$1"; shift
153   pushd "$directory" &>/dev/null
154   retval=0  # normally successful.
155
156   # only update if we see a repository living there.
157   if [ -d ".svn" ]; then
158     # this action so far only makes sense and is needed for svn.
159     bash $FEISTY_MEOW_SCRIPTS/rev_control/svnapply.sh \? echo
160   fi
161
162   popd &>/dev/null
163   return $retval
164 }
165
166 # checks in all the folders in a specified list.
167 function checkin_list()
168 {
169   local list=$*
170   for i in $list; do
171     # turn repo list back into an array.
172     eval "repository_list=( ${REPOSITORY_LIST[*]} )"
173     for j in "${repository_list[@]}"; do
174       # add in the directory component.
175       j="$i/$j"
176       if [ ! -d "$j" ]; then continue; fi
177       echo "checking in '$j'..."
178       do_checkin $j
179     done
180   done
181 }
182
183 # takes out the first few carriage returns that are in the input.
184 function squash_first_few_crs()
185 {
186   i=0
187   while read line; do
188     i=$((i+1))
189     if [ $i -le 3 ]; then
190       echo -n "$line  "
191     else
192       echo $line
193     fi
194   done
195   if [ $i -le 3 ]; then
196     # if we're still squashing eols, make sure we don't leave them hanging.
197     echo
198   fi
199 }
200
201 # selects the checkout method based on where we are (the host the script runs on).
202 function do_update()
203 {
204   directory="$1"; shift
205  
206   pushd "$directory" &>/dev/null
207   if [ -d "CVS" ]; then
208     cvs update . | squash_first_few_crs
209   elif [ -d ".svn" ]; then
210     svn update . | squash_first_few_crs
211   elif [ -d ".git" ]; then
212     git pull 2>&1 | grep -v "X11 forwarding request failed" | squash_first_few_crs
213   else
214     echo unknown repository for $directory...
215   fi
216   popd &>/dev/null
217 }
218
219 # gets all the updates for a list of folders under revision control.
220 function checkout_list {
221   list=$*
222   for i in $list; do
223     # turn repo list back into an array.
224     eval "repository_list=( ${REPOSITORY_LIST[*]} )"
225     for j in "${repository_list[@]}"; do
226       # add in the directory for our purposes here.
227       j="$i/$j"
228       if [ ! -d $j ]; then
229         if [ ! -z "$SHELL_DEBUG" ]; then
230           echo "No directory called $j exists."
231         fi
232         continue
233       fi
234
235       echo -n "retrieving '$j'...  "
236       do_update $j
237     done
238   done
239 }
240
241 # provides a list of absolute paths of revision control directories
242 # that are located under the directory passed as the first parameter.
243 function generate_rev_ctrl_filelist()
244 {
245   local dir="$1"; shift
246   pushd "$dir" &>/dev/null
247   local dirhere="$(\pwd)"
248   local tempfile=$(mktemp /tmp/zz_rev_checkin.XXXXXX)
249   echo >$tempfile
250   find $dirhere -maxdepth 4 -type d -iname ".svn" -exec echo {}/.. ';' >>$tempfile
251   find $dirhere -maxdepth 4 -type d -iname ".git" -exec echo {}/.. ';' >>$tempfile
252 #CVS is not well behaved, and we seldom use it anymore.
253 #  find $dirhere -maxdepth 3 -type d -iname "CVS" -exec echo {}/.. ';' >>$tempfile
254   popd &>/dev/null
255   echo "$tempfile"
256 }
257
258 # iterates across a list of directories contained in a file (first parameter).
259 # on each directory name, it performs the action (second parameter) provided.
260 function perform_action_on_file()
261 {
262   local tempfile="$1"; shift
263   local action="$1"; shift
264
265   dirs=($(cat $tempfile))
266
267   for dirname in ${dirs[@]}; do
268     if [ -z "$dirname" ]; then continue; fi
269     pushd $dirname &>/dev/null
270     echo "[$(pwd)]"
271     $action .
272     echo "======="
273     popd &>/dev/null
274   done 
275
276   rm $tempfile
277 }
278
279