Cosmetics + a bit of paranoia regarding the created files.
[selector.git] / bash-selector.sh
1 #!/bin/bash
2
3 #  selector is a simple command line utility for selection of strings
4 #  with a dynamic pattern-matching.
5 #
6 #  Copyright (c) 2011, 2012 Francois Fleuret
7 #  Written by Francois Fleuret <francois@fleuret.org>
8 #
9 #  This file is part of selector.
10 #
11 #  selector is free software: you can redistribute it and/or modify
12 #  it under the terms of the GNU General Public License version 3 as
13 #  published by the Free Software Foundation.
14 #
15 #  selector is distributed in the hope that it will be useful, but
16 #  WITHOUT ANY WARRANTY; without even the implied warranty of
17 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18 #  General Public License for more details.
19 #
20 #  You should have received a copy of the GNU General Public License
21 #  along with selector.  If not, see <http://www.gnu.org/licenses/>.
22
23 #  This script installs two key-bindings:
24 #
25 #  Alt-r for selector-based command history
26 #
27 #  Alt-c for selector-based directoy history
28 #
29 #  Note that you have to call it with "source bash-selector.sh"
30 #  otherwise the key-bindings will not be effective in your current
31 #  bash
32
33 if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
34     echo "This script must be called with 'source $(basename $0)'" >&2
35     exit 1
36 fi
37
38 ######################################################################
39 # Selector-based command history
40 ######################################################################
41
42 function selector-history () {
43     selector --bash -j -y -u -c 7,4,0,3 -q <(history)
44 }
45
46 ######################################################################
47 # Selector-based directory history
48 ######################################################################
49
50 # The file where we will keep track of the directories and how many
51 # lines to keep in there
52
53 export SELECTOR_CD_HISTORY
54
55 [[ "${SELECTOR_CD_HISTORY}" ]] || SELECTOR_CD_HISTORY=${HOME}/.selector-cd-history
56
57 export SELECTOR_CD_HISTORY_SIZE
58
59 [[ "${SELECTOR_CD_HISTORY_SIZE}" ]] || SELECTOR_CD_HISTORY_SIZE=1000
60
61 # The function to use in place of the standard "cd"
62
63 function selector-cd () {
64     umask 077
65
66     if [[ -z "$1" ]]; then
67         cd
68     else
69         cd "$@"
70     fi
71
72     if [[ -f "${SELECTOR_CD_HISTORY}" ]]; then
73         TMP=$(mktemp /tmp/selector-cd.XXXXXX)
74         tail -$((SELECTOR_CD_HISTORY_SIZE-1)) < "${SELECTOR_CD_HISTORY}" > "${TMP}"
75         cat "${TMP}" > "${SELECTOR_CD_HISTORY}"
76         rm -f "${TMP}"
77     fi
78
79     echo "${PWD}" | sed -e "s!^${HOME}!~!" >> "${SELECTOR_CD_HISTORY}"
80 }
81
82 function selector-cd-search () {
83     umask 077
84
85     if [[ -f "${SELECTOR_CD_HISTORY}" ]]; then
86         PATH_TEMP=$(mktemp /tmp/selector-cd-path.XXXXXX)
87         selector -j -y -u -t "cd" -l "${SELECTOR_CD_HISTORY_SIZE}" -d -i -c 7,2,0,3 -o "${PATH_TEMP}" -q "${SELECTOR_CD_HISTORY}"
88         NEW_PATH="$(cat "${PATH_TEMP}" | sed -e 's!~!'${HOME}'!')"
89         if [[ -d "${NEW_PATH}" ]]; then
90             selector-cd "$(cat "${PATH_TEMP}" | sed -e 's!^~!'${HOME}'!')"
91         fi
92         \rm "${PATH_TEMP}"
93     else
94         echo "No cd history file '${SELECTOR_CD_HISTORY}'." >&2
95     fi
96 }
97
98 alias cd=selector-cd
99
100 ######################################################################
101 # The key-bindings themselves
102 ######################################################################
103
104     # M-t appends the selected history line and the end of the current
105     # one bind '"\C-[t":"\C-a\C-kselector-history\C-m\C-a\C-y\C-e"'
106
107 if [[ "$1" ]]; then
108
109     while [[ "$1" ]]; do
110
111         case "$1" in
112
113             --hist)
114                 # M-r puts the selected history line in place of the current one
115                 bind '"\C-[r":"\C-a\C-kselector-history\C-m"'
116                 ;;
117
118             --cd)
119                 # M-c provides a dynamic list of directories to cd into
120                 bind '"\C-[c":"\C-a\C-kselector-cd-search\C-m"'
121                 ;;
122
123             *)
124                 echo "Unknown argument $1" >&2
125                 ;;
126         esac
127
128         shift
129
130     done
131
132 else
133
134     echo "source bash-selector.sh <--hist|--cd> [...]"
135     echo
136     echo "Defines bash functions, and installs key-bindings and aliases to use selector"
137     echo "for history search with M-r and/or intelligent cd history with M-c."
138
139 fi