#!/bin/bash
CURDIR=$(pwd)
PB2DIR=$CURDIR/pb2
OUTDIR=../trezorlib/messages
INDEX=$OUTDIR/__init__.py

mkdir -p $OUTDIR
mkdir -p $PB2DIR
touch $PB2DIR/__init__.py

rm -f $INDEX
echo '# Automatically generated by pb2py' >> $INDEX
echo 'from __future__ import absolute_import' >> $INDEX
echo '' >> $INDEX

for i in types messages storage ; do
    # Compile .proto files to python2 modules using google protobuf library
    cd $CURDIR/../../trezor-common/protob
    protoc --python_out=$PB2DIR -I/usr/include -I. $i.proto
done

# hack to make output python 3 compatible
sed -i 's/^import types_pb2/from . import types_pb2/g' $CURDIR/pb2/messages_pb2.py
sed -i 's/^import types_pb2/from . import types_pb2/g' $CURDIR/pb2/storage_pb2.py

for i in types messages storage ; do
    # Convert google protobuf library to trezor's internal format
    cd $CURDIR
    ./pb2py -p $CURDIR -l $INDEX $i $OUTDIR
done

rm -rf $PB2DIR