From 08bc7667fc45311cd7daf9c23f684892cb6342fd Mon Sep 17 00:00:00 2001
From: Andrew Camilleri <evilkukka@gmail.com>
Date: Mon, 17 Sep 2018 10:20:49 +0200
Subject: [PATCH] allow to specify generator image when building

---
 build.ps1 | 3 ++-
 build.sh  | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/build.ps1 b/build.ps1
index 840d9a4..89ed908 100755
--- a/build.ps1
+++ b/build.ps1
@@ -1,6 +1,7 @@
 # This script will run docker-compose-generator in a container to generate the yml files
+if (-not ($BTCPAYGEN_DOCKER_IMAGE)) { $BTCPAYGEN_DOCKER_IMAGE = "btcpayserver/docker-compose-generator" }
 
-docker pull btcpayserver/docker-compose-generator
+docker pull $BTCPAYGEN_DOCKER_IMAGE:
 docker run -v "$(Get-Location)\Generated:/app/Generated" `
            -v "$(Get-Location)\docker-compose-generator\docker-fragments:/app/docker-fragments" `
            -e "BTCPAYGEN_CRYPTO1=$BTCPAYGEN_CRYPTO1" `
diff --git a/build.sh b/build.sh
index 62b1a65..eba13bd 100755
--- a/build.sh
+++ b/build.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # This script will run docker-compose-generator in a container to generate the yml files
-docker pull btcpayserver/docker-compose-generator
+docker pull ${BTCPAYGEN_DOCKER_IMAGE:-"btcpayserver/docker-compose-generator"}
 docker run -v "$(pwd)/Generated:/app/Generated" \
            -v "$(pwd)/docker-compose-generator/docker-fragments:/app/docker-fragments" \
            -e "BTCPAYGEN_CRYPTO1=$BTCPAYGEN_CRYPTO1" \