cmake_minimum_required(VERSION 3.13)
project(jcc C)
set(CMAKE_C_STANDARD 11)

set(CMAKE_COMPILE_WARNING_AS_ERROR ON)
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

include(CheckCCompilerFlag)

# set(
#   CMAKE_C_CLANG_TIDY
#   clang-tidy
#   --use-color
#   --checks=-*,cppcoreguidelines-avoid-non-const-global-variables
# )

set(
  SOURCE_FILES
  src/driver.c
  src/args.c

  src/thrd.c

  src/util.c
  src/vector.c
  src/hashtbl.c
  src/deque.c
  src/hash.c
  src/bitset.c
  src/fs.c
  src/json.c
  src/syscmd.c

  src/profile.c

  src/alloc.c
  src/log.c
  src/graphwriter.c

  src/ap_val.c

  src/lsp/lsp.c
  src/lsp/lsp_types.c
  src/lsp/map.c

  src/compiler.c
  src/diagnostics.c
  src/builtins.c
  src/lex.c
  src/io.c
  src/parse.c
  src/preproc.c
  src/program.c
  src/lower.c
  src/liveness.c
  src/disasm.c
  src/codegen.c
  src/var_table.c
  src/typechk.c
  src/lsra.c
  src/graphcol.c
  src/
  src/ir/ir.c
  src/ir/build.c
  src/ir/var_refs.c
  src/ir/eliminate_phi.c
  src/ir/prettyprint.c
  src/ir/validate.c
  src/ir/rw.c
  src/ir/interp.c

  src/opts/opts.c
  src/opts/cnst_fold.c
  src/opts/cnst_branches.c
  src/opts/instr_comb.c
  src/opts/promote.c
  src/opts/inline.c

  src/aarch64.c
  src/x64.c
  src/rv32i.c

  src/macos/mach-o.c
  src/macos/link.c

  src/linux/elf.c
  src/linux/link.c
)

# build for all platforms (support cross compilation)
# TODO: add a more granular system for including targets

if("${CMAKE_C_COMPILER_ID}" STREQUAL "")
    set(CMAKE_C_COMPILER_ID "JCC")
endif()

if(NOT DEFINED ARCHITECTURES OR "${ARCHITECTURES}" STREQUAL "")
  message(STATUS "Enabling all architectures")
  add_definitions(-DJCC_ALL)
  add_definitions(-DJCC_AARCH64)
  add_definitions(-DJCC_X64)
  add_definitions(-DJCC_RV32I)

  list(APPEND SOURCE_FILES
    src/aarch64/codegen.c
    src/aarch64/lower.c
    src/aarch64/emit.c
    src/aarch64/emitter.c

    src/x64/codegen.c
    src/x64/lower.c
    src/x64/emit.c
    src/x64/emitter.c

    src/rv32i/codegen.c
    src/rv32i/lower.c
    src/rv32i/emit.c
    src/rv32i/emitter.c
    src/rv32i/object.c
  )
endif()

if (DEFINED INTERP)
  message(STATUS "Enabling interpreter")
  add_definitions(-DJCC_INTERP)
endif()

foreach(arch IN LISTS ARCHITECTURES)
  if("${arch}" STREQUAL "aarch64")
    add_definitions(-DJCC_AARCH64)

    list(APPEND SOURCE_FILES
      src/aarch64/codegen.c
      src/aarch64/lower.c
      src/aarch64/emit.c
      src/aarch64/emitter.c
    )
  elseif("${arch}" STREQUAL "x64")
    add_definitions(-DJCC_X64)

    list(APPEND SOURCE_FILES
      src/x64/codegen.c
      src/x64/lower.c
      src/x64/emit.c
      src/x64/emitter.c
    )
  elseif("${arch}" STREQUAL "rv32i")
    add_definitions(-DJCC_RV32I)

    list(APPEND SOURCE_FILES
      src/rv32i/codegen.c
      src/rv32i/lower.c
      src/rv32i/emit.c
      src/rv32i/emitter.c
      src/rv32i/object.c
    )
  else()
    message(FATAL_ERROR "Unsupported architecture: ${arch}")
  endif()
endforeach()

# currently unused
# eep.c
# eep/codegen.c
# eep/lower.c
# eep/emit.c
# eep/emitter.c
# eep/object.c
# eep/disasm.c

# TODO: use hwaddress if possible instead of addresssanitizer (plus if hwaddress supported use hwaddress + thread)

set(SANITIZER_TYPE "none" CACHE STRING "Sanitizer type")
set_property(CACHE SANITIZER_TYPE PROPERTY STRINGS none address memory thread)

if(CMAKE_BUILD_TYPE STREQUAL "Debug")
  message(STATUS "Debug build detected")
  
  if(NOT SANITIZER_TYPE STREQUAL "none")
    if(SANITIZER_TYPE STREQUAL "address")
      if(CMAKE_C_COMPILER_ID STREQUAL "GNU")
        message(WARNING "Debug mode may be VERY SLOW on GCC due to ASan problems - Clang w/ ASan is significantly faster")
      endif()

      # try enable leak san

      check_c_compiler_flag("-fsanitize=leak" HAS_C_LEAK_SAN)
      if(HAS_C_LEAK_SAN)
        add_link_options(-fsanitize=leak)
        add_compile_options(-fsanitize=leak)
      endif()
    endif()

    if(SANITIZER_TYPE STREQUAL "memory")
      add_compile_options(-fsanitize-memory-track-origins)
    endif()

    if(NOT CMAKE_C_COMPILER_ID STREQUAL "JCC")
      add_link_options(-fsanitize=${SANITIZER_TYPE},undefined)
      add_compile_options(-fsanitize=${SANITIZER_TYPE},undefined -fno-sanitize-recover=all)
    endif()
  else()
    message(STATUS "Sanitisers disabled")
  endif()
elseif(CMAKE_BUILD_TYPE STREQUAL "Release" OR CMAKE_BUILD_TYPE STREQUAL "RelWithDebInfo")
  add_compile_options(
    -O3
    -mcpu=native
  )

  add_link_options(
    -O3
  )

  # add_compile_options(-flto)
  # add_link_options(-flto)
  add_compile_options(-fno-lto)
  add_link_options(-fno-lto)
endif()


if(MSVC)
  add_compile_option(/W4 /WX)
elseif(CMAKE_C_COMPILER_ID STREQUAL "JCC")
  add_compile_options(
    -Werror
  )
elseif(CMAKE_C_COMPILER_ID STREQUAL "AppleClang")
  message(STATUS "Building with AppleClang")

  add_compile_options(
    -Wall
    -Wextra
    -Wpedantic
    -Weverything

    # -Wformat-security
    # -Wformat
    # -Wformat=2
    # -Wconversion
    -Wimplicit-fallthrough
    -Wswitch

    # -Wimplicit
    # -Wincompatible-pointer-types
    # -Wint-conversion

    -Wno-error=unreachable-code

    -Wno-switch-default

    -Wzero-as-null-pointer-constant
    -Wbool-conversion

    -Wmissing-prototypes
    -Wsign-conversion
    -Wimplicit-fallthrough
    -Wstrict-prototypes
    -Wmissing-field-initializers

    # i would _like_ to enable this, but it causes errors everywhere flag enums are used
    # -Wsign-conversion
    -Wno-sign-conversion

    # we use recursive macros
    -Wno-disabled-macro-expansion

    -Wno-missing-noreturn
    -Wno-unused-macros
    -Wno-switch-enum
    -Wno-bad-function-cast
    -Wno-undef
    -Wno-padded
    -Wno-declaration-after-statement
    -Wno-class-varargs
    -Wno-conditional-uninitialized
    -Wno-shorten-64-to-32
    -Wno-assign-enum
    # TODO: remove binary literals in source
    -Wno-gnu-binary-literal

    -fstrict-enums
  )

  check_c_compiler_flag("-Wno-pre-c11-compat" HAS_C11_COMPAT_WARN)
  if(HAS_C11_COMPAT_WARN)
    add_compile_options(-Wno-pre-c11-compat)
  endif()

  check_c_compiler_flag("-Wno-c23-extensions" HAS_C23_EXT_WARN)
  if(HAS_C23_EXT_WARN)
    add_compile_options(-Wno-c23-extensions)
  endif()
elseif(CMAKE_C_COMPILER_ID STREQUAL "Clang")
  # nothing for now
elseif(CMAKE_C_COMPILER_ID STREQUAL "GNU")
  # nothing for now
else()
  message(WARNING "Unknown compiler: ${CMAKE_C_COMPILER_ID}")
endif()

# libjcc
add_library(libjcc STATIC ${SOURCE_FILES})
set_target_properties(libjcc PROPERTIES OUTPUT_NAME jcc)

# jcc
add_executable(jcc src/main.c)
target_link_libraries(jcc PRIVATE libjcc m pthread)


if(NOT CMAKE_C_COMPILER_ID STREQUAL "JCC")
  # currently can't bootstrap test due to atomic usage

  # test
  add_executable(test src/test.c)
  target_link_libraries(test PRIVATE libjcc m pthread)
endif()