forked from jshiffer/matterbridge
70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
with open("old-types.txt") as f:
|
||
|
|
old_types = [line.rstrip("\n") for line in f]
|
||
|
|
with open("old-enums.txt") as f:
|
||
|
|
old_enums = [line.rstrip("\n") for line in f]
|
||
|
|
|
||
|
|
os.chdir("../../proto")
|
||
|
|
|
||
|
|
new_protos = {}
|
||
|
|
for dir in os.listdir("."):
|
||
|
|
if not dir.startswith("wa"):
|
||
|
|
continue
|
||
|
|
for file in os.listdir(dir):
|
||
|
|
if file.endswith(".pb.go"):
|
||
|
|
with open(f"{dir}/{file}") as f:
|
||
|
|
new_protos[dir] = f.read()
|
||
|
|
break
|
||
|
|
|
||
|
|
match_type_map = {
|
||
|
|
"HandshakeServerHello": "HandshakeMessage_ServerHello",
|
||
|
|
"HandshakeClientHello": "HandshakeMessage_ClientHello",
|
||
|
|
"HandshakeClientFinish": "HandshakeMessage_ClientFinish",
|
||
|
|
"InteractiveMessage_Header_JpegThumbnail": "InteractiveMessage_Header_JPEGThumbnail",
|
||
|
|
}
|
||
|
|
|
||
|
|
print("// DO NOT MODIFY: Generated by generatelegacy.sh")
|
||
|
|
print()
|
||
|
|
print("package proto")
|
||
|
|
print()
|
||
|
|
print("import (")
|
||
|
|
for proto in new_protos.keys():
|
||
|
|
print(f'\t"go.mau.fi/whatsmeow/proto/{proto}"')
|
||
|
|
print(")")
|
||
|
|
print()
|
||
|
|
|
||
|
|
print("// Deprecated: use new packages directly")
|
||
|
|
print("type (")
|
||
|
|
for type in old_types:
|
||
|
|
match_type = match_type_map.get(type, type)
|
||
|
|
for mod, proto in new_protos.items():
|
||
|
|
if f"type {match_type} " in proto:
|
||
|
|
print(f"\t{type} = {mod}.{match_type}")
|
||
|
|
break
|
||
|
|
elif f"type ContextInfo_{match_type} " in proto:
|
||
|
|
print(f"\t{type} = {mod}.ContextInfo_{match_type}")
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
print(f"{type} not found")
|
||
|
|
sys.exit(1)
|
||
|
|
print(")")
|
||
|
|
print()
|
||
|
|
|
||
|
|
print("// Deprecated: use new packages directly")
|
||
|
|
print("const (")
|
||
|
|
for type in old_enums:
|
||
|
|
for mod, proto in new_protos.items():
|
||
|
|
if f"\t{type} " in proto:
|
||
|
|
print(f"\t{type} = {mod}.{type}")
|
||
|
|
break
|
||
|
|
elif f"\tContextInfo_{type} " in proto:
|
||
|
|
print(f"\t{type} = {mod}.ContextInfo_{type}")
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
print(f"{type} not found")
|
||
|
|
sys.exit(1)
|
||
|
|
print(")")
|