Enhance UX and add error handling to the system setup script.
This commit is contained in:
Nick Hepler
parent
e9e19af507
commit
27237a866b
167
setup.sh
167
setup.sh
@ -8,9 +8,16 @@ timezone="America/New_York"
|
||||
sshd_config="/etc/ssh/sshd_config"
|
||||
docker_installed=false
|
||||
|
||||
# Function to prompt for hostname
|
||||
# Function to prompt for hostname with validation
|
||||
prompt_for_hostname() {
|
||||
while true; do
|
||||
read -p "Please enter the desired hostname: " hostname
|
||||
if [[ -z "$hostname" ]]; then
|
||||
echo "Hostname cannot be empty. Please try again."
|
||||
else
|
||||
break
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
# Function to add hostname and IP to /etc/hosts
|
||||
@ -29,22 +36,33 @@ update_hosts_file() {
|
||||
fi
|
||||
}
|
||||
|
||||
# Function to prompt for Docker installation
|
||||
# Function to prompt for Docker installation with validation
|
||||
prompt_for_docker_install() {
|
||||
while true; do
|
||||
read -p "Would you like to install Docker CE? (y/n): " install_docker
|
||||
if [[ "$install_docker" =~ ^[Yy]$ ]]; then
|
||||
install_docker
|
||||
else
|
||||
echo "Skipping Docker installation."
|
||||
fi
|
||||
case "$install_docker" in
|
||||
[Yy]*) install_docker ;;
|
||||
[Nn]*) echo "Skipping Docker installation."; break ;;
|
||||
*) echo "Invalid input. Please enter 'y' or 'n'." ;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
# Function to install Docker
|
||||
install_docker() {
|
||||
echo "Installing Docker CE..."
|
||||
curl -fsSL https://get.docker.com -o get-docker.sh
|
||||
if [ $? -ne 0 ]; then
|
||||
echo "Error downloading Docker installation script."
|
||||
exit 1
|
||||
fi
|
||||
sh get-docker.sh
|
||||
|
||||
if [ $? -ne 0 ]; then
|
||||
echo "Error installing Docker."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Start and enable Docker service
|
||||
systemctl start docker
|
||||
systemctl enable docker
|
||||
@ -52,12 +70,19 @@ install_docker() {
|
||||
|
||||
usermod -aG docker "$username"
|
||||
echo "User $username added to the docker group."
|
||||
|
||||
}
|
||||
|
||||
# Function to create a limited user account
|
||||
# Function to create a limited user account with validation
|
||||
create_user_account() {
|
||||
while true; do
|
||||
read -p "Please enter the username for the new user account: " username
|
||||
if [[ -z "$username" ]]; then
|
||||
echo "Username cannot be empty. Please try again."
|
||||
else
|
||||
break
|
||||
fi
|
||||
done
|
||||
|
||||
read -sp "Please enter the password for the new user account: " password
|
||||
echo
|
||||
|
||||
@ -71,49 +96,54 @@ create_user_account() {
|
||||
fi
|
||||
}
|
||||
|
||||
# Check if the script is run as root
|
||||
if [ "$(id -u)" -ne 0 ]; then
|
||||
# Function to check if running as root
|
||||
check_root() {
|
||||
if [ "$(id -u)" -ne 0 ]; then
|
||||
echo "This script must be run as root. Please use sudo or switch to root."
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# Update the package manager
|
||||
echo "Updating package manager..."
|
||||
if command -v dnf &> /dev/null; then
|
||||
# Function to update package manager cache
|
||||
update_package_manager() {
|
||||
echo "Updating package manager..."
|
||||
if command -v dnf &> /dev/null; then
|
||||
dnf -y makecache
|
||||
elif command -v yum &> /dev/null; then
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum -y makecache
|
||||
else
|
||||
else
|
||||
echo "Neither dnf nor yum found. This script only works on RHEL-based distributions."
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# Upgrade installed packages
|
||||
echo "Upgrading installed packages..."
|
||||
if command -v dnf &> /dev/null; then
|
||||
# Function to upgrade installed packages
|
||||
upgrade_packages() {
|
||||
echo "Upgrading installed packages..."
|
||||
if command -v dnf &> /dev/null; then
|
||||
dnf -y upgrade
|
||||
elif command -v yum &> /dev/null; then
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum -y update
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# Change the timezone
|
||||
echo "Setting timezone to $timezone..."
|
||||
timedatectl set-timezone "$timezone"
|
||||
# Function to change the timezone
|
||||
set_timezone() {
|
||||
echo "Setting timezone to $timezone..."
|
||||
timedatectl set-timezone "$timezone"
|
||||
}
|
||||
|
||||
# Change the hostname
|
||||
prompt_for_hostname
|
||||
echo "Setting hostname to $hostname..."
|
||||
hostnamectl set-hostname "$hostname"
|
||||
# Function to change the hostname
|
||||
set_hostname() {
|
||||
prompt_for_hostname
|
||||
echo "Setting hostname to $hostname..."
|
||||
hostnamectl set-hostname "$hostname"
|
||||
}
|
||||
|
||||
# Update /etc/hosts
|
||||
update_hosts_file
|
||||
|
||||
# Create a limited user account
|
||||
create_user_account
|
||||
|
||||
# Modify sshd_config
|
||||
echo "Modifying SSH configuration..."
|
||||
if [ -f "$sshd_config" ]; then
|
||||
# Function to modify sshd_config
|
||||
modify_sshd_config() {
|
||||
echo "Modifying SSH configuration..."
|
||||
if [ -f "$sshd_config" ]; then
|
||||
# Set PermitRootLogin to no
|
||||
sed -i 's/^PermitRootLogin .*/PermitRootLogin no/' "$sshd_config" || echo "PermitRootLogin no" >> "$sshd_config"
|
||||
|
||||
@ -130,41 +160,52 @@ if [ -f "$sshd_config" ]; then
|
||||
else
|
||||
sed -i 's/^AddressFamily .*/AddressFamily inet/' "$sshd_config" || echo "AddressFamily inet" >> "$sshd_config"
|
||||
fi
|
||||
else
|
||||
else
|
||||
echo "sshd_config file not found. Exiting."
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# Get the IP address of the machine
|
||||
ip_address=$(hostname -I | awk '{print $1}')
|
||||
# Function to restart SSH service
|
||||
restart_ssh() {
|
||||
echo "Restarting SSH service..."
|
||||
systemctl restart sshd
|
||||
}
|
||||
|
||||
# Prompt the user to copy their public key
|
||||
echo "Please copy your public SSH key to the server using the following command:"
|
||||
echo "ssh-copy-id $username@$ip_address"
|
||||
|
||||
# Wait until the authorized_keys file is no longer empty
|
||||
echo "Waiting for your public key to be copied..."
|
||||
while true; do
|
||||
# Function to wait for SSH key copy
|
||||
wait_for_key_copy() {
|
||||
echo "Waiting for your public key to be copied..."
|
||||
while true; do
|
||||
if [ -s "/home/$username/.ssh/authorized_keys" ]; then
|
||||
echo "Public key has been successfully copied."
|
||||
break
|
||||
fi
|
||||
sleep 2
|
||||
done
|
||||
done
|
||||
}
|
||||
|
||||
# Restart SSH service to apply changes
|
||||
echo "Restarting SSH service..."
|
||||
systemctl restart sshd
|
||||
|
||||
# Prompt for Docker installation
|
||||
prompt_for_docker_install
|
||||
|
||||
# Clean up
|
||||
echo "Cleaning up..."
|
||||
if command -v dnf &> /dev/null; then
|
||||
# Function to clean up unused packages
|
||||
cleanup() {
|
||||
echo "Cleaning up..."
|
||||
if command -v dnf &> /dev/null; then
|
||||
dnf -y autoremove
|
||||
elif command -v yum &> /dev/null; then
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum -y autoremove
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# Main execution
|
||||
check_root
|
||||
update_package_manager
|
||||
upgrade_packages
|
||||
set_timezone
|
||||
set_hostname
|
||||
update_hosts_file
|
||||
create_user_account
|
||||
modify_sshd_config
|
||||
restart_ssh
|
||||
wait_for_key_copy
|
||||
prompt_for_docker_install
|
||||
cleanup
|
||||
|
||||
echo "System update complete! Timezone set to $timezone, hostname set to $hostname, limited user created, sshd_config modified, and Docker installation completed if selected."
|
||||
Loading…
Reference in New Issue
Block a user