shell/setup.sh

   1 #!/bin/sh
   2
   3 # Setup script for shell configuration files.
   4
   5 # Copyright (C) 2011-2018  Simon Ruderich
   6 #
   7 # This program is free software: you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation, either version 3 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20
  21 set -eu
  22
  23 . ../lib.sh
  24
  25
  26 # Helper functions.
  27 terminal_info() {
  28     infocmp "$@" 2>&1
  29 }
  30 terminal_available() {
  31     terminal_info "$@" > /dev/null
  32 }
  33 # Usage: <file> <name> <grep-string> <value>
  34 apply_optional_replacement() {
  35     if test -n "$4"; then
  36         printf '%s: using %s %s\n' "$1" "$2" "$4"
  37         generate "$1" '' simple_cpp \
  38             "$3" -- "$4"
  39     else
  40         printf '%s: removing %s display\n' "$1" "$2"
  41         grep_i -v "$3" "$1"
  42     fi
  43 }
  44 # Check if `infocmp` is available.
  45 if ! infocmp >/dev/null 2>&1; then
  46     echo 'Warning: `infocmp` not available! 256color checks will fail.'
  47     echo
  48 fi
  49
  50
  51 # DIRECTORY SETUP
  52
  53 # Create private temporary directory used by many tools (including GNU screen
  54 # and tmux).
  55 mkdir -p ~/.tmp
  56 chmod 0700 ~/.tmp
  57
  58 # Create rlwrap history directory.
  59 mkdir -p shell/rlwrap
  60 # Create zsh cache directory.
  61 mkdir -p zsh/cache
  62
  63
  64 # FILE SETUP
  65
  66 # Generate ~/.less with lesskey. Prevent cluttering ~/ by storing the history
  67 # file in this directory; this requires replacing the constant HISTORY_PATH in
  68 # lesskey.
  69 echo 'lesskey: generating .lesskey'
  70 cp lesskey lesskey.tmp
  71 if ! less --help | grep rscroll= > /dev/null; then
  72     sed_i 's/--rscroll=[^ ]*//' lesskey.tmp
  73 fi
  74 simple_cpp < lesskey.tmp \
  75     HISTORY_PATH -- "`pwd`/lesshistory" \
  76     | lesskey -
  77 rm lesskey.tmp
  78 chmod 0600 ~/.less
  79
  80 # Custom colors for GNU ls.
  81 if installed dircolors; then
  82     echo '# WARNING! DO NOT EDIT THIS FILE!' >shell/dircolors
  83     dircolors -b shell/dircolors.in >>shell/dircolors
  84 fi
  85
  86 # Find the required options to get colored ls output. GNU ls is preferred. See
  87 # shell/aliases.in for details. Doing this here instead of in shell/aliases
  88 # speeds up shell starts.
  89 #
  90 # GNU ls with colors available.
  91 if ls --color >/dev/null 2>&1; then
  92     ls_args='command ls --color'
  93 # Normal (BSD) ls with colors available.
  94 elif ls -G >/dev/null 2>&1; then
  95     # Don't display hidden files by default when running as root (-I), I use
  96     # `la` for that.
  97     ls_args='CLICOLOR_FORCE=1 command ls -G -I'
  98 # OpenBSD has no colored ls, use colorls instead.
  99 elif colorls -G >/dev/null 2>&1; then
 100     ls_args='CLICOLOR_FORCE=1 colorls -G'
 101 # Simple ls with no colors.
 102 else
 103     ls_args='command ls'
 104 fi
 105
 106 # Also check if `column -t` is available.
 107 if echo test | column -t >/dev/null 2>&1; then
 108     column=' | column -t'
 109 else
 110     column=
 111 fi
 112
 113 generate shell/aliases .in simple_cpp \
 114     LS_ARGS COLUMN -- \
 115     "$ls_args" "$column"
 116
 117 # If `tig` is not available use my simple replacement.
 118 if ! installed tig; then
 119     echo "alias tig='git tig'" >>shell/aliases
 120 fi
 121
 122 # Check if grep supports --color=auto.
 123 if echo test | grep --color=auto test >/dev/null 2>&1; then
 124     :
 125 else
 126     echo 'shell/aliases: removing grep --color=auto'
 127     sed_i '/^alias grep=/ s/^/#/' shell/aliases
 128 fi
 129
 130 if ! lsblk --merge >/dev/null 2>&1; then
 131     echo 'shell/aliases: removing lsblk --merge'
 132     sed_i 's/lsblk --merge/lsblk/' shell/aliases
 133 fi
 134
 135 generate screenrc .in cat
 136 # As screen-256color is not widely supported use it only on machines where the
 137 # matching terminfo entry is available. This also requires a terminal emulator
 138 # which supports 256 colors. Also used for tmux.
 139 use_256colors=
 140 if terminal_available screen-256color; then
 141     # GNU/Linux's virtual terminal doesn't support 256 colors. If setup.sh is
 142     # run one one, assume this user is mostly used from the terminal.
 143     if test x"$TERM" = xlinux || test x"$TERM" = xscreen.linux; then
 144         echo 'screenrc: running on virtual terminal, disabling 256 colors'
 145     # Called through SSH connection, assume the local system supports 256
 146     # colors.
 147     elif test -n "${SSH_CONNECTION:+set}"; then
 148         use_256colors=1
 149     # We have rxvt-unicode installed, check if it supports 256 colors.
 150     elif installed urxvt; then
 151         # Thanks to deryni in #rxvt-unicode on Freenode (2012-10-14 22:54
 152         # CEST) for the strings/grep idea. The grep check is for "correct" 256
 153         # rxvt-unicode binaries (e.g. Debian's rxvt-unicode-256color), the
 154         # terminal_info check for manual installations which modify
 155         # rxvt-unicode's terminfo entry.
 156         urxvt_path=`installed_path urxvt`
 157         urxvt_grep=`strings "$urxvt_path" | grep '^TERM=rxvt-'`
 158         if test x"$urxvt_grep" = 'xTERM=rxvt-unicode-256color' \
 159                 || terminal_info rxvt-unicode \
 160                     | grep -F 'colors#256' >/dev/null; then
 161             use_256colors=1
 162         fi
 163     # Check if XTerm supports 256 colors (not a perfect check, but most XTerm
 164     # support 256 colors).
 165     elif terminal_available xterm-256color; then
 166         use_256colors=1
 167     fi
 168 fi
 169 if test -z "$use_256colors"; then
 170     echo 'screenrc: removing 256 colors'
 171     sed_i 's/Enable 256 color/Disable 256 color/;
 172            s/screen-256color/screen/' screenrc
 173 fi
 174 # Rxvt doesn't need the attrcolor "fix". As I prefer rxvt assume I use it when
 175 # it's installed.
 176 if installed rxvt; then
 177     echo 'screenrc: removing attrcolor "fix"'
 178     sed_i 's/attrcolor b ".I"/#attrcolor b ".I"/' screenrc
 179 fi
 180 # Display current battery charge on computers with a battery. Necessary lines
 181 # are marked as "(BATTERY)". Also used for Tmux.
 182 battery=
 183 for x in /sys/class/power_supply/BAT*; do
 184     test -d "$x" || continue
 185     battery="$x"
 186 done
 187 apply_optional_replacement screenrc \
 188     battery BATTERY "$battery"
 189 # Display current temperature. Necessary lines are marked as "(TEMPERATURE)".
 190 # Also used for Tmux.
 191 temperature=/sys/devices/platform/coretemp.0
 192 if ! test -d "$temperature"; then
 193     temperature=
 194 fi
 195 apply_optional_replacement screenrc \
 196     temperature TEMPERATURE "$temperature"
 197 # Settings for local hosts.
 198 if test -n "${SSH_CONNECTION:+set}"; then
 199     echo 'screenrc: using dark blue for remote screens'
 200     sed_i 's/sorendition = Bw/sorendition = bw/' screenrc
 201 else
 202     echo 'screenrc: removing remote options'
 203     grep_i -v '(REMOTE)' screenrc
 204 fi
 205
 206
 207 if installed tmux; then
 208     generate tmux.conf .in cat
 209
 210     apply_optional_replacement tmux.conf \
 211         battery BATTERY "$battery"
 212     apply_optional_replacement tmux.conf \
 213         temperature TEMPERATURE "$temperature"
 214
 215     # Old Tmux versions can't handle that.
 216     generate tmux.conf '' ./bin/remove-continuation.pl
 217
 218     # Add mappings to switch to windows 10-29 quickly. See tmux-window.pl for
 219     # details.
 220     perl ./tmux-window.pl 1 "`pwd`/tmux-window2.conf" >tmux-window1.conf
 221     perl ./tmux-window.pl 2                           >tmux-window2.conf
 222     # Set absolute path to tmux-window1.conf in tmux.conf.
 223     generate tmux.conf '' simple_cpp \
 224         TMUX_WINDOW_PATH -- "`pwd`/tmux-window1.conf"
 225
 226     # 256 colors not available.
 227     if test -z "$use_256colors"; then
 228         echo 'tmux.conf: removing 256 colors'
 229         sed_i 's/Enable 256 color/Disable 256 color/;
 230                s/screen-256color/screen/' tmux.conf
 231     fi
 232     # Tmux doesn't display a warning if the shell wasn't found!
 233     if test ! -x '/bin/zsh'; then
 234         echo 'tmux.conf: removing /bin/zsh as shell'
 235         sed_i 's/zsh/sh/' tmux.conf
 236     fi
 237 fi
 238
 239 # Htop overwrites the comments in its configuration file.
 240 generate htoprc .in cat
 241 if installed nproc && test "$(nproc)" -gt 8; then
 242     sed_i 's/LeftCPUs/LeftCPUs2/; s/RightCPUs/RightCPUs2/' htoprc
 243 fi
 244
 245 if installed dig; then
 246     # dig doesn't support any comments in digrc.
 247     grep -v -E '^#' digrc.in >digrc
 248 fi
 249
 250 _iptables='/usr/share/zsh/functions/Completion/Linux/_iptables'
 251 if test -f "$_iptables"; then
 252     sed '/^#compdef/ s/iptables/& ip6tables/' \
 253         < "$_iptables" \
 254         > zsh/functions/_iptables
 255 fi
 256
 257 # LINK SETUP
 258
 259 # Link setup for shells.
 260 link shell ~/.shell
 261 link bash ~/.bash
 262 link bash/rc ~/.bashrc
 263 link bash/profile ~/.bash_profile
 264 link bash/logout ~/.bash_logout
 265 if installed csh; then
 266     link csh/rc ~/.cshrc
 267 fi
 268 link zsh ~/.zsh
 269 link zsh/env ~/.zshenv
 270 link zsh/rc ~/.zshrc
 271 link zsh/logout ~/.zlogout
 272
 273 if installed tmux; then
 274     link terminfo ~/.terminfo
 275 fi
 276
 277 # Link setup for additional files.
 278 if installed crontab; then
 279     link crontab.d ~/.crontab.d
 280 fi
 281 link lessfilter ~/.lessfilter
 282 if installed colordiff; then
 283     link colordiffrc ~/.colordiffrc
 284 fi
 285 link inputrc ~/.inputrc
 286 if installed ghci; then
 287     link haskeline ~/.haskeline
 288 fi
 289 link screenrc ~/.screenrc
 290 if installed tmux; then
 291     link tmux.conf ~/.tmux.conf
 292 fi
 293 if installed htop; then
 294     link htoprc ~/.htoprc
 295     # New location for htoprc. Use both for compatibility.
 296     mkdir -p ~/.config/htop
 297     link htoprc ~/.config/htop/htoprc
 298 fi
 299 if test -d ~/.ssh && test -O ~/.ssh; then
 300     mkdir -p ~/.ssh/master
 301     link ssh_config ~/.ssh/config
 302 fi
 303 if installed dig; then
 304     link digrc ~/.digrc
 305 fi