From 199e3b73c712a74e36bbb75eebf9e9418f1b1341 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Tue, 14 Aug 2018 14:45:56 +0300
Subject: fields are part of the structure

---
 crates/libsyntax2/src/ast/generated.rs | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

(limited to 'crates/libsyntax2/src/ast')

diff --git a/crates/libsyntax2/src/ast/generated.rs b/crates/libsyntax2/src/ast/generated.rs
index 80670ce71..1ec05c950 100644
--- a/crates/libsyntax2/src/ast/generated.rs
+++ b/crates/libsyntax2/src/ast/generated.rs
@@ -249,6 +249,25 @@ impl<R: TreeRoot> AstNode<R> for NameRef<R> {
 
 impl<R: TreeRoot> NameRef<R> {}
 
+// NamedField
+#[derive(Debug, Clone, Copy)]
+pub struct NamedField<R: TreeRoot = Arc<SyntaxRoot>> {
+    syntax: SyntaxNode<R>,
+}
+
+impl<R: TreeRoot> AstNode<R> for NamedField<R> {
+    fn cast(syntax: SyntaxNode<R>) -> Option<Self> {
+        match syntax.kind() {
+            NAMED_FIELD => Some(NamedField { syntax }),
+            _ => None,
+        }
+    }
+    fn syntax(&self) -> &SyntaxNode<R> { &self.syntax }
+}
+
+impl<R: TreeRoot> ast::NameOwner<R> for NamedField<R> {}
+impl<R: TreeRoot> NamedField<R> {}
+
 // NeverType
 #[derive(Debug, Clone, Copy)]
 pub struct NeverType<R: TreeRoot = Arc<SyntaxRoot>> {
@@ -436,7 +455,13 @@ impl<R: TreeRoot> AstNode<R> for StructDef<R> {
 }
 
 impl<R: TreeRoot> ast::NameOwner<R> for StructDef<R> {}
-impl<R: TreeRoot> StructDef<R> {}
+impl<R: TreeRoot> StructDef<R> {
+    pub fn fields<'a>(&'a self) -> impl Iterator<Item = NamedField<R>> + 'a {
+        self.syntax()
+            .children()
+            .filter_map(NamedField::cast)
+    }
+}
 
 // TraitDef
 #[derive(Debug, Clone, Copy)]
-- 
cgit v1.2.3