pokeapi/Resources/scripts/updater.sh

#!/bin/bash

set -o pipefail

org='PokeAPI'
data_repo='api-data'
engine_repo='pokeapi'
branch_name='updated-data'
username='pokeapi-machine-user'
email='pokeapi.co@gmail.com'

function cleanexit {
	echo "Exiting"
	echo "$2"
	exit "$1"
}

# Create and use a personal folder
prepare() {
  mkdir -p ./repositories
  cd repositories || cleanexit 1 "Failed to cd"
}

# Not used
# Check and return the number of the Pull Request that started this job
get_invokator_pr_number() {
  if [ -z "$CIRCLE_PULL_REQUEST" ]; then
    echo "${CIRCLE_PULL_REQUEST##*/}"
  fi
}

# Clone the repository containing the static JSON files
clone() {
  git clone "https://github.com/${org}/${data_repo}.git" "$data_repo"
}

# Configure git to use the supplied user when committing
configure_git() {
  git config --global user.name "$username"
  git config --global user.email "$email"
}

pr_input_notification_comment() {
  cat <<EOF
{
  "body": "A [PokeAPI/api-data](https://github.com/PokeAPI/api-data) refresh has started. If everything works out in 30 minutes a Pull Request will be created and assigned to the PokeAPI Core team to be reviewed. If approved and merged new data will soon be available worldwide."
}
EOF
}

# If the job was started by a Pull Request add a comment notifying the users that a deployment process has started
notify_updater_start() {
  if ! [ -z "$CIRCLE_PULL_REQUEST" ]; then
    pr_number="${CIRCLE_PULL_REQUEST##*/}"
    curl -f -H "Authorization: token $MACHINE_USER_GITHUB_API_TOKEN" -X POST --data "$(pr_input_notification_comment)" "https://api.github.com/repos/$org/$engine_repo/issues/${pr_number}/comments"
  fi
}

# Run the updater script (https://github.com/PokeAPI/api-data/blob/master/updater/cmd.bash) which will generate the new pokeapi data and push it to the api-data repository under a new branch
run_updater() {
  cd "${data_repo}/updater" || cleanexit 1 "Failed to cd"

  # Wait to be sure PokeAPI/pokeapi:origin/master has been updated on Github with the lastest merged PR content
  sleep 10
  
  # Switch to test branch # TODO: don't checkout
  git fetch
  git checkout test 

  # Build the updater image
  docker build -t pokeapi-updater .
  if [ $? -ne 0 ]; then
    cleanexit 1 "Failed to build the pokeapi-updater image"
  fi

  # Run the updater
  docker run --privileged -e COMMIT_EMAIL="$email" -e COMMIT_NAME="$username" -e BRANCH_NAME="$branch_name" -e REPO_POKEAPI="https://github.com/${org}/${engine_repo}.git" -e REPO_DATA="https://${MACHINE_USER_GITHUB_API_TOKEN}@github.com/${org}/${data_repo}.git" pokeapi-updater
  if [ $? -ne 0 ]; then
    cleanexit 2 "Failed to run the pokeapi-updater container"
  fi

  cd .. || cleanexit 1 "Failed to cd"
}

# Check if the updater script has pushed the data to a new branch
check_remote_branch() {
  # Wait for Github to update origin/${branch_name}
  sleep 10

  curl -f -H "Authorization: token $MACHINE_USER_GITHUB_API_TOKEN" -X GET "https://api.github.com/repos/$org/$data_repo/branches/$1"
  if [ $? -ne 0 ]; then
    cleanexit 1 "The updater script failed to push the new data"
  fi
}

pr_input_content() {
  cat <<EOF
{
  "title": "API data update",
  "body": "Incoming data generated by https://github.com/${org}/${engine_repo} CircleCI worker",
  "head": "$branch_name",
  "base": "master",
  "assignees": [
    "Naramsim"
  ],
  "labels": [
    "api-data-update"
  ]
}
EOF
}

# Create a Pull Request to merge the branch recently pushed by the updater with the master branch
create_pr() {
  pr_number=$(curl -H "Authorization: token $MACHINE_USER_GITHUB_API_TOKEN" -X POST --data "$(pr_input_content)" "https://api.github.com/repos/$org/$data_repo/pulls" | jq '.number')
  if [[ "$pr_number" = "null" ]]; then
    cleanexit 1 "Couldn't create the Pull Request"
  fi
  echo "$pr_number"
}

pr_input_assignees_and_labels() {
  cat <<EOF
{
  "assignees": [
    "Naramsim"
  ],
  "labels": [
    "api-data-update"
  ]
}
EOF
}

# Assign the PR to Naramsim and add a label
customize_pr() {
  # Wait for Github to open the PR
  sleep 10
  
  pr_number=$1
  curl -H "Authorization: token $MACHINE_USER_GITHUB_API_TOKEN" -X PATCH --data "$(pr_input_assignees_and_labels)" "https://api.github.com/repos/$org/$data_repo/issues/$pr_number"
  if [ $? -ne 0 ]; then
		echo "Couldn't add Assignees and Labes to the Pull Request"
	fi
}

pr_input_reviewers() { # TODO: Add core team
  cat <<EOF
{
  "reviewers": [
    "Naramsim"
  ]
}
EOF
}

# Request the Core team to review the Pull Request
add_reviewers_to_pr() {
  pr_number=$1
  curl -H "Authorization: token $MACHINE_USER_GITHUB_API_TOKEN" -X POST --data "$(pr_input_reviewers)" "https://api.github.com/repos/$org/$data_repo/pulls/$pr_number/requested_reviewers"
  if [ $? -ne 0 ]; then
    echo "Couldn't add Reviewers to the Pull Request"
  fi
}

prepare
clone
configure_git
notify_updater_start
run_updater
check_remote_branch "$branch_name"
pr_number=$(create_pr)
customize_pr "$pr_number"
add_reviewers_to_pr "$pr_number"