#!/bin/sh
#
# 2012-2013 Steven Armstrong (steven-cdist at armstrong.cc)
#
# This file is part of cdist.
#
# cdist 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.
#
# cdist 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 cdist. If not, see <http://www.gnu.org/licenses/>.
#

owner="$(cat "$__object/parameter/owner" 2>/dev/null || echo "$__object_id")"
if [ -f "$__object/parameter/file" ]; then
   file="$(cat "$__object/parameter/file")"
else
   home="$(cut -d':' -f 6 "$__object/explorer/passwd")"
   file="$home/.ssh/authorized_keys"
fi

entry="$__object/files/entry"
if [ ! -s "$__object/explorer/entry" ]; then
   state_is='absent'
else
   state_is=$(diff -q "$entry" "$__object/explorer/entry" >/dev/null \
      && echo present \
      || echo changed
   )
fi

state_should="$(cat "$__object/parameter/state" 2>/dev/null || echo present)"
if [ "$state_should" = "$state_is" ]; then
   # Nothing to do, move along
   exit 0
fi

remove_entry() {
   # NOTE: keep variables in sync in manifest/explorer/gencode-*
   prefix="#cdist:$__object_name"
   suffix="#/cdist:$__object_name"
   cat << DONE
tmpfile=\$(mktemp ${file}.cdist.XXXXXXXXXX)
# preserve ownership and permissions by copying existing file over tmpfile
cp -p "$file" "\$tmpfile"
awk -v prefix="$prefix" -v suffix="$suffix" '
{
   if (index(\$0,prefix)) {
      triggered=1
   }
   if (triggered) {
      if (index(\$0,suffix)) {
         triggered=0
      }
   } else {
      print
   }
}' "$file" > "\$tmpfile"
mv -f "\$tmpfile" "$file"
DONE
}

case "$state_should" in
   present)
      if [ "$state_is" = "changed" ]; then
         remove_entry
      fi
      cat << DONE
cat >> "$file" << ${__type##*/}_DONE
$(cat "$entry")
${__type##*/}_DONE
DONE
   ;;
   absent)
      remove_entry
   ;;
esac