Changes for duimper

This commit is contained in:
xavo95 2024-09-24 20:59:18 +02:00
parent d93c61cc92
commit 7d9d502a22
Signed by: xavo95
GPG key ID: CBF8ADED6DEBB783
3 changed files with 67 additions and 4 deletions

4
Cargo.lock generated
View file

@ -1745,10 +1745,14 @@ version = "0.1.0"
dependencies = [
"byteorder",
"crc32fast",
"prettyplease",
"prost",
"prost-build",
"quote",
"serde",
"serde_json",
"shorekeeper-protocol-derive",
"syn",
"thiserror",
]

View file

@ -8,8 +8,12 @@ byteorder.workspace = true
crc32fast.workspace = true
thiserror.workspace = true
prost.workspace = true
serde.workspace = true
serde_json.workspace = true
shorekeeper-protocol-derive.workspace = true
[build-dependencies]
prost-build.workspace = true
quote = "1.0.35"
syn = "2.0.76"
prettyplease = "0.2.22"

View file

@ -3,8 +3,9 @@ use std::{
io::{self, BufRead},
path::Path,
};
use std::io::Write;
use quote::quote;
use quote::{format_ident, quote};
const CODEGEN_OUT_DIR: &str = "generated/";
@ -26,9 +27,11 @@ pub fn main() {
prost_build::Config::new()
.out_dir(CODEGEN_OUT_DIR)
.type_attribute(".", "#[derive(shorekeeper_protocol_derive::MessageID)]")
.type_attribute(".", "#[derive(serde::Serialize,serde::Deserialize)]")
.compile_protos(&[proto_file], &["shorekeeper"])
.unwrap();
impl_dumper(Path::new("generated/shorekeeper.rs")).unwrap();
impl_message_id(Path::new("generated/shorekeeper.rs")).unwrap();
}
@ -92,10 +95,11 @@ pub fn impl_proto_config(csv_path: &Path, codegen_path: &Path) -> io::Result<()>
}
}
}
}
.to_string();
};
fs::write(codegen_path, generated_code.as_bytes())?;
let syntax_tree = syn::parse2(generated_code).unwrap();
let formatted = prettyplease::unparse(&syntax_tree);
fs::write(codegen_path, formatted.as_bytes())?;
Ok(())
}
@ -126,3 +130,54 @@ fn make_message_id_attr(line: &str) -> Option<String> {
let id = line.trim_start().split(' ').nth(2)?.parse::<u16>().ok()?;
Some(format!("#[message_id({id})]"))
}
pub fn impl_dumper(codegen_path: &Path) -> io::Result<()> {
let file = fs::File::open(codegen_path)?;
let reader = io::BufReader::new(file);
let mut match_arms = quote! {};
let mut id = None;
for line in reader.lines() {
let line = line?;
if line.contains("MessageId:") {
id = Some(
line.trim_start().split(' ').nth(2).unwrap().parse::<u16>().ok().unwrap()
);
} else if line.contains("pub struct") {
if let Some(id) = id.take() {
let name = format_ident!(
"{}", line.trim_start().split(' ').nth(2).unwrap().to_string()
);
match_arms = quote! {
#match_arms
#id => serde_json::to_string(&#name::decode(data).unwrap()),
};
}
}
}
let generated_code = quote! {
pub mod proto_dumper {
use prost::Message;
use crate::*;
use crate::ai::*;
use crate::combat_message::*;
use crate::debug::*;
use crate::summon::*;
pub fn get_debug_info(id: u16, data: &[u8]) -> String {
match id {
#match_arms
_ => Ok("".to_string()),
}.unwrap()
}
}
};
let syntax_tree = syn::parse2(generated_code).unwrap();
let formatted = prettyplease::unparse(&syntax_tree);
let mut file = fs::OpenOptions::new().append(true).open(codegen_path)?;
file.write(formatted.as_bytes())?;
Ok(())
}