feat!: Refactor, Update to version 2.5.x, Add Battle Scepter

proto/build.rs

@@ -1,3 +1,9 @@
+use std::{
+    fs,
+    io::{self, BufRead},
+    path::Path,
+};
+
 pub fn main() {
     let proto_file = "StarRail.proto";
     if std::path::Path::new(proto_file).exists() {
@@ -5,7 +11,38 @@ pub fn main() {
 
         prost_build::Config::new()
             .out_dir("out/")
+            .type_attribute(".", "#[derive(proto_derive::CmdID)]")
             .compile_protos(&[proto_file], &["."])
             .unwrap();
+
+        impl_message_id(Path::new("out/_.rs")).unwrap();
     }
 }
+
+pub fn impl_message_id(path: &Path) -> io::Result<()> {
+    let file = fs::File::open(path)?;
+    let reader = io::BufReader::new(file);
+    let mut output = Vec::new();
+
+    let mut attr = None;
+    for line in reader.lines() {
+        let line = line?;
+
+        if line.contains("CmdID:") {
+            attr = Some(make_message_id_attr(&line).unwrap());
+        } else {
+            output.push(line);
+            if let Some(attr) = attr.take() {
+                output.push(attr);
+            }
+        }
+    }
+
+    fs::write(path, output.join("\n").as_bytes())?;
+    Ok(())
+}
+
+fn make_message_id_attr(line: &str) -> Option<String> {
+    let id = line.trim_start().split(' ').nth(2)?.parse::<u16>().ok()?;
+    Some(format!("#[cmdid({id})]"))
+}