#!/bin/sh

# This script copies source file lists from src/Makefile.am to cmake files.

proc get_variable_value {
  local FILENAME=$1
  local VARNAME=$2
  awk "
    BEGIN { start = 0; }
    /^$VARNAME =/ { start = 1; }
    { if (start) { print \$0; } }
    /\\\\\$/ { next; }
    { start = 0; }
  " $FILENAME \
    | sed "s/^$VARNAME =//" \
    | sed "s/[ \\]//g" \
    | grep -v "^\\$" \
    | grep -v "^$'" \
    | LC_ALL=C' sort | uniq
}

proc get_header_files {
  get_variable_value $[join(ARGV)] | grep '\.h$'
}

proc get_source_files {
  get_variable_value $[join(ARGV)] | grep "cc$"
}

proc get_proto_files_blacklisted {
  get_proto_files $[join(ARGV)] | sed '/^google\/protobuf\/unittest_enormous_descriptor.proto$/d'
}

proc get_proto_files {
  get_variable_value $[join(ARGV)] | grep "pb.cc$" | sed "s/pb.cc/proto/"
}

proc sort_files {
  for FILE in $@ {
    echo $FILE
  }' | LC_ALL=C' sort | uniq
}

setvar MAKEFILE = "src/Makefile.am"

test -f $MAKEFILE || do {
  echo "Cannot find: $MAKEFILE"
  exit 1
}

# Extract file lists from src/Makefile.am
setvar GZHEADERS = $(get_variable_value $MAKEFILE GZHEADERS)
setvar HEADERS = $(get_variable_value $MAKEFILE nobase_include_HEADERS)
setvar PUBLIC_HEADERS = $(sort_files $GZHEADERS $HEADERS)
setvar LIBPROTOBUF_LITE_SOURCES = $(get_source_files $MAKEFILE libprotobuf_lite_la_SOURCES)
setvar LIBPROTOBUF_SOURCES = $(get_source_files $MAKEFILE libprotobuf_la_SOURCES)
setvar LIBPROTOC_SOURCES = $(get_source_files $MAKEFILE libprotoc_la_SOURCES)
setvar LITE_PROTOS = $(get_proto_files $MAKEFILE protoc_lite_outputs)
setvar PROTOS = $(get_proto_files $MAKEFILE protoc_outputs)
setvar PROTOS_BLACKLISTED = $(get_proto_files_blacklisted $MAKEFILE protoc_outputs)
setvar WKT_PROTOS = $(get_variable_value $MAKEFILE nobase_dist_proto_DATA)
setvar COMMON_TEST_SOURCES = $(get_source_files $MAKEFILE COMMON_TEST_SOURCES)
setvar COMMON_LITE_TEST_SOURCES = $(get_source_files $MAKEFILE COMMON_LITE_TEST_SOURCES)
setvar TEST_SOURCES = $(get_source_files $MAKEFILE protobuf_test_SOURCES)
setvar LITE_TEST_SOURCES = $(get_source_files $MAKEFILE protobuf_lite_test_SOURCES)
setvar LITE_ARENA_TEST_SOURCES = $(get_source_files $MAKEFILE protobuf_lite_arena_test_SOURCES)
setvar TEST_PLUGIN_SOURCES = $(get_source_files $MAKEFILE test_plugin_SOURCES)

################################################################################
# Update cmake files.
################################################################################

setvar CMAKE_DIR = 'cmake'
setvar EXTRACT_INCLUDES_BAT = "cmake/extract_includes.bat.in"
test -d $CMAKE_DIR || do {
  echo "Cannot find: $CMAKE_DIR"
  exit 1
}

test -f $EXTRACT_INCLUDES_BAT || do {
  echo "Cannot find: $EXTRACT_INCLUDES_BAT"
  exit 1
}

proc set_cmake_value {
  local FILENAME=$1
  local VARNAME=$2
  local PREFIX=$3
  shift
  shift
  shift
  awk -v values="$[join(ARGV)]" -v prefix="$PREFIX" "
    BEGIN { start = 0; }
    /^set\\($VARNAME/ {
      start = 1;
      print \$0;
      len = split(values, vlist, \" \");
      for (i = 1; i <= len; ++i) {
        printf(\"  %s%s\\n\", prefix, vlist[i]);
      }
      next;
    }
    start && /^\\)/ {
      start = 0;
    }
    !start {
      print \$0;
    }
  " $FILENAME > /tmp/$$
  cp /tmp/$$ $FILENAME
}


# Replace file lists in cmake files.
setvar CMAKE_PREFIX = ""\${protobuf_source_dir}/src/""
set_cmake_value $CMAKE_DIR/libprotobuf-lite.cmake libprotobuf_lite_files $CMAKE_PREFIX $LIBPROTOBUF_LITE_SOURCES
set_cmake_value $CMAKE_DIR/libprotobuf.cmake libprotobuf_files $CMAKE_PREFIX $LIBPROTOBUF_SOURCES
set_cmake_value $CMAKE_DIR/libprotoc.cmake libprotoc_files $CMAKE_PREFIX $LIBPROTOC_SOURCES
set_cmake_value $CMAKE_DIR/tests.cmake lite_test_protos "" $LITE_PROTOS
set_cmake_value $CMAKE_DIR/tests.cmake tests_protos "" $PROTOS_BLACKLISTED
set_cmake_value $CMAKE_DIR/tests.cmake common_test_files $CMAKE_PREFIX $COMMON_TEST_SOURCES
set_cmake_value $CMAKE_DIR/tests.cmake common_lite_test_files $CMAKE_PREFIX $COMMON_LITE_TEST_SOURCES
set_cmake_value $CMAKE_DIR/tests.cmake tests_files $CMAKE_PREFIX $TEST_SOURCES
set_cmake_value $CMAKE_DIR/tests.cmake lite_test_files $CMAKE_PREFIX $LITE_TEST_SOURCES
set_cmake_value $CMAKE_DIR/tests.cmake lite_arena_test_files $CMAKE_PREFIX $LITE_ARENA_TEST_SOURCES

# Generate extract_includes.bat
echo "mkdir include" > $EXTRACT_INCLUDES_BAT
for HEADER in $PUBLIC_HEADERS {
  setvar HEADER_DIR = $(dirname $HEADER)
  while test ! $HEADER_DIR = "." {
    echo $HEADER_DIR | sed "s/\\//\\\\/g"
    setvar HEADER_DIR = $(dirname $HEADER_DIR)
  }
} | sort | uniq | sed "s/^/mkdir include\\\\/" >> $EXTRACT_INCLUDES_BAT
for HEADER in $PUBLIC_HEADERS {
  setvar WINPATH = $(echo $HEADER | sed 's;/;\\;g')
  echo "copy \${PROTOBUF_SOURCE_WIN32_PATH}\\..\\src\\$WINPATH include\\$WINPATH" >> $EXTRACT_INCLUDES_BAT
}

################################################################################
# Update bazel BUILD files.
################################################################################

proc set_bazel_value {
  local FILENAME=$1
  local VARNAME=$2
  local PREFIX=$3
  shift
  shift
  shift
  awk -v values="$[join(ARGV)]" -v prefix="$PREFIX" "
    BEGIN { start = 0; }
    /# AUTOGEN\\($VARNAME\\)/ {
      start = 1;
      print \$0;
      # replace \$0 with indent.
      sub(/#.*/, \"\", \$0)
      len = split(values, vlist, \" \");
      for (i = 1; i <= len; ++i) {
        printf(\"%s\\\"%s%s\\\",\n\", \$0, prefix, vlist[i]);
      }
      next;
    }
    start && /\]/ {
      start = 0
    }
    !start {
      print \$0;
    }
  " $FILENAME > /tmp/$$
  cp /tmp/$$ $FILENAME
}


setvar BAZEL_BUILD = "./BUILD"
setvar BAZEL_PREFIX = ""src/""
if test -f $BAZEL_BUILD {
  set_bazel_value $BAZEL_BUILD protobuf_lite_srcs $BAZEL_PREFIX $LIBPROTOBUF_LITE_SOURCES
  set_bazel_value $BAZEL_BUILD protobuf_srcs $BAZEL_PREFIX $LIBPROTOBUF_SOURCES
  set_bazel_value $BAZEL_BUILD protoc_lib_srcs $BAZEL_PREFIX $LIBPROTOC_SOURCES
  set_bazel_value $BAZEL_BUILD lite_test_protos "" $LITE_PROTOS
  set_bazel_value $BAZEL_BUILD well_known_protos "" $WKT_PROTOS
  set_bazel_value $BAZEL_BUILD test_protos "" $PROTOS
  set_bazel_value $BAZEL_BUILD common_test_srcs $BAZEL_PREFIX $COMMON_TEST_SOURCES
  set_bazel_value $BAZEL_BUILD test_srcs $BAZEL_PREFIX $TEST_SOURCES
  set_bazel_value $BAZEL_BUILD test_plugin_srcs $BAZEL_PREFIX $TEST_PLUGIN_SOURCES
} else {
  echo "Skipped BUILD file update."
}