CMAKE_MINIMUM_REQUIRED(VERSION 2.6.3)

# ROOT_DIR should be set to root of the repository (where to find the .svn or .hg directory)
# SOURCE_DIR should be set to root of your code (where to find CMakeLists.txt)
# BINARY_DIR should be set to root of your build directory

IF(SOURCE_DIR)
  # Replace spaces by semi-columns
  IF(CMAKE_MODULE_PATH)
    STRING(REPLACE " " ";" CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH})
  ENDIF()

  SET(CMAKE_MODULE_PATH ${SOURCE_DIR}/CMakeModules ${CMAKE_MODULE_PATH})

  IF(NOT ROOT_DIR AND SOURCE_DIR)
    SET(ROOT_DIR ${SOURCE_DIR})
  ENDIF()

  IF(NOT SOURCE_DIR AND ROOT_DIR)
    SET(SOURCE_DIR ${ROOT_DIR})
  ENDIF()
ELSE()
  SET(SOURCE_DIR ${CMAKE_SOURCE_DIR})
  SET(ROOT_DIR ${CMAKE_SOURCE_DIR}/..)

  # convert ROOT_DIR to absolute path
  GET_FILENAME_COMPONENT(ROOT_DIR ${ROOT_DIR} ABSOLUTE)
ENDIF()

MACRO(NOW RESULT)
  IF(CMAKE_VERSION VERSION_GREATER "2.8.10")
    STRING(TIMESTAMP ${RESULT} "%Y-%m-%d %H:%M:%S")
  ELSE()
    IF(WIN32)
      EXECUTE_PROCESS(COMMAND "wmic" "os" "get" "localdatetime" OUTPUT_VARIABLE DATETIME)
      IF(NOT DATETIME MATCHES "ERROR")
        STRING(REGEX REPLACE ".*\n([0-9][0-9][0-9][0-9])([0-9][0-9])([0-9][0-9])([0-9][0-9])([0-9][0-9])([0-9][0-9]).*" "\\1-\\2-\\3 \\4:\\5:\\6" ${RESULT} "${DATETIME}")
      ENDIF()
    ELSEIF(UNIX)
      EXECUTE_PROCESS(COMMAND "date" "+%Y-%m-%d %H:%M:%S" OUTPUT_VARIABLE DATETIME)
      STRING(REGEX REPLACE "([0-9: -]+).*" "\\1" ${RESULT} "${DATETIME}")
    ELSE()
      MESSAGE(SEND_ERROR "date not implemented")
      SET(${RESULT} "0000-00-00 00:00:00")
    ENDIF()
  ENDIF()
ENDMACRO()

IF(EXISTS "${ROOT_DIR}/.svn/")
  FIND_PACKAGE(Subversion QUIET)

  IF(SUBVERSION_FOUND)
    Subversion_WC_INFO(${ROOT_DIR} ER)
    SET(REVISION ${ER_WC_REVISION})
  ENDIF()

  FIND_PACKAGE(TortoiseSVN QUIET)

  IF(TORTOISESVN_FOUND)
    TORTOISESVN_GET_REVISION(${ROOT_DIR} REVISION)
  ENDIF()
ENDIF()

IF(EXISTS "${ROOT_DIR}/.hg/")
  FIND_PACKAGE(Mercurial)

  IF(MERCURIAL_FOUND)
    Mercurial_WC_INFO(${ROOT_DIR} ER)
    SET(REVISION ${ER_WC_REVISION})
    SET(CHANGESET ${ER_WC_CHANGESET})
    SET(BRANCH ${ER_WC_BRANCH})
  ENDIF()
ENDIF()

IF(EXISTS "${ROOT_DIR}/.git/")
  FIND_PACKAGE(Git)

  IF(GIT_FOUND)
    EXECUTE_PROCESS(COMMAND ${GIT_EXECUTABLE} rev-list HEAD --count
      WORKING_DIRECTORY ${ROOT_DIR}
      RESULT_VARIABLE git_exit_code
      OUTPUT_VARIABLE REVISION)
    IF(NOT ${git_exit_code} EQUAL 0)
      message(WARNING "git rev-list failed, unable to include version.")
    ENDIF()
    EXECUTE_PROCESS(COMMAND ${GIT_EXECUTABLE} rev-parse --short=8 HEAD
      WORKING_DIRECTORY ${ROOT_DIR}
      RESULT_VARIABLE git_exit_code
      OUTPUT_VARIABLE CHANGESET)
    IF(NOT ${git_exit_code} EQUAL 0)
      message(WARNING "git rev-parse failed, unable to include version.")
    ENDIF()
    EXECUTE_PROCESS(COMMAND ${GIT_EXECUTABLE} rev-parse --abbrev-ref HEAD
      WORKING_DIRECTORY ${ROOT_DIR}
      RESULT_VARIABLE git_exit_code
      OUTPUT_VARIABLE BRANCH)
    IF(NOT ${git_exit_code} EQUAL 0)
      message(WARNING "git rev-parse failed, unable to include git branch.")
    ENDIF()
    STRING(STRIP ${REVISION} REVISION)
    STRING(STRIP ${CHANGESET} CHANGESET)
    STRING(STRIP ${BRANCH} BRANCH)
  ENDIF()
ENDIF()

# if processing exported sources, use "revision" file if exists
IF(SOURCE_DIR AND NOT DEFINED REVISION)
  SET(REVISION_FILE ${SOURCE_DIR}/revision)
  IF(EXISTS ${REVISION_FILE})
    FILE(STRINGS ${REVISION_FILE} REVISION LIMIT_COUNT 1)
    MESSAGE(STATUS "Read revision ${REVISION} from file")
  ENDIF()
ENDIF()

IF(DEFINED REVISION)
  MESSAGE(STATUS "Found revision ${REVISION}")
ELSE()
  SET(REVISION 0)
ENDIF()