# Copyright 2021 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

# Symlinks built TF import binaries from the specified directory (defaults to
# the appropriate bazel-bin/ subdirectory) into the corresponding Python
# packages. If the binary directory is contained within the root directory, it
# uses a relative symlink, which makes this work when the repository is copied
# or mounted in a Docker container under some other path.

set -euo pipefail

ROOT_DIR="${ROOT_DIR:-$(git rev-parse --show-toplevel)}"
SCRIPT_DIR="$(dirname -- "$( readlink -f -- "$0"; )")";

BINARIES_DIR="${1:-${SCRIPT_DIR}/bazel-bin/iree_tf_compiler}"

function symlink_import_binary() {
  local type="$1"
  local import_binary="${BINARIES_DIR}/iree-import-${type}"
  if [ -f "${import_binary}" ]; then
    local to="${SCRIPT_DIR}/python_projects/iree_${type}/iree/tools/${type}"
    local from="$(realpath --no-symlinks --relative-to=${to} --relative-base="${ROOT_DIR}" "${import_binary}")"
    ln --symbolic --verbose --force "${from}" "${to}"
  fi
}

symlink_import_binary tf
symlink_import_binary tflite
symlink_import_binary xla