#!/usr/bin/env bash
set -Eeuo pipefail

: "${SAMBA:="Y"}"         # Enable Samba
: "${SAMBA_LEVEL:="1"}"   # Logging level
: "${SAMBA_DEBUG:="N"}"   # Disable debug

tmp="/tmp/smb"
rm -rf "$tmp"

rm -f /var/run/wsdd.pid
rm -f /var/run/samba/nmbd.pid
rm -f /var/run/samba/smbd.pid

[[ "$SAMBA" == [Nn]* ]] && return 0
[[ "$NETWORK" == [Nn]* ]] && return 0

if [[ "$DHCP" == [Yy1]* ]]; then
  socket="$IP"
  hostname="$IP"
  interfaces="$VM_NET_DEV"
else
  hostname="host.lan"
  case "${NETWORK,,}" in
    "passt" | "slirp" )
      interfaces="lo"
      socket="127.0.0.1" ;;
    *)
      socket="$VM_NET_IP"
      interfaces="$VM_NET_BRIDGE" ;;
  esac
  if [ -n "${SAMBA_INTERFACE:-}" ]; then
    interfaces+=",$SAMBA_INTERFACE"
  fi
fi

html "Initializing shared folder..."
SAMBA_CONFIG="/etc/samba/smb.conf"
[[ "$DEBUG" == [Yy1]* ]] && echo "Starting Samba daemon..."

addShare() {
  local dir="$1"
  local ref="$2"
  local name="$3"
  local comment="$4"
  local cfg="$5"
  local owner=""

  if ! mkdir -p "$dir"; then
    error "Failed to create directory $dir." && return 1
  fi

  if ! ls -A "$dir" >/dev/null 2>&1; then
    msg="Failed to access directory $dir."
    msg+=" If SELinux is active, you need to add the \":Z\" flag to the bind mount."
    error "$msg" && return 1
  fi

  if [ -z "$(ls -A "$dir")" ]; then
    if ! chmod 2777 "$dir"; then
      error "Failed to set permissions for directory $dir" && return 1
    fi
    owner=$(stat -c %u "$dir")
    if [[ "$owner" == "0" ]]; then
      if ! chown "1000:1000" "$dir"; then
        error "Failed to set ownership for directory $dir" && return 1
      fi
    fi
  fi

  if [[ "$dir" == "$tmp" ]]; then

    {   echo "--------------------------------------------------------"
        echo " $APP for $ENGINE v$(</run/version)..."
        echo " For support visit $SUPPORT"
        echo "--------------------------------------------------------"
        echo ""
        echo "Using this folder you can exchange files with the host machine."
        echo ""
        echo "To select a folder on the host for this purpose, include the following bind mount in your compose file:"
        echo ""
        echo "  volumes:"
        echo "    - \"./example:${ref}\""
        echo ""
        echo "Or in your run command:"
        echo ""
        echo "  -v \"\${PWD:-.}/example:${ref}\""
        echo ""
        echo "Replace the example path ./example with your desired shared folder, which then will become visible here."
        echo ""
    } | unix2dos > "$dir/readme.txt"

  fi

  {     echo ""
        echo "[$name]"
        echo "    path = $dir"
        echo "    comment = $comment"
        echo "    writable = yes"
        echo "    guest ok = yes"
        echo "    guest only = yes"
  } >> "$cfg"

  return 0
}

{       echo "[global]"
        echo "    server string = Dockur"
        echo "    netbios name = $hostname"
        echo "    workgroup = WORKGROUP"
        echo "    interfaces = $interfaces"
        echo "    bind interfaces only = yes"
        echo "    socket address = $socket"
        echo "    security = user"
        echo "    guest account = nobody"
        echo "    map to guest = Bad User"
        echo "    server min protocol = NT1"
        echo "    follow symlinks = yes"
        echo "    wide links = yes"
        echo "    unix extensions = no"
        echo "    inherit owner = yes"
        echo "    create mask = 0666"
        echo "    directory mask = 02777"
        echo "    force user = root"
        echo "    force group = root"
        echo "    force create mode = 0666"
        echo "    force directory mode = 02777"
        echo ""
        echo "    # Disable printing services"
        echo "    load printers = no"
        echo "    printing = bsd"
        echo "    printcap name = /dev/null"
        echo "    disable spoolss = yes"
} > "$SAMBA_CONFIG"

# Add shared folders
share="/shared"
[ ! -d "$share" ] && [ -d "$STORAGE/shared" ] && share="$STORAGE/shared"
[ ! -d "$share" ] && [ -d "/data" ] && share="/data"
[ ! -d "$share" ] && [ -d "$STORAGE/data" ] && share="$STORAGE/data"
[ ! -d "$share" ] && share="$tmp"

! addShare "$share" "/shared" "Data" "Shared" "$SAMBA_CONFIG" && return 0

if [ -d "/shared2" ]; then
  addShare "/shared2" "/shared2" "Data2" "Shared" "$SAMBA_CONFIG" || :
else
  if [ -d "/data2" ]; then
    addShare "/data2" "/shared2" "Data2" "Shared" "$SAMBA_CONFIG" || :
  fi
fi

if [ -d "/shared3" ]; then
  addShare "/shared3" "/shared3" "Data3" "Shared" "$SAMBA_CONFIG" || :
else
  if [ -d "/data3" ]; then
    addShare "/data3" "/shared3" "Data3" "Shared" "$SAMBA_CONFIG" || :
  fi
fi

# Create directories if missing
mkdir -p /var/lib/samba/sysvol
mkdir -p /var/lib/samba/private
mkdir -p /var/lib/samba/bind-dns

# Try to repair Samba permissions
[ -d /run/samba/msg.lock ] && chmod -R 0755 /run/samba/msg.lock 2>/dev/null || :
[ -d /var/log/samba/cores ] && chmod -R 0700 /var/log/samba/cores 2>/dev/null || :
[ -d /var/cache/samba/msg.lock ] && chmod -R 0755 /var/cache/samba/msg.lock 2>/dev/null || :

rm -f /var/log/samba/log.smbd

if ! smbd -l /var/log/samba; then
  SAMBA_DEBUG="Y"
  error "Failed to start Samba daemon!"
fi

if [[ "$SAMBA_DEBUG" == [Yy1]* ]]; then
  tail -fn +0 /var/log/samba/log.smbd --pid=$$ &
fi

case "${NETWORK,,}" in
  "passt" | "slirp" )
    return 0 ;;
esac

if [[ "${BOOT_MODE:-}" == "windows_legacy" ]]; then

  # Enable NetBIOS on Windows 7 and lower
  [[ "$DEBUG" == [Yy1]* ]] && echo "Starting NetBIOS daemon..."

  rm -f /var/log/samba/log.nmbd

  if ! nmbd -l /var/log/samba; then
    SAMBA_DEBUG="Y"
    error "Failed to start NetBIOS daemon!"
  fi

  if [[ "$SAMBA_DEBUG" == [Yy1]* ]]; then
    tail -fn +0 /var/log/samba/log.nmbd --pid=$$ &
  fi

else

  # Enable Web Service Discovery on Vista and up
  [[ "$DEBUG" == [Yy1]* ]] && echo "Starting wsddn daemon..."

  rm -f /var/log/wsddn.log

  if ! wsddn -i "${interfaces%%,*}" -H "$hostname" --unixd --log-file=/var/log/wsddn.log --pid-file=/var/run/wsdd.pid; then
    SAMBA_DEBUG="Y"
    error "Failed to start wsddn daemon!"
  fi

  if [[ "$SAMBA_DEBUG" == [Yy1]* ]]; then
    tail -fn +0 /var/log/wsddn.log --pid=$$ &
  fi

fi

return 0