#!/bin/bash

# Define user keys
declare -A ssh_keys
ssh_keys["alex"]="ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIBsUsUzXcKpdUMYuGs240i/+IbEAbjFoYeUpXB59I8u/ alex@Terminator
ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIFdlYEjdSN9Q5sVVb0S24Ifk8jQL5vLXFRdN/GtayABz alex@spaceship"

# Loop through each user and set up SSH keys
for user in "${!ssh_keys[@]}"; do
    home_dir="/home/$user"

    # Ensure the user exists
    if id "$user" &>/dev/null; then
        echo "Setting up SSH for $user..."


        # Add the SSH key to authorized_keys
        echo "${ssh_keys[$user]}" | sudo tee -a "$home_dir/.ssh/authorized_keys" > /dev/null
        sudo chmod 600 "$home_dir/.ssh/authorized_keys"

        # Set correct ownership
        sudo chown -R "$user:$user" "$home_dir/.ssh"
        
        echo "SSH key added for $user."
    else
        echo "User $user does not exist, skipping..."
    fi
done

echo "All done!"