From 4bb1f4c9f1b6cbd25a9daace7912c3ab443fc4ed Mon Sep 17 00:00:00 2001
From: Theophilus <isahtheophilus@gmail.com>
Date: Fri, 17 Nov 2023 20:02:21 +0100
Subject: [PATCH 1/4] fix: change to config structure made in
 https://github.com/bitcoin-dev-project/sim-ln/pull/152

---
 docker/setup-volume.sh | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/docker/setup-volume.sh b/docker/setup-volume.sh
index b46141da..40ac50e0 100755
--- a/docker/setup-volume.sh
+++ b/docker/setup-volume.sh
@@ -28,11 +28,10 @@ NODE_COUNT=$(cat $SIMFILE_PATH_ON_HOST | jq '.nodes | length')
 
 # Loop Over Each Node
 for (( i=0; i<$NODE_COUNT; i++ )); do
-    NODE_TYPE=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i] | keys[0]") # Determine if it's LND or CLN.
-    NODE_ID=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i].$NODE_TYPE.id") # Extract node ID for directory creation.
-    NODE_TLS_PATH_ON_HOST=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i].$NODE_TYPE.cert") # TLS path
-    NODE_MACAROON_PATH_ON_HOST=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i].$NODE_TYPE.macaroon") # Macaroon path
-    
+    NODE_ID=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i].id") # Extract node ID for directory creation.
+    NODE_TLS_PATH_ON_HOST=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i].cert") # TLS path
+    NODE_MACAROON_PATH_ON_HOST=$(cat $SIMFILE_PATH_ON_HOST | jq -r ".nodes[$i].macaroon") # Macaroon path
+   
     # Create staging directories for each node
     mkdir -p $STAGING_DIR/lnd/$NODE_ID
 

From a495e9d41ca6a7e006188d7adb04cc7747166a06 Mon Sep 17 00:00:00 2001
From: Theophilus <isahtheophilus@gmail.com>
Date: Fri, 17 Nov 2023 20:05:27 +0100
Subject: [PATCH 2/4] feat: update to path specification in
 https://github.com/bitcoin-dev-project/sim-ln/pull/132

---
 Makefile               | 17 ++++++++++-------
 docker/README.md       |  1 +
 docker/entrypoint.sh   |  7 ++++++-
 docker/setup-volume.sh |  6 +++---
 4 files changed, 20 insertions(+), 11 deletions(-)

diff --git a/Makefile b/Makefile
index 68777f3f..c4b779eb 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,6 @@
 LOG_LEVEL ?= info
+DEFAULT_DATA_DIR ?= /data_dir
+DEFAULT_SIMFILE_PATH ?= /data_dir/sim.json
 
 build-docker:
 	docker build -f docker/Dockerfile -t sim-ln .
@@ -15,17 +17,18 @@ help:
 	@echo "stop              Stops the Docker container."
 	@echo ""
 	@echo "Variables:"
-	@echo "SIMFILE_PATH       Path to the sim.json file."
-	@echo "LOG_LEVEL         Set the logging level (default: info) e.g. <make run LOG_LEVEL=debug>"
-	@echo "HELP              Set to true to print the help message (default: false) e.g. <make run HELP=true>"
-	@echo "PRINT_BATCH_SIZE  Set the batch size for printing the results e.g. <make run PRINT_BATCH_SIZE=100>"
-	@echo "TOTAL_TIME        Set the total time for the simulation e.g. <make run TOTAL_TIME=1000>"
+	@echo "SIMFILE_PATH      Path to the sim.json file."
+	@echo "LOG_LEVEL         Set the logging level (default: info) e.g. <make run LOG_LEVEL=debug>."
+	@echo "HELP              Set to true to print the help message (default: false) e.g. <make run HELP=true>."
+	@echo "PRINT_BATCH_SIZE  Set the batch size for printing the results e.g. <make run PRINT_BATCH_SIZE=100>."
+	@echo "TOTAL_TIME        Set the total time for the simulation e.g. <make run TOTAL_TIME=1000>."
+	@echo "DATA_DIR          Set the data directory for the simulation containing simulation files and results e.g. <make run-docker DATA_DIR="/Users/anon/data_dir>"."
 
 run-docker:
-	docker run -d --rm --name sim-ln --init -v simln-data:/data -e SIMFILE_PATH=/data/sim.json -e LOG_LEVEL=$(LOG_LEVEL) -e HELP=${HELP} -e PRINT_BATCH_SIZE=${PRINT_BATCH_SIZE} -e TOTAL_TIME=${TOTAL_TIME} sim-ln
+	docker run -d --rm --name sim-ln --init -v simln-data:${DEFAULT_DATA_DIR} -e SIMFILE_PATH=${DEFAULT_SIMFILE_PATH} -e LOG_LEVEL=$(LOG_LEVEL) -e HELP=${HELP} -e PRINT_BATCH_SIZE=${PRINT_BATCH_SIZE} -e TOTAL_TIME=${TOTAL_TIME} sim-ln
 
 run-interactive:
-	docker run --rm --name sim-ln --init -v simln-data:/data -e SIMFILE_PATH=/data/sim.json -it sim-ln
+	docker run -it --rm --name sim-ln --init -v simln-data:${DEFAULT_DATA_DIR} -e SIMFILE_PATH=${DEFAULT_SIMFILE_PATH} -e DATA_DIR=${DATA_DIR} -e LOG_LEVEL=$(LOG_LEVEL) -e HELP=${HELP} -e PRINT_BATCH_SIZE=${PRINT_BATCH_SIZE} -e TOTAL_TIME=${TOTAL_TIME} sim-ln
 
 stop-docker:
 	docker stop sim-ln
\ No newline at end of file
diff --git a/docker/README.md b/docker/README.md
index 19500934..a9d3c4eb 100644
--- a/docker/README.md
+++ b/docker/README.md
@@ -81,6 +81,7 @@ Other configurable variables include:
 - `HELP`: Set to `true` to print the help message.
 - `PRINT_BATCH_SIZE`: determines the number of payment results that will be written to disk at a time.
 - `TOTAL_TIME`: the total runtime for the simulation expressed in seconds.
+- `DATA_DIR`: Path to a directory containing simulation files, and where simulation results will be stored (default is the `/data_dir` volume directory).
 
 Example usage:
 
diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index e850025c..781b918f 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -1,7 +1,12 @@
 #!/bin/sh
 
 # Define the start command
-START_COMMAND="/usr/local/bin/sim-cli $SIMFILE_PATH"
+START_COMMAND="/usr/local/bin/sim-cli --sim-file $SIMFILE_PATH"
+
+# Check if a custom data directory was provided
+if [[ ! -z ${DATA_DIR} ]]; then
+    START_COMMAND="$START_COMMAND --data-dir $DATA_DIR"
+fi
 
 # Check for version arg
 if [[ ! -z ${VERSION} ]]; then
diff --git a/docker/setup-volume.sh b/docker/setup-volume.sh
index 40ac50e0..57afdf01 100755
--- a/docker/setup-volume.sh
+++ b/docker/setup-volume.sh
@@ -40,10 +40,10 @@ for (( i=0; i<$NODE_COUNT; i++ )); do
     cp $NODE_MACAROON_PATH_ON_HOST $STAGING_DIR/lnd/$NODE_ID/admin.macaroon
 
     # Adjust the paths in the staging sim.json so we don't use the host path
-    sed -i '' 's|'$(dirname $NODE_TLS_PATH_ON_HOST)'/tls.cert|/data/lnd/'$NODE_ID'/tls.cert|' $STAGING_DIR/sim.json
-    sed -i '' 's|'$(dirname $NODE_MACAROON_PATH_ON_HOST)'/admin.macaroon|/data/lnd/'$NODE_ID'/admin.macaroon|' $STAGING_DIR/sim.json
+    sed -i '' 's|'$(dirname $NODE_TLS_PATH_ON_HOST)'/tls.cert|/data_dir/lnd/'$NODE_ID'/tls.cert|' $STAGING_DIR/sim.json
+    sed -i '' 's|'$(dirname $NODE_MACAROON_PATH_ON_HOST)'/admin.macaroon|/data_dir/lnd/'$NODE_ID'/admin.macaroon|' $STAGING_DIR/sim.json
 done
 
 # Create Docker volume and copy the data
 docker volume create $VOLUME_NAME
-docker run --rm -v $VOLUME_NAME:/data -v $STAGING_DIR:/staging alpine sh -c 'cp -r /staging/* /data/'
+docker run --rm -v $VOLUME_NAME:/data_dir -v $STAGING_DIR:/staging alpine sh -c 'cp -r /staging/* /data_dir/'

From bbb168da101ecd7f79977cee08389a54f54ae459 Mon Sep 17 00:00:00 2001
From: Theophilus <isahtheophilus@gmail.com>
Date: Thu, 25 Jan 2024 10:19:36 +0100
Subject: [PATCH 3/4] docs: fix docker run commands

---
 docker/README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/README.md b/docker/README.md
index a9d3c4eb..9de43807 100644
--- a/docker/README.md
+++ b/docker/README.md
@@ -73,7 +73,7 @@ make run-docker
 You can adjust the logging level by providing the `LOG_LEVEL` variable. The default value is `info`. Example:
 
 ```bash
-make run LOG_LEVEL=debug
+make run-docker LOG_LEVEL=debug
 ```
 
 Other configurable variables include:
@@ -86,7 +86,7 @@ Other configurable variables include:
 Example usage:
 
 ```bash
-make run PRINT_BATCH_SIZE=100 TOTAL_TIME=5000
+make run-docker PRINT_BATCH_SIZE=100 TOTAL_TIME=5000
 ```
 
 For an interactive session:

From bbda19fe0e038ce97418e688a2037bd5dd17d823 Mon Sep 17 00:00:00 2001
From: Theophilus <isahtheophilus@gmail.com>
Date: Thu, 25 Jan 2024 11:26:11 +0100
Subject: [PATCH 4/4] chore: update Dockerfile to clean up apt-get cache

This commit updates the Dockerfile to clean up the apt-get cache after installing clang and llvm. This helps reduce the size of the Docker image.
---
 docker/Dockerfile | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 7453bb5f..0a8acdfa 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -10,7 +10,10 @@ ENV TARGET_RUST_ARCH="x86_64-unknown-linux-musl"
 
 FROM base as builder-arm64
 ENV TARGET_RUST_ARCH="aarch64-unknown-linux-musl"
-RUN apt-get update && apt-get install clang llvm -y --no-install-recommends
+RUN apt-get update \
+    && apt-get install clang llvm -y --no-install-recommends \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
 ENV CC_aarch64_unknown_linux_musl=clang
 ENV AR_aarch64_unknown_linux_musl=llvm-ar
 ENV CARGO_TARGET_AARCH64_UNKNOWN_LINUX_MUSL_RUSTFLAGS="-Clink-self-contained=yes -Clinker=rust-lld"