pre-receive

#!/bin/bash

# Puppet attempts to source ~/.puppet and will error if $HOME is not set
if [[ -z $HOME ]]
then
  HOME="$(grep "${USER}:" /etc/passwd | awk -F ':' '{print $6}')"
  export HOME
fi

failures=0
RC=0

hook_dir="$(dirname "$0")"
hook_symlink="$(readlink -f "$0")"

# Figure out where commit hooks are if pre-receive is setup as a symlink
if [[ ! -z "$hook_symlink" ]]; then
  subhook_root="$(dirname "$hook_symlink")/commit_hooks"
else
  subhook_root="$hook_dir/commit_hooks"
fi

tmptree=$(mktemp -d)

# Prevent tput from throwing an error by ensuring that $TERM is always set
if [[ -z "$TERM" ]]; then
    TERM=dumb
fi
export TERM

# Decide if we want puppet-lint
# Decide if we want the puppet future parser (already on puppet 4?)
CHECK_PUPPET_LINT="enabled"
USE_PUPPET_FUTURE_PARSER="enabled"
CHECK_INITIAL_COMMIT="disabled"
if [[ -e ${subhook_root}/config.cfg ]] ; then
    source "${subhook_root}/config.cfg"
fi
# Only puppet 3.2.1 - 3.8 support "--parser future" option.
case $(puppet --version) in
  4*) USE_PUPPET_FUTURE_PARSER="disabled" ;;
esac

while read -r oldrev newrev refname; do
    git archive "$newrev" | tar x -C "$tmptree"

    # for a new branch oldrev is 0{40}, set newrev to branch name and oldrev to parent branch
    if [[ $oldrev == "0000000000000000000000000000000000000000" ]]; then
      newrev=$(git rev-parse --abbrev-ref HEAD)
      oldrev=$(git show-branch | grep '\*' | grep -v "$newrev" | head -n1 | sed 's/.*\[\(.*\)\].*/\1/' | sed 's/[\^~].*//')
    fi

    files_list=''
    if [[ "x$oldrev" == 'x' ]]; then
      if [[ $CHECK_INITIAL_COMMIT != "disabled" ]] ; then
        files_list=$(git ls-tree --full-tree -r HEAD --name-only)
      else
        echo "Skipping file checks this is the initial commit..."
      fi
    else
      files_list=$(git diff --name-only "$oldrev" "$newrev" --diff-filter=ACM)
    fi

    for changedfile in $files_list; do
        tmpmodule="$tmptree/$changedfile"
        [[ -f "$tmpmodule" ]] || continue
        #check puppet manifest syntax
        if type puppet >/dev/null 2>&1; then
            if [[ $(echo "$changedfile" | grep -q '\.*\.epp$'; echo $?) -eq 0 ]]; then
                ${subhook_root}/puppet_epp_syntax_check.sh "$tmpmodule" "${tmptree}/"
                RC=$?
                if [[ $RC -ne 0 ]]; then
                  failures=$((failures + 1))
                fi
            elif [[ $(echo "$changedfile" | grep -q '\.*\.pp$'; echo $?) -eq 0 ]]; then
                ${subhook_root}/puppet_manifest_syntax_check.sh "$tmpmodule" "${tmptree}/" "$USE_PUPPET_FUTURE_PARSER"
                RC=$?
                if [[ $RC -ne 0 ]]; then
                  failures=$((failures + 1))
                fi
            fi
        else
            echo "puppet not installed. Skipping puppet syntax checks..."
        fi

        if type ruby >/dev/null 2>&1; then
            #check ruby syntax
            if [[ $(echo "$changedfile" | grep -q '\.*\.rb$'; echo $?) -eq 0 ]]; then
                ${subhook_root}/ruby_syntax_check.sh "$tmpmodule" "${tmptree}/"
                RC=$?
                if [[ $RC -ne 0 ]]; then
                  failures=$((failures + 1))
                fi
            fi

            #check erb (template file) syntax
            if type erb >/dev/null 2>&1; then
                if [[ $(echo "$changedfile" | grep -q '\.*\.erb$'; echo $?) -eq 0 ]]; then
                    ${subhook_root}/erb_template_syntax_check.sh "$tmpmodule" "${tmptree}/"
                    RC=$?
                    if [[ $RC -ne 0 ]]; then
                      failures=$((failures + 1))
                    fi
                fi
            else
                echo "erb not installed. Skipping erb template checks..."
            fi

            #check hiera data (yaml/yml) syntax
            if [[ $(echo "$changedfile" | grep -q '\.*\.yaml$\|\.*\.yml$'; echo $?) -eq 0 ]]; then
                ${subhook_root}/yaml_syntax_check.sh "$tmpmodule" "${tmptree}/"
                RC=$?
                if [[ $RC -ne 0 ]]; then
                  failures=$((failures + 1))
                fi
            fi

            #check hiera data (json) syntax
            if [[ $(echo "$changedfile" | grep -q '\.*\.json$'; echo $?) -eq 0 ]]; then
                ${subhook_root}/json_syntax_check.sh "$tmpmodule" "${tmptree}/"
                RC=$?
                if [[ $RC -ne 0 ]]; then
                  failures=$((failures + 1))
                fi
            fi
        else
            echo "ruby not installed. Skipping erb/yaml/ruby checks..."
        fi

        #puppet manifest styleguide compliance
        if [[ $CHECK_PUPPET_LINT != "disabled" ]] ; then
            if type puppet-lint >/dev/null 2>&1; then
                if [[ $(echo "$changedfile" | grep -q '\.*\.pp$' ; echo $?) -eq 0 ]]; then
                    ${subhook_root}/puppet_lint_checks.sh "$CHECK_PUPPET_LINT" "$tmpmodule" "${tmptree}/"
                    RC=$?
                    if [[ $RC -ne 0 ]]; then
                      failures=$((failures + 1))
                    fi
                fi
            else
                echo "puppet-lint not installed. Skipping puppet-lint tests..."
            fi
        fi
        #r10k puppetfile syntax check
        if hash r10k >/dev/null 2>&1; then
          if [ "$changedfile" = "Puppetfile" ]; then
                ${subhook_root}/r10k_syntax_check.sh $tmptree/$changedfile
                RC=$?
                if [ "$RC" -ne 0 ]; then
                       failures=`expr $failures + 1`
                fi
          fi
        else
            echo "r10k not installed. Skipping r10k Puppetfile test..."
        fi
    done
done
rm -rf "$tmptree"

#summary
if [[ $failures -ne 0 ]]; then
    echo -e "$(tput setaf 1)Error: $failures subhooks failed. Declining push.$(tput sgr0)"
    exit 1
fi

exit 0