#!/bin/bash

# Copyright 2021 The Android Open Source Project
#
# Licensed 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.

# Fail fast on any error.
set -e

HOST_TOOLS_DIR=$(realpath $(dirname $0))
source ${HOST_TOOLS_DIR}/common.sh

RELEASE_ID=${1}
if [ -z ${RELEASE_ID} ]; then
  echo "Usage:"
  echo "  ${0} <release ID>"
  echo
  echo "e.g. ${0} 2020d"
  exit 1
fi

SCRIPT_PATH=$(realpath $0)

# -f = report failures
# -L = follow redirects
CURL_CMD="curl -f -L"

TZBB_PROJECT_URL=https://github.com/evansiroky/timezone-boundary-builder/
SERVER_RELEASE_BASE_URL=${TZBB_PROJECT_URL}releases/download
SERVER_SOURCE_BASE_URL=https://raw.githubusercontent.com/evansiroky/timezone-boundary-builder/master

INPUT_DATA_FILE=input-data.zip
LICENSE_FILE=DATA_LICENSE

function download() {
  set -e
  SERVER_BASE_URL=${1}
  FILE_NAME=${2}
  LOCAL_DIR=${3}

  URL=${SERVER_BASE_URL}/${FILE_NAME}
  LOCAL_FILE_NAME=${LOCAL_DIR}/${FILE_NAME}
  echo Downloading ${FILE_NAME} from ${URL}...
  ${CURL_CMD} ${URL} --output ${LOCAL_FILE_NAME}

  DOWNLOAD_DATE=$(date)
  SHA=$(sha1sum ${LOCAL_FILE_NAME} | awk '{ print $1}')

  # Add a METADATA file URL entry for every file we download.
  cat << EOF >> ${METADATA_FILE}
  url {
    type: OTHER
    value: "${FILE_NAME} downloaded from ${URL} on ${DOWNLOAD_DATE}, SHA1=${SHA}"
  }
EOF
}

mkdir -p ${HOST_INPUTS_DIR}
echo Removing existing data files...
rm -f ${HOST_INPUTS_DIR}/*

METADATA_FILE=${HOST_INPUTS_DIR}/METADATA
# Start the METADATA file
cat << EOF > ${METADATA_FILE}
// Generated file DO NOT EDIT
// Run download_input_files.sh
name: "android_inputs"
description:
    "Assorted files downloaded from ${TZBB_PROJECT_URL}"

third_party {
  url {
    type: HOMEPAGE
    value: "${TZBB_PROJECT_URL}"
  }
EOF

download ${SERVER_SOURCE_BASE_URL} ${LICENSE_FILE} ${HOST_INPUTS_DIR}
download ${SERVER_RELEASE_BASE_URL}/${RELEASE_ID} ${INPUT_DATA_FILE} ${HOST_INPUTS_DIR}

PROTO_UPGRADE_DATE=$(date +'{ year: '%Y' month: '%-m' day: '%-d' }')

# Finish the METADATA file
cat << EOF >> ${METADATA_FILE}
  version: "${RELEASE_ID}"
  last_upgrade_date ${PROTO_UPGRADE_DATE}
  license_type: RESTRICTED
}
EOF

ln -sr ${HOST_INPUTS_DIR}/DATA_LICENSE ${HOST_INPUTS_DIR}/LICENSE

# Handle the unzip step to HOST_DOWNLOADS_DIR
unzip -d ${HOST_DOWNLOADS_DIR} ${HOST_INPUTS_DIR}/${INPUT_DATA_FILE}

echo Look in ${HOST_INPUTS_DIR} for input files....