################################################################################
#
#  Program: 3D Slicer
#
#  Copyright (c) Kitware Inc.
#
#  See COPYRIGHT.txt
#  or http://www.slicer.org/copyright/copyright.txt for details.
#
#  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.
#
#  This file was originally developed by Jean-Christophe Fillion-Robin, Kitware Inc.
#  and was partially funded by NIH grant 3P41RR013218-12S1
#
################################################################################

include(CMakeParseArguments)

if(NOT DEFINED Slicer_CMAKE_DIR)
  set(Slicer_CMAKE_DIR ${CMAKE_CURRENT_LIST_DIR}/../../CMake)
endif()
include(${Slicer_CMAKE_DIR}/ListToString.cmake)

if(NOT DEFINED Slicer_EXTENSIONS_CMAKE_DIR)
  set(Slicer_EXTENSIONS_CMAKE_DIR ${CMAKE_CURRENT_LIST_DIR})
endif()

function(slicerFunctionGenerateExtensionDescription)
  set(options)
  set(oneValueArgs
    DESTINATION_DIR
    EXTENSION_BUILD_SUBDIRECTORY
    EXTENSION_CATEGORY
    EXTENSION_DESCRIPTION
    EXTENSION_DICOM_SUPPORT_RULE
    EXTENSION_ENABLED
    EXTENSION_HOMEPAGE
    EXTENSION_ICONURL
    EXTENSION_NAME
    EXTENSION_STATUS
    EXTENSION_TIER
    EXTENSION_WC_REVISION
    EXTENSION_WC_ROOT
    EXTENSION_WC_TYPE
    EXTENSION_WC_URL
    SLICER_REVISION
    SLICER_WC_ROOT
    )
  set(multiValueArgs
    EXTENSION_CONTRIBUTORS
    EXTENSION_DEPENDS
    EXTENSION_SCREENSHOTURLS
    EXTENSION_RECOMMENDS
    EXTENSION_KEYWORDS
    )
  cmake_parse_arguments(MY "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})

  # Sanity checks
  set(expected_nonempty_vars
    EXTENSION_NAME
    EXTENSION_WC_REVISION
    EXTENSION_WC_ROOT
    EXTENSION_WC_TYPE
    EXTENSION_WC_URL
    SLICER_REVISION
    SLICER_WC_ROOT
    )
  foreach(var ${expected_nonempty_vars})
    if("${MY_${var}}" STREQUAL "")
      message(FATAL_ERROR "CMake variable ${var} is empty !")
    endif()
  endforeach()

  function(_convert_items_to_s4ext _items _separator _output_var)
    # Remove newlines
    string(REPLACE "\n" "" _items "${_items}")
    # Strip leading and trailing spaces of each element
    list(TRANSFORM _items STRIP)
    # Strip contiguous spaces
    string(REGEX REPLACE " +" " " _items "${_items}")
    # Strip leading and trailing spaces
    string(STRIP "${_items}" _items)
    # Convert to space separated list
    list_to_string("${_separator}" "${_items}" _items)

    set(${_output_var} "${_items}" PARENT_SCOPE)
  endfunction()

  # contributors: Convert to comma separated list
  _convert_items_to_s4ext("${MY_EXTENSION_CONTRIBUTORS}" ", " MY_EXTENSION_CONTRIBUTORS)

  # description: Replace newlines with "<br>"
  string(REPLACE "\n" "<br>" MY_EXTENSION_DESCRIPTION "${MY_EXTENSION_DESCRIPTION}")

  # screenshoturls: Convert to space separated list
  _convert_items_to_s4ext("${MY_EXTENSION_SCREENSHOTURLS}" " " MY_EXTENSION_SCREENSHOTURLS)

  # Convert to space separated list
  list_to_string(" " "${MY_EXTENSION_DEPENDS}" MY_EXTENSION_DEPENDS)
  list_to_string(" " "${MY_EXTENSION_RECOMMENDS}" MY_EXTENSION_RECOMMENDS)
  list_to_string(" " "${MY_EXTENSION_KEYWORDS}" MY_EXTENSION_KEYWORDS)

  set(expected_existing_vars DESTINATION_DIR)
  foreach(var ${expected_existing_vars})
    if(NOT EXISTS "${MY_${var}}")
      message(FATAL_ERROR "error: ${var} CMake variable points to a inexistent file or directory: ${MY_${var}}")
    endif()
  endforeach()

  include(${Slicer_EXTENSIONS_CMAKE_DIR}/SlicerExtensionDescriptionSpec.cmake)
  slicer_extension_metadata_set_defaults(MY_EXTENSION)

  set(filename ${MY_DESTINATION_DIR}/${MY_EXTENSION_NAME}.s4ext)

  set(scm_type ${MY_EXTENSION_WC_TYPE})
  set(scm_url ${MY_EXTENSION_WC_URL})

  #message(MY_SLICER_WC_ROOT:${MY_SLICER_WC_ROOT})
  #message(MY_SLICER_REVISION:${MY_SLICER_REVISION})
  #message(MY_EXTENSION_WC_TYPE:${MY_EXTENSION_WC_TYPE})
  #message(MY_EXTENSION_WC_ROOT:${MY_EXTENSION_WC_ROOT})
  #message(MY_EXTENSION_WC_REVISION:${MY_EXTENSION_WC_REVISION})

  # If both Root and Revision matches, let's assume both Slicer source and Extension source
  # are checkout on the same filesystem.
  # This is useful for testing purposes
  if(${MY_EXTENSION_WC_TYPE} STREQUAL "local" OR (${MY_EXTENSION_WC_ROOT} STREQUAL ${MY_SLICER_WC_ROOT}
     AND ${MY_EXTENSION_WC_REVISION} STREQUAL ${MY_SLICER_REVISION}))
    set(scm_type local)
    set(scm_url ${CMAKE_CURRENT_SOURCE_DIR})
  endif()

  configure_file(
    ${Slicer_EXTENSIONS_CMAKE_DIR}/../../Utilities/Templates/Extensions/extension_description.s4ext.in
    ${filename}
    )

  message(STATUS "Extension description has been written to: ${filename}")
endfunction()

################################################################################
# Testing
################################################################################

function(_check_generated_description_file)
  set(options
    )
  set(oneValueArgs
    BASELINE
    GENERATED
    )
  set(multiValueArgs
    )
  cmake_parse_arguments(MY "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
  execute_process(
    COMMAND ${CMAKE_COMMAND} -E compare_files --ignore-eol
      ${MY_GENERATED}
      ${MY_BASELINE}
    RESULT_VARIABLE result
    )
  if(NOT result EQUAL 0)
    message(FATAL_ERROR "The generated and baseline files are different but are expected to match. Generated [${MY_GENERATED}]. Baseline [${MY_BASELINE}]")
  endif()
endfunction()

#
# cmake -DTEST_<testfunction>:BOOL=ON -P <this_script>.cmake
#

function(slicer_generate_extension_description_test)

  if(NOT DEFINED Slicer_SOURCE_DIR)
    set(Slicer_SOURCE_DIR ${CMAKE_CURRENT_LIST_DIR}/../..)
  endif()

  set(destination_dir ${CMAKE_CURRENT_BINARY_DIR})
  set(common_args
    DESTINATION_DIR ${destination_dir}
    EXTENSION_DESCRIPTION "The SlicerToKiwiExporter module provides Slicer user with any easy way to export models into a KiwiViewer scene file.
This is a line of text.<br>And another one."
    EXTENSION_CATEGORY "Exporter"
    EXTENSION_CONTRIBUTORS "Jean-Christophe Fillion-Robin (Kitware), Pat Marion (Kitware), \
      Steve Pieper (Isomics), Atsushi Yamada (Shiga University of Medical Science)"
    EXTENSION_HOMEPAGE "https://www.slicer.org/w/index.php/Documentation/Nightly/Extensions/SlicerToKiwiExporter"
    EXTENSION_ICONURL "https://www.slicer.org/w/images/6/64/SlicerToKiwiExporterLogo.png"
    EXTENSION_NAME "SlicerToKiwiExporter"
    EXTENSION_SCREENSHOTURLS "https://www.slicer.org/w/images/9/9e/SlicerToKiwiExporter_Kiwiviewer_8.PNG \
      https://www.slicer.org/w/images/a/ab/SlicerToKiwiExporter_Kiwiviewer_9.PNG"
      "https://www.slicer.org/w/images/9/9a/SlicerToKiwiExporter_SaveDialog_Select-file-format_1.png"
    EXTENSION_STATUS "Development"
    EXTENSION_WC_REVISION "9d7341e978df954a2c875240290833d7528ef29c"
    EXTENSION_WC_ROOT "https://github.com/jcfr/SlicerToKiwiExporter.git"
    EXTENSION_WC_TYPE "git"
    EXTENSION_WC_URL "https://github.com/jcfr/SlicerToKiwiExporter.git"
    SLICER_REVISION "29196"
    SLICER_WC_ROOT "https://github.com/Slicer/Slicer.git"
    EXTENSION_DICOM_SUPPORT_RULE "Modality=\"SEG\""
    EXTENSION_KEYWORDS "KeyFoo" "KeyBar"
    EXTENSION_RECOMMENDS "RecFoo" "RecBar"
    )

  # Generate description file of an extension *without* dependencies
  slicerFunctionGenerateExtensionDescription(
    ${common_args}
    EXTENSION_TIER "5"
    )
  _check_generated_description_file(
    GENERATED "${destination_dir}/SlicerToKiwiExporter.s4ext"
    BASELINE "${Slicer_SOURCE_DIR}/Extensions/CMake/Testing/extension_description_without_depends.s4ext"
    )

  # Generate description file of an extension *without* dependencies again
  # and test that contributors can also be passed in as a list
  set(contributors_list
    "Jean-Christophe  Fillion-Robin (Kitware)"
    "Pat Marion (Kitware), Steve Pieper (Isomics)  "
    "  Atsushi Yamada \
      (Shiga University of Medical Science)"
    )
  # replace the contributors arg string value with the list value
  list(FIND common_args "EXTENSION_CONTRIBUTORS" contributors_args_idx)
  math(EXPR contributors_val_idx "${contributors_args_idx}+1")
  list(REMOVE_AT common_args ${contributors_val_idx})
  list(INSERT common_args ${contributors_val_idx} ${contributors_list})
  # regenerate the description
  slicerFunctionGenerateExtensionDescription(
    ${common_args}
    EXTENSION_TIER "5"
    )
  _check_generated_description_file(
    GENERATED "${destination_dir}/SlicerToKiwiExporter.s4ext"
    BASELINE "${Slicer_SOURCE_DIR}/Extensions/CMake/Testing/extension_description_without_depends.s4ext"
    )

  # Generate description file of an extension *with* dependencies
  # where EXTENSION_DEPENDS is a space separated string
  slicerFunctionGenerateExtensionDescription(
    ${common_args}
    EXTENSION_BUILD_SUBDIRECTORY "inner/inner-inner-build"
    EXTENSION_DEPENDS "Foo Bar"
    EXTENSION_ENABLED 0
    )
  _check_generated_description_file(
    GENERATED "${destination_dir}/SlicerToKiwiExporter.s4ext"
    BASELINE "${Slicer_SOURCE_DIR}/Extensions/CMake/Testing/extension_description_with_depends.s4ext"
    )

  # Generate description file of an extension *with* dependencies
  # where EXTENSION_DEPENDS is a list
  slicerFunctionGenerateExtensionDescription(
    ${common_args}
    EXTENSION_BUILD_SUBDIRECTORY "inner/inner-inner-build"
    EXTENSION_DEPENDS Foo Bar
    EXTENSION_ENABLED 0
    EXTENSION_STATUS ""
    )
  _check_generated_description_file(
    GENERATED "${destination_dir}/SlicerToKiwiExporter.s4ext"
    BASELINE "${Slicer_SOURCE_DIR}/Extensions/CMake/Testing/extension_description_with_depends.s4ext"
    )

  message("SUCCESS")
endfunction()
if(TEST_slicer_generate_extension_description_test)
  slicer_generate_extension_description_test()
endif()
