124 lines
3.4 KiB
Rust
124 lines
3.4 KiB
Rust
// Copyright: Ankitects Pty Ltd and contributors
|
|
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
use std::env;
|
|
use std::fmt::Write;
|
|
use std::path::PathBuf;
|
|
|
|
use anki_io::create_dir_all;
|
|
use anki_io::write_file_if_changed;
|
|
use anyhow::Result;
|
|
use inflections::Inflect;
|
|
use itertools::Itertools;
|
|
|
|
use crate::extract::Module;
|
|
use crate::extract::Variable;
|
|
use crate::extract::VariableKind;
|
|
|
|
pub fn write_ts_interface(modules: &[Module]) -> Result<()> {
|
|
let mut dts_out = header();
|
|
let mut js_out = header();
|
|
write_translate_method(&mut js_out);
|
|
dts_out.push_str("export declare const funcs: any;\n");
|
|
|
|
render_module_map(modules, &mut dts_out, &mut js_out);
|
|
render_methods(modules, &mut dts_out, &mut js_out);
|
|
|
|
if let Ok(path) = env::var("STRINGS_JS") {
|
|
let path = PathBuf::from(path);
|
|
create_dir_all(path.parent().unwrap())?;
|
|
write_file_if_changed(path, js_out)?;
|
|
}
|
|
if let Ok(path) = env::var("STRINGS_DTS") {
|
|
let path = PathBuf::from(path);
|
|
create_dir_all(path.parent().unwrap())?;
|
|
write_file_if_changed(path, dts_out)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn render_module_map(modules: &[Module], dts_out: &mut String, js_out: &mut String) {
|
|
dts_out.push_str("export declare enum ModuleName {\n");
|
|
js_out.push_str("export const ModuleName = {};\n");
|
|
for module in modules {
|
|
let name = &module.name;
|
|
let upper = name.to_upper_case();
|
|
writeln!(dts_out, r#" {upper} = "{name}","#).unwrap();
|
|
writeln!(js_out, r#"ModuleName["{upper}"] = "{name}";"#).unwrap();
|
|
}
|
|
dts_out.push('}');
|
|
}
|
|
|
|
fn render_methods(modules: &[Module], dts_out: &mut String, js_out: &mut String) {
|
|
for module in modules {
|
|
for translation in &module.translations {
|
|
let text = &translation.text;
|
|
let key = &translation.key;
|
|
let func_name = key.replace('-', "_").to_camel_case();
|
|
let arg_types = get_arg_types(&translation.variables);
|
|
let maybe_args = if translation.variables.is_empty() {
|
|
""
|
|
} else {
|
|
"args"
|
|
};
|
|
writeln!(
|
|
dts_out,
|
|
"
|
|
/** {text} */
|
|
export declare function {func_name}({arg_types}): string;",
|
|
)
|
|
.unwrap();
|
|
writeln!(
|
|
js_out,
|
|
r#"
|
|
export function {func_name}({maybe_args}) {{
|
|
return translate("{key}", {maybe_args})
|
|
}}"#,
|
|
)
|
|
.unwrap();
|
|
}
|
|
}
|
|
}
|
|
|
|
fn write_translate_method(buf: &mut String) {
|
|
buf.push_str(
|
|
"
|
|
// tslib is responsible for injecting getMessage helper in
|
|
export const funcs = {};
|
|
|
|
function translate(key, args = {}) {
|
|
return funcs.getMessage(key, args) ?? `missing key: ${key}`;
|
|
}
|
|
",
|
|
);
|
|
}
|
|
|
|
fn header() -> String {
|
|
"// Copyright: Ankitects Pty Ltd and contributors
|
|
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
"
|
|
.to_string()
|
|
}
|
|
|
|
fn get_arg_types(args: &[Variable]) -> String {
|
|
let args = args
|
|
.iter()
|
|
.map(|arg| format!("{}: {}", arg.name.to_camel_case(), arg_kind(&arg.kind)))
|
|
.join(", ");
|
|
if args.is_empty() {
|
|
"".into()
|
|
} else {
|
|
format!("args: {{{args}}}",)
|
|
}
|
|
}
|
|
|
|
fn arg_kind(kind: &VariableKind) -> &str {
|
|
match kind {
|
|
VariableKind::Int | VariableKind::Float => "number",
|
|
VariableKind::String => "string",
|
|
VariableKind::Any => "number | string",
|
|
}
|
|
}
|