#!/bin/bash
# Copyright (c) 2021 Huawei Device Co., Ltd.
# 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 -e
protoc='protoc'
proto_dir='.'
protoreader_plugin='protoreader_plugin'
case "$OSTYPE" in
  msys*)    out='../../out/windows' protoreader_plugin='protoreader_plugin.exe' protoc='protoc.exe' proto_dir=$( cd "$(dirname "${BASH_SOURCE[0]}")" && pwd) ;;
  darwin*)  out='../../out/macx' proto_dir=$( cd "$(dirname "${BASH_SOURCE[0]}")" && pwd) ;;
  linux*)   out='../../out/linux'  proto_dir=$( cd "$(dirname "${BASH_SOURCE[0]}")" && pwd) ;;
  *)        echo "other: $OSTYPE" ;;
esac
echo $proto_dir
SOURCE="${BASH_SOURCE[0]}"
cd $(dirname ${SOURCE})
if [ ! -f "$out/$protoc" ];then
    echo "no $out/$protoc found, you need to run \"./build.sh protoc\" at root folder, and copy protoc.exe to $out/$protoc"
    exit
fi
echo "begin to generate proto based files"
SOURCE=$(dirname ${SOURCE})
services_dir="$proto_dir/services"
# kernel_version="5.10.79_aarch64"
kernel_version="."
ftrace_data_dir="$proto_dir/types/plugins/ftrace_data/$kernel_version"
memory_data_dir="$proto_dir/types/plugins/memory_data"
hilog_data_dir="$proto_dir/types/plugins/hilog_data"
native_hook_dir="$proto_dir/types/plugins/native_hook"
hidump_data_dir="$proto_dir/types/plugins/hidump_data"
network_data_dir="$proto_dir/types/plugins/network_data"
cpu_data_dir="$proto_dir/types/plugins/cpu_data"
diskio_data_dir="$proto_dir/types/plugins/diskio_data"
process_data_dir="$proto_dir/types/plugins/process_data"
hisysevent_data_dir="$proto_dir/types/plugins/hisysevent_data"
test_data_dir="$proto_dir/types/plugins/test_data"
js_memory_dir="$proto_dir/types/plugins/js_memory"
sph_data_dir="$proto_dir/smartperf_host"
proto_array=("${services_dir}/common_types.proto"
    "$ftrace_data_dir/trace_plugin_result.proto"
    "$ftrace_data_dir/ftrace_event.proto"
    "$ftrace_data_dir/irq.proto"
    "$ftrace_data_dir/vmscan.proto"
    "$ftrace_data_dir/workqueue.proto"
    "$ftrace_data_dir/task.proto"
    "$ftrace_data_dir/power.proto"
    "$ftrace_data_dir/sched.proto"
    "$ftrace_data_dir/filemap.proto"
    "$ftrace_data_dir/i2c.proto"
    "$ftrace_data_dir/kmem.proto"
    "$ftrace_data_dir/block.proto"
    "$ftrace_data_dir/ipi.proto"
    "$ftrace_data_dir/ftrace.proto"
    "$ftrace_data_dir/ext4.proto"
    "$ftrace_data_dir/oom.proto"
    "$ftrace_data_dir/compaction.proto"
    "$ftrace_data_dir/clk.proto"
    "$ftrace_data_dir/cgroup.proto"
    "$ftrace_data_dir/binder.proto"
    "$ftrace_data_dir/signal.proto"
    "$ftrace_data_dir/sunrpc.proto"
    "$ftrace_data_dir/net.proto"
    "$ftrace_data_dir/cpuhp.proto"
    "$ftrace_data_dir/writeback.proto"
    "$ftrace_data_dir/v4l2.proto"
    "$ftrace_data_dir/pagemap.proto"
    "$ftrace_data_dir/dma_fence.proto"
    "$ftrace_data_dir/printk.proto"
    "$ftrace_data_dir/filelock.proto"
    "$ftrace_data_dir/gpio.proto"
    "$ftrace_data_dir/timer.proto"
    "$ftrace_data_dir/raw_syscalls.proto"
    "$ftrace_data_dir/rcu.proto"
    "$memory_data_dir/memory_plugin_common.proto"
    "$memory_data_dir/memory_plugin_config.proto"
    "$memory_data_dir/memory_plugin_result.proto"
    "$hilog_data_dir/hilog_plugin_result.proto"
    "$native_hook_dir/native_hook_result.proto"
    "$native_hook_dir/native_hook_config.proto"
    "$hidump_data_dir/hidump_plugin_result.proto"
    "$network_data_dir/network_plugin_result.proto"
    "$cpu_data_dir/cpu_plugin_result.proto"
    "$diskio_data_dir/diskio_plugin_result.proto"
    "$hisysevent_data_dir/hisysevent_plugin_config.proto"
    "$process_data_dir/process_plugin_result.proto"
    "$hisysevent_data_dir/hisysevent_plugin_result.proto"
    "$js_memory_dir/js_heap_result.proto"
    "$js_memory_dir/js_heap_config.proto"
    "$test_data_dir/test.proto"
    "$sph_data_dir/sph_data.proto")

export LD_LIBRARY_PATH=$out
for ((i = 0; i < ${#proto_array[@]}; i ++))
do
   newpath=$(dirname ${proto_array[$i]})
   tailpath='\'${newpath#$proto_dir}
   newpath=${tailpath:2}
   cppout=../../third_party/protogen/$newpath
   mkdir -p $cppout
   $out/$protoc --proto_path=$sph_data_dir:$memory_data_dir:$native_hook_dir:$hidump_data_dir:$hilog_data_dir:$ftrace_data_dir:$js_memory_dir:$services_dir:$network_data_dir:$cpu_data_dir:$diskio_data_dir:$process_data_dir:$hisysevent_data_dir:$test_data_dir --cpp_out=$cppout ${proto_array[$i]}
   echo "----$out/$protoc --proto_path=$sph_data_dir:$memory_data_dir:$native_hook_dir:$hidump_data_dir:$hilog_data_dir:$ftrace_data_dir:$js_memory_dir:$services_dir:$network_data_dir:$cpu_data_dir:$diskio_data_dir:$process_data_dir:$hisysevent_data_dir:$test_data_dir --plugin=protoc-gen-plugin=$out/$protoreader_plugin --plugin_out=wrapper_namespace=ProtoReader:$cppout ${proto_array[$i]}---"
   $out/$protoc --proto_path=$sph_data_dir:$memory_data_dir:$native_hook_dir:$hidump_data_dir:$hilog_data_dir:$ftrace_data_dir:$js_memory_dir:$services_dir:$network_data_dir:$cpu_data_dir:$diskio_data_dir:$process_data_dir:$hisysevent_data_dir:$test_data_dir --plugin=protoc-gen-plugin=$out/$protoreader_plugin --plugin_out=wrapper_namespace=ProtoReader:$cppout ${proto_array[$i]}
done
echo "generate proto based files over"