aea0a6fcc6
Running and testing should be working on the three platforms, but there's still a fair bit that needs to be done: - Wheel building + testing in a venv still needs to be implemented. - Python requirements still need to be compiled with piptool and pinned; need to compile on all platforms then merge - Cargo deps in cargo/ and rslib/ need to be cleaned up, and ideally unified into one place - Currently using rustls to work around openssl compilation issues on Linux, but this will break corporate proxies with custom SSL authorities; need to conditionally use openssl or use https://github.com/seanmonstar/reqwest/pull/1058 - Makefiles and docs still need cleaning up - It may make sense to reparent ts/* to the top level, as we don't nest the other modules under a specific language. - rspy and pylib must always be updated in lock-step, so merging rspy into pylib as a private module would simplify things. - Merging desktop-ftl and mobile-ftl into the core ftl would make managing and updating translations easier. - Obsolete scripts need removing. - And probably more.
95 lines
2.8 KiB
Rust
95 lines
2.8 KiB
Rust
use std::path::PathBuf;
|
|
use std::{env, fmt::Write};
|
|
|
|
struct CustomGenerator {}
|
|
|
|
fn write_method_enum(buf: &mut String, service: &prost_build::Service) {
|
|
buf.push_str(
|
|
r#"
|
|
use num_enum::TryFromPrimitive;
|
|
#[derive(PartialEq,TryFromPrimitive)]
|
|
#[repr(u32)]
|
|
pub enum BackendMethod {
|
|
"#,
|
|
);
|
|
for (idx, method) in service.methods.iter().enumerate() {
|
|
writeln!(buf, " {} = {},", method.proto_name, idx + 1).unwrap();
|
|
}
|
|
buf.push_str("}\n\n");
|
|
}
|
|
|
|
fn write_method_trait(buf: &mut String, service: &prost_build::Service) {
|
|
buf.push_str(
|
|
r#"
|
|
use prost::Message;
|
|
pub type BackendResult<T> = std::result::Result<T, crate::err::AnkiError>;
|
|
pub trait BackendService {
|
|
fn run_command_bytes2_inner(&self, method: u32, input: &[u8]) -> std::result::Result<Vec<u8>, crate::err::AnkiError> {
|
|
match method {
|
|
"#,
|
|
);
|
|
|
|
for (idx, method) in service.methods.iter().enumerate() {
|
|
write!(
|
|
buf,
|
|
concat!(" ",
|
|
"{idx} => {{ let input = {input_type}::decode(input)?;\n",
|
|
"let output = self.{rust_method}(input)?;\n",
|
|
"let mut out_bytes = Vec::new(); output.encode(&mut out_bytes)?; Ok(out_bytes) }}, "),
|
|
idx = idx + 1,
|
|
input_type = method.input_type,
|
|
rust_method = method.name
|
|
)
|
|
.unwrap();
|
|
}
|
|
buf.push_str(
|
|
r#"
|
|
_ => Err(crate::err::AnkiError::invalid_input("invalid command")),
|
|
}
|
|
}
|
|
"#,
|
|
);
|
|
|
|
for method in &service.methods {
|
|
write!(
|
|
buf,
|
|
concat!(
|
|
" fn {method_name}(&self, input: {input_type}) -> ",
|
|
"BackendResult<{output_type}>;\n"
|
|
),
|
|
method_name = method.name,
|
|
input_type = method.input_type,
|
|
output_type = method.output_type
|
|
)
|
|
.unwrap();
|
|
}
|
|
buf.push_str("}\n");
|
|
}
|
|
|
|
impl prost_build::ServiceGenerator for CustomGenerator {
|
|
fn generate(&mut self, service: prost_build::Service, buf: &mut String) {
|
|
write_method_enum(buf, &service);
|
|
write_method_trait(buf, &service);
|
|
}
|
|
}
|
|
|
|
fn service_generator() -> Box<dyn prost_build::ServiceGenerator> {
|
|
Box::new(CustomGenerator {})
|
|
}
|
|
|
|
pub fn write_backend_proto_rs() {
|
|
let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
|
|
let backend_proto = PathBuf::from(
|
|
env::var("BACKEND_PROTO").unwrap_or_else(|_| "../proto/backend.proto".to_string()),
|
|
);
|
|
let proto_dir = backend_proto.parent().unwrap().to_owned();
|
|
let fluent_proto = out_dir.join("fluent.proto");
|
|
|
|
let mut config = prost_build::Config::new();
|
|
config
|
|
.out_dir(&out_dir)
|
|
.service_generator(service_generator())
|
|
.compile_protos(&[&backend_proto, &fluent_proto], &[&proto_dir, &out_dir])
|
|
.unwrap();
|
|
}
|