include_guard()

set(NPM_TIMEOUT 600 CACHE STRING "The number of seconds to wait for npm to install packages, or 0 to wait indefinitely")

function(find_npm result)
  if(CMAKE_HOST_WIN32)
    find_program(
      npm
      NAMES npm.cmd npm
      REQUIRED
    )
  else()
    find_program(
      npm
      NAMES npm
      REQUIRED
    )
  endif()

  set(${result} "${npm}")

  return(PROPAGATE ${result})
endfunction()

function(node_module_prefix result)
  set(one_value_keywords
    WORKING_DIRECTORY
  )

  cmake_parse_arguments(
    PARSE_ARGV 1 ARGV "" "${one_value_keywords}" ""
  )

  if(ARGV_WORKING_DIRECTORY)
    cmake_path(ABSOLUTE_PATH ARGV_WORKING_DIRECTORY BASE_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}" NORMALIZE)
  else()
    set(ARGV_WORKING_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}")
  endif()

  find_npm(npm)

  execute_process(
    COMMAND ${npm} prefix
    WORKING_DIRECTORY "${ARGV_WORKING_DIRECTORY}"
    OUTPUT_VARIABLE prefix
    OUTPUT_STRIP_TRAILING_WHITESPACE
    RESULT_VARIABLE status
    ERROR_VARIABLE error
  )

  if(NOT status EQUAL 0)
    message(FATAL_ERROR "${error}")
  endif()

  set(${result} "${prefix}")

  return(PROPAGATE ${result})
endfunction()

function(install_node_module specifier)
  set(option_keywords
    SAVE
    FORCE
  )

  set(one_value_keywords
    VERSION
    TIMEOUT
    WORKING_DIRECTORY
  )

  cmake_parse_arguments(
    PARSE_ARGV 1 ARGV "${option_keywords}" "${one_value_keywords}" ""
  )

  if(NOT ARGV_VERSION)
    set(ARGV_VERSION "latest")
  endif()

  if(ARGV_WORKING_DIRECTORY)
    cmake_path(ABSOLUTE_PATH ARGV_WORKING_DIRECTORY BASE_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}" NORMALIZE)
  else()
    set(ARGV_WORKING_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}")
  endif()

  find_npm(npm)

  execute_process(
    COMMAND ${npm} ls --depth=0 ${specifier}@${ARGV_VERSION}
    WORKING_DIRECTORY "${ARGV_WORKING_DIRECTORY}"
    OUTPUT_QUIET
    ERROR_QUIET
    RESULT_VARIABLE status
  )

  if(NOT status EQUAL 0 AND NOT ARGV_VERSION MATCHES "^[~^=<>]?[0-9]")
    execute_process(
      COMMAND ${npm} ls --depth=0 ${specifier}
      WORKING_DIRECTORY "${ARGV_WORKING_DIRECTORY}"
      OUTPUT_QUIET
      ERROR_QUIET
      RESULT_VARIABLE status
    )
  endif()

  if(status EQUAL 0)
    return()
  endif()

  set(args --ignore-scripts --foreground-scripts --allow-git=none)

  if(ARGV_SAVE)
    list(APPEND args --save)
  endif()

  if(ARGV_FORCE)
    list(APPEND args --force)
  endif()

  list(APPEND args ${specifier}@${ARGV_VERSION})

  if(CMAKE_HOST_WIN32)
    find_program(
      sfw
      NAMES sfw.cmd sfw
    )
  else()
    find_program(
      sfw
      NAMES sfw
    )
  endif()

  if(NOT sfw MATCHES "NOTFOUND")
    set(npm "${sfw}" ${npm})
  endif()

  if(NOT DEFINED ARGV_TIMEOUT)
    set(ARGV_TIMEOUT ${NPM_TIMEOUT})
  endif()

  if(ARGV_TIMEOUT GREATER 0)
    set(timeout TIMEOUT ${ARGV_TIMEOUT})
  else()
    set(timeout)
  endif()

  execute_process(
    COMMAND ${npm} install ${args}
    WORKING_DIRECTORY "${ARGV_WORKING_DIRECTORY}"
    OUTPUT_VARIABLE output
    RESULT_VARIABLE status
    ERROR_VARIABLE error
    ${timeout}
  )

  if(NOT status EQUAL 0)
    message(FATAL_ERROR "npm install failed: ${status}\n${error}${output}")
  endif()
endfunction()

function(install_node_modules)
  set(option_keywords
    FORCE
    LOCKFILE
  )

  set(one_value_keywords
    TIMEOUT
    WORKING_DIRECTORY
  )

  cmake_parse_arguments(
    PARSE_ARGV 0 ARGV "${option_keywords}" "${one_value_keywords}" ""
  )

  if(ARGV_WORKING_DIRECTORY)
    cmake_path(ABSOLUTE_PATH ARGV_WORKING_DIRECTORY BASE_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}" NORMALIZE)
  else()
    set(ARGV_WORKING_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}")
  endif()

  set(args --ignore-scripts --foreground-scripts --allow-git=none)

  if(ARGV_FORCE)
    list(APPEND args --force)
  endif()

  if(ARGV_LOCKFILE)
    set(install install-clean)
  else()
    set(install install)
  endif()

  cmake_path(APPEND ARGV_WORKING_DIRECTORY package.json OUTPUT_VARIABLE package_path)

  cmake_path(APPEND ARGV_WORKING_DIRECTORY package-lock.json OUTPUT_VARIABLE package_lock_path)

  find_npm(npm)

  if(CMAKE_HOST_WIN32)
    find_program(
      sfw
      NAMES sfw.cmd sfw
    )
  else()
    find_program(
      sfw
      NAMES sfw
    )
  endif()

  if(NOT sfw MATCHES "NOTFOUND")
    set(npm "${sfw}" ${npm})
  endif()

  if(NOT DEFINED ARGV_TIMEOUT)
    set(ARGV_TIMEOUT ${NPM_TIMEOUT})
  endif()

  if(ARGV_TIMEOUT GREATER 0)
    set(timeout TIMEOUT ${ARGV_TIMEOUT})
  else()
    set(timeout)
  endif()

  execute_process(
    COMMAND ${npm} ${install} ${args}
    WORKING_DIRECTORY "${ARGV_WORKING_DIRECTORY}"
    OUTPUT_VARIABLE output
    RESULT_VARIABLE status
    ERROR_VARIABLE error
    ${timeout}
  )

  if(NOT status EQUAL 0)
    message(FATAL_ERROR "npm ${install} failed: ${status}\n${error}${output}")
  endif()

  set_property(
    DIRECTORY
    APPEND
    PROPERTY CMAKE_CONFIGURE_DEPENDS
      "${package_path}"
      "${package_lock_path}"
  )
endfunction()

function(resolve_node_module specifier result)
  set(one_value_keywords
    WORKING_DIRECTORY
  )

  cmake_parse_arguments(
    PARSE_ARGV 2 ARGV "" "${one_value_keywords}" ""
  )

  if(ARGV_WORKING_DIRECTORY)
    cmake_path(ABSOLUTE_PATH ARGV_WORKING_DIRECTORY BASE_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}" NORMALIZE)
  else()
    set(ARGV_WORKING_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}")
  endif()

  set(dirname "${ARGV_WORKING_DIRECTORY}")

  if(specifier STREQUAL "." OR specifier STREQUAL ".." OR specifier MATCHES "^[.]?/")
    cmake_path(
      APPEND dirname "${specifier}" package.json
      OUTPUT_VARIABLE target
    )

    cmake_path(NORMAL_PATH target)

    if(EXISTS ${target})
      cmake_path(GET target PARENT_PATH ${result})
    else()
      set(${result} "${specifier}-NOTFOUND")
    endif()

    return(PROPAGATE ${result})
  endif()

  cmake_path(GET dirname ROOT_PATH root)

  while(TRUE)
    cmake_path(
      APPEND dirname node_modules "${specifier}" package.json
      OUTPUT_VARIABLE target
    )

    cmake_path(NORMAL_PATH target)

    if(EXISTS ${target})
      cmake_path(GET target PARENT_PATH ${result})

      return(PROPAGATE ${result})
    endif()

    if(dirname PATH_EQUAL root)
      break()
    endif()

    cmake_path(GET dirname PARENT_PATH dirname)
  endwhile()

  set(${result} "${specifier}-NOTFOUND")

  return(PROPAGATE ${result})
endfunction()

function(list_node_modules result)
  set(option_keywords
    DEVELOPMENT
  )

  set(one_value_keywords
    WORKING_DIRECTORY
  )

  cmake_parse_arguments(
    PARSE_ARGV 1 ARGV "${option_keywords}" "${one_value_keywords}" ""
  )

  if(ARGV_WORKING_DIRECTORY)
    cmake_path(ABSOLUTE_PATH ARGV_WORKING_DIRECTORY BASE_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}" NORMALIZE)
  else()
    set(ARGV_WORKING_DIRECTORY "${CMAKE_CURRENT_LIST_DIR}")
  endif()

  if(ARGV_DEVELOPMENT)
    set(DEVELOPMENT DEVELOPMENT)
  endif()

  cmake_path(APPEND ARGV_WORKING_DIRECTORY package.json OUTPUT_VARIABLE package_path)

  file(READ "${package_path}" package)

  list(APPEND properties dependencies optionalDependencies)

  if(ARGV_DEVELOPMENT)
    list(APPEND properties devDependencies)
  endif()

  foreach(property ${properties})
    string(JSON dependencies ERROR_VARIABLE error GET "${package}" ${property})

    if(error MATCHES "NOTFOUND")
      string(JSON len LENGTH "${dependencies}")

      foreach(i RANGE ${len})
        if(NOT i EQUAL len)
          string(JSON specifier MEMBER "${dependencies}" ${i})

          resolve_node_module(${specifier} resolved WORKING_DIRECTORY ${ARGV_WORKING_DIRECTORY})

          if("${resolved}" MATCHES "NOTFOUND" OR "${resolved}" IN_LIST ${result})
            continue()
          endif()

          list(APPEND ${result} "${resolved}")

          list_node_modules(${result} WORKING_DIRECTORY ${resolved} ${DEVELOPMENT})
        endif()
      endforeach()
    endif()
  endforeach()

  return(PROPAGATE ${result})
endfunction()