# clang: https://clang.llvm.org/docs/CommandGuide/clang.html
# clang-cl: https://clang.llvm.org/docs/UsersManual.html#id11

# --- Macros ---

macro(setb variable)
  if(${variable})
    set(${variable} ON)
  else()
    set(${variable} OFF)
  endif()
endmacro()

set(targets WIN32 APPLE UNIX LINUX)

foreach(target ${targets})
  setb(${target})
endforeach()

# --- CPU target ---
if(CMAKE_SYSTEM_PROCESSOR MATCHES "arm|ARM|arm64|ARM64|aarch64|AARCH64")
  if(APPLE)
    register_compiler_flags(-mcpu=apple-m1)
  elseif(WIN32)
    # Windows ARM64: use /clang: prefix for clang-cl, skip for MSVC cl.exe subprojects
    # These flags are only understood by clang-cl, not MSVC cl.exe
    register_compiler_flags(/clang:-march=armv8-a+crc /clang:-mtune=ampere1)
  else()
    register_compiler_flags(-march=armv8-a+crc -mtune=ampere1)
  endif()
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64|X86_64|x64|X64|amd64|AMD64")
  if(ENABLE_BASELINE)
    register_compiler_flags(-march=nehalem)
  else()
    register_compiler_flags(-march=haswell)
  endif()
else()
  unsupported(CMAKE_SYSTEM_PROCESSOR)
endif()

# --- MSVC runtime ---
if(WIN32)
  register_compiler_flags(
    DESCRIPTION "Use static MSVC runtime"
    /MTd ${DEBUG}
    /MT ${RELEASE}
    /U_DLL
  )
endif()

if(ENABLE_ASAN)
  register_compiler_flags(
    DESCRIPTION "Enable AddressSanitizer"
    -fsanitize=address
  )
endif()

if(ENABLE_FUZZILLI)
  register_compiler_flags(
    DESCRIPTION "Enable coverage instrumentation for fuzzing"
    -fsanitize-coverage=trace-pc-guard
  )

  register_linker_flags(
    DESCRIPTION "Link coverage instrumentation"
    -fsanitize-coverage=trace-pc-guard
  )

  register_compiler_flags(
    DESCRIPTION "Enable fuzzilli-specific code"
    -DFUZZILLI_ENABLED
  )
endif()

# --- Optimization level ---
if(DEBUG)
  register_compiler_flags(
    DESCRIPTION "Disable optimization"
    /Od ${WIN32}
    -O0 ${UNIX}
  )
elseif(ENABLE_SMOL)
  register_compiler_flags(
    DESCRIPTION "Optimize for size"
    /Os ${WIN32}
    -Os ${UNIX}
  )
else()
  register_compiler_flags(
    DESCRIPTION "Optimize for speed"
    /O2 ${WIN32} # TODO: change to /0t (same as -O3) to match macOS and Linux?
    -O3 ${UNIX}
  )
endif()

# --- Debug level ---
if(WIN32)
  register_compiler_flags(
    DESCRIPTION "Enable debug symbols (.pdb)"
    /Z7
  )
elseif(APPLE)
  register_compiler_flags(
    DESCRIPTION "Enable debug symbols (.dSYM)"
    -gdwarf-4
  )
endif()

if(UNIX)
  # Nix LLVM doesn't support zstd compression, use zlib instead
  if(DEFINED ENV{NIX_CC})
    register_compiler_flags(
      DESCRIPTION "Enable debug symbols (zlib-compressed for Nix)"
      -g3 -gz=zlib ${DEBUG}
      -g1 ${RELEASE}
    )
  else()
    register_compiler_flags(
      DESCRIPTION "Enable debug symbols (zstd-compressed)"
      -g3 -gz=zstd ${DEBUG}
      -g1 ${RELEASE}
    )
  endif()

  register_compiler_flags(
    DESCRIPTION "Optimize debug symbols for LLDB"
    -glldb
  )
endif()

# TODO: consider other debug options
# -fdebug-macro # Emit debug info for macros
# -fstandalone-debug # Emit debug info for non-system libraries
# -fno-eliminate-unused-debug-types # Don't eliminate unused debug symbols

# --- C/C++ flags ---
register_compiler_flags(
  DESCRIPTION "Disable C/C++ exceptions"
  -fno-exceptions ${UNIX}
  /EHsc ${WIN32} # (s- disables C++, c- disables C)
)

register_compiler_flags(
  DESCRIPTION "Disable C++ static destructors"
  LANGUAGES CXX
  -Xclang ${WIN32}
  -fno-c++-static-destructors
)

register_compiler_flags(
  DESCRIPTION "Disable runtime type information (RTTI)"
  /GR- ${WIN32}
  -fno-rtti ${UNIX}
)

register_compiler_flags(
  DESCRIPTION "Keep frame pointers"
  /Oy- ${WIN32}
  -fno-omit-frame-pointer ${UNIX}
  -mno-omit-leaf-frame-pointer ${UNIX}
)

if(UNIX)
  register_compiler_flags(
    DESCRIPTION "Set C/C++ visibility to hidden"
    -fvisibility=hidden
    -fvisibility-inlines-hidden
  )

  register_compiler_flags(
    DESCRIPTION "Disable unwind tables"
    -fno-unwind-tables
    -fno-asynchronous-unwind-tables
  )

  # needed for libuv stubs because they use
  # C23 feature which lets you define parameter without
  # name
  register_compiler_flags(
    DESCRIPTION "Allow C23 extensions"
    -Wno-c23-extensions
  )
endif()

register_compiler_flags(
  DESCRIPTION "Place each function in its own section"
  -ffunction-sections ${UNIX}
  /Gy ${WIN32}
)

register_compiler_flags(
  DESCRIPTION "Place each data item in its own section"
  -fdata-sections ${UNIX}
  /Gw ${WIN32}
)

# having this enabled in debug mode on macOS >=14 causes libarchive to fail to configure with the error:
# > pid_t doesn't exist on this platform?
if((DEBUG AND LINUX) OR((NOT DEBUG) AND UNIX))
  register_compiler_flags(
    DESCRIPTION "Emit an address-significance table"
    -faddrsig
  )
endif()

if(WIN32)
  register_compiler_flags(
    DESCRIPTION "Enable string pooling"
    /GF
  )

  register_compiler_flags(
    DESCRIPTION "Assume thread-local variables are defined in the executable"
    /GA
  )
endif()

# --- Linker flags ---
if(LINUX)
  register_linker_flags(
    DESCRIPTION "Disable relocation read-only (RELRO)"
    -Wl,-z,norelro
  )
  register_compiler_flags(
    DESCRIPTION "Disable semantic interposition"
    -fno-semantic-interposition
  )
endif()

# --- Assertions ---

# Note: This is a helpful guide about assertions:
# https://best.openssf.org/Compiler-Hardening-Guides/Compiler-Options-Hardening-Guide-for-C-and-C++
if(ENABLE_ASSERTIONS)
  register_compiler_flags(
    DESCRIPTION "Do not eliminate null-pointer checks"
    -fno-delete-null-pointer-checks
  )
endif()

# --- Diagnostics ---
if(UNIX)
  register_compiler_flags(
    DESCRIPTION "Enable color diagnostics"
    -fdiagnostics-color=always
  )
endif()

if(WIN32)
  register_compiler_flags(
    DESCRIPTION "Set C/C++ error limit"
    /clang:-ferror-limit=${ERROR_LIMIT}
  )
else()
  register_compiler_flags(
    DESCRIPTION "Set C/C++ error limit"
    -ferror-limit=${ERROR_LIMIT}
  )
endif()

# --- LTO ---
if(ENABLE_LTO)
  register_compiler_flags(
    DESCRIPTION "Enable link-time optimization (LTO)"
    -flto=full ${UNIX}
    -flto ${WIN32}
  )

  if(UNIX)
    register_compiler_flags(
      DESCRIPTION "Enable virtual tables"
      LANGUAGES CXX
      -fforce-emit-vtables
      -fwhole-program-vtables
    )

    register_linker_flags(
      DESCRIPTION "Enable link-time optimization (LTO)"
      -flto=full
      -fwhole-program-vtables
      -fforce-emit-vtables
    )
  endif()
endif()

# --- Remapping ---
if(UNIX AND CI)
  register_compiler_flags(
    DESCRIPTION "Remap source files"
    -ffile-prefix-map=${CWD}=.
    -ffile-prefix-map=${VENDOR_PATH}=vendor
    -ffile-prefix-map=${CACHE_PATH}=cache
  )
endif()

# --- Other ---

# Workaround for CMake and clang-cl bug.
# https://github.com/ninja-build/ninja/issues/2280
if(WIN32 AND NOT CMAKE_CL_SHOWINCLUDES_PREFIX)
  set(CMAKE_CL_SHOWINCLUDES_PREFIX "Note: including file:")
endif()

# WebKit uses -std=gnu++20 on non-macOS non-Windows.
# If we do not set this, it will crash at startup on the first memory allocation.
if(NOT WIN32 AND NOT APPLE)
  set(CMAKE_CXX_EXTENSIONS ON)
  set(CMAKE_POSITION_INDEPENDENT_CODE OFF)
endif()