From ca2c5a727329e984d1122700bf74fb529e091024 Mon Sep 17 00:00:00 2001
From: Sijawusz Pur Rahnama <sija@sija.pl>
Date: Sat, 29 Oct 2022 18:07:52 +0200
Subject: [PATCH] Small refactor to `DocFinder#visit`

---
 src/contrib/read_type_doc.cr | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/contrib/read_type_doc.cr b/src/contrib/read_type_doc.cr
index d02b12e8..53ec0e2d 100644
--- a/src/contrib/read_type_doc.cr
+++ b/src/contrib/read_type_doc.cr
@@ -1,7 +1,7 @@
 require "compiler/crystal/syntax/*"
 
 private class DocFinder < Crystal::Visitor
-  getter type_name : String?
+  getter type_name : String
   getter doc : String?
 
   def initialize(nodes, @type_name)
@@ -11,11 +11,8 @@ private class DocFinder < Crystal::Visitor
   def visit(node : Crystal::ASTNode)
     return false if @doc
 
-    if node.responds_to?(:name) &&
-       (name = node.name) &&
-       name.is_a?(Crystal::Path) &&
-       name.names.last? == @type_name
-      @doc = node.doc
+    if node.responds_to?(:name) && (name = node.name).is_a?(Crystal::Path)
+      @doc = node.doc if name.names.last? == @type_name
     end
 
     true