From d319c68a1082af0152e7eab671f190885f651111 Mon Sep 17 00:00:00 2001
From: Dan Caseley <dan@caseley.me.uk>
Date: Mon, 20 Dec 2021 11:16:40 +0000
Subject: [PATCH] Update blocking to ignore fixed names

Docker Compose may use hyphens or underscores, depending on platform or version
---
 block_federation.sh        | 2 +-
 block_node_from_cluster.sh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/block_federation.sh b/block_federation.sh
index 70b7ec1..2613899 100755
--- a/block_federation.sh
+++ b/block_federation.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-CONTAINER_TO_REMOVE=openfire-docker-compose-xmpp2-1
+CONTAINER_TO_REMOVE=$(docker ps --filter status=running --format "{{.Names}}" | grep -E openfire-docker.+xmpp2.1)
 CONTAINER_NAME=pumba_node2
 
 echo "About to be blocked from communication: $CONTAINER_TO_REMOVE"
diff --git a/block_node_from_cluster.sh b/block_node_from_cluster.sh
index 60da2cb..10c4124 100755
--- a/block_node_from_cluster.sh
+++ b/block_node_from_cluster.sh
@@ -4,7 +4,7 @@
 #Example usage: `./remove_node_from_cluster.sh 1` to remove node xmpp1
 
 NODE=$1
-CONTAINER_TO_REMOVE=openfire-docker-compose-xmpp"$NODE"-1
+CONTAINER_TO_REMOVE=$(docker ps --filter status=running --format "{{.Names}}" | grep -E openfire-docker.+xmpp"$NODE".1)
 CONTAINER_NAME=pumba_node"$NODE"
 
 echo "About to be removed from cluster: $CONTAINER_TO_REMOVE"