Merge branch 'master' of git+ssh://yeti.esilibrary.com/home/evergreen/evergreen-equin...
[working/Evergreen.git] / build / tools / update_db.sh
1 #!/bin/bash
2 #
3 # Based on a script by Bill Erickson.
4 #
5 # TODO:
6 #   ADD OPTIONS:
7 #     ~ single-step mode that calls psql -f once per file
8 #          (but also prompts for password once per file).
9 #     ~ help/usage option
10
11 DB_HOST=$1
12 DB_USER=$2
13 DB_NAME=$3
14
15 function usage() {
16     cat <<END_OF_USAGE
17 usage: $0  db_host  db_user  db_name
18
19 Automtically update the DB with all numbered updates beyond the last installed one.
20
21 ALL parameters are required to access the postgres database.
22
23 PARAMETERS:
24   db_host - database host system (e.g. "localhost" or "10.121.99.6")
25   db_user - database username
26   db_name - database name
27     
28 Run from your source repository root or Open-ILS/src/sql/Pg directory.
29
30 You will be prompted for the postgres password if necessary.
31
32 END_OF_USAGE
33 }
34
35 function die() {
36     echo "ERROR: $1" >&2;
37     exit 1;
38 }
39
40 function usage_die() {
41     exec >&2;
42     echo;
43     echo "ERROR: $1";
44     echo;
45     usage;
46     exit 1;
47 }
48
49 function feedback() {
50     #TODO: add a test and verbose mode that use this.
51     echo "Updating database $DB_NAME on $DB_HOST as user $DB_USER";
52 }
53
54 [ -z "$DB_HOST" -o -z "$DB_USER" -o -z "$DB_NAME" ] && usage_die "Need all DB parameters";
55
56 PSQL_ACCESS="-h $DB_HOST -U $DB_USER $DB_NAME";
57
58 # Need to avoid versions like '1.6.0.4' from throwing off the upgrade
59 VERSION=$(psql -c "SELECT MAX(version) FROM config.upgrade_log WHERE version ~ E'^\\\\d+$'" -t $PSQL_ACCESS);
60 [  $? -gt 0  ] && die "Database access failed.";
61 # [ $VERBOSE ] && echo RAW VERSION: $VERSION     # TODO: for verbose mode
62 VERSION=$(echo $VERSION | sed -e 's/^ *0*//');    # This is a separate step so we can check $? above.
63 [ -z "$VERSION" ] && usage_die "config.upgrade_log missing ANY installed version data!";
64 echo "* Last installed version  ->  $VERSION";
65
66 if [ -d ./Open-ILS/src/sql/Pg ] ; then
67     cd ./Open-ILS/src/sql/Pg ;
68 fi
69 [ -d ./upgrade ] || usage_die "No ./upgrade directory found.  Please run from Open-ILS/src/sql/Pg";
70
71 MAX=$(ls upgrade/[0-9][0-9][0-9][0-9]* 2>/dev/null | tail -1 | cut -c9-12 );   # could take an optional arg to set this, if we wanted.
72 echo "* Last upgrade file found -> $MAX";
73 MAX=$(echo $MAX | sed -e 's/^ *0*//');      # remove leading zeroes
74
75 declare -a FILES;
76 declare -a SKIPPED;
77 while true; do
78     VERSION=$(($VERSION + 1));
79     [ $VERSION -gt $MAX ] && break;
80     PREFIX=$(printf "%0.4d" $VERSION);
81     FILE=$(ls upgrade/$PREFIX* 2>/dev/null);
82     if [ -f "$FILE" ] ; then
83         # Note: we only report skipped files once we find the next one.  
84         # Otherwise, we'd report everything from $VERSION+1 to $MAX
85         for skip in ${SKIPPED[@]} ; do
86             echo "* WARNING: Upgrade $skip NOT FOUND.  Skipping it."; 
87         done
88         SKIPPED=();                     # After we reported them, reset array.
89         FILES[${#FILES[@]}]=$FILE;      # "push" onto FILES array
90         # echo "* Pending $FILE";
91     else
92         SKIPPED[${#SKIPPED[@]}]=$PREFIX; # "push" onto SKIPPED array
93     fi
94 done;
95
96 COUNT=${#FILES[@]};
97
98 if [ $COUNT -gt 0 ] ; then
99     echo "* $COUNT update scripts to apply."
100     exec 3>&1;  # our copy of STDOUT
101     for (( i=0; i<$COUNT; i++ )) ; do
102         echo "* Applying ${FILES[$i]}" >&3;   # to the main script STDOUT
103         cat ${FILES[$i]};                     # to the psql pipe
104     done | psql $PSQL_ACCESS ;
105 else
106     echo "* Nothing to update";
107 fi