fix extract info from trait

Author: bambamboole

src/FunctionalDocBlockExtractor.php

@@ -37,6 +37,9 @@ public function enterNode(Node $node): null
         if ($node instanceof Node\Stmt\Class_ && $node->name) {
             $this->currentClassName = $node->name->toString();
         }
+        if ($node instanceof Node\Stmt\Trait_ && $node->name) {
+            $this->currentClassName = $node->name->toString();
+        }
 
         // Process the doc comment if it exists
         if ($node->getDocComment()) {
@@ -63,6 +66,9 @@ public function leaveNode(Node $node): null
         if ($node instanceof Node\Stmt\Class_) {
             $this->currentClassName = null;
         }
+        if ($node instanceof Node\Stmt\Trait_) {
+            $this->currentClassName = null;
+        }
         if ($node instanceof Node\Stmt\Namespace_) {
             $this->currentNamespace = null;
         }
@@ -230,6 +236,11 @@ private function getOwnerIdentifier(Node $node): string
 
             return ltrim($fqcn, '\\');
         }
+        if ($node instanceof Node\Stmt\Trait_) {
+            $fqcn = $namespace.$node->name->toString();
+
+            return ltrim($fqcn, '\\');
+        }
         if ($node instanceof ClassMethod && $this->currentClassName) {
             $fqcn = $namespace.$this->currentClassName.'::'.$node->name->toString();
 
@@ -249,6 +260,9 @@ private function getDefaultTitleForNode(Node $node): string
         if ($node instanceof Node\Stmt\Class_ && $node->name) {
             return $node->name->toString();
         }
+        if ($node instanceof Node\Stmt\Trait_ && $node->name) {
+            return $node->name->toString();
+        }
         if ($node instanceof ClassMethod && $this->currentClassName) {
             return $this->currentClassName.'::'.$node->name->toString();
         }

tests/Unit/FunctionalDocBlockExtractorTest.php

@@ -378,3 +378,87 @@ class ChildService {}
     expect($doc['navId'])->toBe('child-service');
     expect($doc['navParent'])->toBe('parent-service');
 });
+
+it('can extract functional documentation from trait docblocks', function () {
+    $extractor = new FunctionalDocBlockExtractor;
+    $extractor->setCurrentFilePath('/test/path/TestTrait.php');
+
+    $phpCode = <<<'PHP'
+<?php
+namespace App\Traits;
+
+/**
+ * Timestamp handling trait
+ *
+ * @functional
+ * This trait provides timestamp functionality for models.
+ *
+ * # Key Features
+ * - Automatic timestamp management
+ * - Custom timestamp formats
+ *
+ * @nav Traits / Timestamp Handler
+ * @uses \Carbon\Carbon
+ * @link https://example.com/traits
+ */
+trait TimestampHandler
+{
+    public function updateTimestamps() {}
+}
+PHP;
+
+    $parser = (new ParserFactory)->createForNewestSupportedVersion();
+    $traverser = new NodeTraverser;
+    $traverser->addVisitor($extractor);
+
+    $ast = $parser->parse($phpCode);
+    $traverser->traverse($ast);
+
+    expect($extractor->foundDocs)->toHaveCount(1);
+
+    $doc = $extractor->foundDocs[0];
+    expect($doc['owner'])->toBe('App\Traits\TimestampHandler');
+    expect($doc['navPath'])->toBe('Traits / Timestamp Handler');
+    expect($doc['description'])->toContain('This trait provides timestamp functionality');
+    expect($doc['uses'])->toContain('\Carbon\Carbon');
+    expect($doc['links'])->toContain('https://example.com/traits');
+    expect($doc['sourceFile'])->toBe('/test/path/TestTrait.php');
+});
+
+it('can extract functional documentation from trait methods', function () {
+    $extractor = new FunctionalDocBlockExtractor;
+    $extractor->setCurrentFilePath('/test/path/TestTrait.php');
+
+    $phpCode = <<<'PHP'
+<?php
+namespace App\Traits;
+
+trait TimestampHandler
+{
+    /**
+     * Update timestamps method
+     *
+     * @functional
+     * This method updates the created_at and updated_at timestamps.
+     *
+     * @nav Traits / Update Timestamps Process
+     * @uses \Carbon\Carbon
+     */
+    public function updateTimestamps() {}
+}
+PHP;
+
+    $parser = (new ParserFactory)->createForNewestSupportedVersion();
+    $traverser = new NodeTraverser;
+    $traverser->addVisitor($extractor);
+
+    $ast = $parser->parse($phpCode);
+    $traverser->traverse($ast);
+
+    expect($extractor->foundDocs)->toHaveCount(1);
+
+    $doc = $extractor->foundDocs[0];
+    expect($doc['owner'])->toBe('App\Traits\TimestampHandler::updateTimestamps');
+    expect($doc['navPath'])->toBe('Traits / Update Timestamps Process');
+    expect($doc['description'])->toContain('This method updates the created_at and updated_at timestamps');
+});