#!/bin/bash

# NiceIP (c) Nik 2017

PROGNAME=$(basename $0)
TITLE="NiceIP $HOSTNAME"
CURRENT_TIME=$(date +"%x %r %Z")
TIMESTAMP="Generated $CURRENT_TIME, by $USER"
interactive=
filename="$HOME/.niceip.db"

bold=$(tput bold)
underline=$(tput sgr 0 1)
reset=$(tput sgr0)

purple=$(tput setaf 171)
red=$(tput setaf 1)
green=$(tput setaf 76)
tan=$(tput setaf 3)
blue=$(tput setaf 38)

bottom=0
nocomment=0

# Functions
#e_known() { printf "${green}✔ %s${reset}\n" "$@"
#}
#e_unknown() { printf "${red}✖ %s${reset}\n" "$@"
#}
#ask_confirm() {
#    printf "\n${bold}$1${reset}"
#    read -p " (y/n) " -n 1 
#    printf "\n"
#}

containsElement () {
    local e match="$1"
    shift
    for e; do [[ "$e" == "$match" ]] && return 0; done
    return 1
}

showDB () {
   getcomment=$(sqlite3 $filename "SELECT * FROM iplist;")
    echo "$getcomment"
}

usage () {
    echo "$PROGNAME: usage: $PROGNAME [options]"
    echo " Mode 1: In a Pipe: ${bold}yourcommand | $PROGNAME [options]${reset}"
    echo "  -b | --bottom  -  Displays list of IPs on the bottom"
    echo "  -n | --nocomment  -  Turns off prompt for comments at the scripts end"
    echo " "
    echo " Mode 2: Interactive mode: ${bold} $PROGNAME ${reset}"
    echo "   Displays the contents of the DB"
}


# Process Parameters
#started in a Posix shell with no arguments in interactive mode
if [ $# -eq 0 ] && [ -t 0 ]; then
     echo "nada"
     #No arguments and pipes
     showDB
     exit
fi
#else


while [ -n "$1" ]; do
    case $1 in
        -b | --bottom)      bottom=1
        #                   filename=$1
                            ;;
        -n | --nocomment)   nocomment=1
        #                   filename=$1
                            ;;
        -h | --help)        usage
                            exit
                            ;;
        *)                  usage >&2
                            exit 1
                            ;;
    esac
    shift
done

#  Yourcode here

#find current terminal
PPID=`ps -o ppid -p $$` 
echo parent PID $PPID
PTTY=`ps -p $PPID -o tty --no-headers`
echo parent TTY $PTTY 

#Empty arrays to be filled
iparray=()
niparray=()
outarray=()

#Create new table 'iplist' and drop error to stdout if it already exists
tablenew=$(sqlite3 $filename "create table iplist (id INTEGER PRIMARYKEY,ip TEXT,name TEXT,comment TEXT);" 2>&1)
if [[ -z "${tablenew// }" ]]; then
    echo "$filename : New IP table created"
else
    echo "$filename : Table already exists"
fi

#Read all the piped output into arraylines
#while read line; do  #does toomuch field splitting ... better:
while IFS= read -r line; do
    outarray+=("${line}") 
    #check if there is an IPv4 put it to array
    foundip=$(echo $line | grep -oE "\b([0-9]{1,3}\.){3}[0-9]+\b")
    #check if it is already in
    if [[ ! " ${iparray[@]} " =~ " ${foundip} " ]]; then
        iparray+=($(echo $line | grep -oE "\b([0-9]{1,3}\.){3}[0-9]+\b"))
    fi
done

#step through array of ip's
for u in "${iparray[@]}"  
do  
   #check if IP $u is already in table
   getcomment=$(sqlite3 $filename "SELECT comment FROM iplist WHERE ip = '$u';")
    if [[ -z "${getcomment// }" ]]; then
        
        #getcomment comes back empty 
        #colour the IP in red and add it to db with "empty" comment
        replacement="${red}"
        getcomment=$(sqlite3 $filename "INSERT INTO iplist (ip,comment) VALUES ('$u','empty');")
        niparray+=("${red}$u${reset} is yet unknown")
        #echo "${red}$u${reset} is yet unknown"
    else
        #if no usercomment has been set colour it tan
        if [[ "${getcomment// }" == "empty" ]]; then
            replacement="${tan}"
            niparray+=("${tan}$u${reset} is known but unidentified")
            #echo "$u is known but unidentified"
        else
            #use green and display comment
            replacement="${green}"
            niparray+=("${green}$u${reset} is $getcomment")
            #echo "$u is $getcomment"
        fi
    fi
    
    replacement+=$u
    replacement+="${reset}"
    
    ITER=0
    for x in "${outarray[@]}"  
    do  
       #outarray+="$x"
       outarray[$ITER]="${x/$u/$replacement}"
       ITER=$(expr $ITER + 1)
       # echo "$x"
    done  
done  


#Echo all outlines from arrays
if [ $bottom -eq 0 ]; then
    echo "   "
    for o in "${niparray[@]}"  
    do  
        echo "$o"
    done  
fi

echo "   "

for v in "${outarray[@]}"  
do  
    echo "$v"
done  

if [ $bottom -eq 1 ]; then
    echo "   "
    for o in "${niparray[@]}"  
    do  
        echo "$o"
    done  
fi

#Ask for Comments
if [ $nocomment -eq 0 ]; then

    while true; do
        IFS= read -r -p "Do you want to identify the IPs? [y/N] " holder && response=$holder
    
        case $response in
            [yY][eE][sS]|[yY])  echo "Enter New Comments or leave empty for unchanged comments"      
                                break
                                ;;
            [nN][oO]|[nN])      #echo "You said $response"      
                                exit
                                #break
                                ;;
            *)                  #usage >&2
                                #exit 1
                                ;;
        esac
        shift
    done < /dev/$PTTY

    ITER=0
    for v in "${iparray[@]}"  
    do  
        while true; do
            echo "${niparray[$ITER]}"
            IFS= read -r -p "$v : " holder && comm=$holder
            if [[ ! -z "${comm// }" ]]; then
                putcomment=$(sqlite3 $filename "UPDATE iplist SET comment = '$comm' WHERE ip = '$v';")
                
                #echo "Not Empty :'$comm'"    
                break
            else
                #echo "Empty :'$comm'"    
                break
            fi
        done < /dev/$PTTY
        ITER=$(expr $ITER + 1)
        
    done  
    
fi
#fi
#ask_confirm() &