#!/bin/bash

# Get DataDir location
DATA_DIR="/data"
case "$(ubnt-device-info firmware || true)" in
1*)
  DATA_DIR="/mnt/data"
  ;;
2*)
  DATA_DIR="/data"
  ;;
3*)
  DATA_DIR="/data"
  ;;
*)
  echo "ERROR: No persistent storage found." 1>&2
  exit 1
  ;;
esac

# Check if the directory exists
if [ ! -d "${DATA_DIR}/scripts" ]; then
  # If it does not exist, create the directory
  mkdir -p "${DATA_DIR}/scripts"
  echo "Directory '${DATA_DIR}/scripts' created."
else
  # If it already exists, print a message
  echo "Directory '${DATA_DIR}/scripts' already exists. Moving on."
fi

set -e

if ! iptables-save | grep -e '\-A UBIOS_.* \--log-prefix "\[' >/dev/null; then
  ${DATA_DIR}/scripts/ipt-enable-logs.sh | iptables-restore -c
else
  echo "iptables already contains USER log prefixes, ignoring."
fi