From f1aa1db3fb402694f9b544734125f410eb657cdc Mon Sep 17 00:00:00 2001
From: Jonas Pedersen <jonas@pedersen.ninja>
Date: Mon, 24 Jun 2019 17:06:29 +0200
Subject: [PATCH] Add support for DOMAIN and WORKGROUP.

---
 README.md     |  7 +++++++
 docker-cmd.sh | 12 +++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 5e13e2d..0695b92 100644
--- a/README.md
+++ b/README.md
@@ -3,6 +3,13 @@ Docker image for wsdd.py.
 
 wsdd implements a Web Service Discovery host daemon. This enables (Samba) hosts, like your local NAS device or Linux server, to be found by Web Service Discovery Clients like Windows.
 
+## Supported environment variables
+HOSTNAME: Samba Netbios name to report.
+
+WORKGROUP: Workgroup name
+
+DOMAIN: Report being a member of an AD DOMAIN. Disables WORKGROUP if set. 
+
 ## Running container
 ### From command line
 ```
diff --git a/docker-cmd.sh b/docker-cmd.sh
index 6abc6fe..edf915e 100755
--- a/docker-cmd.sh
+++ b/docker-cmd.sh
@@ -3,11 +3,21 @@
 args=( )
 
 if [ ! -z "${HOSTNAME}" ]; then
-	args+=( "-n $HOSTNAME")
+	args+=( "-n $HOSTNAME ")
 else 
 	echo "HOSTNAME environment variable must be set."
 	exit 1
 fi
 
+if  [ ! -z "${WORKGROUP}" ]; then
+	args+=( "-w $WORKGROUP " )
+fi
+
+if  [ ! -z "${DOMAIN}" ]; then
+	args+=( "-d $DOMAIN " )
+fi
+
+
+
 python wsdd.py ${args}