X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=modules%2Froles%2Ffiles%2Fstatic-mirroring%2Fstatic-master-ssh-wrap;h=da162159ac51d4935c03228012b0cbb3f1120524;hb=49428e397bab9eb9ccc420a3195cdb69b5bee86b;hp=94a41f407ded12c4bd3f48ebb8dc5410a111b1de;hpb=5cdbc2499b81d21abd0bcdf4df203f57a507a6ae;p=mirror%2Fdsa-puppet.git

diff --git a/modules/roles/files/static-mirroring/static-master-ssh-wrap b/modules/roles/files/static-mirroring/static-master-ssh-wrap
index 94a41f407..da162159a 100755
--- a/modules/roles/files/static-mirroring/static-master-ssh-wrap
+++ b/modules/roles/files/static-mirroring/static-master-ssh-wrap
@@ -72,22 +72,26 @@ lock() {
 
 do_rsync() {
 	local remote_host="$1"; shift
-
-	local args="--server --sender -vltrze.iLsf --safe-links ."
-	if [ "$*" = "$args -new-/" ] ; then
-		local path="$BASEDIR/current-push"
-		info "serving $remote_host with $path"
-		rsync $args "$path"
-	elif [ "$*" = "$args . -live-/" ] ; then
-		local path="$BASEDIR/current-live"
-		info "host $remote_host wants $path, acquiring lock"
-		lock 200 "$path" 0
-		rsync $args "$path"
-	else
-		info "NOT allowed for $remote_host: rsync $*"
-		echo >&2 "This rsync command ($@) not allowed."
-		exit 1
-	fi
+	local args="--server --sender -vlHtrze.iLsf --safe-links ."
+
+	for component in $(awk -v this_host="$(hostname -f)" '$1 == this_host {print $2}' $COMPONENTLIST); do
+	  if [ "$*" = "$args $component/-new-/" ] || [ "$*" = "$args ./$component/-new-/" ] ; then
+		  local path="$BASEDIR/$component-current-push"
+		  info "serving $remote_host with $path"
+		  rsync $args "$path/."
+		  return
+	  elif [ "$*" = "$args . $component/-live-/" ] || [ "$*" = "$args . ./$component/-live-/" ] ; then
+		  local path="$BASEDIR/$component-current-live"
+		  info "host $remote_host wants $path, acquiring lock"
+		  lock 200 "$path" 0
+		  rsync $args "$path/."
+		  return
+	  fi
+	done
+
+	info "NOT allowed for $remote_host: rsync $*"
+	echo >&2 "This rsync command ($@) not allowed."
+	exit 1
 }
 
 do_update_component() {
@@ -97,7 +101,23 @@ do_update_component() {
 	component="$1"
 	shift
 
-	hit="$(awk -v component="$component" -v host="$remote_host" '$1 == component && $2 == host {print $3; exit}' "$COMPONENTLIST")"
+	hit="$(
+		awk -v this_host="$(hostname -f)" -v component="$component" -v host="$remote_host" '
+		  $1 == this_host && $2 == component {
+			  if ($3 == host) {
+				  print $4
+				  exit
+			  }
+			  split($5,extra,",")
+			  for (i in extra) {
+				  if (host == extra[i]) {
+					  printf "%s:%s\n", $3, $4
+					  exit
+				  }
+			  }
+			  exit
+		  }' "$COMPONENTLIST"
+		)"
 	if [ -n "$hit" ]; then
 		exec static-master-update-component "$component"
 		echo >&2 "Exec failed"