#!/usr/bin/env bash

#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

# Wrapper script that runs the Spark tests then reports QA results
# to github via its API.
# Environment variables are populated by the code here:
#+ https://github.com/jenkinsci/ghprb-plugin/blob/master/src/main/java/org/jenkinsci/plugins/ghprb/GhprbTrigger.java#L139

# Go to the Spark project root directory
FWDIR="$(cd `dirname $0`/..; pwd)"
cd "$FWDIR"

COMMENTS_URL="https://api.github.com/repos/apache/spark/issues/$ghprbPullId/comments"
PULL_REQUEST_URL="https://github.com/apache/spark/pull/$ghprbPullId"

COMMIT_URL="https://github.com/apache/spark/commit/${ghprbActualCommit}"
# GitHub doesn't auto-link short hashes when submitted via the API, unfortunately. :(
SHORT_COMMIT_HASH="${ghprbActualCommit:0:7}"

TESTS_TIMEOUT="120m" # format: http://linux.die.net/man/1/timeout

function post_message () {
  local message=$1
  local data="{\"body\": \"$message\"}"
  local HTTP_CODE_HEADER="HTTP Response Code: "
  
  echo "Attempting to post to Github..."
  
  local curl_output=$(
    curl `#--dump-header -` \
      --silent \
      --user x-oauth-basic:$GITHUB_OAUTH_KEY \
      --request POST \
      --data "$data" \
      --write-out "${HTTP_CODE_HEADER}%{http_code}\n" \
      --header "Content-Type: application/json" \
      "$COMMENTS_URL" #> /dev/null #| "$FWDIR/dev/jq" .id #| head -n 8
  )
  local curl_status=${PIPESTATUS[0]}

  if [ "$curl_status" -ne 0 ]; then
      echo "Failed to post message to GitHub." >&2
      echo " > curl_status: ${curl_status}" >&2
      echo " > curl_output: ${curl_output}" >&2
      echo " > data: ${data}" >&2
      # exit $curl_status
  fi
  
  local api_response=$(
    echo "${curl_output}" \
    | grep -v -e "^${HTTP_CODE_HEADER}"
  )
  
  local http_code=$(
    echo "${curl_output}" \
    | grep -e "^${HTTP_CODE_HEADER}" \
    | sed -r -e "s/^${HTTP_CODE_HEADER}//g"
  )

  if [ -n "$http_code" ] && [ "$http_code" -ne "201" ]; then
      echo " > http_code: ${http_code}." >&2
      echo " > api_response: ${api_response}" >&2
      echo " > data: ${data}" >&2
  fi
  
  if [ "$curl_status" -eq 0 ] && [ "$http_code" -eq "201" ]; then
    echo " > Post successful."
  fi
}

# check PR merge-ability and check for new public classes
{
  if [ "$sha1" == "$ghprbActualCommit" ]; then
    merge_note=" * This patch **does not** merge cleanly!"
  else
    merge_note=" * This patch merges cleanly."

    source_files=$(
        git diff master... --name-only              `# diff patch against master from branch point` \
      | grep -v -e "\/test"                         `# ignore files in test directories` \
      | grep -e "\.py$" -e "\.java$" -e "\.scala$"  `# include only code files` \
      | tr "\n" " "
    )
    new_public_classes=$(
        git diff master... ${source_files}      `# diff patch against master from branch point` \
      | grep "^\+"                              `# filter in only added lines` \
      | sed -r -e "s/^\+//g"                    `# remove the leading +` \
      | grep -e "trait " -e "class "            `# filter in lines with these key words` \
      | grep -e "{" -e "("                      `# filter in lines with these key words, too` \
      | grep -v -e "\@\@" -e "private"          `# exclude lines with these words` \
      | grep -v -e "^// " -e "^/\*" -e "^ \* "  `# exclude comment lines` \
      | sed -r -e "s/\{.*//g"                   `# remove from the { onwards` \
      | sed -r -e "s/\}//g"                     `# just in case, remove }; they mess the JSON` \
      | sed -r -e "s/\"/\\\\\"/g"               `# escape double quotes; they mess the JSON` \
      | sed -r -e "s/^(.*)$/\`\1\`/g"           `# surround with backticks for style` \
      | sed -r -e "s/^/  \* /g"                 `# prepend '  *' to start of line` \
      | sed -r -e "s/$/\\\n/g"                  `# append newline to end of line` \
      | tr -d "\n"                              `# remove actual LF characters`
    )

    if [ "$new_public_classes" == "" ]; then
      public_classes_note=" * This patch adds no public classes."
    else
      public_classes_note=" * This patch adds the following public classes _(experimental)_:"
      public_classes_note="${public_classes_note}\n${new_public_classes}"
    fi
  fi
}

# post start message
{
  start_message="\
  [QA tests have started](${BUILD_URL}consoleFull) for \
  PR $ghprbPullId at commit [\`${SHORT_COMMIT_HASH}\`](${COMMIT_URL})."
  
  start_message="${start_message}\n${merge_note}"
  # start_message="${start_message}\n${public_classes_note}"
  
  post_message "$start_message"
}

# run tests
{
  timeout "${TESTS_TIMEOUT}" ./dev/run-tests
  test_result="$?"

  if [ "$test_result" -eq "124" ]; then
    fail_message="**[Tests timed out](${BUILD_URL}consoleFull)** \
    for PR $ghprbPullId at commit [\`${SHORT_COMMIT_HASH}\`](${COMMIT_URL}) \
    after a configured wait of \`${TESTS_TIMEOUT}\`."

    post_message "$fail_message"
    exit $test_result
  else
    if [ "$test_result" -eq "0" ]; then
      test_result_note=" * This patch **passes** unit tests."
    else
      test_result_note=" * This patch **fails** unit tests."
    fi
  fi
}

# post end message
{
  result_message="\
  [QA tests have finished](${BUILD_URL}consoleFull) for \
  PR $ghprbPullId at commit [\`${SHORT_COMMIT_HASH}\`](${COMMIT_URL})."

  result_message="${result_message}\n${test_result_note}"
  result_message="${result_message}\n${merge_note}"
  result_message="${result_message}\n${public_classes_note}"

  post_message "$result_message"
}

exit $test_result