#!/bin/bash

GFX_VERSION=$(sudo rocminfo 2>/dev/null | grep -oP 'gfx\K[0-9]+[a-z]?' | head -1)
MAJOR=${GFX_VERSION:0:1}; MINOR=${GFX_VERSION:1:1}; PATCH=${GFX_VERSION:2:1}
if [[ "$PATCH" =~ [a-z] ]]; then PATCH=0; fi
HSA_VERSION="$MAJOR.$MINOR.$PATCH"

NOMECONTAINER="frigate-amd"

docker_repo="ghcr.io/blakeblackshear/frigate:stable-rocm"
imagem="frigate-amd"

CustmN2="local_ip"
CustmN3="servidor_dns"
CustmN4="shm_size"
CustmN5="path_media"
CustmN6="gpu_device"
CustmN7="environment"
CustmN8="tmpfs_size"
CustmN9="api_key?"
CustmN10=""

VALUE2="0.0.0.0"
VALUE3="8.8.4.4"
VALUE4="128mb"
VALUE5="/srv/containers/$NOMECONTAINER/media"
VALUE6="--device /dev/dri/renderD128:/dev/dri/renderD128 --device /dev/kfd:/dev/kfd"
VALUE7="-e LIBVA_DRIVER_NAME=radeonsi -e HSA_OVERRIDE_GFX_VERSION=$HSA_VERSION"
VALUE8="1000000000"
VALUE9="your_key"
VALUE10=""

export NOMECONTAINER docker_repo imagem
export CustmN2 CustmN3 CustmN4 CustmN5 CustmN6 CustmN7 CustmN8 CustmN9 CustmN10
export VALUE2 VALUE3 VALUE4 VALUE5 VALUE6 VALUE7 VALUE8 VALUE9 VALUE10

source /root/.index.sh

if [ -f /tmp/common-functions.sh ]; then
  source /tmp/common-functions.sh
else
  echo "ERRO: common-functions.sh nao encontrado!"
  exit 1
fi

function set_mkdir {
  sudo mkdir -p /srv/containers/"$NOMECONTAINER"/{media,config}
:; }

function docker_create {
  local ip_regex="^([0-9]{1,3}\.){3}[0-9]{1,3}$"

  # Se NAO for numerico, execute como host!
  if [[ ! "$VALUE2" =~ $ip_regex ]]; then

    docker run -d --privileged --cap-add=CAP_PERFMON --name="$NOMECONTAINER" --hostname="$NOMECONTAINER" \
    --network host \
    --no-healthcheck --restart=unless-stopped --stop-timeout 30 -v /etc/localtime:/etc/localtime:ro \
    --mount type=tmpfs,target=/tmp/cache,tmpfs-size="$VALUE8" --shm-size="$VALUE4" \
    --log-driver json-file --log-opt max-size=10m --log-opt max-file=3 \
    -e PLUS_API_KEY="$VALUE9" \
    -p 8971:8971 -p 8554:8554 -p 8555:8555/tcp -p 8555:8555/udp \
    -v /srv/containers/"$NOMECONTAINER"/config:/config \
    -v $VALUE5/media:/media/frigate \
    $VALUE6 \
    $VALUE7 \
    "$docker_repo"

    return
  fi

  docker run -d --privileged --cap-add=CAP_PERFMON --name="$NOMECONTAINER" --hostname="$NOMECONTAINER" \
  --network macvlan --ip="$VALUE2" --dns="$VALUE3" \
  --no-healthcheck --restart=unless-stopped --stop-timeout 30 -v /etc/localtime:/etc/localtime:ro \
  --mount type=tmpfs,target=/tmp/cache,tmpfs-size="$VALUE8" --shm-size="$VALUE4" \
  --log-driver json-file --log-opt max-size=10m --log-opt max-file=3 \
  -e PLUS_API_KEY="$VALUE9" \
  -v /srv/containers/"$NOMECONTAINER"/config:/config \
  -v $VALUE5/media:/media/frigate \
  $VALUE6 \
  $VALUE7 \
  "$docker_repo"

#-v "/srv/containers/$NOMECONTAINER/web:/opt/frigate/web" \
}

function docker_extras {
  sudo apt install -y vlc rocminfo rocm-smi hipcc radeontop opencl-clhpp-headers opencl-headers
  docker restart "$NOMECONTAINER"
:; }

execute_main

exit 0