134 lines
3.3 KiB
TypeScript
134 lines
3.3 KiB
TypeScript
import htmlData from "@wzlin/html-data";
|
|
import { writeFileSync } from "fs";
|
|
import { join } from "path";
|
|
import { RUST_OUT_DIR } from "./_common";
|
|
|
|
const rsTagAttr = ({
|
|
boolean = false,
|
|
caseInsensitive = false,
|
|
collapse = false,
|
|
defaultValue,
|
|
redundantIfEmpty = false,
|
|
trim = false,
|
|
}: {
|
|
boolean?: boolean;
|
|
caseInsensitive?: boolean;
|
|
collapse?: boolean;
|
|
defaultValue?: string;
|
|
redundantIfEmpty?: boolean;
|
|
trim?: boolean;
|
|
}) =>
|
|
`
|
|
AttributeMinification {
|
|
boolean: ${boolean},
|
|
case_insensitive: ${caseInsensitive},
|
|
collapse: ${collapse},
|
|
default_value: ${
|
|
defaultValue == undefined ? "None" : `Some(b"${defaultValue}")`
|
|
},
|
|
redundant_if_empty: ${redundantIfEmpty},
|
|
trim: ${trim},
|
|
}
|
|
`;
|
|
|
|
let code = `
|
|
use lazy_static::lazy_static;
|
|
use rustc_hash::FxHashMap;
|
|
use crate::common::spec::tag::ns::Namespace;
|
|
|
|
pub struct AttributeMinification {
|
|
pub boolean: bool,
|
|
pub case_insensitive: bool,
|
|
pub collapse: bool,
|
|
pub default_value: Option<&'static [u8]>,
|
|
pub redundant_if_empty: bool,
|
|
pub trim: bool,
|
|
}
|
|
|
|
pub enum AttrMapEntry {
|
|
AllNamespaceElements(AttributeMinification),
|
|
SpecificNamespaceElements(FxHashMap<&'static [u8], AttributeMinification>),
|
|
}
|
|
|
|
pub struct ByNamespace {
|
|
// Make pub so this struct can be statically created in gen/attrs.rs.
|
|
pub html: Option<AttrMapEntry>,
|
|
pub svg: Option<AttrMapEntry>,
|
|
}
|
|
|
|
impl ByNamespace {
|
|
fn get(&self, ns: Namespace) -> Option<&AttrMapEntry> {
|
|
match ns {
|
|
Namespace::Html => self.html.as_ref(),
|
|
Namespace::Svg => self.svg.as_ref(),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub struct AttrMap(FxHashMap<&'static [u8], ByNamespace>);
|
|
|
|
impl AttrMap {
|
|
pub const fn new(map: FxHashMap<&'static [u8], ByNamespace>) -> AttrMap {
|
|
AttrMap(map)
|
|
}
|
|
|
|
pub fn get(&self, ns: Namespace, tag: &[u8], attr: &[u8]) -> Option<&AttributeMinification> {
|
|
self.0.get(attr).and_then(|namespaces| namespaces.get(ns)).and_then(|entry| match entry {
|
|
AttrMapEntry::AllNamespaceElements(min) => Some(min),
|
|
AttrMapEntry::SpecificNamespaceElements(map) => map.get(tag),
|
|
})
|
|
}
|
|
}
|
|
|
|
`;
|
|
|
|
code += `
|
|
lazy_static! {
|
|
pub static ref ATTRS: AttrMap = {
|
|
let mut m = FxHashMap::<&'static [u8], ByNamespace>::default();
|
|
${[...Object.entries(htmlData.attributes)]
|
|
.map(
|
|
([attr_name, namespaces]) => ` m.insert(b\"${attr_name}\", ByNamespace {
|
|
${(["html", "svg"] as const)
|
|
.map(
|
|
(ns) =>
|
|
` ${ns}: ` +
|
|
(() => {
|
|
const tagsMap = namespaces[ns];
|
|
if (!tagsMap) {
|
|
return "None";
|
|
}
|
|
const globalAttr = tagsMap["*"];
|
|
if (globalAttr) {
|
|
return `Some(AttrMapEntry::AllNamespaceElements(${rsTagAttr(
|
|
globalAttr
|
|
)}))`;
|
|
}
|
|
const entries = Object.entries(tagsMap);
|
|
return `Some({
|
|
let ${
|
|
entries.length ? "mut" : ""
|
|
} m = FxHashMap::<&'static [u8], AttributeMinification>::default();
|
|
${entries
|
|
.map(
|
|
([tagName, tagAttr]) =>
|
|
` m.insert(b\"${tagName}\", ${rsTagAttr(tagAttr)});`
|
|
)
|
|
.join("\n")}
|
|
AttrMapEntry::SpecificNamespaceElements(m)
|
|
})`;
|
|
})() +
|
|
","
|
|
)
|
|
.join("\n")}
|
|
});
|
|
|
|
`
|
|
)
|
|
.join("")}
|
|
AttrMap::new(m)
|
|
};
|
|
}`;
|
|
|
|
writeFileSync(join(RUST_OUT_DIR, "attrs.rs"), code);
|