#!/usr/bin/env bash

# gates-of-tartaros – Minimal replacement for agetty with SSH support
# 
# Copyright © 2013, 2014, 2015  Mattias Andrée (m@maandree.se)
# 
# 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 3 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, see <http://www.gnu.org/licenses/>.


# Read the parameters: the first is the tty, the rest are exports
tty="$1"
shift 1
if test ! $# = 0; then
	export "$@"
fi

# Change TTY
exec 0<>"@dev@/${tty}"
exec 1>&0
exec 2>&0


# Store TTY settings so they can be restorted for the next run
tty_settings="$(stty --save)"


# Function used for printing information on the terminal for the user
display_function=_display
_display () {
	printf '\033[H\033[2J\033[?0c'
	printf "\033[%sm%s (%s)\033[0m\n" "${ANSI_COLOR}" "${NAME}" "${tty}"
	printf '\nLogin: '
}

# Function for reading the username form the terminal
read_function=_read
_read () {
	read user
}

# Function for logging in the user
login_function=_login
_login () {
	if test ! "${user%@*}" = "${user}"; then
		$_use_ssh_function
	elif test -n "${user}" && test -z "${user%%-*}"; then
		$_lock_tty_function
	else
		$_use_login_function
	fi
}

# Function for locking the TTY
_lock_tty_function=_lock_tty
_lock_tty () {
	stty -icanon -echo -isig -ixon -ixoff -ixany
	printf '\033[01;31mNice try! Sleeping for 10 minutes...\033[0m\n'
	sleep 600
}

# Function for performing the log in with `ssh`
_use_ssh_function=_use_ssh
_use_ssh () {
	setsid -c ssh ${user} || sleep 3
	# The sleep allows you to see any error you made
}

# Function for performing the log in with `login`
_use_login_function=_use_login
_use_login () {
	setsid -c login -- "${user}"
}


# Load OS information and extensions
NAME="$(uname -o)"
ANSI_COLOR='01;34' # must be exactly ANSI_COLOR as it is defined in /etc/os-release
if test -f "@etc@/os-release"; then
	. "@etc@/os-release"
fi
if test -f "@etc@/@command@rc"; then
	.  "@etc@/@command@rc"
fi

# Let the user log in
$display_function
$read_function
$login_function

# Restore the TTY settings for the next run
stty "${tty_settings}"