#!/usr/bin/env bash

#
# SPDX-License-Identifier: Apache-2.0
#
# The OpenSearch Contributors require contributions made to
# this file be licensed under the Apache-2.0 license or a
# compatible open source license.
#
# Modifications Copyright OpenSearch Contributors. See
# GitHub history for details.
#

#
# Licensed to Elasticsearch B.V. under one or more contributor license
# agreements. See the NOTICE file distributed with this work for additional
# information regarding copyright ownership. Elasticsearch B.V. 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.
#

set -e
set +x

# Two stage script: first it compiles using Node Docker container, then it runs
# itself from within another docker container to deploy to GCP.

# Usage:
# * Compile and deploy:          ./scripts/deploy/deploy_docs
# * Deploy only without docker:  ./scripts/deploy/deploy_docs nodocker

# Expected env variables:
# * GPROJECT - GCE project name, e.g. elastic-bekitzur
# * GCE_ACCOUNT - credentials for the google service account (JSON blob)
# * GIT_BRANCH - current Git branch or tag (e.g. "refs/heads/main", "v18.2.1")

if [[ -z "${GPROJECT}" ]]; then
    echo "GPROJECT is not set, e.g. 'GPROJECT=elastic-bekitzur'"
    exit 1
fi
if [[ -z "${GCE_ACCOUNT}" ]]; then
  echo "GCE_ACCOUNT is not set. Expected Google service account JSON blob."
  exit 1
fi
if [[ -z "${GIT_BRANCH}" ]]; then
  echo "GIT_BRANCH is not set, e.g. 'GIT_BRANCH=refs/heads/main'"
  exit 1
fi

post_comment_to_gh()
{
  if [[ -z "${GITHUB_TOKEN}" ]]; then
    echo "Warning: GITHUB_TOKEN is not set; cannot publish PR docs preview URL to Github."
  else
    printf '\nAdding comment to GitHub Pull Request: %i\n' "$pull_request_id"

    comment="Preview documentation changes for this PR: https://eui.elastic.co/pr_$pull_request_id/"

    set +x
    curl \
      --silent \
      --location \
      --header "Authorization: token ${GITHUB_TOKEN}" \
      --request POST \
      --data "$(jq --null-input --arg comment "$comment" '{ body: $comment }')" \
      "https://api.github.com/repos/elastic/eui/issues/${pull_request_id}/comments"
  fi
}

publish_to_bucket()
{
  echo "Copying $PWD/docs/* to $full_bucket_path"
  gsutil -m \
    -h "Cache-Control:public, max-age=$max_age" \
    cp \
    -r \
    -a public-read \
    -z js,css,html \
    "$PWD/docs/*" "$full_bucket_path"
}

if [[ "$1" != "nodocker" ]]; then
  ./scripts/deploy/build_docs

  # we look this up to inject into the next docker run so it can decide whether
  # we need to overwrite the root /index.html file with a new one.
  echo "Looking up most-current release info"
  CURRENT_RELEASE=$(docker run \
    --rm -i \
    --volume $PWD:/app \
    --workdir /app \
    node:10 \
    bash -c 'git tag | xargs npx semver' | tail -n 1)

  # Run this script from inside the docker container, using google/cloud-sdk image
  echo "Deploying to bucket"
  docker run \
    --rm -i \
    --env GCE_ACCOUNT \
    --env GIT_BRANCH \
    --env GPROJECT \
    --env pull_request_id \
    --env CURRENT_RELEASE="$CURRENT_RELEASE" \
    --env HOME=/tmp \
    --volume $PWD:/app \
    --user=$(id -u):$(id -g) \
    --workdir /app \
    'google/cloud-sdk:slim' \
    /app/scripts/deploy/deploy_docs nodocker "$@"
  unset GCE_ACCOUNT

  if [[ ! -z "${pull_request_id}" ]]; then
    post_comment_to_gh
  fi
  unset GITHUB_TOKEN
else
  # Copying files to the bucket
  # Login to the cloud with the service account
  gcloud auth activate-service-account --key-file <(echo "$GCE_ACCOUNT")
  unset GCE_ACCOUNT

  # All branches go into correspondingly named dirs.
  # remove remote "origin/", leaving just the branch name
  BRANCH="${GIT_BRANCH#*/}"

  # Copy files
  OUI_DOCS_PROJECT=oui-docs-live
  BUCKET=${GPROJECT}-${OUI_DOCS_PROJECT}

  max_age="1800"

  if [[ ! -z "${pull_request_id}" ]]; then
    full_bucket_path="gs://$BUCKET/pr_$pull_request_id/"
    publish_to_bucket
  else
    full_bucket_path="gs://$BUCKET/$BRANCH/"
    max_age="604800"
    publish_to_bucket

    # if this is the highest release number to date, also publish the docs to the
    # root directory
    if [[ "v$CURRENT_RELEASE" == "$GIT_BRANCH" ]]; then
      echo "This is the current release."
      full_bucket_path="gs://$BUCKET/"
      publish_to_bucket
    fi
  fi
fi