diff --git a/bootstrap.sh b/bootstrap.sh index fb6b3b0..ad6c98e 100644 --- a/bootstrap.sh +++ b/bootstrap.sh @@ -1,67 +1,122 @@ #!/bin/bash +# Hardcode base URL for your raw repository files. +# Set the value to your own fork, replacing the `authorized_keys` file with your own +# i.e., "https://raw.githubusercontent.com///" +REPO_RAW_URL="https://git.krislamo.org/kris/bootstrap/raw/branch/main" +AUTH_KEY_FILE="/authorized_keys" + +# Optional debianzfs install script download +DEBIANZFS="https://git.krislamo.org/kris/debianzfs/raw/branch/main/debianzfs.sh" +DEBIANZFS_BIN="/usr/local/bin/debianzfs" + +############################## +######## STOP EDITING ######## +############################## + # Root required if [ $EUID -ne 0 ]; then - echo "You must run this script as root" - exit 1 + echo "You must run this script as root" + exit 1 fi +# Options +while getopts ':bceg:h:i:r:suz' OPTION; do + case "$OPTION" in + b) BOOT_CYCLE=1;; + c) CDROM_REMOVE=1;; + e) ENABLE_SSH=1;; + g) GATEWAY="$OPTARG";; + h) HOST="$OPTARG";; + i) IP="$OPTARG";; + r) REPO="$OPTARG";; + s) SSH=1;; + u) UPDATE=1;; + z) ZFSINSTALL=1;; + ?) + usage + exit 1;; + esac +done + +# Allow override but use default repo if not set +[ -z "$REPO" ] && REPO="$REPO_RAW_URL" + # Get current date, hostname, create a temporary directory, -# and set the location to the remote repository DATE=$(date '+%Y%m%d') -TMP_DIR=$(mktemp -d) CUR_HOSTNAME=$(hostname) -GIT_LOC="https://github.com/krislamo/bootstrap.git" - -# Get user input for hostname and IP -echo "Enter name server's new hostname:" -read NEW_HOSTNAME - -echo "Enter a static IP address (e.g. 192.168.1.2/24):" -read STATIC_IP - -if [ ! -z "$STATIC_IP" ]; then - echo "Enter the gateway IP (default: 192.168.1.1):" - read GATEWAY_IP - - [ -z "$GATEWAY_IP" ] && GATEWAY_IP="192.168.1.1" - echo "Gateway set to $GATEWAY_IP" -fi # Remove CD sources -cp /etc/apt/sources.list /etc/apt/sources.list.$DATE -sed -i '/deb cdrom/d' /etc/apt/sources.list +if [ "$CDROM_REMOVE" -eq 1 ]; then + echo "NOTICE: Backing up /etc/apt/sources.list => /etc/apt/sources.list.$DATE" + sed -i."$DATE" '/deb cdrom/d' /etc/apt/sources.list +fi # Upgrade software -apt-get update -y -apt-get upgrade -y - -# Install git, clone this repo, and navigate to it -apt-get install git -y -cd $TMP_DIR -git clone $GIT_LOC -cd bootstrap +if [ "$UPDATE" -eq 1 ]; then + echo "NOTICE: Upgrading system" + apt-get update + apt-get upgrade -y +fi # Install personal SSH keys under root and install the OpenSSH server -mkdir -p /root/.ssh/ -cp --update authorized_keys /root/.ssh/authorized_keys -apt-get install openssh-server -y +if [ "$SSH" -eq 1 ]; then + # Does authorized_keys file already exist? + if [ -f /root/.ssh/authorized_keys ]; then + echo "ERROR: /root/.ssh/authorized_keys file already exists" + exit 1 + fi -# If STATIC_IP is set, backup interfaces and configure static IP -if [ ! -z "$STATIC_IP" ]; then - cp /etc/network/interfaces /etc/network/interfaces.$DATE - sed -i "s/dhcp/static/g" /etc/network/interfaces - if ! grep -q "address" /etc/network/interfaces; then - echo " address $STATIC_IP" >> /etc/network/interfaces - echo " gateway $GATEWAY_IP" >> /etc/network/interfaces - fi + echo "NOTICE: Installing root's authorized_keys and the OpenSSH server" + mkdir -p /root/.ssh/ + chmod 700 /root/.ssh/ + wget "${REPO}${AUTH_KEY_FILE}" -O /root/.ssh/authorized_keys + chmod 644 /root/.ssh/authorized_keys + apt-get install openssh-server -y + + if [ "$ENABLE_SSH" -eq 1 ]; then + echo "NOTICE: Enabling the OpenSSH server" + systemctl start ssh + fi +fi + +# If IP is set, backup interfaces and configure static IP +if [ -n "$IP" ]; then + if [ -z "$GATEWAY" ]; then + echo "ERROR: IP set without a GATEWAY address. See option -g" + exit 1 + fi + + echo "NOTICE: Backing up network interfaces file and installing a new static one" + sed -i."$DATE" "s/dhcp/static/g" /etc/network/interfaces + if ! grep -q "address" /etc/network/interfaces; then + echo " address $IP" >> /etc/network/interfaces + echo " gateway $GATEWAY_IP" >> /etc/network/interfaces + else + echo "ERROR: Address already set" + exit 1 + fi +fi + +# Download DebianZFS script +if [ "$ZFSINSTALL" -eq 1 ]; then + echo "NOTICE: Installing DebianZFS installation script" + wget "$DEBIANZFS" -O "$DEBIANZFS_BIN" + chmod u+x "$DEBIANZFS_BIN" fi # If NEW_HOSTNAME is set, configure new hostname and backup /etc/hosts -if [ ! -z "$NEW_HOSTNAME" ]; then - hostnamectl set-hostname $NEW_HOSTNAME - cp /etc/hosts /etc/hosts.$DATE - sed -i "s/$CUR_HOSTNAME/$NEW_HOSTNAME/g" /etc/hosts - read -p "Press [enter] to restart this machine" - systemctl reboot +if [ -n "$HOST" ]; then + hostnamectl set-hostname "$HOST" + echo "NOTICE: Backing up /etc/hosts and setting new hostname to '$HOST'" + sed -i."$DATE" "s/$CUR_HOSTNAME/$HOST/g" /etc/hosts +fi + +# Restart +if [ "$BOOT_CYCLE" -eq 1 ]; then + echo "NOTICE: Restarting the machine in 10 seconds..." + sleep 9 + echo "NOTICE: Restarting!" + sleep 1 + systemctl reboot fi