109 lines
3.3 KiB
Bash
109 lines
3.3 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
: "${STRAP_SENTRY_DSN:=https://4d089076433c4a7aa31bbb2741f053fe@sentry.aenow.com/3}"
|
|
|
|
USR_HOME="${HOME:-/root}"
|
|
ZSHRC="$USR_HOME/.zshrc"
|
|
|
|
append_unique() {
|
|
local line="$1" file="$2"
|
|
touch "$file"
|
|
grep -qxF "$line" "$file" || echo "$line" >> "$file"
|
|
}
|
|
|
|
clone_plugin() {
|
|
local repo="$1" dest="$2"
|
|
[ -d "$dest" ] || git clone --depth=1 "$repo" "$dest"
|
|
}
|
|
|
|
usage() {
|
|
echo "Usage: zsh-setup [digitalocean|ovh|none] [ae-sentry|none]" >&2
|
|
}
|
|
|
|
install_omz() {
|
|
if [ ! -d "$USR_HOME/.oh-my-zsh" ]; then
|
|
sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended
|
|
fi
|
|
if [ -f "$ZSHRC" ]; then
|
|
sed -i -e 's@ZSH_THEME="robbyrussell"@ZSH_THEME="pygmalion"@' "$ZSHRC"
|
|
fi
|
|
}
|
|
|
|
ae_sentry() {
|
|
append_unique "export SENTRY_DSN=$STRAP_SENTRY_DSN" "$ZSHRC"
|
|
}
|
|
|
|
digitalocean() {
|
|
# shellcheck disable=SC2016
|
|
append_unique 'export PUBLIC_IPV4=$(curl -s http://169.254.169.254/metadata/v1/interfaces/public/0/ipv4/address)' "$ZSHRC"
|
|
# shellcheck disable=SC2016
|
|
append_unique 'export PUBLIC_IPV6=$(curl -s http://169.254.169.254/metadata/v1/interfaces/public/0/ipv6/address)' "$ZSHRC"
|
|
}
|
|
|
|
ovh() {
|
|
echo "No OVH-specific zsh configuration."
|
|
}
|
|
|
|
configure_plugins() {
|
|
local custom="${ZSH_CUSTOM:-$USR_HOME/.oh-my-zsh/custom}"
|
|
clone_plugin https://github.com/zsh-users/zsh-autosuggestions "$custom/plugins/zsh-autosuggestions"
|
|
clone_plugin https://github.com/zsh-users/zsh-syntax-highlighting.git "$custom/plugins/zsh-syntax-highlighting"
|
|
clone_plugin https://github.com/zsh-users/zsh-history-substring-search "$custom/plugins/zsh-history-substring-search"
|
|
clone_plugin https://github.com/popstas/zsh-command-time.git "$custom/plugins/command-time"
|
|
|
|
append_unique 'ZSH_COMMAND_TIME_MIN_SECONDS=3' "$ZSHRC"
|
|
append_unique 'ZSH_COMMAND_TIME_MSG="Execution time: %s"' "$ZSHRC"
|
|
append_unique 'ZSH_COMMAND_TIME_COLOR="cyan"' "$ZSHRC"
|
|
append_unique 'ZSH_COMMAND_TIME_EXCLUDE=(vim mcedit nano ctop ssh)' "$ZSHRC"
|
|
|
|
local plugin_line='plugins=(git cp colored-man-pages docker docker-compose extract iterm2 python rsync safe-paste transfer ubuntu zsh-navigation-tools zsh-autosuggestions zsh-syntax-highlighting zsh-history-substring-search command-time universalarchive)'
|
|
if grep -q '^plugins=(git)$' "$ZSHRC"; then
|
|
sed -i -e "s@^plugins=(git)\$@$plugin_line@" "$ZSHRC"
|
|
elif ! grep -qxF "$plugin_line" "$ZSHRC"; then
|
|
echo "$plugin_line" >> "$ZSHRC"
|
|
fi
|
|
}
|
|
|
|
clean_profile_remnants() {
|
|
if [ -f "$USR_HOME/.profile" ]; then
|
|
sed -i -e '/^zsh-setup$/d' -e '/zsh-setup\.sh/d' "$USR_HOME/.profile"
|
|
fi
|
|
}
|
|
|
|
install_omz
|
|
|
|
case "${1:-none}" in
|
|
digitalocean) digitalocean ;;
|
|
ovh) ovh ;;
|
|
none) ;;
|
|
*) usage ;;
|
|
esac
|
|
|
|
case "${2:-none}" in
|
|
ae-sentry) ae_sentry ;;
|
|
none) ;;
|
|
*) usage ;;
|
|
esac
|
|
|
|
append_unique 'export DOCKER_BUILDKIT=1' "$ZSHRC"
|
|
append_unique 'export COMPOSE_DOCKER_CLI_BUILD=1' "$ZSHRC"
|
|
|
|
configure_plugins
|
|
clean_profile_remnants
|
|
|
|
ZSH_BIN="$(command -v zsh || true)"
|
|
if [ -n "$ZSH_BIN" ]; then
|
|
USER_NAME="$(id -un)"
|
|
CURRENT_SHELL="$(getent passwd "$USER_NAME" | cut -d: -f7 || true)"
|
|
if [ "$CURRENT_SHELL" != "$ZSH_BIN" ]; then
|
|
chsh -s "$ZSH_BIN" "$USER_NAME" || true
|
|
fi
|
|
fi
|
|
|
|
echo "Bootstrap complete. Relog to start zsh."
|
|
|
|
if [ -n "$ZSH_BIN" ] && [ -t 0 ] && [ -t 1 ]; then
|
|
exec "$ZSH_BIN"
|
|
fi
|