1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
use super::*; #[derive(Debug, Clone)] pub struct ClassSpec { pub name: String, pub decorators: Vec<DecoratorSpec>, pub elements: Elements, pub extends: Vec<Name>, } impl ClassSpec { pub fn new(name: &str) -> ClassSpec { ClassSpec { name: name.to_owned(), decorators: Vec::new(), elements: Elements::new(), extends: Vec::new(), } } pub fn push_decorator<D>(&mut self, decorator: D) where D: Into<DecoratorSpec> { self.decorators.push(decorator.into()); } pub fn push<E>(&mut self, element: E) where E: Into<Element> { self.elements.push(element); } pub fn extends<N>(&mut self, name: N) where N: Into<Name> { self.extends.push(name.into()); } } impl From<ClassSpec> for Element { fn from(value: ClassSpec) -> Element { let mut out = Elements::new(); for decorator in value.decorators { out.push(decorator); } let mut decl = Statement::new(); decl.push("class "); decl.push(value.name); if !value.extends.is_empty() { decl.push("("); let mut extends = Statement::new(); for extend in value.extends { extends.push(extend); } decl.push(extends.join(", ")); decl.push(")"); } decl.push(":"); out.push(decl); if value.elements.is_empty() { out.push_nested("pass"); } else { out.push_nested(value.elements.join(Spacing)); } out.into() } }