#!/bin/sh -e
#
# 2011-2012 Nico Schottelius (nico-cdist at schottelius.org)
# 2013 Steven Armstrong (steven-cdist 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/>.
#

destination="/$__object_id"
state_should="$(cat "$__object/parameter/state")"
type="$(cat "$__object/explorer/type")"

[ "$state_should" = "exists" ] && [ "$type" = "file" ] && exit 0 # nothing to do

if [ "$state_should" = "pre-exists" ]; then
   if [ -f "$__object/parameter/source" ]; then
      echo "--source cannot be used with --state pre-exists"
      exit 1
   fi

   case $type in
      file)
         # nothing to do
         exit 0
      ;;
      none)
         printf 'File "%s" does not exist\n' "$destination" >&2
         exit 1
      ;;
      directory|symlink)
         printf 'File "%s" exists and is a %s, but should be a regular file\n' "$destination" "$type" >&2
         exit 1
      ;;
      *)
         printf 'File or directory "%s" is in an unknown state\n' "$destination" >&2
         exit 1
      ;;
   esac
fi

upload_file=
create_file=
if [ "$state_should" = "present" ] || [ "$state_should" = "exists" ]; then
   if [ ! -f "$__object/parameter/source" ]; then
      remote_stat="$(cat "$__object/explorer/stat")"
      if [ -z "$remote_stat" ]; then
         create_file=1
         echo create >> "$__messages_out"
      fi
   else
      source="$(cat "$__object/parameter/source")"
      if [ "$source" = "-" ]; then
         source="$__object/stdin"
      fi
      if [ ! -f "$source" ]; then
         echo "Source \"$source\" does not exist." >&2
         exit 1
      else
         if [ "$type" != "file" ]; then
            # destination is not a regular file, upload source to replace it
            upload_file=1
         else
            local_cksum="$(cksum < "$source")"
            remote_cksum="$(cat "$__object/explorer/cksum")"
            if [ "$local_cksum" != "$remote_cksum" ]; then
               # destination is a regular file, but not the right one
               upload_file=1
            fi
         fi
      fi
   fi
   if [ "$create_file" ] || [ "$upload_file" ]; then
      # tell gencode-remote that we created or uploaded a file and that it must
      # set all attributes no matter what the explorer retreived
      mkdir "$__object/files"
      touch "$__object/files/set-attributes"

      # upload file to temp location
      tempfile_template="${destination}.cdist.XXXXXXXXXX"
      cat << DONE
destination_upload="\$($__remote_exec $__target_host "mktemp $tempfile_template")"
DONE
      if [ "$upload_file" ]; then
         echo upload >> "$__messages_out"
         # IPv6 fix
         if echo "${__target_host}" | grep -q -E '^[0-9a-fA-F:]+$'
         then
             my_target_host="[${__target_host}]"
         else
             my_target_host="${__target_host}"
         fi
         cat << DONE
$__remote_copy "$source" "${my_target_host}:\$destination_upload"
DONE
      fi
# move uploaded file into place
cat << DONE
$__remote_exec $__target_host "rm -rf \"$destination\"; mv \"\$destination_upload\" \"$destination\""
DONE
   fi
fi
