setup.sh: Use automic 256color check for GNU screen.

[config/dotfiles.git] / setup.sh

diff --git a/setup.sh b/setup.sh
index 9137fa6464f5c24e7b135e97cc1c30f866093286..c2a56d574475118bd5a32b8545609ca3c45e9c38 100755 (executable)
--- a/setup.sh
+++ b/setup.sh
@@ -2,26 +2,95 @@
 
 # Setup script for shell configuration files.
 
+# Copyright (C) 2011-2012  Simon Ruderich
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
 
 . ../lib.sh
 
+# Helper functions.
+terminal_info() {
+    infocmp "$@" 2>&1
+}
+terminal_available() {
+    terminal_info "$@" > /dev/null
+}
+# Check if `infocmp` is available.
+if ! infocmp > /dev/null 2>&1; then
+    echo 'Warning: `infocmp` not available! 256color checks will fail.'
+    echo
+fi
+
+
 hostname=`hostname`
 
+# Create private temporary directory used by many tools (including GNU screen
+# and tmux).
+mkdir -p ~/.tmp
+chmod 0700 ~/.tmp
+
 # Generate ~/.less with lesskey.
 lesskey lesskey
 chmod 0600 ~/.less
 
-# As screen-256color is not widely supported use it only on machines known to
-# work.
 generate cat screenrc .in
-if [ $hostname != asp -a $hostname != systemofadown ]; then
+# As screen-256color is not widely supported use it only on machines where the
+# matching terminfo entry is available. This also requires a terminal emulator
+# which supports 256 colors. Also used for tmux.
+use_256colors=
+if terminal_available screen-256color; then
+    # Called through SSH connection, assume the local system supports 256
+    # colors.
+    if test -n "$SSH_CONNECTION"; then
+        use_256colors=1
+    # We have rxvt-unicode installed, check if it supports 256 colors.
+    elif installed urxvt; then
+        # Thanks to deryni in #rxvt-unicode on Freenode (2012-10-14 22:54
+        # CEST) for the strings/grep idea. The grep check is for "correct" 256
+        # rxvt-unicode binaries (e.g. Debian's rxvt-unicode-256color), the
+        # terminal_info check for manual installations which modify
+        # rxvt-unicode's terminfo entry.
+        urxvt_path=`which urxvt`
+        urxvt_grep=`strings "$urxvt_path" | grep '^TERM=rxvt-')`
+        if test x"$urxvt_grep" = 'xTERM=rxvt-unicode-256color' \
+                || terminal_info rxvt-unicode \
+                    | grep -F 'colors#256' >/dev/null; then
+            use_256colors=1
+        fi
+    # Check if XTerm supports 256 colors (not a perfect check, but most XTerm
+    # support 256 colors).
+    elif terminal_available xterm-256color; then
+        use_256colors=1
+    fi
+fi
+if test -z "$use_256colors"; then
+    echo screenrc: removing 256 colors
     sed 's/Enable 256 color/Disable 256 color/;
          s/screen-256color/screen/' screenrc > screenrc.tmp
     mv screenrc.tmp screenrc
 fi
+# Some options are only necessary when running as root. They are marked as
+# "(ROOT)".
+if [ x`id -u` != x0 ]; then
+    echo screenrc: removing root options
+    grep -v '(ROOT)' screenrc > screenrc.tmp
+    mv screenrc.tmp screenrc
+fi
 # I use some features of screen which are only in Git. Drop them on machines
 # which have older versions. They are marked as "(GIT)".
-if [ $hostname != asp ]; then
+if [ $hostname != asp -o x`id -u` = x0 ]; then
     echo screenrc: removing Git features
     grep -v '(GIT)' screenrc > screenrc.tmp
     mv screenrc.tmp screenrc
@@ -41,12 +110,18 @@ if [ ! -d /sys/class/power_supply/BAT0 ]; then
     mv screenrc.tmp screenrc
 fi
 
+# Htop overwrites the comments in its configuration file.
+generate cat htoprc .in
+
 # Link setup for shells.
 link shell ~/.shell
 link bash ~/.bash
 link bash/rc ~/.bashrc
 link bash/profile ~/.bash_profile
 link bash/logout ~/.bash_logout
+if installed csh; then
+    link csh/rc ~/.cshrc
+fi
 link zsh ~/.zsh
 link zsh/env ~/.zshenv
 link zsh/rc ~/.zshrc
@@ -59,6 +134,13 @@ if installed colordiff; then
 fi
 link inputrc ~/.inputrc
 link screenrc ~/.screenrc
+if installed tmux; then
+    generate perl tmux.conf ./bin/remove-continuation.pl
+    link tmux.conf ~/.tmux.conf
+fi
+if installed htop; then
+    link htoprc ~/.htoprc
+fi
 
 # Create rlwrap history directory.
 mkdir -p shell/rlwrap