# SPDX-License-Identifier: BSD-3-Clause
# mrv2
# Copyright Contributors to the mrv2 Project. All rights reserved.

cmake_policy(SET CMP0009 NEW)

message( STATUS "----------------------------------------prepackage------------------------------------" )
message( STATUS "CMAKE_CURRENT_BINARY_DIR=${CMAKE_CURRENT_BINARY_DIR}" )
message( STATUS "CMAKE_CURRENT_SOURCE_DIR=${CMAKE_CURRENT_SOURCE_DIR}" )
message( STATUS "CMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}" )

#
# Find mrv2's root dir by brute force, looking for directory that has
# our "cmake/version.cmake" file.
#
set( _current_dir ${CMAKE_CURRENT_BINARY_DIR})
set( _found_root_dir FALSE)
while( NOT _found_root_dir )
    set(_current_dir "${_current_dir}/..")
    file(REAL_PATH ${_current_dir} _current_dir )
    set(_mrv2_version "${_current_dir}/cmake/version.cmake" )
    if (EXISTS ${_mrv2_version})
	set(ROOT_DIR ${_current_dir})
	set(_found_root_dir TRUE)
    endif()
endwhile()

file(REAL_PATH ${ROOT_DIR} ROOT_DIR )
message( STATUS "mrv2 ROOT_DIR=${ROOT_DIR}" )


include( "${ROOT_DIR}/cmake/version.cmake" )
include( "${ROOT_DIR}/cmake/functions.cmake" )

if( UNIX AND NOT APPLE )
    #
    # \@bug: Linux CMAKE_INSTALL_PREFIX is broken and not pointing to
    #        pre-packaging directory!!!
    #
    if (EXISTS "${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_INSTALL_PREFIX}")
	set( CPACK_PREPACKAGE "${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_INSTALL_PREFIX}" )
    else()
	set( CPACK_PREPACKAGE "${CMAKE_INSTALL_PREFIX}" )
    endif()
else()
    set( CPACK_PREPACKAGE "${CMAKE_INSTALL_PREFIX}" )
endif()

message( STATUS "CPACK_PREPACKAGE=${CPACK_PREPACKAGE}" )

set(mrv2_NAME mrv2)
if(CPACK_PREPACKAGE MATCHES ".*vmrv2.*")
    set(mrv2_NAME vmrv2)
    set(MRV2_BACKEND "VK")
endif()

#
# Remove usd directory from lib/ directory on Windows
#
if(WIN32)
    message( STATUS "Removing ${CPACK_PREPACKAGE}/lib/usd")
    file( REMOVE_RECURSE "${CPACK_PREPACKAGE}/lib/usd" )
endif()

#
# Remove .a, .lib and .dll files from packaging lib/ directory
#
message( STATUS "Removing static and DLLs from ${CPACK_PREPACKAGE}/lib")
file( GLOB STATIC_LIBS "${CPACK_PREPACKAGE}/lib/*.a"
		       "${CPACK_PREPACKAGE}/lib/*.lib"
		       "${CPACK_PREPACKAGE}/lib/*.dll" )
		   

if ( NOT "${STATIC_LIBS}" STREQUAL "" )
    file( REMOVE ${STATIC_LIBS} )
endif()


#
# Remove include files from packaging directory
#
message( STATUS "Removing ${CPACK_PREPACKAGE}/include" )
file( REMOVE_RECURSE "${CPACK_PREPACKAGE}/include" )

#
# List of python module globs to remove if present
# 
set( _pythonBuiltinModules 
    test*
    config*
    ctypes/test*
    distutils/test*
    idlelib/idle_test*
    lib2to3/test*
    sqlite3/test*
    tkinter/test*
    setuptools*
    unittest*
)

set( _pythonSiteModules 
    alabaster*
    babel*
    click*
    colorama*
    docutils*
    filelock*
    fltk14/*.cpp
    fltk14/*.h
    fsspec*
    huggingface-hub*
    imagesize*
    Jinja*
    joblib*
    libretranslatepy*
    lxml*
    MarkupSafe*
    meson*
    mpmath*
    numpy*
    networkx*
    pkg_resources*
    pillow*
    PIL*
    polib*
    pyyaml*
    pyYAML*
    pygments*
    Pygments*
    pytz*
    regex*
    sacremoses*
    safetensors*
    sentencepiece*
    setuptools*
    snowballstemmer*
    sphinx*
    sympy*
    tokenizers*
    torch*
    transformers*
    translate*
    typing-extensions*
    tqdm*
    unittest*
)

#
# Install system .SO dependencies
#
if(UNIX)

    #
    # Under Rocky Linux, DSOs sometimes go to lib64/.
    # Copy them to lib/ directory
    #
    set(linux_lib64_dir "${CPACK_PREPACKAGE}/lib64")
    if (EXISTS "${linux_lib64_dir}" )
	# For pyFLTK we need to install all libfltk DSOs including those we
	# do not use, like forms.
	message( NOTICE "${linux_lib64_dir} exists...")
	file(GLOB fltk_dsos "${linux_lib64_dir}/libfltk*.so")
	file(INSTALL
	    DESTINATION "${CMAKE_INSTALL_PREFIX}/lib"
	    TYPE SHARED_LIBRARY
	    FOLLOW_SYMLINK_CHAIN
	    FILES ${fltk_dsos}
	)
	file(REMOVE_RECURSE ${linux_lib64_dir})
    else()
	message( NOTICE "${linux_lib64_dir} does not exist...")
    endif()
	
    #
    # Glob python directories as we don't know the version of python in this
    # script, as it does not inherit variables from the main CMakeLists.
    #
    file(GLOB MRV2_PYTHON_LIB_DIRS "${CPACK_PREPACKAGE}/lib/python*")

    #
    # Grab the last version, in case there are several
    #
    set(MRV2_PYTHON_LIBDIR "")
    if (MRV2_PYTHON_LIB_DIRS STREQUAL "")
	message(WARNING "Could not locate any python version in:\n${CPACK_PREPACKAGE}/lib/python*")
    else()
	list(GET MRV2_PYTHON_LIB_DIRS -1 MRV2_PYTHON_LIB_DIR)
    endif()
    
    set(MRV2_PYTHON_SITE_PACKAGES_DIR "${MRV2_PYTHON_LIB_DIR}/site-packages")

    set( MRV2_EXES "${CPACK_PREPACKAGE}/bin/mrv2" )

    if (EXISTS "${CPACK_PREPACKAGE}/bin/hdr")
	list(APPEND MRV2_EXES "${CPACK_PREPACKAGE}/bin/hdr" )
    endif()
    
    #
    # We need to get the dependencies of the python DSOs to avoid
    # issues like openssl and libcrypto changing between Rocky Linux
    # and Ubuntu 22.04.5 or macOS's libb2.
    #
    set(MRV2_PYTHON_DSO_DIR "${MRV2_PYTHON_LIB_DIR}/lib-dynload")
    file(GLOB python_dsos "${MRV2_PYTHON_DSO_DIR}/*.so")
    list(APPEND MRV2_EXES ${python_dsos} )
	
    if ( APPLE )
	#
	# Get DYLIB dependencies of components
	#
	get_macos_runtime_dependencies( "${MRV2_EXES}" )
	
    else()
	#
	# Get DSO dependencies of components
	#
	get_runtime_dependencies( "${MRV2_EXES}" )
    endif()
elseif(WIN32)
    #
    # Set python's site-packages dir for .exe installer.
    #
    # When building an .exe installer on Windows, the site-packages will
    # be inside an applications component directory.
    #
    set(MRV2_APP_DIR "${CPACK_PREPACKAGE}/applications")

    if (EXISTS "${MRV2_APP_DIR}/bin/mrv2.exe")
	file(CREATE_LINK
	    "${MRV2_APP_DIR}/bin/mrv2.exe"
	    "${MRV2_APP_DIR}/bin/mrv2-v${mrv2_VERSION}.exe") 
    endif()
    
    
    set(MRV2_PYTHON_LIB_DIR "${MRV2_APP_DIR}/bin/Lib")
    set(MRV2_PYTHON_SITE_PACKAGES_DIR
	"${MRV2_PYTHON_LIB_DIR}/site-packages")
    
    #
    # Don't pack sphinx and other auxiliary documentation libs in .exe
    #

    set(MRV2_PYTHON_GLOBS 
	"${MRV2_APP_DIR}/lib/Lib/")

    # 1. Collect all paths into one list first
    set(DIRS_TO_DELETE "")
    
    list(APPEND DIRS_TO_DELETE "${MRV2_APP_DIR}/bin/DLLs/*.pdb")

    foreach(_mod ${_pythonBuiltinModules})
	list(APPEND DIRS_TO_DELETE "${MRV2_PYTHON_LIB_DIR}/${_mod}")
    endforeach()

    foreach(_mod ${_pythonSiteModules})
	list(APPEND DIRS_TO_DELETE "${MRV2_PYTHON_SITE_PACKAGES_DIR}/${_mod}")
    endforeach()

    # 2. Use GLOB once for the patterns that actually need it (the ones with *)
    file(GLOB MATCHED_DIRS ${DIRS_TO_DELETE})

    # 3. Batch remove everything in one go
    if(MATCHED_DIRS)
	message(STATUS "Cleaning up NSIS application's Python modules...")
	foreach (_dir ${MATCHED_DIRS})
	    message(STATUS ${_dir})
	endforeach()
	file(REMOVE_RECURSE ${MATCHED_DIRS})
    endif()

    # 4. Final pycache sweep (one single scan)
    file(GLOB_RECURSE ALL_PYCACHE "${CPACK_PREPACKAGE}/**/__pycache__")
    file(REMOVE_RECURSE ${ALL_PYCACHE})
    
    
    #
    # Set python's site-packages and lib dir for .zip.
    #
    # When building an .exe on Windows, the site-packages will
    # be inside an application directory.
    #
    set(MRV2_PYTHON_LIB_DIR "${CPACK_PREPACKAGE}/bin/Lib")
    set(MRV2_PYTHON_SITE_PACKAGES_DIR
	"${CPACK_PREPACKAGE}/bin/Lib/site-packages")
endif()


#
# Don't pack sphinx and other auxiliary documentation libs nor the tests
# for the libraries.
#

# 1. Collect all paths into one list first
set(DIRS_TO_DELETE "")

list(APPEND DIRS_TO_DELETE "${CPACK_PREPACKAGE}/bin/DLLs/*.pdb")

foreach(_mod ${_pythonBuiltinModules})
    list(APPEND DIRS_TO_DELETE "${MRV2_PYTHON_LIB_DIR}/${_mod}")
endforeach()

foreach(_mod ${_pythonSiteModules})
    list(APPEND DIRS_TO_DELETE "${MRV2_PYTHON_SITE_PACKAGES_DIR}/${_mod}")
endforeach()

# 2. Use GLOB once for the patterns that actually need it (the ones with *)
file(GLOB MATCHED_DIRS ${DIRS_TO_DELETE})

# 3. Batch remove everything in one go
if(MATCHED_DIRS)
    message(STATUS "Cleaning up ZIP's Python modules...")
    foreach (_dir ${MATCHED_DIRS})
	message(STATUS ${_dir})
    endforeach()
    file(REMOVE_RECURSE ${MATCHED_DIRS})
endif()

# 4. Final pycache sweep (one single scan)
file(GLOB_RECURSE ALL_PYCACHE "${CPACK_PREPACKAGE}/**/__pycache__")
file(REMOVE_RECURSE ${ALL_PYCACHE})

if (APPLE)
    include( "${ROOT_DIR}/cmake/prepackage_macos.cmake" )
endif()