shlib.sh

   1 #!/bin/bash
   2 #
   3 # This is generic shell library for all the scripts used by Girocco;
   4 # most importantly, it introduces all the $cfg_* shell variables.
   5
   6
   7 # Import all the variables from Girocco::Config to the local environment,
   8 # prefixing them with 'cfg_'. E.g. $cfg_admin is admin's mail address now.
   9 __girocco_conf="$GIROCCO_CONF"
  10 [ -n "$__girocco_conf" ] || __girocco_conf="Girocco::Config"
  11 [ -z "$basedir" ] || __girocco_extrainc="-I$basedir"
  12 eval $(perl -I@basedir@ $__girocco_extrainc -M$__girocco_conf -le \
  13         'foreach (keys %Girocco::Config::) {
  14                 my $val = ${$Girocco::Config::{$_}}; $val ||= "";
  15                 print "cfg_$_=\"$val\"";
  16         }')
  17
  18
  19 git() { "$cfg_git_bin" "$@"; }
  20
  21 _addrlist() {
  22         _list=
  23         for _addr in "$@"; do
  24                 [ -z "$_list" ] || _list="$_list, "
  25                 _list="$_list$_addr"
  26         done
  27         echo "$_list"
  28 }
  29
  30 _sendmail() {
  31         _mailer="${cfg_sendmail_bin:-/usr/sbin/sendmail}"
  32         if [ -n "$cfg_sender" ]; then
  33                 "$_mailer" -i -f "$cfg_sender" "$@"
  34         else
  35                 "$_mailer" -i "$@"
  36         fi
  37 }
  38
  39 mail() {
  40         _subject=
  41         if [ "$1" = "-s" ]; then
  42                 shift
  43                 _subject="$1"
  44                 shift
  45         fi
  46         {
  47                 echo "From: \"$cfg_name\" ($cfg_title) <$cfg_admin>"
  48                 echo "To: $(_addrlist "$@")"
  49                 [ -z "$_subject" ] || echo "Subject: $_subject"
  50                 echo "Auto-Submitted: auto-generated"
  51                 echo ""
  52                 cat
  53         } | _sendmail "$@"
  54 }
  55
  56 # Supports both the bang CMD... and bang_eval CMD... functions
  57 # Called when the command fails
  58 # Should only be called from bang and bang_eval functions
  59 bang_err() {
  60         if ! [ -e .banged ]; then
  61                 bangmailok=true
  62                 ! [ -f HEAD -a -f config -a -d objects ] ||
  63                 bangmailok="$(GIT_DIR=. git config --bool gitweb.statusupdates 2>/dev/null || echo true)"
  64                 bangaddrs=''
  65                 [ "$bangmailok" = "false" -o -z "$mail" ] || bangaddrs="$mail"
  66                 [ -z "$cfg_admincc" -o "$cfg_admincc" = "0" -o -z "$cfg_admin" ] ||
  67                 if [ -z "$bangaddrs" ]; then bangaddrs="$cfg_admin"; else bangaddrs="$bangaddrs,$cfg_admin"; fi
  68                 [ -z "$bangaddrs" ] ||
  69                 {
  70                         echo "$* failed with error code $errcode"
  71                         echo ""
  72                         [ ! -n "$bang_once" ] || echo "you will not receive any more notifications until recovery"
  73                         echo "this status message may be disabled on the project admin page"
  74                         echo ""
  75                         echo "Log follows:"
  76                         echo ""
  77                         cat "$bang_log"
  78                 } | mail -s "[$cfg_name] $proj $bang_action failed" "$bangaddrs"
  79         fi
  80         touch .banged
  81         bang_trap
  82         exit 1
  83 }
  84
  85 # bang CMD... will execute the command with well-defined failure mode;
  86 # set bang_action to string of the failed action ('clone', 'update', ...);
  87 # pre-set bang_once=1 to make sure jobs banging on a repo repeatedly will
  88 # not spam the owner; re-define the bang_trap() function to do custom
  89 # cleanup before bailing out
  90 bang() {
  91         if [ -n "$show_progress" ]; then
  92                 exec 3>&1
  93                 errcode=
  94                 read -r errcode <<-EOT || :
  95                 $(
  96                         exec 4>&3 3>&1 1>&4 4>&-
  97                         { "$@" 3>&- || echo $? >&3; } 2>&1 | tee -a "$bang_log"
  98                 )
  99                 EOT
 100                 exec 3>&-
 101                 if [ -z "$errcode" ]; then
 102                         # All right. Cool.
 103                         return;
 104                 fi
 105         else
 106                 if "$@" >>"$bang_log" 2>&1; then
 107                         # All right. Cool.
 108                         return;
 109                 else
 110                         errcode="$?"
 111                 fi
 112         fi
 113         bang_err "$@"
 114 }
 115
 116 # bang_eval CMD... will evaluate the command with well-defined failure mode;
 117 # Identical to bang CMD... except the command is eval'd instead of executed.
 118 bang_eval() {
 119         if [ -n "$show_progress" ]; then
 120                 exec 3>&1
 121                 errcode=
 122                 read -r errcode <<-EOT || :
 123                 $(
 124                         exec 4>&3 3>&1 1>&4 4>&-
 125                         { eval "$*" 3>&- || echo $? >&3; } 2>&1 | tee -a "$bang_log"
 126                 )
 127                 EOT
 128                 exec 3>&-
 129                 if [ -z "$errcode" ]; then
 130                         # All right. Cool.
 131                         return;
 132                 fi
 133         else
 134                 if eval "$*" >>"$bang_log" 2>&1; then
 135                         # All right. Cool.
 136                         return;
 137                 else
 138                         errcode="$?"
 139                 fi
 140         fi
 141         bang_err "$@"
 142 }
 143
 144 # Default bang settings:
 145 bang_setup() {
 146         bang_action="lame_programmer"
 147         bang_once=
 148         bang_trap() { :; }
 149         bang_log="$(mktemp -t repomgr-XXXXXX)"
 150         trap "rm \"\$bang_log\"" EXIT
 151 }
 152
 153
 154 # Progress report - if show_progress is set, shows the given message.
 155 progress() {
 156         [ ! -n "$show_progress" ] || echo "$@"
 157 }
 158
 159
 160 # Project config accessors; must be run in project directory
 161 config_get() {
 162         git config "gitweb.$1"
 163 }
 164
 165 config_set() {
 166         git config "gitweb.$1" "$2" && chgrp repo config && chmod g+w config
 167 }
 168
 169 config_set_raw() {
 170         git config "$1" "$2"        && chgrp repo config && chmod g+w config
 171 }
 172
 173
 174 # Tool for checking whether given number of seconds has not passed yet
 175 check_interval() {
 176         od="$(config_get "$1" || :)"
 177         [ -n "$od" ] || return 1
 178         os="$(perl -I@basedir@ -MGirocco::Util -e "print parse_rfc2822_date('$od')")"
 179         [ -n "$os" ] || return 1
 180         ns="$(date +%s)"
 181         [ $ns -lt $(($os+$2)) ]
 182 }
 183
 184
 185 # List all Git repositories, with given prefix if specified, one-per-line
 186 get_repo_list() {
 187         if [ -n "$1" ]; then
 188                 cut -d : -f 1,3 "$cfg_chroot"/etc/group | grep "^$1"
 189         else
 190                 cut -d : -f 1,3 "$cfg_chroot"/etc/group
 191         fi | while IFS=: read name id; do
 192                 [ $id -lt 65536 ] || echo "$name"
 193         done
 194 }