#!/bin/bash
cat > /opt/data-mmdvm.py <<- "EOFXC"
import os
import subprocess

APRSHOSTS = "/opt/data-files/APRSHosts.txt"
DCSHOSTS = "/opt/data-files/DCS_Hosts.txt"
DExtraHOSTS = "/opt/data-files/DExtra_Hosts.txt"
DMRIDFILE = "/opt/data-files/DMRIds.dat"
DMRHOSTS = "/opt/data-files/DMR_Hosts.txt"
DPlusHOSTS = "/opt/data-files/DPlus_Hosts.txt"
P25HOSTS = "/opt/data-files/P25Hosts.txt"
M17HOSTS = "/opt/data-files/M17Hosts.txt"
YSFHOSTS = "/opt/data-files/YSFHosts.txt"
FCSHOSTS = "/opt/data-files/FCSHosts.txt"
XLXHOSTS = "/opt/data-files/XLXHosts.txt"
NXDNIDFILE = "/opt/data-files/NXDN.csv"
NXDNHOSTS = "/opt/data-files/NXDNHosts.txt"
TGLISTBM = "/opt/data-files/TGList_BM.txt"
TGLISTDMR = "/opt/data-files/TGList-DMR.txt"
TGLISTP25 = "/opt/data-files/TGList_P25.txt"
TGLISTNXDN = "/opt/data-files/TGList_NXDN.txt"
TGLISTYSF = "/opt/data-files/TGList_YSF.txt"

# How many backups
FILEBACKUP = 1

# Check if the script is running as root
if os.geteuid() != 0:
    print("This script must be run as root")
    exit(1)

# Create backup of old files
if FILEBACKUP != 0:
    files_to_backup = [
        APRSHOSTS, DCSHOSTS, DExtraHOSTS, DMRIDFILE, DMRHOSTS,
        DPlusHOSTS, P25HOSTS, M17HOSTS, YSFHOSTS, FCSHOSTS,
        XLXHOSTS, NXDNIDFILE, NXDNHOSTS, TGLISTBM, TGLISTDMR,
        TGLISTP25, TGLISTNXDN, TGLISTYSF
    ]

    for file in files_to_backup:
        backup_filename = f"{file}.{subprocess.check_output(['date', '+%Y%m%d']).decode().strip()}"
        subprocess.run(['cp', file, backup_filename])

# Prune backups
for file in files_to_backup:
    backup_count = len([f for f in os.listdir() if f.startswith(f"{file}.")])
    backups_to_delete = backup_count - FILEBACKUP

    if backup_count > FILEBACKUP:
        for backup_file in sorted(os.listdir(), key=lambda f: os.path.getctime(f)):
            if backup_file.startswith(f"{file}.") and backups_to_delete > 0:
                os.remove(backup_file)
                backups_to_delete -= 1

# Generate Host Files
files_to_generate = [
    (APRSHOSTS, "http://www.pistar.uk/downloads/APRS_Hosts.txt"),
    (DCSHOSTS, "http://www.pistar.uk/downloads/DCS_Hosts.txt"),
    (DMRHOSTS, "http://www.pistar.uk/downloads/DMR_Hosts.txt"),
    (DPlusHOSTS, "http://www.pistar.uk/downloads/DPlus_Hosts.txt"),
    (DExtraHOSTS, "http://www.pistar.uk/downloads/DExtra_Hosts.txt"),
    (DMRIDFILE, "http://www.pistar.uk/downloads/DMRIds.dat"),
    (P25HOSTS, "http://www.pistar.uk/downloads/P25_Hosts.txt"),
    (M17HOSTS, "http://www.pistar.uk/downloads/M17_Hosts.txt"),
    (YSFHOSTS, "http://www.pistar.uk/downloads/YSF_Hosts.txt"),
    (FCSHOSTS, "http://www.pistar.uk/downloads/FCS_Hosts.txt"),
    (XLXHOSTS, "http://www.pistar.uk/downloads/XLXHosts.txt"),
    (NXDNIDFILE, "http://www.pistar.uk/downloads/NXDN.csv"),
    (NXDNHOSTS, "http://www.pistar.uk/downloads/NXDN_Hosts.txt"),
    (TGLISTBM, "http://www.pistar.uk/downloads/TGList_BM.txt"),
    (TGLISTDMR, "http://www.pistar.uk/downloads/TGList_BM.txt"),
    (TGLISTP25, "http://www.pistar.uk/downloads/TGList_P25.txt"),
    (TGLISTNXDN, "http://www.pistar.uk/downloads/TGList_NXDN.txt"),
    (TGLISTYSF, "http://www.pistar.uk/downloads/TGList_YSF.txt")
]

for dest_file, source_url in files_to_generate:
    subprocess.run(['curl', '--fail', '-o', dest_file, '-s', source_url])

# Fix up ircDDBGateway Host Files on v4
if os.path.exists("/opt/data-files/ircddbgateway"):
    ircddbgateway_files = [
        ("DCS_Hosts.txt", DCSHOSTS),
        ("DExtra_Hosts.txt", DExtraHOSTS),
        ("DPlus_Hosts.txt", DPlusHOSTS),
        ("CCS_Hosts.txt", "/opt/data-files/CCS_Hosts.txt"),
    ]

    for ircddbgateway_file, symlink_target in ircddbgateway_files:
        ircddbgateway_path = f"/opt/data-files/ircddbgateway/{ircddbgateway_file}"
        if os.path.exists(ircddbgateway_path) and not os.path.islink(ircddbgateway_path):
            os.remove(ircddbgateway_path)
            os.symlink(symlink_target, ircddbgateway_path)

exit(0)

EOFXC
###
if [ ! -d "/opt/data-files" ];
then
   mkdir /opt/data-files
   chmod 777 /opt/data-files
fi
python3 /opt/data-mmdvm.py