mirror of
https://github.com/vicliu624/trail-mate.git
synced 2026-07-02 01:51:42 +00:00
liu weikai
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Generate nanopb C headers from Meshtastic .proto files
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
def find_nanopb_generator():
|
|
"""Find nanopb_generator.py in the nanopb library"""
|
|
# Try to find in PlatformIO libs
|
|
lib_paths = [
|
|
os.path.join(os.getcwd(), ".pio", "libdeps", "tlora_pager_sx1262", "nanopb", "generator", "nanopb_generator.py"),
|
|
os.path.join(os.getcwd(), ".pio", "libdeps", "tlora_pager_sx1262", "nanopb", "generator", "nanopb_generator.py"),
|
|
]
|
|
|
|
for path in lib_paths:
|
|
if os.path.exists(path):
|
|
return path
|
|
|
|
# Try system nanopb
|
|
try:
|
|
result = subprocess.run(["which", "nanopb_generator.py"], capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
return result.stdout.strip()
|
|
except:
|
|
pass
|
|
|
|
return None
|
|
|
|
def generate_protobuf():
|
|
"""Generate C headers from .proto files"""
|
|
proto_dir = os.path.join(os.getcwd(), "lib", "meshtastic_protobufs")
|
|
output_dir = os.path.join(os.getcwd(), "src", "chat", "infra", "meshtastic", "generated")
|
|
|
|
# Create output directory
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
|
|
# Find nanopb generator
|
|
generator = find_nanopb_generator()
|
|
if not generator:
|
|
print("ERROR: nanopb_generator.py not found!")
|
|
print("Please install nanopb library first: pio lib install nanopb")
|
|
return False
|
|
|
|
# Generate for each .proto file
|
|
proto_files = [
|
|
"meshtastic/portnums.proto",
|
|
"meshtastic/channel.proto",
|
|
"meshtastic/mesh.proto",
|
|
]
|
|
|
|
for proto_file in proto_files:
|
|
proto_path = os.path.join(proto_dir, proto_file)
|
|
if not os.path.exists(proto_path):
|
|
print(f"WARNING: {proto_path} not found, skipping")
|
|
continue
|
|
|
|
print(f"Generating from {proto_file}...")
|
|
cmd = [
|
|
sys.executable,
|
|
generator,
|
|
"-I", proto_dir,
|
|
"-D", output_dir,
|
|
proto_path
|
|
]
|
|
|
|
try:
|
|
result = subprocess.run(cmd, capture_output=True, text=True, cwd=proto_dir)
|
|
if result.returncode != 0:
|
|
print(f"ERROR generating {proto_file}:")
|
|
print(result.stderr)
|
|
return False
|
|
except Exception as e:
|
|
print(f"ERROR: {e}")
|
|
return False
|
|
|
|
print("Protobuf generation complete!")
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
if not generate_protobuf():
|
|
sys.exit(1)
|