diff --git a/packager/third_party/protobuf/CMakeLists.txt b/packager/third_party/protobuf/CMakeLists.txt index 0e7e4d511b..467103fca2 100644 --- a/packager/third_party/protobuf/CMakeLists.txt +++ b/packager/third_party/protobuf/CMakeLists.txt @@ -30,3 +30,6 @@ add_definitions(-DNDEBUG) # With these set in scope of this folder, load the library's own CMakeLists.txt. add_subdirectory(source) + +# Build generated python sources for the protobuf library. +include("protobuf_py.cmake") diff --git a/packager/third_party/protobuf/protobuf_py.cmake b/packager/third_party/protobuf/protobuf_py.cmake new file mode 100644 index 0000000000..488aab4da9 --- /dev/null +++ b/packager/third_party/protobuf/protobuf_py.cmake @@ -0,0 +1,63 @@ +# Copyright 2022 Google LLC. All rights reserved. +# +# Use of this source code is governed by a BSD-style +# license that can be found in the LICENSE file or at +# https://developers.google.com/open-source/licenses/bsd + +# Build generated python sources for the protobuf library. +set(protobuf_py_proto_path ${CMAKE_CURRENT_SOURCE_DIR}/source/src) +set(protobuf_py_library_path ${CMAKE_CURRENT_SOURCE_DIR}/source/python) +set(protobuf_py_output_path ${CMAKE_CURRENT_BINARY_DIR}/py) + +# Copy the library without the generated sources. +add_custom_command( + DEPENDS + ${protobuf_py_proto_path} + OUTPUT + ${protobuf_py_output_path}/google + COMMAND + ${CMAKE_COMMAND} -E copy_directory + ${CMAKE_CURRENT_SOURCE_DIR}/source/python/google + ${protobuf_py_output_path}/google) + +# This list was taken from packager/third_party/protobuf/source/python/setup.py +set(protobuf_py_filenames + google/protobuf/descriptor + google/protobuf/compiler/plugin + google/protobuf/any + google/protobuf/api + google/protobuf/duration + google/protobuf/empty + google/protobuf/field_mask + google/protobuf/source_context + google/protobuf/struct + google/protobuf/timestamp + google/protobuf/type + google/protobuf/wrappers) + +# For each proto in the list, generate a _pb2.py file to add to the library. +set(protobuf_py_outputs "") +foreach (filename ${protobuf_py_filenames}) + set(proto_path "${protobuf_py_proto_path}/${filename}.proto") + set(output_path "${protobuf_py_output_path}/${filename}_pb2.py") + + list(APPEND protobuf_py_outputs "${output_path}") + + add_custom_command( + DEPENDS + protoc + ${proto_path} + ${protobuf_py_output_path}/google + OUTPUT + ${output_path} + COMMAND + protoc + ARGS + -I${protobuf_py_proto_path} + --python_out=${protobuf_py_output_path} + ${proto_path}) +endforeach () + +# The entire python protobuf library (repo source and generated) to the output +# folder. +add_custom_target(protobuf_py ALL DEPENDS ${protobuf_py_outputs}) diff --git a/packager/tools/pssh/CMakeLists.txt b/packager/tools/pssh/CMakeLists.txt index 0b981cec29..9ec3a39d26 100644 --- a/packager/tools/pssh/CMakeLists.txt +++ b/packager/tools/pssh/CMakeLists.txt @@ -14,6 +14,7 @@ set(PSSH_BOX_OUTPUTS add_custom_command( DEPENDS + protobuf_py widevine_protos pssh-box.py OUTPUT ${PSSH_BOX_OUTPUTS} @@ -25,6 +26,10 @@ add_custom_command( ${CMAKE_BINARY_DIR}/packager/media/base/widevine_common_encryption_pb2.py ${CMAKE_BINARY_DIR}/packager/media/base/widevine_pssh_data_pb2.py ${CMAKE_BINARY_DIR}/packager/pssh-box-protos/ + COMMAND + ${CMAKE_COMMAND} -E copy_directory + ${CMAKE_BINARY_DIR}/packager/third_party/protobuf/py/google + ${CMAKE_BINARY_DIR}/packager/pssh-box-protos/google COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/pssh-box.py diff --git a/packager/tools/pssh/pssh-box.py b/packager/tools/pssh/pssh-box.py index 2fc211f7a9..b7a9aba61e 100755 --- a/packager/tools/pssh/pssh-box.py +++ b/packager/tools/pssh/pssh-box.py @@ -1,4 +1,4 @@ -#!/usr/bin/python +#!/usr/bin/python3 # Copyright 2016 Google LLC. All rights reserved. # # Use of this source code is governed by a BSD-style @@ -11,8 +11,6 @@ # the filename: pssh-box.py # pylint: disable=invalid-name -from __future__ import print_function - import argparse import base64 import itertools