#!/bin/bash
########################################################################
# Author: Fred (support@qo-op.com)
# Version: 2020.03.24
# License: AGPL-3.0 (https://choosealicense.com/licenses/agpl-3.0/)
########################################################################
MY_PATH="`dirname \"$0\"`"              # relative
MY_PATH="`( cd \"$MY_PATH\" && pwd )`"  # absolutized and normalized
ME="${0##*/}"
########################################################################
# \\///
# qo-op
############# '$MY_PATH/$ME'
########################################################################
# ex: ./'$ME'
########################################################################
# This script is controling IPFS swarm
# Sync friends peers ~/.zen/ipfs/ into ~/.zen/ipfs_swarm/.IPFSNODEID's/ directories
# add not friend_of_mine IPs to fail2ban
########################################################################
# TODO remove because now it is in~/.bashrc
export YOU=$(ps auxf --sort=+utime | grep -w ipfs | grep -v -E 'color=auto|grep' | tail -n 1 | cut -d " " -f 1)
export IPFSNODEID=$(ipfs id -f='<id>\n')
########################################################################
mkdir -p ~/.zen/ipfs_swarm
echo '
___ _  _ __  __         _       _  _ _ _  _ __    
 | |_)|_(_  (_\    //\ |_)|\/| |_)|_|_|_)|_(_ |_| 
_|_|  | __) __)\/\//--\| \|  | | \|_| | \|___)| | 

'
echo "I am $IPFSNODEID"
timestamp=$(date -u +%s%N | cut -b1-13)

echo "REFRESHING /home/$YOU/.zen/ipfs_swarm/ from my SWARM peers"
rm -Rf /home/$YOU/.zen/ipfs_swarm/.12D3KooW*
rm -Rf /home/$YOU/.zen/ipfs_swarm/.Qm*
rm -Rf /home/$YOU/.zen/ipfs_swarm/*

count=1
for peerline in $(ipfs swarm peers && cat ~/.zen/A_swarm_map.txt);
do
    ipfsnodeid=$(echo "$peerline" | awk -F '/' '{print $8}')
    [[ "$ipfsnodeid" == "" ]] && continue
    whoisg1=$(~/.zen/astrXbian/zen/tools/ipfs_to_g1.py $ipfsnodeid)

    ip=$(echo "$peerline" | awk -F '/' '{print $3}')
    isLAN=$(echo $ip | cut -f3 -d '/' | grep -E "(^127\.)|(^192\.168\.)|(^fd42\:)|(^10\.)|(^172\.1[6-9]\.)|(^172\.2[0-9]\.)|(^172\.3[0-1]\.)|(^::1$)|(^[fF][cCdD])/")

    nowdate=$(date)

    echo "### ANALYSING $whoisg1 = $ipfsnodeid ($ip) ###"
    echo "${peerline}"
	myfriendpeer=$(cat ~/.zen/A_my_swarm.txt | grep $ipfsnodeid )
	if [[ "$myfriendpeer" != "" ]]; then
		[[ ! $(ipfs swarm peers | grep $ipfsnodeid) ]] && ipfs swarm connect "$peerline" && ipfs bootstrap add "$peerline"
		echo "OK Friend REFRESH /ipns/$ipfsnodeid INTO ~/.zen/ipfs_swarm/"
		$MY_PATH/tools/timeout.sh -t 12 ipfs get --output=/home/$YOU/.zen/ipfs_swarm/ /ipns/$ipfsnodeid
		if [ $? != 0 ]; then
			echo "TIMEOUT REACHED ___ REMOVE $ipfsnodeid FROM ~/.zen/A_my_swarm.txt";
			grep -vEi $ip ~/.zen/A_my_swarm.txt
		fi
	else
		DEFCON=$(cat ~/.zen/astrXbian/DEFCON)
		echo "!!! NOT MY FRIEND !!! ___________________ # DEFCON : $DEFCON "
		if [[ ( $DEFCON < 5 ) && ! $isLAN ]]
		then
			### IMPORTANT ADD TO /etc/sudoers ###
			# $USER ALL=(ALL:ALL) NOPASSWD:/usr/bin/fail2ban-client
			sudo fail2ban-client add recidive 2>/dev/null 1>&2
			sudo fail2ban-client start recidive
			sudo fail2ban-client set recidive banip $ip
			# Show ALL banned IP 
			sudo fail2ban-client status recidive
		fi
		ipfs swarm disconnect "$peerline"
		ipfs bootstrap rm "$peerline"
		echo "INFORM _FAIL2BAN STATUS $ip IN ~/.zen/ipfs_swarm/.$ipfsnodeid/_FAIL2BAN"
		# INFORM _FAIL2BAN STATUS in ipfs_swarm/.$ipfsnodeid
		mkdir -p ~/.zen/ipfs_swarm/.$ipfsnodeid
		echo "$ip" > ~/.zen/ipfs_swarm/.$ipfsnodeid/_FAIL2BAN
		# REMOVE IT (TODO)
		rm ~/.zen/ipfs_swarm/.$ipfsnodeid/_FAIL2BAN
	fi
done

	
########################################################################
########################################################################
echo "RETRIEVING MY OWN DATA FROM IPNS TO ~/.zen/ipfs_swarm"
ipfs get --output=/home/$YOU/.zen/ipfs_swarm/ /ipns/$IPFSNODEID

########################################################################
# REFRESH xbian Film Serie Anime ASTRXBIAN index
########################################################################
PREFIX="F"
cat ~/.zen/ipfs*/xbian/${PREFIX}* | sort | uniq > ~/.zen/ipfs/.${IPFSNODEID}/${PREFIX}ASTRXBIAN

PREFIX="S"
cat ~/.zen/ipfs*/xbian/${PREFIX}* | sort | uniq > ~/.zen/ipfs/.${IPFSNODEID}/${PREFIX}ASTRXBIAN

PREFIX="A"
cat ~/.zen/ipfs*/xbian/${PREFIX}* | sort | uniq > ~/.zen/ipfs/.${IPFSNODEID}/${PREFIX}ASTRXBIAN

########################################################################

########################################################################
echo "PUBLISHING ~/.zen/ipfs SELF /ipns/$IPFSNODEID"
echo "$(date -u +%s%N | cut -b1-13)" > ~/.zen/ipfs/.${IPFSNODEID}/_timestamp
MIPFS=$(ipfs add -rHq ~/.zen/ipfs | tail -n 1)
echo $MIPFS > ~/.zen/ipfs/.${IPFSNODEID}/_chain
NODEIPNS=$(ipfs name publish --quieter /ipfs/$MIPFS)

exit 0