#!/bin/bash
set -e

source ./env.sh

BUILD_TYPE=Release

if [[ $1 == "musl" ]]; then
    echo "*** Build musl-libc grpc_ratls ***"
    CC=occlum-gcc
    CXX=occlum-g++
    DCAP_LIB_PATH="/opt/occlum/toolchains/dcap_lib/musl"
    INSTALL_PREFIX="/usr/local/occlum/x86_64-linux-musl"
    GRPC_INSTALL_PATH="/tmp/grpc_ratls/musl"
else
    echo "*** Build glibc grpc_ratls ***"
    CC=gcc
    CXX=g++
    DCAP_LIB_PATH="/opt/occlum/toolchains/dcap_lib/glibc"
    INSTALL_PREFIX="/usr/local"
    GRPC_INSTALL_PATH="/tmp/grpc_ratls/glibc"
fi

# Build and install cJSON
function build_cjson() {
    pushd cJSON-${CJSON_VER}
    rm -rf build && mkdir build && cd build
    cmake -DENABLE_CJSON_UTILS=On -DENABLE_CJSON_TEST=Off -DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX} \
        -DCMAKE_C_COMPILER=${CC} ..
    make install
    popd
}

function build_grpc_ratls() {
    rm -rf ${GRPC_INSTALL_PATH} && mkdir -p ${GRPC_INSTALL_PATH}
    # Copy occlum dcap lib first to ease linking
    cp ${DCAP_LIB_PATH}/libocclum_dcap.so* ${INSTALL_PREFIX}/lib

    # Copy ratls added/updated files to grpc source
    cp -rf grpc/${GRPC_VERSION}/* ${GRPC_PATH}/

    ABSEIL_PATH=${GRPC_PATH}/third_party/abseil-cpp

    # build and install abseil library
    # https://abseil.io/docs/cpp/quickstart-cmake.html
    pushd ${ABSEIL_PATH}
    rm -rf build && mkdir build && cd build
    cmake -DCMAKE_CXX_STANDARD=11 -DCMAKE_POSITION_INDEPENDENT_CODE=TRUE \
            -DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DCMAKE_INSTALL_PREFIX=${GRPC_INSTALL_PATH} \
            -DCMAKE_CXX_COMPILER=${CXX} -DCMAKE_C_COMPILER=${CC} ..
    make -j `nproc`
    make install
    popd

    # Build grpc + ratls
    pushd ${GRPC_PATH}
    rm -rf build && mkdir build && cd build
    cmake -DgRPC_INSTALL=ON -DgRPC_ABSL_PROVIDER=package -DgRPC_BUILD_TESTS=OFF \
        -DgRPC_BUILD_CSHARP_EXT=OFF -DgRPC_BUILD_GRPC_CSHARP_PLUGIN=OFF \
        -DgRPC_BUILD_GRPC_PHP_PLUGIN=OFF -DgRPC_BUILD_GRPC_RUBY_PLUGIN=OFF \
        -DDEFINE_SGX_RA_TLS_OCCLUM_BACKEND=ON \
        -DCMAKE_CXX_COMPILER=${CXX} -DCMAKE_C_COMPILER=${CC} \
        -DCMAKE_PREFIX_PATH=${GRPC_INSTALL_PATH} \
        -DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DCMAKE_INSTALL_PREFIX=${GRPC_INSTALL_PATH} ..
    make -j `nproc`
    make install
    popd

    # Build grpc ratls client and server demo
    pushd ${GRPC_PATH}/examples/cpp/ratls
    rm -rf build && mkdir -p build
    cd build
    cmake -DCMAKE_PREFIX_PATH=${GRPC_INSTALL_PATH} \
        -DCMAKE_BUILD_TYPE=${BUILD_TYPE} \
        -DCMAKE_CXX_COMPILER=${CXX} -DCMAKE_C_COMPILER=${CC} ..
    make -j `nproc`
    popd

    # Clean temp occlum dcap lib
    rm ${INSTALL_PREFIX}/lib/libocclum_dcap.so*
}

build_cjson
build_grpc_ratls