From ff188179312a593ca2cca8ac14ee0861ab3600d0 Mon Sep 17 00:00:00 2001
From: David Personette <dperson@gmail.com>
Date: Fri, 12 Feb 2016 18:13:20 -0500
Subject: [PATCH] samba.sh add option to set the workgroup/domain

---
 README.md |  4 ++++
 samba.sh  | 17 +++++++++++++++--
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 6d0e474..b888504 100644
--- a/README.md
+++ b/README.md
@@ -49,6 +49,9 @@ OR set local storage:
                     required arg: "<username>;<passwd>"
                     <username> for user
                     <password> for user
+        -w "<workgroup>"       Configure the workgroup (domain) samba should use
+                    required arg: "<workgroup>"
+                    <workgroup> for samba
 
     The 'command' (if provided and valid) will be run instead of samba
 
@@ -56,6 +59,7 @@ ENVIRONMENT VARIABLES (only available with `docker run`)
 
  * `NMBD` - As above, enable nmbd
  * `TZ` - As above, set a zoneinfo timezone, IE `EST5EDT`
+ * `WORKGROUP` - As above, set workgroup
 
 **NOTE**: if you enable nmbd (via `-n` or the `NMBD` environment variable), you
 will also want to expose port 137 with `-p 137:137`.
diff --git a/samba.sh b/samba.sh
index d306886..fbdec3d 100755
--- a/samba.sh
+++ b/samba.sh
@@ -82,6 +82,14 @@ user() { local name="${1}" passwd="${2}"
     echo "$passwd" | tee - | smbpasswd -s -a "$name"
 }
 
+### workgroup: set the workgroup
+# Arguments:
+#   workgroup) the name to set
+# Return: configure the correct workgroup
+user() { local workgroup="${1}" file=/etc/samba/smb.conf
+    sed -i 's/^\( *workgroup = \).*/\1'"$workgroup"'/' $file
+}
+
 ### usage: Help
 # Arguments:
 #   none)
@@ -109,20 +117,24 @@ Options (fields in '[]' are optional, '<>' are required):
                 required arg: \"<username>;<passwd>\"
                 <username> for user
                 <password> for user
+    -w \"<workgroup>\"       Configure the workgroup (domain) samba should use
+                required arg: \"<workgroup>\"
+                <workgroup> for samba
 
 The 'command' (if provided and valid) will be run instead of samba
 " >&2
     exit $RC
 }
 
-while getopts ":hi:nt:u:s:" opt; do
+while getopts ":hi:ns:t:u:w:" opt; do
     case "$opt" in
         h) usage ;;
         i) import "$OPTARG" ;;
         n) NMBD="true" ;;
         s) eval share $(sed 's/^\|$/"/g; s/;/" "/g' <<< $OPTARG) ;;
-        u) eval user $(sed 's/;/ /g' <<< $OPTARG) ;;
         t) timezone "$OPTARG" ;;
+        u) eval user $(sed 's/;/ /g' <<< $OPTARG) ;;
+        w) workgroup "$OPTARG" ;;
         "?") echo "Unknown option: -$OPTARG"; usage 1 ;;
         ":") echo "No argument value for option: -$OPTARG"; usage 2 ;;
     esac
@@ -130,6 +142,7 @@ done
 shift $(( OPTIND - 1 ))
 
 [[ "${TZ:-""}" ]] && timezone "$TZ"
+[[ "${WORKGROUP:-""}" ]] && workgroup "$WORKGROUP"
 
 if [[ $# -ge 1 && -x $(which $1 2>&-) ]]; then
     exec "$@"