# Setup SSH server
echo '### Setting up SSH ###'
sudo systemctl enable sshd
sudo ufw allow 22/tcp
sudo ufw allow 8123/tcp
sudo ufw allow 8123/udp
sudo ufw allow 3000/tcp
sudo ufw allow 3001/tcp


# Setup Docker
echo '### Setting up Docker ###'
sudo pacman -Sy --needed docker docker-compose docker-buildx
sudo systemctl enable --now docker.service
if ! getent group docker > /dev/null 2>&1; then
    sudo groupadd docker
fi
if ! id -nG "$USER" | grep -qw "docker"; then
    sudo usermod -aG docker "$USER"
fi
sudo docker network create homelogic 2>/dev/null || true

# Setup ZSH
echo '### Setting up ZSH ###'
if [ "${SHELL##*/}" != "zsh" ]; then
    chsh -s /usr/bin/zsh
fi

sed -i '/# --- HOMELOGIC ALIASES START ---/,/# --- HOMELOGIC ALIASES END ---/d' ~/.zshrc 2>/dev/null || true
sed -i '/# Custom aliases/,/alias restart=/d' ~/.zshrc 2>/dev/null || true
{
    echo '# --- HOMELOGIC ALIASES START ---'
    echo '# Custom aliases'
    echo 'alias la="ls -a"'
    echo 'alias lah="ls -lah"'
    echo "alias treesize='sudo du -sh \\* | sort -rh'"
    echo "alias up='docker compose up -d'"
    echo "alias down='docker compose down'"
    echo "alias pull='docker compose pull && up'"
    echo "alias token='openssl rand -hex 32'"
    echo "alias shortToken='openssl rand -hex 16'"
    echo "alias size='sudo du -sxh .'"
    echo "alias ,,='..'"
    echo "alias restart='down && up'"
    echo '# --- HOMELOGIC ALIASES END ---'
} >> ~/.zshrc

# Setup Tailscale
echo '### Setting up Tailscale ###'
if ! command -v tailscale > /dev/null 2>&1; then
    curl -fsSL https://tailscale.com/install.sh | sh
    sudo tailscale up
fi