#!/bin/bash

# Copyright 2020 The Kubernetes Authors.
#
# 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.

set -eo pipefail

function cleanup {
  echo 'pkill -f nfsplugin'
  pkill -f nfsplugin
  echo 'Uninstalling NFS server on localhost'
  docker rm nfs -f
}
trap cleanup EXIT

function install_csc_bin {
  echo 'Installing CSC binary...'
  export set GOPATH=$HOME/go
  export set PATH=$PATH:$GOPATH/bin
  go get github.com/rexray/gocsi/csc
  pushd $GOPATH/src/github.com/rexray/gocsi/csc
  make build
  popd
}

function provision_nfs_server {
  echo 'Installing NFS server on localhost'
  apt-get update -y
  apt-get install -y nfs-common
  docker run -d --name nfs --privileged -p 2049:2049 -v $(pwd)/nfsshare:/nfsshare -e SHARED_DIRECTORY=/nfsshare itsthenetwork/nfs-server-alpine:latest
}

provision_nfs_server
install_csc_bin

readonly volname="citest-$(date +%s)"
readonly volsize="2147483648"
readonly endpoint="tcp://127.0.0.1:10000"
readonly target_path="/tmp/targetpath"
readonly params="server=127.0.0.1,share=/"

nodeid='CSINode'
if [[ "$#" -gt 0 ]] && [[ -n "$1" ]]; then
  nodeid="$1"
fi

# Run CSI driver as a background service
bin/nfsplugin --endpoint "$endpoint" --nodeid "$nodeid" -v=5 &
sleep 5

echo 'Begin to run integration test...'

# Begin to run CSI functions one by one
echo "Create volume test:"
value="$(csc controller new --endpoint "$endpoint" --cap 1,block "$volname" --req-bytes "$volsize" --params "$params")"
sleep 15

volumeid="$(echo "$value" | awk '{print $1}' | sed 's/"//g')"
echo "Got volume id: $volumeid"

csc controller validate-volume-capabilities --endpoint "$endpoint" --cap 1,block "$volumeid"

echo "stage volume test:"
csc node stage --endpoint "$endpoint" --cap 1,block --staging-target-path "$staging_target_path" "$volumeid"

echo "publish volume test:"
csc node publish --endpoint "$endpoint" --cap 1,block --vol-context "$params" --target-path "$target_path" "$volumeid"
sleep 2

echo "unpublish volume test:"
csc node unpublish --endpoint "$endpoint" --target-path "$target_path" "$volumeid"
sleep 2

echo "unstage volume test:"
csc node unstage --endpoint "$endpoint" --staging-target-path "$staging_target_path" "$volumeid"
sleep 2

echo "Delete volume test:"
csc controller del --endpoint "$endpoint" "$volumeid"
sleep 15

csc identity plugin-info --endpoint "$endpoint"
csc node get-info --endpoint "$endpoint"

echo "Integration test is completed."