autocommit 2022-04-08 23:02:38 CEST

src/tree/walker.rs

@@ -108,6 +108,68 @@ impl<T: NodeVisitorBase> NodeVisitor for T {
     }
 }
 
+pub struct GenericNodeVisitor<F> {
+    f: F,
+}
+
+pub fn generic_node_visitor<F: FnMut(Rc<dyn Node>)>(f: F)  -> GenericNodeVisitor<F> {
+    GenericNodeVisitor {
+        f,
+    }
+}
+
+impl<F: FnMut(Rc<dyn Node>)> NodeVisitor for GenericNodeVisitor<F> {
+    fn visit_surface(&mut self, node: &Rc<WlSurface>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_container(&mut self, node: &Rc<ContainerNode>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_toplevel(&mut self, node: &Rc<XdgToplevel>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_popup(&mut self, node: &Rc<XdgPopup>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_display(&mut self, node: &Rc<DisplayNode>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_output(&mut self, node: &Rc<OutputNode>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_float(&mut self, node: &Rc<FloatNode>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_workspace(&mut self, node: &Rc<WorkspaceNode>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_layer_surface(&mut self, node: &Rc<ZwlrLayerSurfaceV1>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+
+    fn visit_xwindow(&mut self, node: &Rc<Xwindow>) {
+        (self.f)(node.clone());
+        node.visit_children(self);
+    }
+}
+
 // pub fn visit_containers<F: FnMut(&Rc<ContainerNode>)>(f: F) -> impl NodeVisitor {
 //     struct V<F>(F);
 //     impl<F: FnMut(&Rc<ContainerNode>)> NodeVisitorBase for V<F> {