#!/usr/bin/env bash

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF 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

BASEDIR=$(dirname "$0")/..
TEMPDIR="$BASEDIR"/temp

. "$BASEDIR"/dependencies.sh

if [[ ! -d "$TEMPDIR" ]]; then
  mkdir -p "$TEMPDIR"
else
  rm -rf "${TEMPDIR:?}"/*
fi

curl -sLo "$TEMPDIR"/collect-protocol.tgz https://github.com/apache/skywalking-data-collect-protocol/archive/"${COLLECT_PROTOCOL_SHA}".tar.gz

if [[ ! -d "$TEMPDIR"/collect-protocol ]]; then
  mkdir "$TEMPDIR"/collect-protocol
else
  rm -rf "$TEMPDIR"/collect-protocol/*
fi

tar -zxf "$TEMPDIR"/collect-protocol.tgz -C "$TEMPDIR"/collect-protocol --strip 1

find "$TEMPDIR"/collect-protocol -name "*Compat.proto" -exec rm {} \;

rm -rf "$BASEDIR"/{agent,common,event,language,logging,management,servicemesh}

go get -u google.golang.org/protobuf/cmd/protoc-gen-go@v1.26.0
go get -u google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.1.0

"$BASEDIR"/scripts/protoc.sh \
  --proto_path="$TEMPDIR"/collect-protocol \
  --go_out="$BASEDIR" \
  --go-grpc_out="$BASEDIR" \
  "$TEMPDIR"/collect-protocol/*/*.proto

cp -R "$BASEDIR"/skywalking/network/* "$BASEDIR"/ && rm -rf "$BASEDIR"/skywalking

go mod tidy