#
# 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.
#
cmake_minimum_required(VERSION 3.16)

project(thrift_compiler_tests)

set(THRIFT_COMPILER_SOURCE_DIR
    ${CMAKE_CURRENT_SOURCE_DIR}/..
)

# don't generate ZERO_CHECK
set(CMAKE_SUPPRESS_REGENERATION true)

# version.h now handled via veralign.sh
#configure_file(${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/version.h.in ${CMAKE_CURRENT_BINARY_DIR}/thrift/version.h)
if(MSVC)
    # The winflexbison generator outputs some macros that conflict with the Visual Studio 2010 copy of stdint.h
    # This might be fixed in later versions of Visual Studio, but an easy solution is to include stdint.h first
    if(HAVE_STDINT_H)
        add_definitions(-D__STDC_LIMIT_MACROS)
        add_definitions(/FI"stdint.h")
    endif(HAVE_STDINT_H)
endif()

find_package(FLEX REQUIRED)
find_package(BISON REQUIRED)

if(NOT TARGET parse)
    # create directory for thrifty and thriftl
    file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/thrift/)

    # Create flex and bison files and build the lib parse static library
    BISON_TARGET(thrifty ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/thrifty.yy ${CMAKE_CURRENT_BINARY_DIR}/thrift/thrifty.cc COMPILE_FLAGS "--file-prefix-map=${CMAKE_BINARY_DIR}='' --file-prefix-map=${CMAKE_SOURCE_DIR}=''")
    FLEX_TARGET(thriftl ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/thriftl.ll ${CMAKE_CURRENT_BINARY_DIR}/thrift/thriftl.cc)
    ADD_FLEX_BISON_DEPENDENCY(thriftl thrifty)

    set(parse_SOURCES
        ${CMAKE_CURRENT_BINARY_DIR}/thrift/thrifty.cc
        ${CMAKE_CURRENT_BINARY_DIR}/thrift/thriftl.cc
        ${CMAKE_CURRENT_BINARY_DIR}/thrift/thrifty.hh
    )

    add_library(parse STATIC ${parse_SOURCES})
endif()

# Thrift compiler tests
set(thrift_compiler_tests
)

# you can add some files manually there 
set(thrift_compiler_tests_manual_SOURCES
    # tests file to avoid main in every test file
    ${CMAKE_CURRENT_SOURCE_DIR}/tests_main.cc
)

# set variable for tests sources - will be filled later
set(thrift_compiler_tests_SOURCES
)

set(thrift_compiler_SOURCES
    ${CMAKE_CURRENT_SOURCE_DIR}/thrift_test_globals.cc
    ${CMAKE_CURRENT_SOURCE_DIR}/thrift_test_parser_support.cc
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/logging.cc # we use logging instead of main to avoid breaking compillation (2 main v)
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/audit/t_audit.cpp
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/common.cc
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/t_generator.cc
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/validator_parser.cc
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/validator_parser.h
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/parse/t_typedef.cc
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/parse/parse.cc
    ${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/version.h
)

# This macro adds an option THRIFT_COMPILER_${NAME}
# that allows enabling or disabling certain languages
macro(THRIFT_ADD_COMPILER name description initial)
    string(TOUPPER "THRIFT_COMPILER_${name}" enabler)
    set(src "${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/t_${name}_generator.cc")
    option(${enabler} ${description} ${initial})
    if(${enabler})
        list(APPEND thrift_compiler_SOURCES ${src})
        file(GLOB temp_test_sources
            "${CMAKE_CURRENT_SOURCE_DIR}/${name}/*.c*"
            "${CMAKE_CURRENT_SOURCE_DIR}/${name}/*.thrift"
        )
        list(APPEND thrift_compiler_tests_SOURCES ${temp_test_sources})
    endif()
endmacro()

# This macro adds an option THRIFT_VALIDATOR_COMPILER_${NAME}
# that allows enabling or disabling certain languages
macro(THRIFT_ADD_VALIDATOR_COMPILER name description initial)
    string(TOUPPER "THRIFT_VALIDATOR_COMPILER_${name}" enabler)
    set(src "${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/${name}_validator_generator.cc")
    list(APPEND "${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/${name}_validator_generator.h")
    option(${enabler} ${description} ${initial})
    if(${enabler})
        list(APPEND thrift_compiler_SOURCES ${src})
    endif()
endmacro()

# The following compiler with unit tests can be enabled or disabled
THRIFT_ADD_COMPILER(c_glib  "Enable compiler for C with Glib" OFF)
THRIFT_ADD_COMPILER(cl      "Enable compiler for Common LISP" OFF)
THRIFT_ADD_COMPILER(cpp     "Enable compiler for C++" ON)
THRIFT_ADD_COMPILER(d       "Enable compiler for D" OFF)
THRIFT_ADD_COMPILER(dart    "Enable compiler for Dart" OFF)
THRIFT_ADD_COMPILER(delphi  "Enable compiler for Delphi" OFF)
THRIFT_ADD_COMPILER(erl     "Enable compiler for Erlang" OFF)
THRIFT_ADD_COMPILER(go      "Enable compiler for Go" OFF)
THRIFT_ADD_COMPILER(gv      "Enable compiler for GraphViz" OFF)
THRIFT_ADD_COMPILER(haxe    "Enable compiler for Haxe" OFF)
THRIFT_ADD_COMPILER(html    "Enable compiler for HTML Documentation" OFF)
THRIFT_ADD_COMPILER(java    "Enable compiler for Java"   OFF)
THRIFT_ADD_COMPILER(javame  "Enable compiler for Java ME" OFF)
THRIFT_ADD_COMPILER(js      "Enable compiler for JavaScript" OFF)
THRIFT_ADD_COMPILER(json    "Enable compiler for JSON" OFF)
THRIFT_ADD_COMPILER(lua     "Enable compiler for Lua" OFF)
THRIFT_ADD_COMPILER(netstd  "Enable compiler for .NET Standard" ON)
THRIFT_ADD_COMPILER(ocaml   "Enable compiler for OCaml" ON)
THRIFT_ADD_COMPILER(perl    "Enable compiler for Perl" OFF)
THRIFT_ADD_COMPILER(php     "Enable compiler for PHP" OFF)
THRIFT_ADD_COMPILER(py      "Enable compiler for Python 2.0" OFF)
THRIFT_ADD_COMPILER(rb      "Enable compiler for Ruby" OFF)
THRIFT_ADD_COMPILER(rs      "Enable compiler for Rust" OFF)
THRIFT_ADD_COMPILER(st      "Enable compiler for Smalltalk" OFF)
THRIFT_ADD_COMPILER(swift   "Enable compiler for Swift" OFF)
THRIFT_ADD_COMPILER(xml     "Enable compiler for XML" OFF)
THRIFT_ADD_COMPILER(xsd     "Enable compiler for XSD" OFF)

# The following compiler can be enabled or disabled by enabling or disabling certain languages
THRIFT_ADD_VALIDATOR_COMPILER(go           "Enable validator compiler for Go" ON)

# OCaml tests include the implementation .cc directly, so compiling it into the
# thrift_compiler lib would cause duplicate definitions (LNK2005).
list(REMOVE_ITEM thrift_compiler_SOURCES
  "${THRIFT_COMPILER_SOURCE_DIR}/src/thrift/generate/t_ocaml_generator.cc"
)

# Thrift is looking for include files in the src directory
# We also add:
# - the current binary directory for locally generated files (standalone tests build)
# - the top-level binary directory for generated files when reusing targets from the parent build
include_directories(${CMAKE_CURRENT_BINARY_DIR} ${CMAKE_BINARY_DIR} ${THRIFT_COMPILER_SOURCE_DIR}/src ${CMAKE_CURRENT_SOURCE_DIR}/catch)

add_library(thrift_compiler STATIC ${thrift_compiler_SOURCES})

#link parse lib to thrift_compiler lib
target_link_libraries(thrift_compiler parse)

# add tests executable
add_executable(thrift_compiler_tests ${thrift_compiler_tests_manual_SOURCES} ${thrift_compiler_tests_SOURCES})

# if generates for Visual Studio set thrift_compiler_tests as default project
if(MSVC)
    set_property(TARGET thrift_compiler_tests PROPERTY VS_STARTUP_PROJECT thrift_compiler_tests)
endif()

set_target_properties(thrift_compiler_tests PROPERTIES RUNTIME_OUTPUT_DIRECTORY bin/)
set_target_properties(thrift_compiler_tests PROPERTIES OUTPUT_NAME thrift_compiler_tests)

# Ensure generator registration translation units are linked in.
# Many generators register themselves via static initialization; linkers may
# otherwise discard those objects from static libraries.
if(MSVC)
    target_link_libraries(thrift_compiler_tests PRIVATE thrift_compiler)
    target_link_options(thrift_compiler_tests PRIVATE
        "/WHOLEARCHIVE:$<TARGET_LINKER_FILE:thrift_compiler>"
    )
elseif(APPLE)
    target_link_libraries(thrift_compiler_tests PRIVATE thrift_compiler)
    target_link_options(thrift_compiler_tests PRIVATE
        "-Wl,-force_load,$<TARGET_LINKER_FILE:thrift_compiler>"
    )
else()
    target_link_libraries(thrift_compiler_tests PRIVATE
        "-Wl,--whole-archive" thrift_compiler "-Wl,--no-whole-archive"
    )
endif()

# Compile-check generated C++ output for fixture thrift files.
# This ensures the generator output is compileable (no link step required).
# Note: These checks require Boost headers and are optional.
if(TARGET thrift-compiler)
    # Try to find Boost for the compile checks
    find_package(Boost QUIET)
    set(_compile_check_boost_include_dirs "")
    if(Boost_FOUND)
        set(_compile_check_boost_include_dirs ${Boost_INCLUDE_DIRS})
    elseif(DEFINED BOOST_ROOT)
        if(EXISTS "${BOOST_ROOT}/include/boost")
            set(_compile_check_boost_include_dirs "${BOOST_ROOT}/include")
        elseif(EXISTS "${BOOST_ROOT}/boost")
            set(_compile_check_boost_include_dirs "${BOOST_ROOT}")
        endif()
    endif()

    # Only create compile-check targets if Boost is available
    if(NOT _compile_check_boost_include_dirs STREQUAL "")
        set(_private_optional_thrift
            "${CMAKE_CURRENT_SOURCE_DIR}/cpp/test_private_optional.thrift"
        )
        set(_private_optional_gen_out_dir
            "${CMAKE_CURRENT_BINARY_DIR}/generated-private-optional"
        )
        set(_private_optional_gen_cpp_dir
            "${_private_optional_gen_out_dir}/gen-cpp"
        )
        set(_private_optional_types_cpp
            "${_private_optional_gen_cpp_dir}/test_private_optional_types.cpp"
        )

        add_custom_command(
            OUTPUT "${_private_optional_types_cpp}"
            COMMAND ${CMAKE_COMMAND} -E make_directory "${_private_optional_gen_out_dir}"
            COMMAND ${CMAKE_COMMAND} -E chdir "${_private_optional_gen_out_dir}"
                $<TARGET_FILE:thrift-compiler>
                --gen cpp:private_optional
                -o "${_private_optional_gen_out_dir}"
                "${_private_optional_thrift}"
            DEPENDS thrift-compiler "${_private_optional_thrift}"
            VERBATIM
        )

        set_source_files_properties(
            "${_private_optional_types_cpp}"
            PROPERTIES GENERATED TRUE
        )

        add_library(thrift_compiler_generated_private_optional STATIC
            "${_private_optional_types_cpp}"
        )

        target_include_directories(thrift_compiler_generated_private_optional PRIVATE
            "${_private_optional_gen_cpp_dir}"
            "${THRIFT_COMPILER_SOURCE_DIR}/../../lib/cpp/src"
            "${CMAKE_CURRENT_BINARY_DIR}"
            "${CMAKE_BINARY_DIR}"
            ${_compile_check_boost_include_dirs}
        )

        set(_enum_class_thrift
            "${CMAKE_CURRENT_SOURCE_DIR}/cpp/test_enum_class.thrift"
        )
        set(_enum_class_gen_out_dir
            "${CMAKE_CURRENT_BINARY_DIR}/generated-enum-class"
        )
        set(_enum_class_gen_cpp_dir
            "${_enum_class_gen_out_dir}/gen-cpp"
        )
        set(_enum_class_types_cpp
            "${_enum_class_gen_cpp_dir}/test_enum_class_types.cpp"
        )

        add_custom_command(
            OUTPUT "${_enum_class_types_cpp}"
            COMMAND ${CMAKE_COMMAND} -E make_directory "${_enum_class_gen_out_dir}"
            COMMAND ${CMAKE_COMMAND} -E chdir "${_enum_class_gen_out_dir}"
                $<TARGET_FILE:thrift-compiler>
                --gen cpp:pure_enums=enum_class
                -o "${_enum_class_gen_out_dir}"
                "${_enum_class_thrift}"
            DEPENDS thrift-compiler "${_enum_class_thrift}"
            VERBATIM
        )

        set_source_files_properties(
            "${_enum_class_types_cpp}"
            PROPERTIES GENERATED TRUE
        )

        add_library(thrift_compiler_generated_enum_class STATIC
            "${_enum_class_types_cpp}"
        )

        target_include_directories(thrift_compiler_generated_enum_class PRIVATE
            "${_enum_class_gen_cpp_dir}"
            "${THRIFT_COMPILER_SOURCE_DIR}/../../lib/cpp/src"
            "${CMAKE_CURRENT_BINARY_DIR}"
            "${CMAKE_BINARY_DIR}"
            ${_compile_check_boost_include_dirs}
        )

        # Build the compile-check as part of the standard test build.
        add_dependencies(thrift_compiler_tests thrift_compiler_generated_private_optional)
        add_dependencies(thrift_compiler_tests thrift_compiler_generated_enum_class)
    else()
        message(STATUS "Skipping generated code compile-checks (Boost headers not found)")
    endif()
else()
    message(STATUS "Skipping generated code compile-checks (no thrift-compiler target)")
endif()

enable_testing()
add_test(NAME ThriftCompilerTests COMMAND thrift_compiler_tests)
