mirror of
https://github.com/a2x/cs2-dumper.git
synced 2026-04-17 22:59:59 +08:00
Refactor code writer and fix minor things
Also added the ability to specify which files should be generated based on their file type. E.g. `cs2-dumper.exe -f hpp,json`
This commit is contained in:
a2x
@@ -5,33 +5,113 @@ use heck::{AsPascalCase, AsSnakeCase};
|
||||
|
||||
use serde_json::json;
|
||||
|
||||
use super::{CodeGen, Formatter, Results, SchemaMap};
|
||||
use super::{slugify, CodeWriter, Formatter, SchemaMap};
|
||||
|
||||
use crate::analysis::ClassMetadata;
|
||||
use crate::error::Result;
|
||||
|
||||
impl CodeGen for SchemaMap {
|
||||
fn to_cs(&self, results: &Results, indent_size: usize) -> Result<String> {
|
||||
self.write_content(results, indent_size, |fmt| {
|
||||
fmt.block("namespace CS2Dumper.Schemas", false, |fmt| {
|
||||
impl CodeWriter for SchemaMap {
|
||||
fn write_cs(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
|
||||
fmt.block("namespace CS2Dumper.Schemas", false, |fmt| {
|
||||
for (module_name, (classes, enums)) in self {
|
||||
writeln!(fmt, "// Module: {}", module_name)?;
|
||||
writeln!(fmt, "// Classes count: {}", classes.len())?;
|
||||
writeln!(fmt, "// Enums count: {}", enums.len())?;
|
||||
|
||||
fmt.block(
|
||||
&format!("public static class {}", AsPascalCase(slugify(module_name))),
|
||||
false,
|
||||
|fmt| {
|
||||
for enum_ in enums {
|
||||
let type_name = match enum_.alignment {
|
||||
1 => "byte",
|
||||
2 => "ushort",
|
||||
4 => "uint",
|
||||
8 => "ulong",
|
||||
_ => continue,
|
||||
};
|
||||
|
||||
writeln!(fmt, "// Alignment: {}", enum_.alignment)?;
|
||||
writeln!(fmt, "// Members count: {}", enum_.size)?;
|
||||
|
||||
fmt.block(
|
||||
&format!("public enum {} : {}", slugify(&enum_.name), type_name),
|
||||
false,
|
||||
|fmt| {
|
||||
// TODO: Handle the case where multiple members share
|
||||
// the same value.
|
||||
let members = enum_
|
||||
.members
|
||||
.iter()
|
||||
.map(|member| {
|
||||
format!("{} = {:#X}", member.name, member.value)
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(",\n");
|
||||
|
||||
writeln!(fmt, "{}", members)
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
for class in classes {
|
||||
let parent_name = class
|
||||
.parent
|
||||
.as_ref()
|
||||
.map(|parent| slugify(&parent.name))
|
||||
.unwrap_or_else(|| "None".to_string());
|
||||
|
||||
writeln!(fmt, "// Parent: {}", parent_name)?;
|
||||
writeln!(fmt, "// Fields count: {}", class.fields.len())?;
|
||||
|
||||
write_metadata(fmt, &class.metadata)?;
|
||||
|
||||
fmt.block(
|
||||
&format!("public static class {}", slugify(&class.name)),
|
||||
false,
|
||||
|fmt| {
|
||||
for field in &class.fields {
|
||||
writeln!(
|
||||
fmt,
|
||||
"public const nint {} = {:#X}; // {}",
|
||||
field.name, field.offset, field.type_name
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
})
|
||||
}
|
||||
|
||||
fn write_hpp(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
|
||||
writeln!(fmt, "#pragma once\n")?;
|
||||
writeln!(fmt, "#include <cstddef>\n")?;
|
||||
|
||||
fmt.block("namespace cs2_dumper", false, |fmt| {
|
||||
fmt.block("namespace schemas", false, |fmt| {
|
||||
for (module_name, (classes, enums)) in self {
|
||||
writeln!(fmt, "// Module: {}", module_name)?;
|
||||
writeln!(fmt, "// Classes count: {}", classes.len())?;
|
||||
writeln!(fmt, "// Enums count: {}", enums.len())?;
|
||||
|
||||
fmt.block(
|
||||
&format!(
|
||||
"public static class {}",
|
||||
AsPascalCase(Self::slugify(module_name))
|
||||
),
|
||||
&format!("namespace {}", AsSnakeCase(slugify(module_name))),
|
||||
false,
|
||||
|fmt| {
|
||||
for enum_ in enums {
|
||||
let type_name = match enum_.alignment {
|
||||
1 => "byte",
|
||||
2 => "ushort",
|
||||
4 => "uint",
|
||||
8 => "ulong",
|
||||
1 => "uint8_t",
|
||||
2 => "uint16_t",
|
||||
4 => "uint32_t",
|
||||
8 => "uint64_t",
|
||||
_ => continue,
|
||||
};
|
||||
|
||||
@@ -39,12 +119,8 @@ impl CodeGen for SchemaMap {
|
||||
writeln!(fmt, "// Members count: {}", enum_.size)?;
|
||||
|
||||
fmt.block(
|
||||
&format!(
|
||||
"public enum {} : {}",
|
||||
Self::slugify(&enum_.name),
|
||||
type_name
|
||||
),
|
||||
false,
|
||||
&format!("enum class {} : {}", slugify(&enum_.name), type_name),
|
||||
true,
|
||||
|fmt| {
|
||||
// TODO: Handle the case where multiple members share
|
||||
// the same value.
|
||||
@@ -66,7 +142,7 @@ impl CodeGen for SchemaMap {
|
||||
let parent_name = class
|
||||
.parent
|
||||
.as_ref()
|
||||
.map(|parent| Self::slugify(&parent.name))
|
||||
.map(|parent| slugify(&parent.name))
|
||||
.unwrap_or_else(|| "None".to_string());
|
||||
|
||||
writeln!(fmt, "// Parent: {}", parent_name)?;
|
||||
@@ -75,13 +151,13 @@ impl CodeGen for SchemaMap {
|
||||
write_metadata(fmt, &class.metadata)?;
|
||||
|
||||
fmt.block(
|
||||
&format!("public static class {}", Self::slugify(&class.name)),
|
||||
&format!("namespace {}", slugify(&class.name)),
|
||||
false,
|
||||
|fmt| {
|
||||
for field in &class.fields {
|
||||
writeln!(
|
||||
fmt,
|
||||
"public const nint {} = {:#X}; // {}",
|
||||
"constexpr std::ptrdiff_t {} = {:#X}; // {}",
|
||||
field.name, field.offset, field.type_name
|
||||
)?;
|
||||
}
|
||||
@@ -97,107 +173,11 @@ impl CodeGen for SchemaMap {
|
||||
}
|
||||
|
||||
Ok(())
|
||||
})?;
|
||||
|
||||
Ok(())
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
fn to_hpp(&self, results: &Results, indent_size: usize) -> Result<String> {
|
||||
self.write_content(results, indent_size, |fmt| {
|
||||
writeln!(fmt, "#pragma once\n")?;
|
||||
writeln!(fmt, "#include <cstddef>\n")?;
|
||||
|
||||
fmt.block("namespace cs2_dumper", false, |fmt| {
|
||||
fmt.block("namespace schemas", false, |fmt| {
|
||||
for (module_name, (classes, enums)) in self {
|
||||
writeln!(fmt, "// Module: {}", module_name)?;
|
||||
writeln!(fmt, "// Classes count: {}", classes.len())?;
|
||||
writeln!(fmt, "// Enums count: {}", enums.len())?;
|
||||
|
||||
fmt.block(
|
||||
&format!("namespace {}", AsSnakeCase(Self::slugify(module_name))),
|
||||
false,
|
||||
|fmt| {
|
||||
for enum_ in enums {
|
||||
let type_name = match enum_.alignment {
|
||||
1 => "uint8_t",
|
||||
2 => "uint16_t",
|
||||
4 => "uint32_t",
|
||||
8 => "uint64_t",
|
||||
_ => continue,
|
||||
};
|
||||
|
||||
writeln!(fmt, "// Alignment: {}", enum_.alignment)?;
|
||||
writeln!(fmt, "// Members count: {}", enum_.size)?;
|
||||
|
||||
fmt.block(
|
||||
&format!(
|
||||
"enum class {} : {}",
|
||||
Self::slugify(&enum_.name),
|
||||
type_name
|
||||
),
|
||||
true,
|
||||
|fmt| {
|
||||
// TODO: Handle the case where multiple members share
|
||||
// the same value.
|
||||
let members = enum_
|
||||
.members
|
||||
.iter()
|
||||
.map(|member| {
|
||||
format!("{} = {:#X}", member.name, member.value)
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(",\n");
|
||||
|
||||
writeln!(fmt, "{}", members)
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
for class in classes {
|
||||
let parent_name = class
|
||||
.parent
|
||||
.as_ref()
|
||||
.map(|parent| Self::slugify(&parent.name))
|
||||
.unwrap_or_else(|| "None".to_string());
|
||||
|
||||
writeln!(fmt, "// Parent: {}", parent_name)?;
|
||||
writeln!(fmt, "// Fields count: {}", class.fields.len())?;
|
||||
|
||||
write_metadata(fmt, &class.metadata)?;
|
||||
|
||||
fmt.block(
|
||||
&format!("namespace {}", Self::slugify(&class.name)),
|
||||
false,
|
||||
|fmt| {
|
||||
for field in &class.fields {
|
||||
writeln!(
|
||||
fmt,
|
||||
"constexpr std::ptrdiff_t {} = {:#X}; // {}",
|
||||
field.name, field.offset, field.type_name
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
})
|
||||
})?;
|
||||
|
||||
Ok(())
|
||||
})
|
||||
}
|
||||
|
||||
fn to_json(&self, _results: &Results, _indent_size: usize) -> Result<String> {
|
||||
fn write_json(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
|
||||
let content: BTreeMap<_, _> = self
|
||||
.iter()
|
||||
.map(|(module_name, (classes, enums))| {
|
||||
@@ -231,7 +211,7 @@ impl CodeGen for SchemaMap {
|
||||
.collect();
|
||||
|
||||
(
|
||||
Self::slugify(&class.name),
|
||||
slugify(&class.name),
|
||||
json!({
|
||||
"parent": class.parent.as_ref().map(|parent| &parent.name),
|
||||
"fields": fields,
|
||||
@@ -259,7 +239,7 @@ impl CodeGen for SchemaMap {
|
||||
};
|
||||
|
||||
(
|
||||
Self::slugify(&enum_.name),
|
||||
slugify(&enum_.name),
|
||||
json!({
|
||||
"alignment": enum_.alignment,
|
||||
"type": type_name,
|
||||
@@ -279,110 +259,106 @@ impl CodeGen for SchemaMap {
|
||||
})
|
||||
.collect();
|
||||
|
||||
serde_json::to_string_pretty(&content).map_err(Into::into)
|
||||
fmt.write_str(&serde_json::to_string_pretty(&content).expect("failed to serialize"))
|
||||
}
|
||||
|
||||
fn to_rs(&self, results: &Results, indent_size: usize) -> Result<String> {
|
||||
self.write_content(results, indent_size, |fmt| {
|
||||
writeln!(
|
||||
fmt,
|
||||
"#![allow(non_upper_case_globals, non_camel_case_types, non_snake_case, unused)]\n"
|
||||
)?;
|
||||
fn write_rs(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
|
||||
writeln!(
|
||||
fmt,
|
||||
"#![allow(non_upper_case_globals, non_camel_case_types, non_snake_case, unused)]\n"
|
||||
)?;
|
||||
|
||||
fmt.block("pub mod cs2_dumper", false, |fmt| {
|
||||
fmt.block("pub mod schemas", false, |fmt| {
|
||||
for (module_name, (classes, enums)) in self {
|
||||
writeln!(fmt, "// Module: {}", module_name)?;
|
||||
writeln!(fmt, "// Classes count: {}", classes.len())?;
|
||||
writeln!(fmt, "// Enums count: {}", enums.len())?;
|
||||
fmt.block("pub mod cs2_dumper", false, |fmt| {
|
||||
fmt.block("pub mod schemas", false, |fmt| {
|
||||
for (module_name, (classes, enums)) in self {
|
||||
writeln!(fmt, "// Module: {}", module_name)?;
|
||||
writeln!(fmt, "// Classes count: {}", classes.len())?;
|
||||
writeln!(fmt, "// Enums count: {}", enums.len())?;
|
||||
|
||||
fmt.block(
|
||||
&format!("pub mod {}", AsSnakeCase(Self::slugify(module_name))),
|
||||
false,
|
||||
|fmt| {
|
||||
for enum_ in enums {
|
||||
let type_name = match enum_.alignment {
|
||||
1 => "u8",
|
||||
2 => "u16",
|
||||
4 => "u32",
|
||||
8 => "u64",
|
||||
_ => continue,
|
||||
};
|
||||
fmt.block(
|
||||
&format!("pub mod {}", AsSnakeCase(slugify(module_name))),
|
||||
false,
|
||||
|fmt| {
|
||||
for enum_ in enums {
|
||||
let type_name = match enum_.alignment {
|
||||
1 => "u8",
|
||||
2 => "u16",
|
||||
4 => "u32",
|
||||
8 => "u64",
|
||||
_ => continue,
|
||||
};
|
||||
|
||||
writeln!(fmt, "// Alignment: {}", enum_.alignment)?;
|
||||
writeln!(fmt, "// Members count: {}", enum_.size)?;
|
||||
writeln!(fmt, "// Alignment: {}", enum_.alignment)?;
|
||||
writeln!(fmt, "// Members count: {}", enum_.size)?;
|
||||
|
||||
fmt.block(
|
||||
&format!(
|
||||
"#[repr({})]\npub enum {}",
|
||||
type_name,
|
||||
Self::slugify(&enum_.name),
|
||||
),
|
||||
false,
|
||||
|fmt| {
|
||||
// TODO: Handle the case where multiple members share
|
||||
// the same value.
|
||||
let members = enum_
|
||||
.members
|
||||
.iter()
|
||||
.map(|member| {
|
||||
format!(
|
||||
"{} = {}",
|
||||
member.name,
|
||||
if member.value == -1 {
|
||||
format!("{}::MAX", type_name)
|
||||
} else {
|
||||
format!("{:#X}", member.value)
|
||||
}
|
||||
)
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(",\n");
|
||||
fmt.block(
|
||||
&format!(
|
||||
"#[repr({})]\npub enum {}",
|
||||
type_name,
|
||||
slugify(&enum_.name),
|
||||
),
|
||||
false,
|
||||
|fmt| {
|
||||
// TODO: Handle the case where multiple members share
|
||||
// the same value.
|
||||
let members = enum_
|
||||
.members
|
||||
.iter()
|
||||
.map(|member| {
|
||||
format!(
|
||||
"{} = {}",
|
||||
member.name,
|
||||
if member.value == -1 {
|
||||
format!("{}::MAX", type_name)
|
||||
} else {
|
||||
format!("{:#X}", member.value)
|
||||
}
|
||||
)
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(",\n");
|
||||
|
||||
writeln!(fmt, "{}", members)
|
||||
},
|
||||
)?;
|
||||
}
|
||||
writeln!(fmt, "{}", members)
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
for class in classes {
|
||||
let parent_name = class
|
||||
.parent
|
||||
.as_ref()
|
||||
.map(|parent| Self::slugify(&parent.name))
|
||||
.unwrap_or_else(|| "None".to_string());
|
||||
for class in classes {
|
||||
let parent_name = class
|
||||
.parent
|
||||
.as_ref()
|
||||
.map(|parent| slugify(&parent.name))
|
||||
.unwrap_or_else(|| "None".to_string());
|
||||
|
||||
writeln!(fmt, "// Parent: {}", parent_name)?;
|
||||
writeln!(fmt, "// Fields count: {}", class.fields.len())?;
|
||||
writeln!(fmt, "// Parent: {}", parent_name)?;
|
||||
writeln!(fmt, "// Fields count: {}", class.fields.len())?;
|
||||
|
||||
write_metadata(fmt, &class.metadata)?;
|
||||
write_metadata(fmt, &class.metadata)?;
|
||||
|
||||
fmt.block(
|
||||
&format!("pub mod {}", Self::slugify(&class.name)),
|
||||
false,
|
||||
|fmt| {
|
||||
for field in &class.fields {
|
||||
writeln!(
|
||||
fmt,
|
||||
"pub const {}: usize = {:#X}; // {}",
|
||||
field.name, field.offset, field.type_name
|
||||
)?;
|
||||
}
|
||||
fmt.block(
|
||||
&format!("pub mod {}", slugify(&class.name)),
|
||||
false,
|
||||
|fmt| {
|
||||
for field in &class.fields {
|
||||
writeln!(
|
||||
fmt,
|
||||
"pub const {}: usize = {:#X}; // {}",
|
||||
field.name, field.offset, field.type_name
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
Ok(())
|
||||
},
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
})
|
||||
})?;
|
||||
|
||||
Ok(())
|
||||
Ok(())
|
||||
})
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user