#!/bin/bash

# color definitions
C_RESET=$'\e[0m'
C_GRAY=$'\e[90m'
C_CYAN=$'\e[36m'
C_WHITE=$'\e[97m'
C_BYELLOW_BOLD=$'\e[1;93m'
C_BWHITE_BOLD=$'\e[1;97m'
C_DIMYELLOW_ITALIC=$'\e[3;93m'

# logo definition
LOGO_ARCH="                $C_GRAY/$C_RESET
$C_CYAN       /\\       $C_GRAY/$C_RESET
$C_CYAN      /  \\      $C_GRAY/$C_RESET
$C_CYAN     /    \\     $C_GRAY/$C_RESET
$C_CYAN    /      \\    $C_GRAY/$C_RESET
$C_CYAN   /   ,,   \\   $C_GRAY/$C_RESET
$C_CYAN  /   |  |   \\  $C_GRAY/$C_RESET
$C_CYAN /_-''    ''-_\\ $C_GRAY/$C_RESET
                $C_GRAY/$C_RESET"
LOGO="               $C_GRAY/$C_RESET
$C_CYAN  ..:::.       $C_GRAY/$C_RESET
$C_CYAN     ..-==-    $C_GRAY/$C_RESET
$C_CYAN         .+#:  $C_GRAY/$C_RESET
$C_CYAN          =@@  $C_GRAY/$C_RESET
$C_CYAN       :+%@#:  $C_GRAY/$C_RESET
$C_CYAN .:=+#@@%*:    $C_GRAY/$C_RESET
$C_CYAN #@@@#=:       $C_GRAY/$C_RESET
               $C_GRAY/$C_RESET"

LOGO_GENTOO="             $C_GRAY/$C_RESET
$C_WHITE  _-----_    $C_GRAY/$C_RESET
$C_WHITE (       \   $C_GRAY/$C_RESET
$C_WHITE \    0   \  $C_GRAY/$C_RESET
$C_WHITE  \        ) $C_GRAY/$C_RESET
$C_WHITE  /      _/  $C_GRAY/$C_RESET
$C_WHITE (     _-    $C_GRAY/$C_RESET
$C_WHITE \____-      $C_GRAY/$C_RESET
             $C_GRAY/$C_RESET"

# all the info fields
USER=$USER
HOSTNAME=$(cat /etc/hostname)
CPU_MODELNAME=$(cat /proc/cpuinfo | awk -F ":" '/model name/ {print $2}' | awk -F " " '{print $3}' | head -1)
CPU_THREADS_NUM=$(nproc)

RAM_TOTAL=$(awk '/MemTotal:/ { printf "%.1f\n", $2 / 1000000 }' /proc/meminfo)
RAM_FREE=$(awk '/MemAvailable:/ { printf "%.1f\n", $2 / 1000000 }' /proc/meminfo)
RAM="$(awk -v total="${RAM_TOTAL}" -v free="${RAM_FREE}" 'BEGIN { print (total - free) }') G / ${RAM_TOTAL} G"

UPTIME=$(awk '{ s=int($1); printf "%dd %dh %dm\n", s/86400, (s%86400)/3600, (s%3600)/60 }' /proc/uptime)
KERNEL=$(uname -r)
WM="sway"
TERM=$TERM
SHELL=$(echo -n $SHELL | awk -F "/" '{print $NF}')


horizontal_n=$((${#USER} + ${#HOSTNAME} + 3))
horizontal_separator_char="-"
HORIZONTAL_SEPARATOR=$(printf '%*s' "$horizontal_n" ' ' | tr ' ' "$horizontal_separator_char")

# -----
# output login

# convert the logo into an array of "lines" (strings)
mapfile -t LOGO_LINES <<< "$LOGO"

# array of strings as well for the info fields
INFO_LINES=(
"${C_BYELLOW_BOLD}$USER${C_BWHITE_BOLD} @ ${C_BYELLOW_BOLD}$HOSTNAME${C_RESET}"
${C_GRAY}$HORIZONTAL_SEPARATOR${C_RESET}
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $CPU_MODELNAME ($CPU_THREADS_NUM threads)${C_RESET}"
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $RAM${C_RESET}"
#"GPU: $GPU"
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $UPTIME${C_RESET}"
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $KERNEL${C_RESET}"
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $WM${C_RESET}"
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $TERM${C_RESET}"
"${C_CYAN}${C_DIMYELLOW_ITALIC} : $SHELL${C_RESET}"
)

for i in "${!LOGO_LINES[@]}"; do
    # 15 chars is the length of the logo
    printf "%-15s" "${LOGO_LINES[$i]}"
    # only print info if it exists for this line
    [ $i -lt ${#INFO_LINES[@]} ] && printf " %s" "${INFO_LINES[$i]}"
    printf "\n"
done
printf "${C_RESET}"