diff --git a/setup.sh b/setup.sh new file mode 100644 index 0000000..56d3a37 --- /dev/null +++ b/setup.sh @@ -0,0 +1,189 @@ +#!/usr/bin/env bash +set -euo pipefail + +HUB_HOST="oily.dad" +HUB_USER="armbian" +ARMBIAN_HOME="/home/armbian" +SSH_DIR="$ARMBIAN_HOME/.ssh" +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +SPOKE_DIR="$SCRIPT_DIR/spoke" +COMPOSE="$SPOKE_DIR/compose.yaml" +START_PORT=11111 + +RED='\033[0;31m' +GREEN='\033[0;32m' +YELLOW='\033[1;33m' +CYAN='\033[0;36m' +NC='\033[0m' + +info() { echo -e "${GREEN}[+]${NC} $*"; } +warn() { echo -e "${YELLOW}[!]${NC} $*"; } +die() { echo -e "${RED}[ERROR]${NC} $*" >&2; exit 1; } +header() { echo -e "\n${CYAN}══════════════════════════════════════════${NC}"; echo -e "${CYAN} $*${NC}"; echo -e "${CYAN}══════════════════════════════════════════${NC}"; } + +[ "$(id -u)" -eq 0 ] || die "Run as root" + +header "TinyBoard Spoke Setup" + +info "Installing packages..." +apt-get update -q +apt-get install -y -q vim autossh docker.io docker-compose git openssh-server + +info "Adding armbian to docker group..." +usermod -aG docker armbian 2>/dev/null || true + +info "Enabling SSH server..." +systemctl enable ssh +systemctl start ssh + +header "Hostname Setup" +CURRENT_HOSTNAME=$(hostname) +echo -e "Current hostname: ${YELLOW}$CURRENT_HOSTNAME${NC}" +read -rp "Enter a hostname for this spoke (e.g. rocky, gouda, camembert): " SPOKE_NAME +SPOKE_NAME="${SPOKE_NAME:-$CURRENT_HOSTNAME}" +hostnamectl set-hostname "$SPOKE_NAME" +echo "$SPOKE_NAME" > /etc/hostname +info "Hostname set to: $SPOKE_NAME" + +header "SSH Key Setup" +echo "How would you like to handle the SSH key for the tunnel to $HUB_HOST?" +echo " 1) Generate a new key automatically" +echo " 2) Use an existing key (paste the private key)" +echo "" +read -rp "Choose [1/2]: " KEY_CHOICE + +case "$KEY_CHOICE" in +1) + KEY_NAME="oilykey" + KEY_PATH="$SSH_DIR/$KEY_NAME" + mkdir -p "$SSH_DIR" + chown armbian:armbian "$SSH_DIR" + chmod 700 "$SSH_DIR" + + if [ -f "$KEY_PATH" ]; then + warn "Key $KEY_PATH already exists, using it." + else + info "Generating new ED25519 key..." + sudo -u armbian ssh-keygen -t ed25519 -f "$KEY_PATH" -N "" + chown armbian:armbian "$KEY_PATH" "$KEY_PATH.pub" + chmod 600 "$KEY_PATH" + fi + + echo "" + echo -e "${YELLOW}══════════════════════════════════════════${NC}" + echo -e "${YELLOW} Send this public key to finn (oily.dad owner)${NC}" + echo -e "${YELLOW} and ask him to add it to armbian@oily.dad authorized_keys:${NC}" + echo -e "${YELLOW}══════════════════════════════════════════${NC}" + cat "$KEY_PATH.pub" + echo -e "${YELLOW}══════════════════════════════════════════${NC}" + echo "" + read -rp "Press ENTER once finn has added the key to oily.dad..." + ;; +2) + read -rp "Enter a name for the key file (e.g. oilykey): " KEY_NAME + KEY_NAME="${KEY_NAME:-oilykey}" + KEY_PATH="$SSH_DIR/$KEY_NAME" + mkdir -p "$SSH_DIR" + chown armbian:armbian "$SSH_DIR" + chmod 700 "$SSH_DIR" + + echo "Paste the private key content below, then press ENTER and CTRL+D:" + KEY_CONTENT=$(cat) + echo "$KEY_CONTENT" > "$KEY_PATH" + chown armbian:armbian "$KEY_PATH" + chmod 600 "$KEY_PATH" + info "Key saved to $KEY_PATH" + ;; +*) + die "Invalid choice" + ;; +esac + +info "Scanning hub host key..." +sudo -u armbian ssh-keyscan -H "$HUB_HOST" >> "$SSH_DIR/known_hosts" 2>/dev/null +chown armbian:armbian "$SSH_DIR/known_hosts" +chmod 600 "$SSH_DIR/known_hosts" + +header "Testing SSH Connection" +info "Testing connection to $HUB_HOST..." +if sudo -u armbian ssh -i "$KEY_PATH" -o BatchMode=yes -o ConnectTimeout=10 "$HUB_USER@$HUB_HOST" exit 2>/dev/null; then + info "SSH connection successful." +else + die "SSH connection to $HUB_HOST failed. Check that finn added your public key." +fi + +header "Finding Available Tunnel Port" +info "Scanning for a free port on $HUB_HOST starting from $START_PORT..." +TUNNEL_PORT="" +for PORT in $(seq "$START_PORT" $((START_PORT + 20))); do + RESULT=$(sudo -u armbian ssh -i "$KEY_PATH" "$HUB_USER@$HUB_HOST" "ss -tlnp | grep :$PORT" 2>/dev/null || true) + if [ -z "$RESULT" ]; then + TUNNEL_PORT="$PORT" + info "Port $TUNNEL_PORT is available." + break + else + warn "Port $PORT is in use, trying next..." + fi +done + +[ -n "$TUNNEL_PORT" ] || die "Could not find a free port between $START_PORT and $((START_PORT + 20)). Ask finn to free up a port." + +header "Configuring compose.yaml" +info "Setting port to $TUNNEL_PORT and key to $KEY_NAME..." + +sed -i "s|-R [0-9]*:localhost:22|-R ${TUNNEL_PORT}:localhost:22|g" "$COMPOSE" +sed -i "s|-i /home/armbian/.ssh/[^ ]*|-i /home/armbian/.ssh/${KEY_NAME}|g" "$COMPOSE" +sed -i "s|/home/armbian/.ssh/oilykey[^:]*:/home/armbian/.ssh/oilykey[^:]*|${SSH_DIR}/${KEY_NAME}:${SSH_DIR}/${KEY_NAME}|g" "$COMPOSE" +sed -i "s|container_name: spoke-autossh|container_name: ${SPOKE_NAME}-autossh|g" "$COMPOSE" +sed -i "s|container_name: spoke-syncthing|container_name: ${SPOKE_NAME}-syncthing|g" "$COMPOSE" +sed -i "s|hostname: spoke-syncthing|hostname: ${SPOKE_NAME}-syncthing|g" "$COMPOSE" +sed -i '/^version:/d' "$COMPOSE" + +SYNCTHING_MOUNT="$ARMBIAN_HOME/st" +mkdir -p "$SYNCTHING_MOUNT" +chown armbian:armbian "$SYNCTHING_MOUNT" + +header "Building Docker Image" +cd "$SPOKE_DIR" +docker build \ + --build-arg UID="$(id -u armbian)" \ + --build-arg GID="$(id -g armbian)" \ + -t spoke-autossh . + +header "Starting Containers" +docker-compose up -d +info "Waiting for tunnel to establish..." +sleep 6 + +LOGS=$(docker logs "${SPOKE_NAME}-autossh" 2>&1 || docker logs spoke-autossh 2>&1 || true) +if echo "$LOGS" | grep -q "remote port forwarding failed"; then + warn "Tunnel failed — port $TUNNEL_PORT may have been taken between check and connect." + warn "Try running: docker-compose down && docker-compose up -d" + warn "Or re-run this script." +else + info "Tunnel is up on port $TUNNEL_PORT." +fi + +header "Setup Complete" +echo -e " Spoke name: ${GREEN}$SPOKE_NAME${NC}" +echo -e " Tunnel port: ${GREEN}$TUNNEL_PORT${NC} on $HUB_HOST" +echo -e " SSH key: ${GREEN}$KEY_PATH${NC}" +echo "" +echo -e "${YELLOW}Finn needs to do the following on oily.dad:${NC}" +echo "" +echo " 1. Generate a hub->spoke key:" +echo " ssh-keygen -t ed25519 -f ~/.ssh/armbian-${SPOKE_NAME}-$(date +%Y%m)" +echo "" +echo " 2. Copy it to this spoke through the tunnel:" +echo " ssh-copy-id -i ~/.ssh/armbian-${SPOKE_NAME}-$(date +%Y%m).pub -p $TUNNEL_PORT armbian@localhost" +echo "" +echo " 3. Add an rclone remote in ~/.config/rclone/rclone.conf:" +echo " [${SPOKE_NAME}-remote]" +echo " type = sftp" +echo " host = localhost" +echo " port = $TUNNEL_PORT" +echo " key_file = /home/armbian/.ssh/armbian-${SPOKE_NAME}-$(date +%Y%m)" +echo " shell_type = unix" +echo " md5sum_command = md5sum" +echo " sha1sum_command = sha1sum" +echo ""