#!/bin/sh
# Copyright (c) 2018-2019 TildeLinux Maintainers
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Email: tildelinux@tildeverse.org

export SSH_ASKPASS="/usr/lib/ssh/x11-ssh-askpass"
set -e
eval "$(xdotool getmouselocation --shell)"
mouse_x=$X
mouse_y=$Y

server_address="tilde.town!tilde.team!yourtilde.com!cosmic.voyage!tilde.institute"
server_name="MyTilde"
login=""
ssh_port="22"
ssh_dir="$HOME/.ssh"
priv_key_path=" "

get_field() {
    field_num="$1"
    info="$2"

    printf "%s" "$info" | cut -f "$field_num" -d '|'
}

error_box() {
    error_text="$1"
    yad \
    --posx="$mouse_x" \
    --posy="$mouse_y" \
    --width=250 \
    --image="dialog-error" \
    --title="Error!" \
    --text="$error_text" \
    --button="OK":0
}

succ_box() {
    succ_text="$1"
    yad \
    --posx="$mouse_x" \
    --posy="$mouse_y" \
    --width=250 \
    --image="dialog-info" \
    --title="Success!" \
    --text="$succ_text" \
    --button="OK":0
}

dec_box() {
    error_text="$1"
    yad \
    --posx="$mouse_x" \
    --posy="$mouse_y" \
    --width=250 \
    --image="dialog-error" \
    --title="Error!" \
    --text="$error_text" \
    --button="No":1 \
    --button="Yes":0
    return $?
}

info_box() {
    info_text="$1"
    yad \
    --posx="$mouse_x" \
    --posy="$mouse_y" \
    --no-buttons \
    --width=250 \
    --image="dialog-info" \
    --title="Info" \
    --text="$info_text"
}

get_info() {
    error_num=0
    error_msg=""

    info=$(yad \
                --posx="$mouse_x" \
                --posy="$mouse_y" \
                --title="New server setup" \
                --text="Add a new server" \
                --form \
                --field="Server":CB \
                --field="Short Name" \
                --field="Login" \
                --field="Port" \
                --field="SSH Directory":DIR \
                --field="Private key":FL \
                --button="Cancel":1 \
                --button="Continue":0 \
                "$server_address" "$server_name" "$login" "$ssh_port" "$ssh_dir" "$priv_key_path")
    [ $? -eq 1 ] && exit 1

    server_address=$(get_field 1 "$info")
    server_name=$(get_field 2 "$info")
    login=$(get_field 3 "$info")
    ssh_port=$(get_field 4 "$info")
    ssh_dir=$(get_field 5 "$info")
    priv_key_path=$(get_field 6 "$info")

    if test -z "$login"
    then
            error_msg="No login provided!"
            error_num=$((error_num + 1))
    fi

    if test -d "$(printf "%s" "$priv_key_path" | sed 's/\/ *$//')"
    then
        error_msg="${error_msg}\\nNo private key provided!"
        error_num=$((error_num + 1))
    fi

    if [ $error_num -gt 0 ]
    then
        error_box "$error_msg" \
        && get_info
    fi

    configure_server
}

configure_server() {
    ssh_current_config="${ssh_dir}/config"
    ssh_temp_config="${ssh_dir}/config.tmp"
    ssh_host_dir="${ssh_dir}/${server_address}"
    key_type=$(head -n 1 "${priv_key_path}" \
                | cut -f 2 -d ' ' \
                | tr '[:upper:]' '[:lower:]')

    test -e "$ssh_current_config" \
    || touch "$ssh_current_config"
    cp "$ssh_current_config" "$ssh_temp_config"

    if grep -qi "Host ${server_name}" "$ssh_current_config"
    then
        dec_box "Host named ${server_name} already exists, overwrite?" \
        || get_info

        if ! dec_box "Keep old config in file? (ssh will use the new one)"
        then
            IFS=""
            in_old_host_block=0
            while read -r line
            do
                if printf "%s" "$line" | grep -q "Host ${server_name}"
                then
                    in_old_host_block=0
                elif printf "%s" "$line" | grep -q "Host "
                then
                    in_old_host_block=1
                fi

                [ $in_old_host_block -eq 1 ] \
                && printf "%s" "$line"
            done < "$ssh_current_config" > "$ssh_temp_config"
        fi
    fi

    mkdir -p "$ssh_host_dir"
    cp "$priv_key_path" "${ssh_host_dir}/id_${key_type}"

    {
        printf "Host %s\\n" "$server_name"
        printf "    HostName %s\\n" "$server_address"
        printf "    Port %s\\n" "$ssh_port"
        printf "    User %s\\n" "$login"
        printf "    IdentityFile %s/id_%s\\n" "$ssh_host_dir" "$key_type"
    } >> "$ssh_temp_config"


    info_box "Attempting to log in with the provided credentials..." &
    info_pid=$(( $! + 2 )) # yikes, is this safe?
    printf "%s\\n" "$ssh_temp_config"
    printf "%s\\n" "$server_address"
    if setsid -w ssh -qF "$ssh_temp_config" "$server_name" exit
    then
        kill $info_pid
        succ_box "Login success!\\nAccount for ${server_address} created!"
        cp "$ssh_temp_config" "$ssh_current_config"
        rm "$ssh_temp_config"
        exit # Without this the "Host already exists" box appears, investigate.
    else
        kill $info_pid
        error_box "Login failed, please try again."
        get_info
    fi
}

get_info