mirror of
https://github.com/lifegpc/msg-tool.git
synced 2026-06-06 04:48:54 +08:00
Add export support
This commit is contained in:
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
/target
|
||||||
|
/testscripts
|
||||||
|
/output
|
||||||
327
Cargo.lock
generated
Normal file
327
Cargo.lock
generated
Normal file
@@ -0,0 +1,327 @@
|
|||||||
|
# This file is automatically @generated by Cargo.
|
||||||
|
# It is not intended for manual editing.
|
||||||
|
version = 4
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anstream"
|
||||||
|
version = "0.6.18"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8acc5369981196006228e28809f761875c0327210a891e941f4c683b3a99529b"
|
||||||
|
dependencies = [
|
||||||
|
"anstyle",
|
||||||
|
"anstyle-parse",
|
||||||
|
"anstyle-query",
|
||||||
|
"anstyle-wincon",
|
||||||
|
"colorchoice",
|
||||||
|
"is_terminal_polyfill",
|
||||||
|
"utf8parse",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anstyle"
|
||||||
|
version = "1.0.10"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "55cc3b69f167a1ef2e161439aa98aed94e6028e5f9a59be9a6ffb47aef1651f9"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anstyle-parse"
|
||||||
|
version = "0.2.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "3b2d16507662817a6a20a9ea92df6652ee4f94f914589377d69f3b21bc5798a9"
|
||||||
|
dependencies = [
|
||||||
|
"utf8parse",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anstyle-query"
|
||||||
|
version = "1.1.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "79947af37f4177cfead1110013d678905c37501914fba0efea834c3fe9a8d60c"
|
||||||
|
dependencies = [
|
||||||
|
"windows-sys",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anstyle-wincon"
|
||||||
|
version = "3.0.7"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ca3534e77181a9cc07539ad51f2141fe32f6c3ffd4df76db8ad92346b003ae4e"
|
||||||
|
dependencies = [
|
||||||
|
"anstyle",
|
||||||
|
"once_cell",
|
||||||
|
"windows-sys",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anyhow"
|
||||||
|
version = "1.0.98"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e16d2d3311acee920a9eb8d33b8cbc1787ce4a264e85f964c2404b969bdcd487"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "cfg-if"
|
||||||
|
version = "1.0.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "clap"
|
||||||
|
version = "4.5.38"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ed93b9805f8ba930df42c2590f05453d5ec36cbb85d018868a5b24d31f6ac000"
|
||||||
|
dependencies = [
|
||||||
|
"clap_builder",
|
||||||
|
"clap_derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "clap_builder"
|
||||||
|
version = "4.5.38"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "379026ff283facf611b0ea629334361c4211d1b12ee01024eec1591133b04120"
|
||||||
|
dependencies = [
|
||||||
|
"anstream",
|
||||||
|
"anstyle",
|
||||||
|
"clap_lex",
|
||||||
|
"strsim",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "clap_derive"
|
||||||
|
version = "4.5.32"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "09176aae279615badda0765c0c0b3f6ed53f4709118af73cf4655d85d1530cd7"
|
||||||
|
dependencies = [
|
||||||
|
"heck",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "clap_lex"
|
||||||
|
version = "0.7.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "f46ad14479a25103f283c0f10005961cf086d8dc42205bb44c46ac563475dca6"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "colorchoice"
|
||||||
|
version = "1.0.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5b63caa9aa9397e2d9480a9b13673856c78d8ac123288526c37d7839f2a86990"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "encoding_rs"
|
||||||
|
version = "0.8.35"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "75030f3c4f45dafd7586dd6780965a8c7e8e285a5ecb86713e63a79c5b2766f3"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "heck"
|
||||||
|
version = "0.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "is_terminal_polyfill"
|
||||||
|
version = "1.70.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "itoa"
|
||||||
|
version = "1.0.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "4a5f13b858c8d314ee3e8f639011f7ccefe71f97f96e50151fb991f267928e2c"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lazy_static"
|
||||||
|
version = "1.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "bbd2bcb4c963f2ddae06a2efc7e9f3591312473c50c6685e1f298068316e66fe"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "memchr"
|
||||||
|
version = "2.7.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "msg_tool"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"anyhow",
|
||||||
|
"clap",
|
||||||
|
"encoding_rs",
|
||||||
|
"lazy_static",
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"windows-sys",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "once_cell"
|
||||||
|
version = "1.21.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro2"
|
||||||
|
version = "1.0.95"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "02b3e5e68a3a1a02aad3ec490a98007cbc13c37cbe84a3cd7b8e406d76e7f778"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "quote"
|
||||||
|
version = "1.0.40"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "ryu"
|
||||||
|
version = "1.0.20"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "28d3b2b1366ec20994f1fd18c3c594f05c5dd4bc44d8bb0c1c632c8d6829481f"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde"
|
||||||
|
version = "1.0.219"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5f0e2c6ed6606019b4e29e69dbaba95b11854410e5347d525002456dbbb786b6"
|
||||||
|
dependencies = [
|
||||||
|
"serde_derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde_derive"
|
||||||
|
version = "1.0.219"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5b0276cf7f2c73365f7157c8123c21cd9a50fbbd844757af28ca1f5925fc2a00"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde_json"
|
||||||
|
version = "1.0.140"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "20068b6e96dc6c9bd23e01df8827e6c7e1f2fddd43c21810382803c136b99373"
|
||||||
|
dependencies = [
|
||||||
|
"itoa",
|
||||||
|
"memchr",
|
||||||
|
"ryu",
|
||||||
|
"serde",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "strsim"
|
||||||
|
version = "0.11.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "2.0.101"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8ce2b7fc941b3a24138a0a7cf8e858bfc6a992e7978a068a5c760deb0ed43caf"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-ident"
|
||||||
|
version = "1.0.18"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "utf8parse"
|
||||||
|
version = "0.2.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows-sys"
|
||||||
|
version = "0.59.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b"
|
||||||
|
dependencies = [
|
||||||
|
"windows-targets",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows-targets"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973"
|
||||||
|
dependencies = [
|
||||||
|
"windows_aarch64_gnullvm",
|
||||||
|
"windows_aarch64_msvc",
|
||||||
|
"windows_i686_gnu",
|
||||||
|
"windows_i686_gnullvm",
|
||||||
|
"windows_i686_msvc",
|
||||||
|
"windows_x86_64_gnu",
|
||||||
|
"windows_x86_64_gnullvm",
|
||||||
|
"windows_x86_64_msvc",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_aarch64_gnullvm"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_aarch64_msvc"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_i686_gnu"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_i686_gnullvm"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_i686_msvc"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_x86_64_gnu"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_x86_64_gnullvm"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_x86_64_msvc"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec"
|
||||||
15
Cargo.toml
Normal file
15
Cargo.toml
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
[package]
|
||||||
|
name = "msg_tool"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
anyhow = "1"
|
||||||
|
clap = { version = "4.5", features = ["derive"] }
|
||||||
|
encoding_rs = "0.8"
|
||||||
|
lazy_static = "1.5.0"
|
||||||
|
serde = { version = "1", features = ["derive"] }
|
||||||
|
serde_json = "1.0.140"
|
||||||
|
|
||||||
|
[target.'cfg(windows)'.dependencies]
|
||||||
|
windows-sys = { version = "0", features = ["Win32_Globalization"] }
|
||||||
69
src/args.rs
Normal file
69
src/args.rs
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
use crate::types::*;
|
||||||
|
use clap::{ArgAction, ArgGroup, Parser, Subcommand};
|
||||||
|
|
||||||
|
/// Tools for export and import scripts
|
||||||
|
#[derive(Parser, Debug)]
|
||||||
|
#[clap(group = ArgGroup::new("encodingg").multiple(false), group = ArgGroup::new("output_encodingg").multiple(false))]
|
||||||
|
#[command(version, about, long_about = None)]
|
||||||
|
pub struct Arg {
|
||||||
|
#[arg(short = 't', long, value_enum, global = true)]
|
||||||
|
/// Script type
|
||||||
|
pub script_type: Option<ScriptType>,
|
||||||
|
#[arg(short = 'T', long, value_enum, global = true)]
|
||||||
|
/// Output script type
|
||||||
|
pub output_type: Option<OutputScriptType>,
|
||||||
|
#[arg(short = 'e', long, value_enum, global = true, group = "encodingg")]
|
||||||
|
/// Script encoding
|
||||||
|
pub encoding: Option<TextEncoding>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
#[arg(short = 'c', long, value_enum, global = true, group = "encodingg")]
|
||||||
|
/// Script code page
|
||||||
|
pub code_page: Option<u32>,
|
||||||
|
#[arg(
|
||||||
|
short = 'E',
|
||||||
|
long,
|
||||||
|
value_enum,
|
||||||
|
global = true,
|
||||||
|
group = "output_encodingg"
|
||||||
|
)]
|
||||||
|
/// Output text encoding
|
||||||
|
pub output_encoding: Option<TextEncoding>,
|
||||||
|
#[cfg(windows)]
|
||||||
|
#[arg(
|
||||||
|
short = 'C',
|
||||||
|
long,
|
||||||
|
value_enum,
|
||||||
|
global = true,
|
||||||
|
group = "output_encodingg"
|
||||||
|
)]
|
||||||
|
/// Output code page
|
||||||
|
pub output_code_page: Option<u32>,
|
||||||
|
#[arg(long, value_enum, global = true)]
|
||||||
|
/// Circus Game
|
||||||
|
pub circus_mes_type: Option<CircusMesType>,
|
||||||
|
#[arg(short, long, action = ArgAction::SetTrue, global = true)]
|
||||||
|
/// Search for script files in the directory recursively
|
||||||
|
pub recursive: bool,
|
||||||
|
#[arg(global = true, action = ArgAction::SetTrue, short, long)]
|
||||||
|
/// Print backtrace on error
|
||||||
|
pub backtrace: bool,
|
||||||
|
#[command(subcommand)]
|
||||||
|
/// Command
|
||||||
|
pub command: Command,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Subcommand, Debug)]
|
||||||
|
/// Commands
|
||||||
|
pub enum Command {
|
||||||
|
/// Extract from script
|
||||||
|
Export {
|
||||||
|
/// Input script file or directory
|
||||||
|
input: String,
|
||||||
|
/// Output file or directory
|
||||||
|
output: Option<String>,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn parse_args() -> Arg {
|
||||||
|
Arg::parse()
|
||||||
|
}
|
||||||
200
src/main.rs
Normal file
200
src/main.rs
Normal file
@@ -0,0 +1,200 @@
|
|||||||
|
pub mod args;
|
||||||
|
pub mod scripts;
|
||||||
|
pub mod types;
|
||||||
|
pub mod utils;
|
||||||
|
|
||||||
|
fn get_encoding(arg: &args::Arg, builder: &Box<dyn scripts::ScriptBuilder + Send + Sync>) -> types::Encoding {
|
||||||
|
match &arg.encoding {
|
||||||
|
Some(enc) => {
|
||||||
|
return match enc {
|
||||||
|
&types::TextEncoding::Default => {
|
||||||
|
builder.default_encoding()
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Auto => {
|
||||||
|
types::Encoding::Auto
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Cp932 => {
|
||||||
|
types::Encoding::Cp932
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Utf8 => {
|
||||||
|
types::Encoding::Utf8
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Gb2312 => {
|
||||||
|
types::Encoding::Gb2312
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
None => {}
|
||||||
|
}
|
||||||
|
#[cfg(windows)]
|
||||||
|
match &arg.code_page {
|
||||||
|
Some(code_page) => {
|
||||||
|
return types::Encoding::CodePage(*code_page);
|
||||||
|
}
|
||||||
|
None => {}
|
||||||
|
}
|
||||||
|
builder.default_encoding()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_output_encoding(arg: &args::Arg) -> types::Encoding {
|
||||||
|
match &arg.output_encoding {
|
||||||
|
Some(enc) => {
|
||||||
|
return match enc {
|
||||||
|
&types::TextEncoding::Default => {
|
||||||
|
types::Encoding::Utf8
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Auto => {
|
||||||
|
types::Encoding::Utf8
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Cp932 => {
|
||||||
|
types::Encoding::Cp932
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Utf8 => {
|
||||||
|
types::Encoding::Utf8
|
||||||
|
}
|
||||||
|
&types::TextEncoding::Gb2312 => {
|
||||||
|
types::Encoding::Gb2312
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
None => {}
|
||||||
|
}
|
||||||
|
#[cfg(windows)]
|
||||||
|
match &arg.code_page {
|
||||||
|
Some(code_page) => {
|
||||||
|
return types::Encoding::CodePage(*code_page);
|
||||||
|
}
|
||||||
|
None => {}
|
||||||
|
}
|
||||||
|
types::Encoding::Utf8
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn parse_script(filename: &str, arg: &args::Arg, config: &types::ExtraConfig) -> anyhow::Result<Box<dyn scripts::Script>> {
|
||||||
|
match &arg.script_type {
|
||||||
|
Some(typ) => {
|
||||||
|
for builder in scripts::BUILDER.iter() {
|
||||||
|
if typ == builder.script_type() {
|
||||||
|
let encoding = get_encoding(arg, builder);
|
||||||
|
return Ok(builder.build_script(filename, encoding, config)?);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
for builder in scripts::BUILDER.iter() {
|
||||||
|
let exts = builder.extensions();
|
||||||
|
for ext in exts {
|
||||||
|
if filename.to_lowercase().ends_with(ext) {
|
||||||
|
let encoding = get_encoding(arg, builder);
|
||||||
|
return Ok(builder.build_script(filename, encoding, config)?);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(anyhow::anyhow!("Unsupported script type"))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn export_script(
|
||||||
|
filename: &str,
|
||||||
|
arg: &args::Arg,
|
||||||
|
config: &types::ExtraConfig,
|
||||||
|
output: &Option<String>,
|
||||||
|
is_dir: bool,
|
||||||
|
) -> anyhow::Result<()> {
|
||||||
|
eprintln!("Exporting {}", filename);
|
||||||
|
let script = parse_script(filename, arg, config)?;
|
||||||
|
// println!("{:?}", script);
|
||||||
|
let mes = script.extract_messages()?;
|
||||||
|
// for m in mes.iter() {
|
||||||
|
// println!("{:?}", m);
|
||||||
|
// }
|
||||||
|
if mes.is_empty() {
|
||||||
|
eprintln!("No messages found");
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
let of = match &arg.output_type {
|
||||||
|
Some(t) => t.clone(),
|
||||||
|
None => script.default_output_script_type(),
|
||||||
|
};
|
||||||
|
let f = if filename == "-" {
|
||||||
|
String::from("-")
|
||||||
|
} else {
|
||||||
|
match output.as_ref() {
|
||||||
|
Some(output) => {
|
||||||
|
if is_dir {
|
||||||
|
let f = std::path::PathBuf::from(filename);
|
||||||
|
let mut pb = std::path::PathBuf::from(output);
|
||||||
|
if let Some(fname) = f.file_name() {
|
||||||
|
pb.push(fname);
|
||||||
|
}
|
||||||
|
pb.set_extension(of.as_ref());
|
||||||
|
pb.to_string_lossy().into_owned()
|
||||||
|
} else {
|
||||||
|
output.clone()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
let mut pb = std::path::PathBuf::from(filename);
|
||||||
|
pb.set_extension(of.as_ref());
|
||||||
|
pb.to_string_lossy().into_owned()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
match of {
|
||||||
|
types::OutputScriptType::Json => {
|
||||||
|
let enc = get_output_encoding(arg);
|
||||||
|
let s = serde_json::to_string_pretty(&mes)?;
|
||||||
|
let b = utils::encoding::encode_string(enc, &s)?;
|
||||||
|
let mut f = utils::files::write_file(&f)?;
|
||||||
|
f.write_all(&b)?;
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
let arg = args::parse_args();
|
||||||
|
if arg.backtrace {
|
||||||
|
unsafe { std::env::set_var("RUST_LIB_BACKTRACE", "1") };
|
||||||
|
}
|
||||||
|
let cfg = types::ExtraConfig {
|
||||||
|
circus_mes_type: arg.circus_mes_type.clone(),
|
||||||
|
};
|
||||||
|
match &arg.command {
|
||||||
|
args::Command::Export { input, output } => {
|
||||||
|
let (scripts, is_dir) = utils::files::collect_files(input, arg.recursive).unwrap();
|
||||||
|
if is_dir {
|
||||||
|
match &output {
|
||||||
|
Some(output) => {
|
||||||
|
let op = std::path::Path::new(output);
|
||||||
|
if op.exists() {
|
||||||
|
if !op.is_dir() {
|
||||||
|
eprintln!("Output path is not a directory");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
std::fs::create_dir_all(op).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
eprintln!("Output path is not specified");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for script in scripts.iter() {
|
||||||
|
let re = export_script(&script, &arg, &cfg, output, is_dir);
|
||||||
|
match re {
|
||||||
|
Ok(_) => {
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
eprintln!("Error exporting {}: {}", script, e);
|
||||||
|
if arg.backtrace {
|
||||||
|
eprintln!("Backtrace: {:?}", e.backtrace());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
23
src/scripts/base.rs
Normal file
23
src/scripts/base.rs
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
use crate::types::*;
|
||||||
|
use anyhow::Result;
|
||||||
|
|
||||||
|
pub trait ScriptBuilder {
|
||||||
|
fn default_encoding(&self) -> Encoding;
|
||||||
|
|
||||||
|
fn build_script(
|
||||||
|
&self,
|
||||||
|
filename: &str,
|
||||||
|
encoding: Encoding,
|
||||||
|
config: &ExtraConfig,
|
||||||
|
) -> Result<Box<dyn Script>>;
|
||||||
|
|
||||||
|
fn extensions(&self) -> &'static [&'static str];
|
||||||
|
|
||||||
|
fn script_type(&self) -> &'static ScriptType;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub trait Script: std::fmt::Debug {
|
||||||
|
fn default_output_script_type(&self) -> OutputScriptType;
|
||||||
|
|
||||||
|
fn extract_messages(&self) -> Result<Vec<Message>>;
|
||||||
|
}
|
||||||
465
src/scripts/circus/info.rs
Normal file
465
src/scripts/circus/info.rs
Normal file
@@ -0,0 +1,465 @@
|
|||||||
|
pub struct Section {
|
||||||
|
beg: u8,
|
||||||
|
end: u8,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::fmt::Debug for Section {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
f.debug_tuple("Section")
|
||||||
|
.field(&self.beg)
|
||||||
|
.field(&self.end)
|
||||||
|
.finish()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Section {
|
||||||
|
pub const fn new(beg: u8, end: u8) -> Self {
|
||||||
|
Section { beg, end }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn its(&self, key: u8) -> bool {
|
||||||
|
return (!(self.beg == self.end && self.beg == 0xFF))
|
||||||
|
&& (key >= self.beg && key <= self.end);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ScriptInfo {
|
||||||
|
pub name: &'static str,
|
||||||
|
pub version: u16,
|
||||||
|
/// \[op: byte\] \[arg1: uint8\] \[arg2: uint8\]
|
||||||
|
pub uint8x2: Section,
|
||||||
|
/// \[op: byte\] \[arg1: uint8\] \[arg2: string\]
|
||||||
|
pub uint8str: Section,
|
||||||
|
/// \[op: byte\] \[arg1: string\]
|
||||||
|
pub string: Section,
|
||||||
|
/// \[op: byte\] \[arg1: encstr\]
|
||||||
|
pub encstr: Section,
|
||||||
|
/// \[op: byte\] \[arg1: uint16\] \[arg2: uint16\] \[arg3: uint16\] \[arg4: uint16\]
|
||||||
|
pub uint16x4: Section,
|
||||||
|
/// the opcode for unencrypted strings in scene text
|
||||||
|
pub optunenc: u8,
|
||||||
|
pub deckey: u8,
|
||||||
|
pub nameopcode: u8,
|
||||||
|
}
|
||||||
|
|
||||||
|
const SCRIPT_INFO: [ScriptInfo; 31] = [
|
||||||
|
ScriptInfo::new(
|
||||||
|
"ffexa",
|
||||||
|
0x7B69,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x43,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"ffexs",
|
||||||
|
0x7B6B,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x4B),
|
||||||
|
(0x4c, 0x4F),
|
||||||
|
(0x50, 0xFF),
|
||||||
|
0x43,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"ef",
|
||||||
|
0x466A,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x2A, 0x2F),
|
||||||
|
(0x30, 0x4A),
|
||||||
|
(0x4B, 0x4E),
|
||||||
|
(0x4F, 0xFF),
|
||||||
|
0x46,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcos",
|
||||||
|
0x315D,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0xFF, 0xFF),
|
||||||
|
(0x2C, 0x45),
|
||||||
|
(0x46, 0x49),
|
||||||
|
(0x4A, 0xFF),
|
||||||
|
0x42,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"ktlep",
|
||||||
|
0x6E69,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcws",
|
||||||
|
0x656C,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x31),
|
||||||
|
(0x32, 0x4C),
|
||||||
|
(0x4D, 0x50),
|
||||||
|
(0x51, 0xFF),
|
||||||
|
0x48,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcsv",
|
||||||
|
0x636C,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x31),
|
||||||
|
(0x32, 0x4C),
|
||||||
|
(0x4D, 0x50),
|
||||||
|
(0x51, 0xFF),
|
||||||
|
0x46,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcpc",
|
||||||
|
0x3D63,
|
||||||
|
(0x00, 0x2C),
|
||||||
|
(0xFF, 0xFF),
|
||||||
|
(0x2D, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x44,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcmems",
|
||||||
|
0x315D,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0xFF, 0xFF),
|
||||||
|
(0x2C, 0x45),
|
||||||
|
(0x46, 0x49),
|
||||||
|
(0x4A, 0xFF),
|
||||||
|
0x42,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcdx",
|
||||||
|
0x7769,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcas",
|
||||||
|
0x4E69,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x43,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dcbs",
|
||||||
|
0x3163,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0xFF, 0xFF),
|
||||||
|
(0x2C, 0x48),
|
||||||
|
(0x49, 0x4C),
|
||||||
|
(0x4D, 0xFF),
|
||||||
|
0xFF,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2fl",
|
||||||
|
0x9C69,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2bs",
|
||||||
|
0x316C,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x31),
|
||||||
|
(0x32, 0x4C),
|
||||||
|
(0x4D, 0x50),
|
||||||
|
(0x51, 0xFF),
|
||||||
|
0xFF,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2dm",
|
||||||
|
0x9D72,
|
||||||
|
(0x00, 0x29),
|
||||||
|
(0x2A, 0x31),
|
||||||
|
(0x32, 0x4C),
|
||||||
|
(0x4D, 0x50),
|
||||||
|
(0x51, 0xFF),
|
||||||
|
0x44,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2fy",
|
||||||
|
0x3866,
|
||||||
|
(0x00, 0x2E),
|
||||||
|
(0xFF, 0xFF),
|
||||||
|
(0x2F, 0x4B),
|
||||||
|
(0x4C, 0x4F),
|
||||||
|
(0x50, 0xFF),
|
||||||
|
0x48,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2cckko",
|
||||||
|
0x026C,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x31),
|
||||||
|
(0x32, 0x4C),
|
||||||
|
(0x4D, 0x50),
|
||||||
|
(0x51, 0xFF),
|
||||||
|
0xFF,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2ccotm",
|
||||||
|
0x016C,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x31),
|
||||||
|
(0x32, 0x4C),
|
||||||
|
(0x4D, 0x50),
|
||||||
|
(0x51, 0xFF),
|
||||||
|
0xFF,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2sc",
|
||||||
|
0x3B69,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2ty",
|
||||||
|
0x5F69,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0xFF,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc2pc",
|
||||||
|
0x5769,
|
||||||
|
(0x00, 0x28),
|
||||||
|
(0x29, 0x2E),
|
||||||
|
(0x2F, 0x49),
|
||||||
|
(0x4A, 0x4D),
|
||||||
|
(0x4E, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc3rx",
|
||||||
|
0x9772,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x33),
|
||||||
|
(0x34, 0x4E),
|
||||||
|
(0x4F, 0x52),
|
||||||
|
(0x53, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc3pp",
|
||||||
|
0x9872,
|
||||||
|
(0x00, 0x2A),
|
||||||
|
(0x2B, 0x32),
|
||||||
|
(0x33, 0x4E),
|
||||||
|
(0x4F, 0x51),
|
||||||
|
(0x52, 0xFF),
|
||||||
|
0x45,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc3wy",
|
||||||
|
0xA09F,
|
||||||
|
(0x00, 0x38),
|
||||||
|
(0x39, 0x41),
|
||||||
|
(0x42, 0x5F),
|
||||||
|
(0x60, 0x63),
|
||||||
|
(0x64, 0xFF),
|
||||||
|
0x55,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc3dd",
|
||||||
|
0xA5A8,
|
||||||
|
(0x00, 0x38),
|
||||||
|
(0x39, 0x43),
|
||||||
|
(0x44, 0x62),
|
||||||
|
(0x63, 0x67),
|
||||||
|
(0x68, 0xFF),
|
||||||
|
0x58,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc4",
|
||||||
|
0xAAB6,
|
||||||
|
(0x00, 0x3A),
|
||||||
|
(0x3B, 0x47),
|
||||||
|
(0x48, 0x68),
|
||||||
|
(0x69, 0x6D),
|
||||||
|
(0x6E, 0xFF),
|
||||||
|
0x5D,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dc4ph",
|
||||||
|
0xABB6,
|
||||||
|
(0x00, 0x3A),
|
||||||
|
(0x3B, 0x47),
|
||||||
|
(0x48, 0x68),
|
||||||
|
(0x69, 0x6D),
|
||||||
|
(0x6E, 0xFF),
|
||||||
|
0x5D,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"ds",
|
||||||
|
0x9F9A,
|
||||||
|
(0x00, 0x38),
|
||||||
|
(0x39, 0x4A),
|
||||||
|
(0x41, 0x5E),
|
||||||
|
(0x5F, 0x62),
|
||||||
|
(0x63, 0xFF),
|
||||||
|
0x54,
|
||||||
|
0x20,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"dsif",
|
||||||
|
0xA1A1,
|
||||||
|
(0x00, 0x39),
|
||||||
|
(0x3A, 0x42),
|
||||||
|
(0x43, 0x60),
|
||||||
|
(0x61, 0x64),
|
||||||
|
(0x65, 0xFF),
|
||||||
|
0x56,
|
||||||
|
0x20,
|
||||||
|
0x62,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"tmpl",
|
||||||
|
0xA6B4,
|
||||||
|
(0x00, 0x3B),
|
||||||
|
(0x3A, 0x46),
|
||||||
|
(0x46, 0x67),
|
||||||
|
(0x68, 0x6E),
|
||||||
|
(0x6D, 0xFF),
|
||||||
|
0x5C,
|
||||||
|
0x20,
|
||||||
|
0x69,
|
||||||
|
),
|
||||||
|
ScriptInfo::new(
|
||||||
|
"nightshade",
|
||||||
|
0x0871,
|
||||||
|
(0x00, 0x2B),
|
||||||
|
(0x2C, 0x33),
|
||||||
|
(0x34, 0x4E),
|
||||||
|
(0x4F, 0x52),
|
||||||
|
(0x53, 0xFF),
|
||||||
|
0x43,
|
||||||
|
0x01,
|
||||||
|
0xFF,
|
||||||
|
),
|
||||||
|
];
|
||||||
|
|
||||||
|
impl ScriptInfo {
|
||||||
|
pub const fn new(
|
||||||
|
name: &'static str,
|
||||||
|
version: u16,
|
||||||
|
uint8x2: (u8, u8),
|
||||||
|
uint8str: (u8, u8),
|
||||||
|
string: (u8, u8),
|
||||||
|
encstr: (u8, u8),
|
||||||
|
uint16x4: (u8, u8),
|
||||||
|
optunenc: u8,
|
||||||
|
deckey: u8,
|
||||||
|
nameopcode: u8,
|
||||||
|
) -> Self {
|
||||||
|
ScriptInfo {
|
||||||
|
name,
|
||||||
|
version,
|
||||||
|
uint8x2: Section::new(uint8x2.0, uint8x2.1),
|
||||||
|
uint8str: Section::new(uint8str.0, uint8str.1),
|
||||||
|
string: Section::new(string.0, string.1),
|
||||||
|
encstr: Section::new(encstr.0, encstr.1),
|
||||||
|
uint16x4: Section::new(uint16x4.0, uint16x4.1),
|
||||||
|
optunenc,
|
||||||
|
deckey,
|
||||||
|
nameopcode,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn query(name: &str) -> Option<&'static ScriptInfo> {
|
||||||
|
for info in SCRIPT_INFO.iter() {
|
||||||
|
if info.name == name {
|
||||||
|
return Some(info);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn query_by_version(version: u16) -> Option<&'static ScriptInfo> {
|
||||||
|
for info in SCRIPT_INFO.iter() {
|
||||||
|
if info.version == version {
|
||||||
|
return Some(info);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
2
src/scripts/circus/mod.rs
Normal file
2
src/scripts/circus/mod.rs
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
mod info;
|
||||||
|
pub mod script;
|
||||||
218
src/scripts/circus/script.rs
Normal file
218
src/scripts/circus/script.rs
Normal file
@@ -0,0 +1,218 @@
|
|||||||
|
use super::info::*;
|
||||||
|
use crate::scripts::base::*;
|
||||||
|
use crate::types::*;
|
||||||
|
use crate::utils::encoding::decode_to_string;
|
||||||
|
use anyhow::Result;
|
||||||
|
|
||||||
|
pub struct CircusMesScriptBuilder {}
|
||||||
|
|
||||||
|
impl CircusMesScriptBuilder {
|
||||||
|
pub const fn new() -> Self {
|
||||||
|
CircusMesScriptBuilder {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ScriptBuilder for CircusMesScriptBuilder {
|
||||||
|
fn default_encoding(&self) -> Encoding {
|
||||||
|
Encoding::Cp932
|
||||||
|
}
|
||||||
|
|
||||||
|
fn build_script(
|
||||||
|
&self,
|
||||||
|
filename: &str,
|
||||||
|
encoding: Encoding,
|
||||||
|
config: &ExtraConfig,
|
||||||
|
) -> Result<Box<dyn Script>> {
|
||||||
|
Ok(Box::new(CircusMesScript::new(
|
||||||
|
filename.as_ref(),
|
||||||
|
encoding,
|
||||||
|
config,
|
||||||
|
)?))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn extensions(&self) -> &'static [&'static str] {
|
||||||
|
&["mes"]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn script_type(&self) -> &'static ScriptType {
|
||||||
|
&ScriptType::Circus
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct Token {
|
||||||
|
offset: usize,
|
||||||
|
length: usize,
|
||||||
|
value: u8,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct CircusMesScript {
|
||||||
|
data: Vec<u8>,
|
||||||
|
encoding: Encoding,
|
||||||
|
is_new_ver: bool,
|
||||||
|
version: u16,
|
||||||
|
info: &'static ScriptInfo,
|
||||||
|
asm_bin_offset: usize,
|
||||||
|
blocks_offset: usize,
|
||||||
|
tokens: Vec<Token>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl CircusMesScript {
|
||||||
|
pub fn new(filename: &str, encoding: Encoding, config: &ExtraConfig) -> Result<Self> {
|
||||||
|
let data = crate::utils::files::read_file(filename)?;
|
||||||
|
let head0 = i32::from_le_bytes(data[0..4].try_into()?);
|
||||||
|
let head1 = i32::from_le_bytes(data[4..8].try_into()?);
|
||||||
|
let mut is_new_ver = false;
|
||||||
|
let mut version = 0;
|
||||||
|
let mut info = config
|
||||||
|
.circus_mes_type
|
||||||
|
.as_ref()
|
||||||
|
.and_then(|name| ScriptInfo::query(name.as_ref()));
|
||||||
|
let mut asm_bin_offset = 0;
|
||||||
|
let mut blocks_offset = 0;
|
||||||
|
if head1 == 0x3 {
|
||||||
|
let offset = head0 * 0x6 + 0x4;
|
||||||
|
if data.len() > offset as usize {
|
||||||
|
if data.len() > offset as usize + 3 {
|
||||||
|
version =
|
||||||
|
u16::from_le_bytes(data[offset as usize..offset as usize + 2].try_into()?);
|
||||||
|
if info.is_none() {
|
||||||
|
info = ScriptInfo::query_by_version(version);
|
||||||
|
}
|
||||||
|
asm_bin_offset = offset as usize + 3;
|
||||||
|
}
|
||||||
|
blocks_offset = 8;
|
||||||
|
}
|
||||||
|
is_new_ver = true;
|
||||||
|
} else {
|
||||||
|
let offset = head0 * 0x4 + 0x4;
|
||||||
|
if data.len() > offset as usize {
|
||||||
|
if data.len() > offset as usize + 2 {
|
||||||
|
version =
|
||||||
|
u16::from_le_bytes(data[offset as usize..offset as usize + 2].try_into()?);
|
||||||
|
if info.is_none() {
|
||||||
|
info = ScriptInfo::query_by_version(version);
|
||||||
|
}
|
||||||
|
asm_bin_offset = offset as usize + 2;
|
||||||
|
}
|
||||||
|
blocks_offset = 4;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let info = info.ok_or(anyhow::anyhow!("Failed to detect version."))?;
|
||||||
|
let mut tokens = Vec::new();
|
||||||
|
let mut offset = 0;
|
||||||
|
let asm_bin_size = if asm_bin_offset == 0 {
|
||||||
|
0
|
||||||
|
} else {
|
||||||
|
data.len() - asm_bin_offset
|
||||||
|
};
|
||||||
|
while offset < asm_bin_size {
|
||||||
|
let value = data[asm_bin_offset + offset];
|
||||||
|
let length = if info.uint8x2.its(value) {
|
||||||
|
0x03
|
||||||
|
} else if info.uint8str.its(value) {
|
||||||
|
let mut len = 0x3;
|
||||||
|
let mut temp = data[asm_bin_offset + offset + len - 1];
|
||||||
|
while temp != 0x00 {
|
||||||
|
len += 0x1;
|
||||||
|
if asm_bin_offset + offset + len >= data.len() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
temp = data[asm_bin_offset + offset + len - 1];
|
||||||
|
}
|
||||||
|
len
|
||||||
|
} else if info.string.its(value) || info.encstr.its(value) {
|
||||||
|
let mut len = 1;
|
||||||
|
let mut temp = data[asm_bin_offset + offset + len - 1];
|
||||||
|
while temp != 0x00 {
|
||||||
|
len += 0x1;
|
||||||
|
if asm_bin_offset + offset + len >= data.len() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
temp = data[asm_bin_offset + offset + len - 1];
|
||||||
|
}
|
||||||
|
len
|
||||||
|
} else if info.uint16x4.its(value) {
|
||||||
|
0x09
|
||||||
|
} else {
|
||||||
|
return Err(anyhow::anyhow!(format!(
|
||||||
|
"Unknown token type: 0x{:02X} at offset {}",
|
||||||
|
value,
|
||||||
|
asm_bin_offset + offset
|
||||||
|
)));
|
||||||
|
};
|
||||||
|
let token = Token {
|
||||||
|
offset,
|
||||||
|
length,
|
||||||
|
value,
|
||||||
|
};
|
||||||
|
offset += length;
|
||||||
|
tokens.push(token);
|
||||||
|
}
|
||||||
|
Ok(CircusMesScript {
|
||||||
|
data,
|
||||||
|
encoding,
|
||||||
|
is_new_ver,
|
||||||
|
version,
|
||||||
|
info,
|
||||||
|
asm_bin_offset,
|
||||||
|
blocks_offset,
|
||||||
|
tokens,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::fmt::Debug for CircusMesScript {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
f.debug_struct("CircusMesScript")
|
||||||
|
.field("encoding", &self.encoding)
|
||||||
|
.field("is_new_ver", &self.is_new_ver)
|
||||||
|
.field("version", &self.version)
|
||||||
|
.field("info", &self.info)
|
||||||
|
.field("asm_bin_offset", &self.asm_bin_offset)
|
||||||
|
.field("blocks_offset", &self.blocks_offset)
|
||||||
|
.field("tokens", &self.tokens)
|
||||||
|
.finish_non_exhaustive()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Script for CircusMesScript {
|
||||||
|
fn default_output_script_type(&self) -> OutputScriptType {
|
||||||
|
OutputScriptType::Json
|
||||||
|
}
|
||||||
|
|
||||||
|
fn extract_messages(&self) -> Result<Vec<Message>> {
|
||||||
|
let mut mes = vec![];
|
||||||
|
let mut name = None;
|
||||||
|
for token in self.tokens.iter() {
|
||||||
|
let mut t = None;
|
||||||
|
if self.info.encstr.its(token.value) {
|
||||||
|
let mut text = self.data[self.asm_bin_offset + token.offset + 1
|
||||||
|
..self.asm_bin_offset + token.offset + token.length]
|
||||||
|
.to_vec();
|
||||||
|
for t in text.iter_mut() {
|
||||||
|
*t = (*t).overflowing_add(self.info.deckey).0;
|
||||||
|
}
|
||||||
|
t = Some(decode_to_string(self.encoding, &text)?);
|
||||||
|
// println!("Token(enc): {:?}, {}", token, t.as_ref().unwrap());
|
||||||
|
} else if token.value == self.info.optunenc {
|
||||||
|
let text = &self.data[self.asm_bin_offset + token.offset + 1
|
||||||
|
..self.asm_bin_offset + token.offset + token.length];
|
||||||
|
t = Some(decode_to_string(self.encoding, text)?);
|
||||||
|
// println!("Token: {:?}, {}", token, t.as_ref().unwrap());
|
||||||
|
}
|
||||||
|
match t {
|
||||||
|
Some(t) => {
|
||||||
|
if token.value == self.info.nameopcode {
|
||||||
|
name = Some(t);
|
||||||
|
} else {
|
||||||
|
let message = Message::new(t, name.take());
|
||||||
|
mes.push(message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(mes)
|
||||||
|
}
|
||||||
|
}
|
||||||
12
src/scripts/mod.rs
Normal file
12
src/scripts/mod.rs
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
pub mod base;
|
||||||
|
pub mod circus;
|
||||||
|
|
||||||
|
pub use base::{Script, ScriptBuilder};
|
||||||
|
|
||||||
|
lazy_static::lazy_static! {
|
||||||
|
pub static ref BUILDER: Vec<Box<dyn ScriptBuilder + Sync + Send>> = vec![
|
||||||
|
Box::new(circus::script::CircusMesScriptBuilder::new()),
|
||||||
|
];
|
||||||
|
pub static ref ALL_EXTS: Vec<String> =
|
||||||
|
BUILDER.iter().flat_map(|b| b.extensions()).map(|s| s.to_string()).collect();
|
||||||
|
}
|
||||||
187
src/types.rs
Normal file
187
src/types.rs
Normal file
@@ -0,0 +1,187 @@
|
|||||||
|
use clap::ValueEnum;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
#[derive(Copy, Clone, Serialize, Deserialize, Debug)]
|
||||||
|
#[serde(untagged, rename_all = "camelCase")]
|
||||||
|
/// Text Encoding
|
||||||
|
pub enum Encoding {
|
||||||
|
/// Automatically detect encoding
|
||||||
|
Auto,
|
||||||
|
/// UTF-8 encoding
|
||||||
|
Utf8,
|
||||||
|
/// Shift-JIS encoding
|
||||||
|
Cp932,
|
||||||
|
/// GB2312 encoding
|
||||||
|
Gb2312,
|
||||||
|
/// Code page encoding (Windows only)
|
||||||
|
#[cfg(windows)]
|
||||||
|
CodePage(u32),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Encoding {
|
||||||
|
fn default() -> Self {
|
||||||
|
Encoding::Utf8
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Debug, ValueEnum, PartialEq, Eq, PartialOrd, Ord)]
|
||||||
|
/// Text Encoding
|
||||||
|
pub enum TextEncoding {
|
||||||
|
/// Use script's default encoding
|
||||||
|
Default,
|
||||||
|
/// Automatically detect encoding
|
||||||
|
Auto,
|
||||||
|
/// UTF-8 encoding
|
||||||
|
Utf8,
|
||||||
|
/// Shift-JIS encoding
|
||||||
|
Cp932,
|
||||||
|
/// GB2312 encoding
|
||||||
|
Gb2312,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Debug, ValueEnum, PartialEq, Eq, PartialOrd, Ord)]
|
||||||
|
/// Script type
|
||||||
|
pub enum OutputScriptType {
|
||||||
|
/// Text script
|
||||||
|
Txt,
|
||||||
|
/// JSON which can be used for GalTransl
|
||||||
|
Json,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsRef<str> for OutputScriptType {
|
||||||
|
fn as_ref(&self) -> &str {
|
||||||
|
match self {
|
||||||
|
OutputScriptType::Txt => "txt",
|
||||||
|
OutputScriptType::Json => "json",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Debug, ValueEnum, PartialEq, Eq, PartialOrd, Ord)]
|
||||||
|
pub enum CircusMesType {
|
||||||
|
/// fortissimo//Akkord:Bsusvier
|
||||||
|
Ffexa,
|
||||||
|
/// fortissimo EXS//Akkord:nächsten Phase
|
||||||
|
Ffexs,
|
||||||
|
/// Eternal Fantasy
|
||||||
|
Ef,
|
||||||
|
/// D.C.〜ダ・カーポ〜 温泉編
|
||||||
|
Dcos,
|
||||||
|
/// ことり Love Ex P
|
||||||
|
Ktlep,
|
||||||
|
/// D.C.WhiteSeason
|
||||||
|
Dcws,
|
||||||
|
/// D.C. Summer Vacation
|
||||||
|
Dcsv,
|
||||||
|
/// Ｄ．Ｃ．Ｐ．Ｃ．(Vista)
|
||||||
|
Dcpc,
|
||||||
|
/// D.C.〜ダ・カーポ〜 MEMORIES DISC
|
||||||
|
Dcmems,
|
||||||
|
/// D.C. Dream X’mas
|
||||||
|
Dcdx,
|
||||||
|
/// D.C.A.S. 〜ダ・カーポ〜アフターシーズンズ
|
||||||
|
Dcas,
|
||||||
|
/// D.C.II 春風のアルティメットバトル！
|
||||||
|
Dcbs,
|
||||||
|
/// D.C.II Fall in Love
|
||||||
|
Dc2fl,
|
||||||
|
/// D.C.II 春風のアルティメットバトル！
|
||||||
|
Dc2bs,
|
||||||
|
/// D.C.II Dearest Marriage
|
||||||
|
Dc2dm,
|
||||||
|
/// D.C.II 〜featuring Yun2〜
|
||||||
|
Dc2fy,
|
||||||
|
/// D.C.II C.C. 月島小恋のらぶらぶバスルーム
|
||||||
|
Dc2cckko,
|
||||||
|
/// D.C.II C.C. 音姫先生のどきどき特別授業
|
||||||
|
Dc2ccotm,
|
||||||
|
/// D.C.II Spring Celebration
|
||||||
|
Dc2sc,
|
||||||
|
/// D.C.II To You
|
||||||
|
Dc2ty,
|
||||||
|
/// D.C.II P.C.
|
||||||
|
Dc2pc,
|
||||||
|
/// D.C.III RX-rated
|
||||||
|
Dc3rx,
|
||||||
|
/// D.C.III P.P.～ダ・カーポIII プラチナパートナー～
|
||||||
|
Dc3pp,
|
||||||
|
/// D.C.III WithYou
|
||||||
|
Dc3wy,
|
||||||
|
/// D.C.III DreamDays
|
||||||
|
Dc3dd,
|
||||||
|
/// D.C.4 ～ダ・カーポ4～
|
||||||
|
Dc4,
|
||||||
|
/// D.C.4 Plus Harmony 〜ダ・カーポ4〜 プラスハーモニー
|
||||||
|
Dc4ph,
|
||||||
|
/// D.S. -Dal Segno-
|
||||||
|
Ds,
|
||||||
|
/// D.S.i.F. -Dal Segno- in Future
|
||||||
|
Dsif,
|
||||||
|
/// てんぷれ！
|
||||||
|
Tmpl,
|
||||||
|
/// 百花百狼/Hyakka Hyakurou
|
||||||
|
Nightshade,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsRef<str> for CircusMesType {
|
||||||
|
fn as_ref(&self) -> &str {
|
||||||
|
match self {
|
||||||
|
CircusMesType::Ffexa => "ffexa",
|
||||||
|
CircusMesType::Ffexs => "ffexs",
|
||||||
|
CircusMesType::Ef => "ef",
|
||||||
|
CircusMesType::Dcos => "dcos",
|
||||||
|
CircusMesType::Ktlep => "ktlep",
|
||||||
|
CircusMesType::Dcws => "dcws",
|
||||||
|
CircusMesType::Dcsv => "dcsv",
|
||||||
|
CircusMesType::Dcpc => "dcpc",
|
||||||
|
CircusMesType::Dcmems => "dcmems",
|
||||||
|
CircusMesType::Dcdx => "dcdx",
|
||||||
|
CircusMesType::Dcas => "dcas",
|
||||||
|
CircusMesType::Dcbs => "dcbs",
|
||||||
|
CircusMesType::Dc2fl => "dc2fl",
|
||||||
|
CircusMesType::Dc2bs => "dc2bs",
|
||||||
|
CircusMesType::Dc2dm => "dc2dm",
|
||||||
|
CircusMesType::Dc2fy => "dc2fy",
|
||||||
|
CircusMesType::Dc2cckko => "dc2cckko",
|
||||||
|
CircusMesType::Dc2ccotm => "dc2ccotm",
|
||||||
|
CircusMesType::Dc2sc => "dc2sc",
|
||||||
|
CircusMesType::Dc2ty => "dc2ty",
|
||||||
|
CircusMesType::Dc2pc => "dc2pc",
|
||||||
|
CircusMesType::Dc3rx => "dc3rx",
|
||||||
|
CircusMesType::Dc3pp => "dc3pp",
|
||||||
|
CircusMesType::Dc3wy => "dc3wy",
|
||||||
|
CircusMesType::Dc3dd => "dc3dd",
|
||||||
|
CircusMesType::Dc4 => "dc4",
|
||||||
|
CircusMesType::Dc4ph => "dc4ph",
|
||||||
|
CircusMesType::Ds => "ds",
|
||||||
|
CircusMesType::Dsif => "dsif",
|
||||||
|
CircusMesType::Tmpl => "tmpl",
|
||||||
|
CircusMesType::Nightshade => "nightshade",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct ExtraConfig {
|
||||||
|
pub circus_mes_type: Option<CircusMesType>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Debug, ValueEnum, PartialEq, Eq, PartialOrd, Ord)]
|
||||||
|
/// Script type
|
||||||
|
pub enum ScriptType {
|
||||||
|
/// Circus MES script
|
||||||
|
Circus,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
|
pub struct Message {
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub name: Option<String>,
|
||||||
|
pub message: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Message {
|
||||||
|
pub fn new(message: String, name: Option<String>) -> Self {
|
||||||
|
Message { message, name }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
106
src/utils/encoding.rs
Normal file
106
src/utils/encoding.rs
Normal file
@@ -0,0 +1,106 @@
|
|||||||
|
use crate::types::*;
|
||||||
|
|
||||||
|
pub fn decode_to_string(encoding: Encoding, data: &[u8]) -> Result<String, anyhow::Error> {
|
||||||
|
match encoding {
|
||||||
|
Encoding::Auto => decode_to_string(Encoding::Utf8, data)
|
||||||
|
.or_else(|_| decode_to_string(Encoding::Cp932, data))
|
||||||
|
.or_else(|_| decode_to_string(Encoding::Gb2312, data)),
|
||||||
|
Encoding::Utf8 => Ok(String::from_utf8(data.to_vec())?),
|
||||||
|
Encoding::Cp932 => {
|
||||||
|
let result = encoding_rs::SHIFT_JIS.decode(data);
|
||||||
|
if result.2 {
|
||||||
|
Err(anyhow::anyhow!("Failed to decode Shift-JIS"))
|
||||||
|
} else {
|
||||||
|
Ok(result.0.to_string())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Encoding::Gb2312 => {
|
||||||
|
let result = encoding_rs::GBK.decode(data);
|
||||||
|
if result.2 {
|
||||||
|
Err(anyhow::anyhow!("Failed to decode GB2312"))
|
||||||
|
} else {
|
||||||
|
Ok(result.0.to_string())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[cfg(windows)]
|
||||||
|
Encoding::CodePage(code_page) => {
|
||||||
|
Ok(super::encoding_win::decode_to_string(code_page, data)?)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn encode_string(encoding: Encoding, data: &str) -> Result<Vec<u8>, anyhow::Error> {
|
||||||
|
match encoding {
|
||||||
|
Encoding::Auto => Ok(data.as_bytes().to_vec()),
|
||||||
|
Encoding::Utf8 => Ok(data.as_bytes().to_vec()),
|
||||||
|
Encoding::Cp932 => {
|
||||||
|
let result = encoding_rs::SHIFT_JIS.encode(data);
|
||||||
|
if result.2 {
|
||||||
|
Err(anyhow::anyhow!("Failed to encode Shift-JIS"))
|
||||||
|
} else {
|
||||||
|
Ok(result.0.to_vec())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Encoding::Gb2312 => {
|
||||||
|
let result = encoding_rs::GBK.encode(data);
|
||||||
|
if result.2 {
|
||||||
|
Err(anyhow::anyhow!("Failed to encode GB2312"))
|
||||||
|
} else {
|
||||||
|
Ok(result.0.to_vec())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[cfg(windows)]
|
||||||
|
Encoding::CodePage(code_page) => {
|
||||||
|
Ok(super::encoding_win::encode_string(code_page, data)?)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_decode_to_string() {
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(
|
||||||
|
Encoding::Utf8,
|
||||||
|
&[228, 184, 173, 230, 150, 135, 230, 181, 139, 232, 175, 149]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
|
"中文测试".to_string()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(
|
||||||
|
Encoding::Cp932,
|
||||||
|
&[
|
||||||
|
130, 171, 130, 225, 130, 215, 130, 194, 130, 187, 130, 211, 130, 198
|
||||||
|
]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
|
"きゃべつそふと".to_string()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(Encoding::Gb2312, &[214, 208, 206, 196]).unwrap(),
|
||||||
|
"中文".to_string()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(
|
||||||
|
Encoding::Auto,
|
||||||
|
&[228, 184, 173, 230, 150, 135, 230, 181, 139, 232, 175, 149]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
|
"中文测试".to_string()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(
|
||||||
|
Encoding::Auto,
|
||||||
|
&[
|
||||||
|
130, 171, 130, 225, 130, 215, 130, 194, 130, 187, 130, 211, 130, 198
|
||||||
|
]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
|
"きゃべつそふと".to_string()
|
||||||
|
);
|
||||||
|
#[cfg(windows)]
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(Encoding::CodePage(936), &[214, 208, 206, 196]).unwrap(),
|
||||||
|
"中文".to_string()
|
||||||
|
);
|
||||||
|
}
|
||||||
121
src/utils/encoding_win.rs
Normal file
121
src/utils/encoding_win.rs
Normal file
@@ -0,0 +1,121 @@
|
|||||||
|
use windows_sys::Win32::Foundation::GetLastError;
|
||||||
|
use windows_sys::Win32::Globalization::{MB_ERR_INVALID_CHARS, MultiByteToWideChar, WideCharToMultiByte};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct WinError {
|
||||||
|
pub code: u32,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl WinError {
|
||||||
|
pub fn new(code: u32) -> Self {
|
||||||
|
WinError { code }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn from_last_error() -> Self {
|
||||||
|
let code = unsafe { GetLastError() };
|
||||||
|
WinError::new(code)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::error::Error for WinError {}
|
||||||
|
|
||||||
|
impl std::fmt::Display for WinError {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
write!(f, "Windows error code: {}", self.code)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn decode_to_string(cp: u32, data: &[u8]) -> Result<String, WinError> {
|
||||||
|
let needed_len = unsafe {
|
||||||
|
MultiByteToWideChar(
|
||||||
|
cp,
|
||||||
|
MB_ERR_INVALID_CHARS,
|
||||||
|
data.as_ptr() as _,
|
||||||
|
data.len() as i32,
|
||||||
|
std::ptr::null_mut(),
|
||||||
|
0,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
if needed_len == 0 {
|
||||||
|
return Err(WinError::from_last_error());
|
||||||
|
}
|
||||||
|
let mut wc = Vec::with_capacity(needed_len as usize);
|
||||||
|
wc.resize(needed_len as usize, 0);
|
||||||
|
let result = unsafe {
|
||||||
|
MultiByteToWideChar(
|
||||||
|
cp,
|
||||||
|
MB_ERR_INVALID_CHARS,
|
||||||
|
data.as_ptr() as _,
|
||||||
|
data.len() as i32,
|
||||||
|
wc.as_mut_ptr(),
|
||||||
|
needed_len,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
if result == 0 {
|
||||||
|
return Err(WinError::from_last_error());
|
||||||
|
}
|
||||||
|
Ok(String::from_utf16_lossy(&wc))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn encode_string(cp: u32, data: &str) -> Result<Vec<u8>, WinError> {
|
||||||
|
let wstr = data.encode_utf16().collect::<Vec<u16>>();
|
||||||
|
let needed_len = unsafe {
|
||||||
|
WideCharToMultiByte(
|
||||||
|
cp,
|
||||||
|
0,
|
||||||
|
wstr.as_ptr(),
|
||||||
|
wstr.len() as i32,
|
||||||
|
std::ptr::null_mut(),
|
||||||
|
0,
|
||||||
|
std::ptr::null_mut(),
|
||||||
|
std::ptr::null_mut(),
|
||||||
|
)
|
||||||
|
};
|
||||||
|
if needed_len == 0 {
|
||||||
|
return Err(WinError::from_last_error());
|
||||||
|
}
|
||||||
|
let mut mb = Vec::with_capacity(needed_len as usize);
|
||||||
|
mb.resize(needed_len as usize, 0);
|
||||||
|
let result = unsafe {
|
||||||
|
WideCharToMultiByte(
|
||||||
|
cp,
|
||||||
|
0,
|
||||||
|
wstr.as_ptr(),
|
||||||
|
wstr.len() as i32,
|
||||||
|
mb.as_mut_ptr(),
|
||||||
|
needed_len,
|
||||||
|
std::ptr::null_mut(),
|
||||||
|
std::ptr::null_mut(),
|
||||||
|
)
|
||||||
|
};
|
||||||
|
if result == 0 {
|
||||||
|
return Err(WinError::from_last_error());
|
||||||
|
}
|
||||||
|
Ok(mb)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_decode_to_string() {
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(
|
||||||
|
65001,
|
||||||
|
&[228, 184, 173, 230, 150, 135, 230, 181, 139, 232, 175, 149]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
|
"中文测试".to_string()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(
|
||||||
|
932,
|
||||||
|
&[
|
||||||
|
130, 171, 130, 225, 130, 215, 130, 194, 130, 187, 130, 211, 130, 198
|
||||||
|
]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
|
"きゃべつそふと".to_string()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
decode_to_string(936, &[214, 208, 206, 196]).unwrap(),
|
||||||
|
"中文".to_string()
|
||||||
|
);
|
||||||
|
}
|
||||||
67
src/utils/files.rs
Normal file
67
src/utils/files.rs
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
use std::fs;
|
||||||
|
use std::io;
|
||||||
|
use std::path::Path;
|
||||||
|
use std::io::{Read, Write};
|
||||||
|
|
||||||
|
use crate::scripts::ALL_EXTS;
|
||||||
|
|
||||||
|
pub fn find_files(path: &String, recursive: bool) -> io::Result<Vec<String>> {
|
||||||
|
let mut result = Vec::new();
|
||||||
|
let dir_path = Path::new(&path);
|
||||||
|
|
||||||
|
if dir_path.is_dir() {
|
||||||
|
for entry in fs::read_dir(dir_path)? {
|
||||||
|
let entry = entry?;
|
||||||
|
let path = entry.path();
|
||||||
|
|
||||||
|
if path.is_file()
|
||||||
|
&& path.extension().map_or(true, |ext| {
|
||||||
|
ALL_EXTS.contains(&ext.to_string_lossy().to_lowercase())
|
||||||
|
})
|
||||||
|
{
|
||||||
|
if let Some(path_str) = path.to_str() {
|
||||||
|
result.push(path_str.to_string());
|
||||||
|
}
|
||||||
|
} else if recursive && path.is_dir() {
|
||||||
|
if let Some(path_str) = path.to_str() {
|
||||||
|
let mut sub_files = find_files(&path_str.to_string(), recursive)?;
|
||||||
|
result.append(&mut sub_files);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn collect_files(path: &String, recursive: bool) -> io::Result<(Vec<String>, bool)> {
|
||||||
|
let pa = Path::new(path);
|
||||||
|
if pa.is_dir() {
|
||||||
|
return Ok((find_files(path, recursive)?, true));
|
||||||
|
}
|
||||||
|
if pa.is_file() {
|
||||||
|
return Ok((vec![path.clone()], false));
|
||||||
|
}
|
||||||
|
Err(io::Error::new(
|
||||||
|
io::ErrorKind::NotFound,
|
||||||
|
format!("Path {} is neither a file nor a directory", pa.display()),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn read_file<F: AsRef<Path> + ?Sized>(f: &F) -> io::Result<Vec<u8>> {
|
||||||
|
let mut content = Vec::new();
|
||||||
|
if f.as_ref() == Path::new("-") {
|
||||||
|
io::stdin().read_to_end(&mut content)?;
|
||||||
|
} else {
|
||||||
|
content = fs::read(f)?;
|
||||||
|
}
|
||||||
|
Ok(content)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn write_file<F: AsRef<Path> + ?Sized>(f: &F) -> io::Result<Box<dyn Write>> {
|
||||||
|
Ok(if f.as_ref() == Path::new("-") {
|
||||||
|
Box::new(io::stdout())
|
||||||
|
} else {
|
||||||
|
Box::new(fs::File::create(f)?)
|
||||||
|
})
|
||||||
|
}
|
||||||
4
src/utils/mod.rs
Normal file
4
src/utils/mod.rs
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
pub mod encoding;
|
||||||
|
#[cfg(windows)]
|
||||||
|
mod encoding_win;
|
||||||
|
pub mod files;
|
||||||
Reference in New Issue
Block a user