Initial work on Protocol generation

This commit is contained in:
Joshua Ashton 2020-08-01 22:53:12 +01:00
parent 70d41ab577
commit 2d64680804
2 changed files with 121 additions and 0 deletions

View File

@ -0,0 +1,75 @@
import hjson
text = ''
def add_text(fmt, *args):
global text
text += (fmt + '\n').format(*args)
def resolve_type(aliases, type):
if type in aliases:
return aliases[type]
return type
def print_states(states):
add_text(' enum class ProtocolState : int32_t')
add_text(' {{')
for state, value in states.items():
add_text(' {} = {},', state, value)
add_text(' }};')
def get_rw_func(primitiveType, aliasedType, read):
prefix = 'Read' if read else 'Write'
if aliasedType == 'varint':
return '{}VarInt<{}>'.format(prefix, primitiveType)
elif aliasedType == 'string':
return '{}String'.format(prefix)
else:
return 'Read<{}>'.format(aliasedType)
def print_messages(list, aliases, primitives, serverbound):
global text
for message_name, message in list.items():
add_text(' struct {}Message', message_name)
add_text(' {{')
add_text(' static constexpr int32_t PacketId = {};', message['id'])
add_text(' static constexpr bool ServerBound = {};', 'true' if serverbound else 'false')
add_text(' static constexpr ProtocolState PacketState = ProtocolState::{};', message['state'])
add_text('')
if serverbound:
add_text(' {}Message(PacketReader& reader)', message_name)
add_text(' {{')
for name, type in message['vars'].items():
add_text(' {} = reader.{}();', name, get_rw_func(resolve_type(primitives, type), resolve_type(aliases, type), True))
add_text(' }}')
add_text('')
for name, type in message['vars'].items():
add_text(' {} {};', resolve_type(primitives, type), name)
add_text(' }};')
def print_protocol():
with open('protocol.hjson') as message_file:
message_scheme = hjson.load(message_file)
print_states(message_scheme['states'])
add_text('')
print_messages(
message_scheme['messages']['serverbound'],
message_scheme['types']['aliases'],
message_scheme['types']['primitives'],
True)
add_text('#pragma once')
add_text('')
add_text('#include <cstdint>')
add_text('')
add_text('namespace Feather::Messages')
add_text('{{')
print_protocol()
add_text('}}')
print(text)

View File

@ -0,0 +1,46 @@
{
types :
{
aliases :
{
ProtocolState : varint
}
primitives :
{
varint : int32_t
}
}
states :
{
Handholding : -1
Play : 0
Status : 1
Login : 2
}
messages :
{
serverbound :
{
Handshake :
{
id : 0
state : Handholding
vars :
{
protocolVersion : varint
serverIP : string
port : uint16_t
intention : ProtocolState
}
}
}
clientbound :
{
}
}
}