2019-09-12 13:42:04 +00:00
|
|
|
#!/usr/bin/env bash
|
2021-11-19 13:09:40 +00:00
|
|
|
set -e
|
2019-05-16 11:28:20 +00:00
|
|
|
cd $(dirname $0)/..
|
|
|
|
|
|
|
|
PROTOB=common/protob
|
|
|
|
|
2022-03-01 16:08:26 +00:00
|
|
|
# Bootloader messages cannot end up in core
|
|
|
|
CORE_PROTOBUF_SOURCES=$(ls $PROTOB/*.proto | grep -v "bootloader")
|
2019-05-16 11:28:20 +00:00
|
|
|
|
2022-03-01 16:08:26 +00:00
|
|
|
# Taking all files for python
|
2019-05-16 11:28:20 +00:00
|
|
|
PYTHON_PROTOBUF_SOURCES=$PROTOB/*.proto
|
|
|
|
|
|
|
|
RETURN=0
|
|
|
|
|
|
|
|
do_rebuild() {
|
2021-03-23 12:31:34 +00:00
|
|
|
local FILE_OR_DIR="$1"
|
2019-05-16 11:28:20 +00:00
|
|
|
shift
|
2021-03-23 12:31:34 +00:00
|
|
|
local OUTPUT="$1"
|
2019-07-01 13:52:36 +00:00
|
|
|
shift
|
2021-03-23 12:31:34 +00:00
|
|
|
local SOURCES="$1"
|
2019-08-26 12:50:11 +00:00
|
|
|
shift
|
2019-05-16 11:28:20 +00:00
|
|
|
|
2021-03-23 12:31:34 +00:00
|
|
|
if [ "$FILE_OR_DIR" == file ]; then
|
|
|
|
local param="--outfile"
|
|
|
|
else
|
|
|
|
local param="--python-outdir"
|
2019-08-26 12:50:11 +00:00
|
|
|
fi
|
|
|
|
|
2021-03-23 12:31:34 +00:00
|
|
|
# note $SOURCES is unquoted - we want wildcard expansion and multiple args
|
|
|
|
$PROTOB/pb2py "$@" $param="$OUTPUT" $SOURCES
|
2019-05-16 11:28:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
do_check() {
|
|
|
|
# rebuild protobuf in tmpdir and check result against specified directory
|
|
|
|
local TMPDIR=$(mktemp -d proto-check.XXXXXX)
|
2021-11-19 13:09:40 +00:00
|
|
|
trap "rm -r $TMPDIR" RETURN
|
2021-03-23 12:31:34 +00:00
|
|
|
|
|
|
|
local FILE_OR_DIR="$1"
|
|
|
|
shift
|
|
|
|
local OUTPUT="$1"
|
2019-05-16 11:28:20 +00:00
|
|
|
shift
|
|
|
|
|
2021-03-23 12:31:34 +00:00
|
|
|
if [ "$FILE_OR_DIR" == file ]; then
|
|
|
|
local TMPDEST="$TMPDIR/testfile"
|
|
|
|
else
|
|
|
|
cp -rT "$OUTPUT" "$TMPDIR"
|
|
|
|
local TMPDEST="$TMPDIR"
|
|
|
|
fi
|
|
|
|
|
|
|
|
do_rebuild "$FILE_OR_DIR" "$TMPDEST" "$@"
|
2021-11-19 13:09:40 +00:00
|
|
|
if ! diff -ur --exclude __pycache__ "$OUTPUT" "$TMPDEST"; then
|
2019-05-16 11:28:20 +00:00
|
|
|
RETURN=1
|
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
|
|
|
if [ "$1" == "--check" ]; then
|
|
|
|
func=do_check
|
|
|
|
else
|
|
|
|
func=do_rebuild
|
|
|
|
fi
|
|
|
|
|
2021-03-23 12:31:34 +00:00
|
|
|
$func dir core/src/trezor/enums "$CORE_PROTOBUF_SOURCES"
|
|
|
|
$func file core/src/trezor/enums/__init__.py "$CORE_PROTOBUF_SOURCES" --template=core/src/trezor/enums/_proto_init.mako
|
|
|
|
$func file core/src/trezor/messages.py "$CORE_PROTOBUF_SOURCES" --template=core/src/trezor/_proto_messages.mako
|
|
|
|
|
|
|
|
$func file python/src/trezorlib/messages.py "$PYTHON_PROTOBUF_SOURCES" \
|
|
|
|
--template=python/src/trezorlib/_proto_messages.mako \
|
|
|
|
--include-deprecated
|
2019-05-16 11:28:20 +00:00
|
|
|
|
|
|
|
exit $RETURN
|