#!/usr/bin/env bash

# Copyright (c) 2021-2025 community-scripts ORG
# Authors: MickLesk (CanbiZ)
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
# Source: https://frigate.video/

source /dev/stdin <<<"$FUNCTIONS_FILE_PATH"
color
verb_ip6
catch_errors
setting_up_container
network_check
update_os

msg_info "Installing Dependencies (Patience)"
$STD apt-get install -y {git,ffmpeg,ca-certificates,automake,build-essential,xz-utils,libtool,ccache,pkg-config,libgtk-3-dev,libavcodec-dev,libavformat-dev,libswscale-dev,libv4l-dev,libxvidcore-dev,libx264-dev,libjpeg-dev,libpng-dev,libtiff-dev,gfortran,openexr,libatlas-base-dev,libssl-dev,libtbbmalloc2,libtbb-dev,libdc1394-dev,libopenexr-dev,libgstreamer-plugins-base1.0-dev,libgstreamer1.0-dev,gcc,gfortran,libopenblas-dev,liblapack-dev,libusb-1.0-0-dev,jq,moreutils}
msg_ok "Installed Dependencies"

msg_info "Setup Python3"
$STD apt-get install -y {python3,python3-dev,python3-setuptools,python3-distutils,python3-pip,python3-venv}
$STD pip install --upgrade pip
msg_ok "Setup Python3"

NODE_VERSION="22" NODE_MODULE="yarn" setup_nodejs
fetch_and_deploy_gh_release "go2rtc" "AlexxIT/go2rtc" "singlefile" "latest" "/usr/local/go2rtc/bin" "go2rtc_linux_amd64"
fetch_and_deploy_gh_release "frigate" "blakeblackshear/frigate" "tarball" "latest" "/opt/frigate"
fetch_and_deploy_gh_release "libusb" "libusb/libusb" "tarball" "v1.0.29" "/opt/frigate/libusb"

# msg_info "Setting Up Hardware Acceleration"
# $STD apt-get -y install {va-driver-all,ocl-icd-libopencl1,intel-opencl-icd,vainfo,intel-gpu-tools}
# if [[ "$CTTYPE" == "0" ]]; then
#   chgrp video /dev/dri
#   chmod 755 /dev/dri
#   chmod 660 /dev/dri/*
# fi
# msg_ok "Set Up Hardware Acceleration"

msg_info "Setting up Python venv"
cd /opt/frigate
python3 -m venv venv
source venv/bin/activate
$STD pip install --upgrade pip wheel --break-system-packages
$STD pip install -r docker/main/requirements.txt --break-system-packages
$STD pip install -r docker/main/requirements-wheels.txt --break-system-packages
$STD pip install -r docker/main/requirements-ov.txt --break-system-packages
msg_ok "Python venv ready"

msg_info "Building Web UI"
cd /opt/frigate/web
$STD npm ci
$STD npm run build
msg_ok "Web UI built"

msg_info "Writing default config"
mkdir -p /opt/frigate/config
cat <<EOF >/opt/frigate/config/config.yml
mqtt:
  enabled: false
cameras:
  test:
    ffmpeg:
      inputs:
        - path: /media/frigate/person-bicycle-car-detection.mp4
          input_args: -re -stream_loop -1 -fflags +genpts
          roles:
            - detect
            - rtmp
    detect:
      height: 1080
      width: 1920
      fps: 5
EOF
mkdir -p /config
ln -sf /opt/frigate/config/config.yml /config/config.yml
mkdir -p /media/frigate
wget -qO /media/frigate/person-bicycle-car-detection.mp4 https://github.com/intel-iot-devkit/sample-videos/raw/master/person-bicycle-car-detection.mp4
cat <<'EOF' >/opt/frigate/frigate/version.py
VERSION = "0.16.0"
EOF
ln -sf /usr/lib/ffmpeg/7.0/bin/ffmpeg /usr/bin/ffmpeg
ln -sf /usr/lib/ffmpeg/7.0/bin/ffprobe /usr/bin/ffprobe
msg_ok "Config ready"

msg_info "Building and Installing libUSB without udev"
wget -qO /tmp/libusb.zip https://github.com/libusb/libusb/archive/v1.0.29.zip
unzip -q /tmp/libusb.zip -d /tmp/
cd /tmp/libusb-1.0.29
./bootstrap.sh
./configure --disable-udev --enable-shared
make -j$(nproc --all)
make install
ldconfig
rm -rf /tmp/libusb.zip /tmp/libusb-1.0.29
msg_ok "Installed libUSB without udev"

# Coral Object Detection Models
msg_info "Installing Coral Object Detection Models"
cd /opt/frigate
export CCACHE_DIR=/root/.ccache
export CCACHE_MAXSIZE=2G

# edgetpu / cpu Modelle
wget -qO edgetpu_model.tflite https://github.com/google-coral/test_data/raw/release-frogfish/ssdlite_mobiledet_coco_qat_postprocess_edgetpu.tflite
wget -qO cpu_model.tflite https://github.com/google-coral/test_data/raw/release-frogfish/ssdlite_mobiledet_coco_qat_postprocess.tflite
cp /opt/frigate/labelmap.txt /labelmap.txt

# Audio-Modelle
wget -qO yamnet-tflite-classification-tflite-v1.tar.gz https://www.kaggle.com/api/v1/models/google/yamnet/tfLite/classification-tflite/1/download
tar xzf yamnet-tflite-classification-tflite-v1.tar.gz
rm -rf yamnet-tflite-classification-tflite-v1.tar.gz
mv 1.tflite cpu_audio_model.tflite
cp /opt/frigate/audio-labelmap.txt /audio-labelmap.txt
msg_ok "Installed Coral Object Detection Models"

# ------------------------------------------------------------
# Tempio installieren
msg_info "Installing Tempio"
sed -i 's|/rootfs/usr/local|/usr/local|g' /opt/frigate/docker/main/install_tempio.sh
export TARGETARCH="amd64"
export DEBIAN_FRONTEND=noninteractive
echo "libedgetpu1-max libedgetpu/accepted-eula select true" | debconf-set-selections
echo "libedgetpu1-max libedgetpu/install-confirm-max select true" | debconf-set-selections
/opt/frigate/docker/main/install_tempio.sh
chmod +x /usr/local/tempio/bin/tempio
ln -sf /usr/local/tempio/bin/tempio /usr/local/bin/tempio
msg_ok "Installed Tempio"

msg_info "Copying model files"
cp /opt/frigate/cpu_model.tflite /
cp /opt/frigate/edgetpu_model.tflite /
cp /opt/frigate/audio-labelmap.txt /
cp /opt/frigate/labelmap.txt /
msg_ok "Copied model files"

# ------------------------------------------------------------
# systemd Units
msg_info "Creating systemd service for go2rtc"
cat <<EOF >/etc/systemd/system/go2rtc.service
[Unit]
Description=go2rtc
After=network.target

[Service]
ExecStart=/usr/local/bin/go2rtc
Restart=always
RestartSec=2
User=root
StandardOutput=journal
StandardError=journal

[Install]
WantedBy=multi-user.target
EOF
systemctl enable -q --now go2rtc
msg_ok "go2rtc service enabled"

msg_info "Creating systemd service for Frigate"
cat <<EOF >/etc/systemd/system/frigate.service
[Unit]
Description=Frigate service
After=go2rtc.service network.target

[Service]
WorkingDirectory=/opt/frigate
Environment="PATH=/opt/frigate/venv/bin"
Environment="PYTHONPATH=/opt/frigate"
ExecStartPre=/bin/mkdir -p /media/frigate/recordings /media/frigate/clips /media/frigate/snapshots
ExecStart=/opt/frigate/venv/bin/python3 -u -m frigate
Restart=always
RestartSec=5
User=root
StandardOutput=journal
StandardError=journal

[Install]
WantedBy=multi-user.target
EOF
systemctl enable -q --now frigate
msg_ok "Frigate service enabled"

msg_info "Setting environmental variables"
cat <<EOF >>/root/.bashrc
export TERM='xterm-256color'
export DEFAULT_FFMPEG_VERSION='7.0'
export NVIDIA_VISIBLE_DEVICES='all'
export ENV NVIDIA_DRIVER_CAPABILITIES='compute,video,utility'
export PATH="/usr/local/go2rtc/bin:/usr/local/tempio/bin:/usr/local/nginx/sbin:${PATH}"
export TOKENIZERS_PARALLELISM=true
export TRANSFORMERS_NO_ADVISORY_WARNINGS=1
export OPENCV_FFMPEG_LOGLEVEL=8
export HAILORT_LOGGER_PATH=NONE
export INCLUDED_FFMPEG_VERSIONS="7.0:5.0"
export S6_LOGGING_SCRIPT="T 1 n0 s10000000 T"
export S6_CMD_WAIT_FOR_SERVICES_MAXTIME=0
export CCACHE_DIR=/root/.ccache
export CCACHE_MAXSIZE=2G
EOF
msg_ok "Environment set"

msg_info "Building Nginx with Custom Modules"
sed -i 's/if \[\[ "\$VERSION_ID" == "12" \]\]; then/if \[\[ "\$VERSION_ID" == "13" \]\]; then/' /opt/frigate/docker/main/build_nginx.sh
$STD bash /opt/frigate/docker/main/build_nginx.sh
sed -e '/s6-notifyoncheck/ s/^#*/#/' -i /opt/frigate/docker/main/rootfs/etc/s6-overlay/s6-rc.d/nginx/run
ln -sf /usr/local/nginx/sbin/nginx /usr/local/bin/nginx
msg_ok "Built Nginx"

motd_ssh
customize

msg_info "Cleaning up"
$STD apt-get -y autoremove
$STD apt-get -y autoclean
msg_ok "Cleaned"