bin/rcup


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235

#!/bin/sh

#set -x

REPLACE_ALL=0

DEST_DIR=$HOME
DEFAULT_DOTFILES_DIR=$HOME/.dotfiles
HOSTNAME=`hostname -s`
DEBUG=:
PRINT=echo
VERBOSE=:
PROMPT=echo
MKDIR=mkdir
LN=ln
RM=rm

pushdir() {
  DIR_STACK="$DIR_STACK:$PWD/$1"
  $DEBUG "cd'ing to $1 from `pwd` with stack $DIR_STACK"
  cd $1
}

popdir() {
  current=`echo $DIR_STACK | sed -e 's/.*://g'`
  prior=`echo $DIR_STACK | sed -e "s|:$current$||" | sed -e 's/.*://g'`
  DIR_STACK=`echo $DIR_STACK | sed -e 's/:[^:]*$//'`
  $DEBUG "cd'ing to $prior from `pwd` with stack $DIR_STACK"
  cd $prior
}

build_path() {
  local dest=$1
  local file=$2
  local dotted=$3

  if [ $dotted -eq 1 ]; then
    echo $dest/$file
  else
    echo $dest/.$file
  fi
}

link_dir() {
  local dir=$1
  local dest_dir=$2
  local dotfiles_dir=$3
  local dotted=$4
  local dest_path=`build_path $dest_dir $dir $dotted`

  $VERBOSE "recurring on $dest_path"
  $MKDIR -p $dest_path
  pushdir $dir
  for f in *; do
    $DEBUG "handling the file $f"
    handle_file $f $dest_path $3/$1 1
  done
  popdir
}

link_file() {
  local file=$1
  local dest_dir=$2
  local dotfiles_dir=$3
  local dotted=$4
  local dest_file=`build_path $dest_dir $file $dotted`

  $PRINT "linking $dest_file"
  if [ -h $dest_file ]; then
    rm -f $dest_file
  fi
  $LN -s $dotfiles_dir/$file $dest_file
}

replace_file() {
  local file=$1
  local dest_dir=$2
  local dotfiles_dir=$3
  local dotted=$4
  local dest_file=`build_path $dest_dir $file $dotted`
  
  $RM -rf $dest_file
  link_file $file $dest_dir $dotfiles_dir $dotted
}

is_identical() {
  diff -q -s $1 $2 > /dev/null
}

handle_file() {
  local file=$1
  local dest_dir=$2
  local dotfiles_dir=$3
  local dotted=$4

  $DEBUG handle_file $1 $2 $3 $4

  if [ ! -e $file ]; then
    $VERBOSE "skipping non-existent file $file"
  elif [ -d $file ]; then
    link_dir $file $dest_dir $dotfiles_dir $dotted
  else
    dest_file=`build_path $dest_dir $file $dotted`
    if [ -e "$dest_file" ]; then
      if is_identical $file $dest_file; then
        $VERBOSE "identical $dest_file"
      elif [ $REPLACE_ALL -eq 1 ]; then
        replace_file $file $dest_dir $dotfiles_dir $dotted
      else
        $PROMPT "overwrite ${dest_file}? [ynaq]"
        read overwrite
        case $overwrite in
          a)
            REPLACE_ALL=1
            replace_file $file $dest_dir $dotfiles_dir $dotted
            ;;
          y) replace_file $file $dest_dir $dotfiles_dir $dotted
            ;;
          q) exit 1
            ;;
          *) $VERBOSE "skipping $dest_file"
            ;;
        esac
      fi
    else
      link_file $file $dest_dir $dotfiles_dir $dotted
    fi
  fi
}

metafile() {
  [ x$2 = 'xhost-' -o x$3 = 'xtag-' -o x$1 = "xRakefile" -o x$1 = "xinstall" ]
}

handle_command_line() {
  arg_tags=""
  verbosity=0
  version=0
  dotfiles_dirs=
  while getopts Vqvt:d: opt; do
    case "$opt" in
      t) arg_tags="$arg_tags $OPTARG";;
      v) verbosity=$(($verbosity + 1));;
      q) verbosity=$(($verbosity - 1));;
      d) dotfiles_dirs="$dotfiles_dirs $OPTARG";;
      V) version=1
    esac
  done
  shift $(($OPTIND-1))

  if [ $version -eq 1 ]; then
    version rcup
    exit 0
  elif [ $verbosity -ge 2 ]; then
    DEBUG=echo
    VERBOSE=echo
    PRINT=echo
  elif [ $verbosity -eq 1 ]; then
    DEBUG=:
    VERBOSE=echo
    PRINT=echo
  elif [ $verbosity -eq 0 ]; then
    DEBUG=:
    VERBOSE=:
    PRINT=echo
  else
    DEBUG=:
    VERBOSE=:
    PRINT=:
  fi

  if [ "x$arg_tags" != "x" ]; then
    TAGS=$arg_tags
  fi

  if [ "x$dotfiles_dirs" != "x" ]; then
    DOTFILES_DIRS=$dotfiles_dirs
  fi

  FILES=$@
}

if [ -e $HOME/.rcrc ]; then
  . $HOME/.rcrc
fi

. `dirname $0`/../libexec/rcm/rcm.sh
handle_command_line $*

if [ "x$DOTFILES_DIRS" = "x" ]; then
  DOTFILES_DIRS="$DOTFILES_DIRS $DEFAULT_DOTFILES_DIR"
fi

for DOTFILES_DIR in $DOTFILES_DIRS; do

  if [ ! -d $DOTFILES_DIR ]; then
    $VERBOSE "skipping non-existent directory: $DOTFILES_DIR" 
    continue
  fi

  cd $DOTFILES_DIR
  DIR_STACK=":$DOTFILES_DIR"

  for file in ${FILES:-*}; do
    host_portion=`echo $file | sed -e 's/host-.*/host-/'`
    tag_portion=`echo $file | sed -e 's/tag-.*/tag-/'`
    if ! metafile $file $host_portion $tag_portion; then
      handle_file $file $DEST_DIR $DOTFILES_DIR 0
    fi
  done

  cd $DOTFILES_DIR

  host_files=$DOTFILES_DIR/host-$HOSTNAME
  if [ -d $host_files ]; then
    pushdir `basename $host_files`
    for file in ${FILES:-*}; do
      handle_file $file $DEST_DIR $host_files 0
    done
    popdir
  fi

  cd $DOTFILES_DIR

  for tag in $TAGS; do
    if [ -d tag-$tag ]; then
      pushdir `basename tag-$tag`
      for file in ${FILES:-*}; do
        handle_file $file $DEST_DIR $DOTFILES_DIR/tag-$tag 0
      done
      popdir
    fi
  done

done