]> ruderich.org/simon Gitweb - config/dotfiles.git/blob - lib.sh
lib.sh: Add simple_cpp() to replace "defines".
[config/dotfiles.git] / lib.sh
1 # Setup functions and settings used in subdirectories.
2 #
3 # Their setup.sh script sources this file.
4
5 # Copyright (C) 2009-2013  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 # csh gives the error "Unknown colorls variable `su'." when used with newer
22 # options supported by Zsh or GNU ls.
23 unset LS_COLORS
24
25
26 # Check if the given program is installed. `type` is portable, `which` is not.
27 installed() {
28     type "$1" >/dev/null 2>&1
29 }
30 # Get the path of the given program. Thanks to Gilles on [1] (read on
31 # 2013-03-10) for the PATH-walking idea. `which` is not portable and `type`
32 # has no well-formed output format.
33 #
34 # [1]: http://unix.stackexchange.com/questions/4988/how-do-i-test-to-see-if-an-application-exists-in-path/4991
35 installed_path() {
36     test -z "$1" && return 1
37
38     # Keep IFS change local.
39     (
40         IFS=:
41         # Walk PATH.
42         for directory in $PATH; do
43             if test -x "$directory/$1"; then
44                 echo "$directory/$1"
45                 return 0
46             fi
47         done
48
49         return 1
50     )
51 }
52
53 # Usage: cmd_i <cmd> ... <file>
54 #
55 # Run <cmd> with all arguments (including the last file) and write the result
56 # to the temporary file <file>.tmp and then renamed that file to <file>. This
57 # can't be done in-place (e.g. cmd <file >file) because it truncates the file.
58 cmd_i() {
59     # Get last argument.
60     last=
61     for x; do
62         last="$x"
63     done
64
65     "$@" >"$last".tmp
66     mv "$last".tmp "$last"
67 }
68
69 # Usage: sed_i ... <file>
70 #
71 # sed -i is not compatible due to different implementations. See cmd_i.
72 sed_i() {
73     cmd_i sed "$@"
74 }
75
76 # Usage: perl_line_filter <cmd> ...
77 #
78 # Run the perl command cmd on each line before printing it.
79 perl_line_filter() {
80     cmd="$1"
81     shift
82
83     # Can't use -pe because it uses <> which treats the arguments as files.
84     perl -e "use strict; use warnings; while (<STDIN>) { $cmd; print; }" "$@"
85 }
86
87 # Usage: simple_cpp <FIRST> <SECOND> .. -- <replacement-for-first> ...
88 #
89 # Replaces each FIRST (on word boundaries) with <replacement-for-first> like a
90 # simple cpp replacement.
91 simple_cpp() {
92     cmd='my $i = 0;'
93
94     for x; do
95         shift
96
97         if test x"$x" = x--; then
98             break
99         fi
100
101         cmd="$cmd s/\b$x\b/\$ARGV[\$i]/g; \$i++;"
102     done
103
104     perl_line_filter "$cmd" -- "$@"
105 }
106
107 # Print the current OS. The following OS are supported at the moment:
108 # - Debian (debian)
109 # - Gentoo (gentoo)
110 # - Mac OS X (darwin)
111 # - Solaris/OpenSolaris (sun)
112 # - FreeBSD (freebsd)
113 # If an unsupported OS is used an error is printed.
114 os() {
115     if test -f /etc/debian_version; then
116         echo debian
117     elif test -f /etc/gentoo-release; then
118         echo gentoo
119     elif test x`uname` = xDarwin; then
120         echo darwin
121     elif test x`uname` = xSunOS; then
122         echo sun
123     elif test x`uname` = xFreeBSD; then
124         echo freebsd
125     else
126         echo unsupported OS! >&2
127         return 1
128     fi
129 }
130
131 # Creates a symbolic link for file $1 in dirname of $2 with name of basename
132 # $2.
133 #
134 # `./link.sh example ~/.examplerc` creates a symbolic link to example
135 # (wherever it is located) in ~/ named .examplerc.
136 link() {
137     local pwd base source target >/dev/null 2>&1 || true
138
139     # Get all necessary paths.
140     pwd=`pwd`
141     base=`echo "$2" | sed "s|\~|$HOME|"` # expand ~, some sh don't do it
142     base=`dirname "$base"`
143     source=`echo "$pwd/$1" | sed "s|$base/||"`
144     target=`basename "$2"`
145
146     # Go to the directory where the link is going to be created.
147     cd "$base"
148
149     # Abort if the target file exists and is no symbolic link. Prevents
150     # overwriting real files.
151     if ( test -f "$target" && test ! -h "$target" ) || \
152             ( test -s "$target" && test ! -h "$target" ); then
153         echo "link(): target '$target' exists already and is no symbolic link!" >&2
154         exit 1
155     fi
156
157     # Make sure the source exists (is file, directory or link).
158     if test ! -f "$source" && test ! -d "$source" && test ! -h "$source"; then
159         echo "link(): source '$source' doesn't exist!" >&2
160         exit 1
161     fi
162
163     # Create the new symbolic link; remove the old one if necessary.
164     echo "link(): linking '$source' to '$target'"
165     rm -f "$target"
166     ln -s "$source" "$target"
167
168     # Go back to the directory where we were before.
169     cd "$pwd"
170 }
171
172 # Generate a file from a source file using a given command. A warning not to
173 # edit it is automatically added to the created file.
174 #
175 # Usage: generated() <file> <extension> <cmd..>
176 #
177 # If an empty extension is provided, the file is modified in-place (through a
178 # temporary file).
179 generate() {
180     local file      >/dev/null 2>&1 || true
181     local file_tmp  >/dev/null 2>&1 || true
182     local extension >/dev/null 2>&1 || true
183
184     # Get command and target file.
185     file="$1"
186     extension="$2"
187     shift
188     shift
189
190     if test -z "$extension"; then
191         file_tmp="$file.tmp"
192     else
193         # We only need this message if we generate a new file.
194         printf "%s: generating from '%s' (%s)\n" \
195             "$file" "$file$extension" "$1"
196
197         echo '###################################'  >"$file"
198         echo '# WARNING! DO NOT EDIT THIS FILE! #' >>"$file"
199         echo '###################################' >>"$file"
200         echo >>"$file"
201         echo "# It was generated from $file$extension on `date`." >>"$file"
202         echo >>"$file"
203
204         file_tmp="$file"
205     fi
206
207     # Generate $file from $file$extension using the given command.
208     "$@" <"$file$extension" >>"$file_tmp"
209
210     if test -z "$extension"; then
211         mv "$file_tmp" "$file"
212     fi
213 }