#!/usr/bin/env bash

if [ "$EUID" -ne 0 ]; then
  echo "Please run as root"
  exit 1
fi

HOSTNAME=$(hostname)
TMPDIR=/tmp/$(hostname)-$(date +"%d-%m-%Y_%H:%M")
MIT="/srv/mitigator"
DATE=$(date '+%Y%m%d_%H%M%S')
DELIMITER=""

trap "rm -rf $TMPDIR" EXIT

test -d $MIT
if [ $? -eq 0 ]; then
cd $MIT
else
echo -e "\033[31mCan't find /srv/mitigator directory! \033[0m"
exit 1
fi

echo "-------------------------------Copying current MITIGATOR configuration files------------------------"
mkdir $TMPDIR

ls -lah $MIT > $TMPDIR/ls.txt

test -f $MIT/data-plane.conf && cp $MIT/data-plane.conf $TMPDIR/data-plane.conf

test -f /var/lib/docker/volumes/mitigator_own_id/_data/own_id.conf && cp /var/lib/docker/volumes/mitigator_own_id/_data/own_id.conf $TMPDIR/own_id.conf

test -f /etc/docker/daemon.json && cp /etc/docker/daemon.json $TMPDIR/daemon.json

test -f $MIT/.env && cp $MIT/.env $TMPDIR/env.txt

test -f /etc/default/grub && cp /etc/default/grub $TMPDIR/grub.txt

test -f /etc/systemd/system/mitigator.service.d/nics.conf && cp /etc/systemd/system/mitigator.service.d/nics.conf $TMPDIR/nics.conf

find $MIT -type f -name '*.yml' -exec cp {} $TMPDIR/ \;

echo -e "\033[32mAll needed files was copied \033[0m"

echo "-------------------------------Getting state and logs of docker containers-------------------------"
docker-compose ps > $TMPDIR/docker_compose_ps.txt
for service in $(docker-compose ps --service)
do
if [ "$service" == 'backend' ]; then
echo "Copying $service logs"
docker-compose logs --no-color --timestamps $service > $TMPDIR/LOGS_$service.txt
else
echo "Copying $service logs"
docker-compose logs --no-color --timestamps --tail=30000 $service > $TMPDIR/LOGS_$service.txt
fi
done

docker-compose exec -Td postgres psql mitigator -c "\copy (select * from backend.journals order by id desc limit 1000) to /tmp/journal.txt;"
docker cp mitigator_postgres_1:/tmp/journal.txt $TMPDIR/journal.txt

docker-compose exec -Td postgres psql mitigator -c "\copy (select * from backend.instances LEFT JOIN backend.leader USING (id)) to /tmp/instances.txt;"
docker cp mitigator_postgres_1:/tmp/instances.txt $TMPDIR/instances.txt

for ports in $(nsenter -t $(docker inspect mitigator_data-plane_1 -f '{{ .State.Pid }}') \
                    -n curl --noproxy '*' -s http://127.0.0.1:8889/info \
                    | grep name: \
                    | awk '{ print substr($2, 2, length($2) - 2) }'); do
    nsenter -t $(docker inspect mitigator_data-plane_1 -f '{{ .State.Pid }}') \
            -n curl \
            --noproxy '*' \
            -s http://127.0.0.1:8889/$ports/stats \
            > $TMPDIR/DATAPLANE_STATS_$ports.txt
done

echo "Collecting inspect files of all containers"
for inspect in $(docker ps --format "{{.Names}}")
do
docker inspect $inspect > $TMPDIR/INSPECT_$inspect.json
done

echo -e "\x1dclose\x0d" | nsenter -t $(docker inspect mitigator_backend_1 -f '{{ .State.Pid }}') -n telnet -d ls.mitigator.ru > $TMPDIR/ls_host.txt 2>/dev/null

echo -e "\x1dclose\x0d" | nsenter -t $(docker inspect mitigator_backend_1 -f '{{ .State.Pid }}') -n telnet -d 109.232.248.251 443 > $TMPDIR/ls_ip.txt 2>/dev/null

echo -e "\033[32mGetting state of containers was successful \033[0m"

echo "-------------------------------Collect ethernet interfaces information------------------------------"
which lshw 2>&1 > /dev/null
if [ $? -eq 0 ]; then
lshw -class network > $TMPDIR/NETWORK_lshw.txt 2>/dev/null
else
echo -e "\033[31mAttention, the utility lshw not installed, if necessary, install and run the script again \033[0m"
fi

which ethtool 2>&1 > /dev/null
if [ $? -eq 0 ]; then
for interfaces in $(ip -o link show | awk -F': ' '{print $2}')
do
echo "Copying information of $interfaces state"
ethtool -i $interfaces > $TMPDIR/NETWORK_$interfaces.txt 2>/dev/null
done
echo -e "\033[32mCollection was successful \033[0m"
else
echo -e "\033[31mAttention, the utility ethtool not installed, if necessary, install and run the script again \033[0m"
fi

echo "-------------------------------Copying current information of system state--------------------------"
which curl 2>&1 > /dev/null
if [ $? -eq 0 ]; then
nsenter -t $(docker inspect mitigator_data-plane_1 -f '{{ .State.Pid }}') -n curl --noproxy '*' -s http://127.0.0.1:8889/settings > $TMPDIR/DATAPLANE_settings.txt
nsenter -t $(docker inspect mitigator_data-plane_1 -f '{{ .State.Pid }}') -n curl --noproxy '*' -s http://127.0.0.1:8889/info > $TMPDIR/DATAPLANE_info.txt
else
echo -e "\033[31mAttention, install curl and run this script again \033[0m"
fi

dpdk-devbind -s > $TMPDIR/dpdk-devbind.txt 2>/dev/null
df -ih > $TMPDIR/df_inodes.txt
df -h > $TMPDIR/df.txt
free -h > $TMPDIR/free.txt
ps aux > $TMPDIR/ps.txt
dmesg -T > $TMPDIR/dmesg.txt
lscpu > $TMPDIR/lscpu.txt
lspci > $TMPDIR/lspci.txt
lsmod > $TMPDIR/lsmod.txt
uname -a > $TMPDIR/uname.txt
uptime > $TMPDIR/uptime.txt
cat /etc/issue > $TMPDIR/issue.txt
cat /etc/*release > $TMPDIR/release.txt
docker ps -a --no-trunc > $TMPDIR/docker_ps.txt
docker system df -v > $TMPDIR/docker_df.txt
ip a > $TMPDIR/ip_a.txt
ip r > $TMPDIR/ip_r.txt
ip neigh > $TMPDIR/ip_neigh.txt
docker version > $TMPDIR/docker_version.txt
docker images > $TMPDIR/docker_images.txt
docker events --since '24h' --until '1s' >$TMPDIR/docker_events.txt
docker stats --no-stream > $TMPDIR/docker_stats.txt
docker-compose --version > $TMPDIR/docker_compose_version.txt
nsenter -t $(docker inspect mitigator_gateway_1 -f '{{ .State.Pid }}') -n wg > $TMPDIR/docker_exec_gateway_wg.txt
cat /proc/meminfo > $TMPDIR/meminfo.txt
date > $TMPDIR/date.txt
cat /etc/modules > $TMPDIR/modules.txt
cat /etc/debian_version > $TMPDIR/debian_version.txt 2>/dev/null
top -b -n 1 > $TMPDIR/top.txt

echo -e "\033[32mCopying was successful \033[0m"

echo $DELIMITER
echo "--------------------------------Creating the final archive------------------------------------------"
tar -czf $MIT/support_archive_$HOSTNAME-$DATE.tgz $TMPDIR/* 2>/dev/null
echo -e "\033[32mDone. The archive located here:\033[33m /srv/mitigator/support_archive_$HOSTNAME-$DATE.tgz \033[0m"