-
Notifications
You must be signed in to change notification settings - Fork 96
/
CMakeLists.txt
94 lines (86 loc) · 4.87 KB
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
cmake_minimum_required(VERSION 3.0)
project(mozilla-pipeline-schemas VERSION 0.0.9 LANGUAGES NONE)
set(CPACK_PACKAGE_DESCRIPTION_SUMMARY "Mozilla data ingestion and data lake schemas")
set(CPACK_PACKAGE_VERSION_MAJOR ${PROJECT_VERSION_MAJOR})
set(CPACK_PACKAGE_VERSION_MINOR ${PROJECT_VERSION_MINOR})
set(CPACK_PACKAGE_VERSION_PATCH ${PROJECT_VERSION_PATCH})
set(CPACK_PACKAGE_VENDOR "Mozilla Services")
set(CPACK_PACKAGE_CONTACT "Mike Trinkala <[email protected]>")
set(CPACK_RESOURCE_FILE_LICENSE "${CMAKE_SOURCE_DIR}/LICENSE.txt")
set(CPACK_RPM_PACKAGE_LICENSE "MPLv2.0")
find_program(PYTHON_EXE python3)
if(NOT PYTHON_EXE)
message(FATAL_ERROR "python3 is required for output formatting and metadata injection")
endif()
find_program(PARQUETFMT_EXE parquetfmt)
if(NOT PARQUETFMT_EXE)
if(WIN32)
file(DOWNLOAD https://github.com/trink/parquetfmt/releases/download/v0.1.0/parquetfmt-win64.zip ./parquetfmt-win64.zip EXPECTED_MD5 d8ceb24e2d42ecb7fe286ede61e36e01)
execute_process(COMMAND ${CMAKE_COMMAND} -E tar x parquetfmt-win64.zip)
elseif(APPLE)
file(DOWNLOAD https://github.com/trink/parquetfmt/releases/download/v0.1.0/parquetfmt-osx.zip ./parquetfmt-osx.zip EXPECTED_MD5 750d88084d3df4fc4f747de7d12a5e27)
execute_process(COMMAND ${CMAKE_COMMAND} -E tar x parquetfmt-osx.zip)
else()
file(DOWNLOAD https://github.com/trink/parquetfmt/releases/download/v0.1.0/parquetfmt-linux.tgz ./parquetfmt-linux.tgz EXPECTED_MD5 a4ecfb3ba777ca698501f5fbe273bb48)
execute_process(COMMAND ${CMAKE_COMMAND} -E tar x parquetfmt-linux.tgz)
endif()
find_program(PARQUETFMT_EXE parquetfmt HINTS ${CMAKE_CURRENT_BINARY_DIR})
endif()
file(GLOB_RECURSE SCHEMA_INCLUDES LIST_DIRECTORIES FALSE "${CMAKE_SOURCE_DIR}/templates/include/*")
foreach(ITEM IN ITEMS ${SCHEMA_INCLUDES})
string(REGEX MATCHALL "/include/([^/]+)/([^/]+)\\.([1-9][0-9]*)\\..+(json|txt)$" PARTS ${ITEM})
if( PARTS )
string(TOUPPER "${CMAKE_MATCH_1}_${CMAKE_MATCH_2}_${CMAKE_MATCH_3}_${CMAKE_MATCH_4}" VARNAME)
set(INCLUDE_TEMPLATES ${INCLUDE_TEMPLATES} ${VARNAME})
file(READ ${ITEM} ${VARNAME})
endif()
endforeach()
foreach(ITEM IN ITEMS ${INCLUDE_TEMPLATES}) # expand the variables in the include templates
while("${${ITEM}}" MATCHES "@.+@")
string(CONFIGURE "${${ITEM}}" ${ITEM} @ONLY)
endwhile()
endforeach()
file(GLOB_RECURSE SCHEMAS LIST_DIRECTORIES FALSE RELATIVE "${CMAKE_SOURCE_DIR}/templates" "${CMAKE_SOURCE_DIR}/templates/*")
foreach(ITEM IN ITEMS ${SCHEMAS})
if(NOT ITEM MATCHES "^(include/|.*/\\.[^/]|.*README\\.md|.*defaults\\.schema\\.json)")
# Consistency in naming of directories/files in this repository is important
# for it to be easily machine-parseable and so that we can consistently
# document how pings are represented at various points in the pipeline,
# so we enforce "kebab-case" naming in this regex.
# The format enforced here must match the recommendations documented in
# https://docs.telemetry.mozilla.org/cookbooks/new_ping.html#choose-a-namespace-and-doctype
string(REGEX MATCH "/([-a-z0-9]+\\.[1-9][0-9]*(\\.schema\\.json|\\.parquetmr\\.txt))$" MATCH ${ITEM})
if(NOT MATCH)
message(FATAL_ERROR "schema name contains invalid characters: " ${ITEM})
endif()
message("generating: schemas/" ${ITEM})
# NOTE the artifacts are written back into the source repository (see the README)
set(IN_FILE "${CMAKE_SOURCE_DIR}/templates/${ITEM}")
set(OUT_FILE "${CMAKE_SOURCE_DIR}/schemas/${ITEM}")
if(ITEM MATCHES "\\.json$") # inject metadata and pretty print
set(TMP_FILE "${OUT_FILE}.tmp")
configure_file(${IN_FILE} ${TMP_FILE} @ONLY)
execute_process(COMMAND ${PYTHON_EXE} ../scripts/inject-metadata ${ITEM} RESULT_VARIABLE INJECT_RV INPUT_FILE ${TMP_FILE} OUTPUT_FILE ${OUT_FILE})
if(NOT INJECT_RV EQUAL 0)
message(FATAL_ERROR "python output error ${INJECT_RV} ${ITEM}")
endif()
file(REMOVE ${TMP_FILE}) # the tmp file is intentionally not removed on failure so it can be examined
elseif(ITEM MATCHES "\\.parquetmr\\.txt$")
set(TMP_FILE "${OUT_FILE}.tmp")
configure_file(${IN_FILE} ${TMP_FILE} @ONLY)
execute_process(COMMAND ${PARQUETFMT_EXE} - RESULT_VARIABLE PFMT_RV INPUT_FILE ${TMP_FILE} OUTPUT_FILE ${OUT_FILE})
if(NOT PFMT_RV EQUAL 0)
message(FATAL_ERROR "parquetfmt output error ${PFMT_RV} ${ITEM}")
endif()
file(REMOVE ${TMP_FILE}) # the tmp file is intentionally not removed on failure so it can be examined
else()
configure_file(${IN_FILE} ${OUT_FILE} @ONLY)
endif()
endif()
endforeach()
include(GNUInstallDirs)
install(DIRECTORY "${CMAKE_SOURCE_DIR}/schemas/" DESTINATION "${CMAKE_INSTALL_DATAROOTDIR}/${PROJECT_NAME}/")
include(CPack)