debian/scripts/koha-rebuild-zebra

   1 #!/bin/sh
   2 #
   3 # koha-rebuild-zebra - Rebuild the Zebra database for Koha instances.
   4 # Copyright 2010  Catalyst IT, Ltd
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18
  19
  20 set -e
  21
  22 die()
  23 {
  24     echo "$@" 1>&2
  25     exit 1
  26 }
  27
  28 warn()
  29 {
  30     echo "$@" 1>&2
  31 }
  32
  33 is_instance()
  34 {
  35     local instancename=$1
  36
  37     if find /etc/koha/sites -mindepth 1 -maxdepth 1 \
  38                          -type d -printf '%f\n'\
  39           | grep -q -x $instancename ; then
  40         return 0
  41     else
  42         return 1
  43     fi
  44 }
  45
  46 toggle_biblios_only()
  47 {
  48     biblios_only="yes"
  49     biblios="yes"
  50     if [ "$authorities_only" != "yes" ]; then
  51         authorities="no"
  52     fi
  53 }
  54
  55 toggle_authorities_only()
  56 {
  57     authorities_only="yes"
  58     authorities="yes"
  59     if [ "$biblios_only" != "yes" ]; then
  60         biblios="no"
  61     fi
  62 }
  63
  64 run_rebuild_zebra()
  65 {
  66     local instancename=$1; shift
  67
  68     # TODO: This comment is here to remind us that we should make
  69     # rebuild_zebra.pl return error codes on failure
  70     if sudo -u "$instancename-koha" -H \
  71         env PERL5LIB=/usr/share/koha/lib \
  72         KOHA_CONF="/etc/koha/sites/$instancename/koha-conf.xml" \
  73         /usr/share/koha/bin/migration_tools/rebuild_zebra.pl $@ ; then
  74         return 0
  75     else
  76         return 1
  77     fi
  78 }
  79
  80 usage()
  81 {
  82     local scriptname=$0
  83     cat <<EOF
  84 Rebuild the Zebra indexes for Koha instances. The default behaviour
  85 is to do an incremental rebuild.
  86
  87 Usage: $scriptname [options] instancename1 instancename2...
  88 Options:
  89     --usmarc|-u       Runs the process as USMARC rather than
  90                       the default of MARCXML.
  91     --authorities|-a  Only run process for authorities.
  92     --biblios|-b      Only run process for biblios.
  93     --full|-f         Does a reindex of the whole collection.
  94     --verbose|-v      Be verbose.
  95     --help|-h         Print this help.
  96
  97
  98 Note: Any other options are passed directly to rebuild_zebra.pl.
  99 EOF
 100 }
 101
 102 # Default parameters
 103 opt_idx="-z"
 104 opt_xml="-x"
 105 opt_verbose=""
 106 opts_other=""
 107 biblios_only="no"
 108 authorities_only="no"
 109 biblios="yes"
 110 authorities="yes"
 111
 112 # Read parameters
 113 while [ -n "$*" ]; do
 114     case "$1" in
 115         -h|--help)
 116             usage ; exit 0
 117             ;;
 118         -b|--biblios)
 119             toggle_biblios_only
 120             ;;
 121         -a|--authorities)
 122             toggle_authorities_only
 123             ;;
 124         -u|--usmarc)
 125             opt_xml=""
 126             ;;
 127         -f|--full)
 128             opt_idx="-r"
 129             ;;
 130         -v|--verbose)
 131             opt_verbose="-v"
 132             ;;
 133         -*)
 134             opts_other="$opts_other $1";
 135             ;;
 136         *)
 137             break
 138             ;;
 139     esac
 140
 141     shift
 142 done
 143
 144 # Parse command line.
 145 [ $# -ge 1 ] || ( usage ; die "Missing instance name..." )
 146
 147 # Loop over instance names
 148 for name in "$@"
 149 do
 150     if is_instance $name; then
 151         if [ "$biblios" = "yes" ]; then
 152             if ! run_rebuild_zebra $name \
 153                 -b $opt_verbose $opt_idx $opt_xml $opts_other; then
 154                 warn "Something went wrong rebuilding biblio indexes for $name"
 155             fi
 156         fi
 157         if [ "$authorities" = "yes" ]; then
 158             if ! run_rebuild_zebra $name \
 159                 -a $opt_verbose $opt_idx $opts_other ; then
 160                 warn "Something went wrong rebuilding authority indexes for $name"
 161             fi
 162         fi
 163     else
 164         warn "Unknown instance $name."
 165     fi
 166 done
 167
 168 exit 0