diff --git a/.gitattributes b/.gitattributes
index 9ea7204..28039e1 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -10,5 +10,6 @@
 /phpunit.xml.dist export-ignore
 /release-latest-versions.sh export-ignore
 /source/ export-ignore
+/src/ export-ignore
 /visitor.php export-ignore
 /tests/ export-ignore
diff --git a/composer.json b/composer.json
index ff357ce..b7c9def 100644
--- a/composer.json
+++ b/composer.json
@@ -25,6 +25,9 @@
     },
     "minimum-stability": "stable",
     "autoload-dev": {
+        "psr-4": {
+            "PhpStubs\\WordPress\\Core\\": "src/"
+        },
         "classmap": [
             "tests/"
         ]
@@ -43,9 +46,9 @@
         "post-update-cmd": "@composer --working-dir=source/ update --no-interaction",
         "cleanup": "git status --short --ignored | sed -n -e 's#^!! ##p' | xargs -r rm -vrf",
         "test": [
-            "@test:cs",
+            "@test:phpunit",
             "@test:phpstan",
-            "@test:phpunit"
+            "@test:cs"
         ],
         "test:cs": "phpcs",
         "test:cs:fix": "phpcbf",
diff --git a/phpcs.xml.dist b/phpcs.xml.dist
index c566d91..749a233 100644
--- a/phpcs.xml.dist
+++ b/phpcs.xml.dist
@@ -2,7 +2,8 @@
 <ruleset>
     <file>finder.php</file>
     <file>functionMap.php</file>
-    <!-- <file>visitor.php</file> -->
+    <file>src/</file>
+    <file>visitor.php</file>
 
     <rule ref="PSR12NeutronRuleset">
         <exclude name="Generic.Files.LineLength"/>
diff --git a/phpstan.neon.dist b/phpstan.neon.dist
index 94f0ee1..a039dff 100644
--- a/phpstan.neon.dist
+++ b/phpstan.neon.dist
@@ -3,6 +3,7 @@ parameters:
         - finder.php
         - functionMap.php
         - visitor.php
+        - src/
         - tests/
     excludePaths:
         - tests/data/
diff --git a/src/Visitor.php b/src/Visitor.php
new file mode 100644
index 0000000..ab78a78
--- /dev/null
+++ b/src/Visitor.php
@@ -0,0 +1,866 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpStubs\WordPress\Core;
+
+use phpDocumentor\Reflection\DocBlock;
+use phpDocumentor\Reflection\DocBlock\Description;
+use phpDocumentor\Reflection\DocBlock\Tags\Param;
+use phpDocumentor\Reflection\DocBlock\Tags\Return_;
+use phpDocumentor\Reflection\DocBlock\Tags\Var_;
+use phpDocumentor\Reflection\Type;
+use phpDocumentor\Reflection\Types\Never_;
+use phpDocumentor\Reflection\Types\Void_;
+use PhpParser\Comment\Doc;
+use PhpParser\ConstExprEvaluator;
+use PhpParser\Node;
+use PhpParser\NodeFinder;
+use PhpParser\Node\Identifier;
+use PhpParser\Node\Name;
+use PhpParser\Node\Expr\Exit_;
+use PhpParser\Node\Expr\FuncCall;
+use PhpParser\Node\Scalar\String_;
+use PhpParser\Node\Stmt\Class_;
+use PhpParser\Node\Stmt\ClassMethod;
+use PhpParser\Node\Stmt\Expression;
+use PhpParser\Node\Stmt\Function_;
+use PhpParser\Node\Stmt\Property;
+use PhpParser\Node\Stmt\Return_ as Stmt_Return;
+use StubsGenerator\NodeVisitor;
+
+class Visitor extends NodeVisitor
+{
+    private \phpDocumentor\Reflection\DocBlockFactory $docBlockFactory;
+
+    /** @var ?array<string,array<int|string,string>> */
+    private $functionMap = null;
+
+    /** @var array<string, array<int, \PhpStubs\WordPress\Core\WordPressTag>> */
+    private $additionalTags = [];
+
+    /** @var array<string, array<int, string>> */
+    private $additionalTagStrings = [];
+
+    private \PhpParser\NodeFinder $nodeFinder;
+
+    public function __construct()
+    {
+        $this->docBlockFactory = \phpDocumentor\Reflection\DocBlockFactory::createInstance();
+        $this->nodeFinder = new NodeFinder();
+    }
+
+    /** @return int|null */
+    public function enterNode(Node $node)
+    {
+        $voidOrNever = $this->voidOrNever($node);
+
+        parent::enterNode($node);
+
+        if (! ($node instanceof Function_) && ! ($node instanceof ClassMethod) && ! ($node instanceof Property) && ! ($node instanceof Class_)) {
+            return null;
+        }
+
+        $docComment = $node->getDocComment();
+
+        if (! ($docComment instanceof Doc)) {
+            return null;
+        }
+
+        $symbolName = self::getNodeName($node);
+
+        if ($node instanceof ClassMethod) {
+            $parent = $this->stack[count($this->stack) - 2];
+            \assert($parent instanceof \PhpParser\Node\Stmt\ClassLike);
+
+            if (isset($parent->name)) {
+                $symbolName = sprintf(
+                    '%1$s::%2$s',
+                    $parent->name->name,
+                    $node->name->name
+                );
+            }
+        }
+        $node->setAttribute('fullSymbolName', $symbolName);
+
+        $additions = $this->generateAdditionalTagsFromDoc($docComment);
+        if (count($additions) > 0) {
+            $this->additionalTags[$symbolName] = $additions;
+        }
+
+        $additions = $this->getAdditionalTagsFromMap($symbolName);
+        if (count($additions) > 0) {
+            $this->additionalTagStrings[$symbolName] = $additions;
+        }
+
+        if ($voidOrNever !== '') {
+            $addition = sprintf(
+                '@phpstan-return %s',
+                $voidOrNever === 'never'
+                    ? (new Never_())->__toString()
+                    : (new Void_())->__toString()
+            );
+            if (
+                ! isset($this->additionalTagStrings[$symbolName])
+                || ! in_array($addition, $this->additionalTagStrings[$symbolName], true)
+            ) {
+                $this->additionalTagStrings[$symbolName][] = $addition;
+            }
+        }
+
+        return null;
+    }
+
+    private static function getNodeName(Node $node): string
+    {
+        if ((($node instanceof Function_) || ($node instanceof ClassMethod) || ($node instanceof Class_)) && $node->name instanceof Identifier) {
+            return $node->name->name;
+        }
+
+        if ($node instanceof Property) {
+            return sprintf(
+                'property_%s',
+                uniqid()
+            );
+        }
+
+        return '';
+    }
+
+    /**
+     * @return array<\PhpParser\Node>
+     */
+    public function getStubStmts(): array
+    {
+        $stmts = parent::getStubStmts();
+
+        foreach ($stmts as $stmt) {
+            $this->postProcessNode($stmt);
+        }
+        return $stmts;
+    }
+
+    private function postProcessNode(Node $node): void
+    {
+        if (isset($node->stmts) && is_array($node->stmts)) {
+            foreach ($node->stmts as $stmt) {
+                $this->postProcessNode($stmt);
+            }
+        }
+
+        if (! ($node instanceof Function_) && ! ($node instanceof ClassMethod) && ! ($node instanceof Property) && ! ($node instanceof Class_)) {
+            return;
+        }
+
+        $name = $node->getAttribute('fullSymbolName');
+
+        if ($name === null) {
+            return;
+        }
+
+        $docComment = $node->getDocComment();
+
+        if (! ($docComment instanceof Doc)) {
+            return;
+        }
+
+        $newDocComment = $this->addTags($name, $docComment);
+
+        if ($newDocComment instanceof Doc) {
+            $node->setDocComment($newDocComment);
+        }
+
+        if (! isset($this->additionalTagStrings[$name])) {
+            return;
+        }
+
+        $docComment = $node->getDocComment();
+
+        if (! ($docComment instanceof Doc)) {
+            return;
+        }
+
+        $newDocComment = $this->addStringTags($name, $docComment);
+
+        if (! ($newDocComment instanceof Doc)) {
+            return;
+        }
+
+        $node->setDocComment($newDocComment);
+    }
+
+    /**
+     * @return array<int, \PhpStubs\WordPress\Core\WordPressTag>
+     */
+    private function generateAdditionalTagsFromDoc(Doc $docComment): array
+    {
+        $docCommentText = $docComment->getText();
+
+        try {
+            $docblock = $this->docBlockFactory->create($docCommentText);
+        } catch (\RuntimeException $e) {
+            return [];
+        } catch (\InvalidArgumentException $e) {
+            return [];
+        }
+
+        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Param> $params*/
+        $params = $docblock->getTagsByName('param');
+
+        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Return_> $returns */
+        $returns = $docblock->getTagsByName('return');
+
+        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Var_> $vars */
+        $vars = $docblock->getTagsByName('var');
+
+        /** @var list<\PhpStubs\WordPress\Core\WordPressTag> $additions */
+        $additions = [];
+
+        foreach ($params as $param) {
+            $addition = self::getAdditionFromParam($param);
+
+            if (! ($addition instanceof WordPressTag)) {
+                continue;
+            }
+
+            $additions[] = $addition;
+        }
+
+        foreach ($returns as $return) {
+            $addition = self::getAdditionFromReturn($return);
+
+            if (! ($addition instanceof WordPressTag)) {
+                continue;
+            }
+
+            $additions[] = $addition;
+        }
+
+        foreach ($vars as $var) {
+            $addition = self::getAdditionFromVar($var);
+
+            if (! ($addition instanceof WordPressTag)) {
+                continue;
+            }
+
+            $additions[] = $addition;
+        }
+
+        return $additions;
+    }
+
+    private function addTags(string $name, Doc $docComment): ?Doc
+    {
+        $additions = $this->additionalTags[$name] ?? [];
+        $docCommentText = $docComment->getText();
+
+        try {
+            $docblock = $this->docBlockFactory->create($docCommentText);
+        } catch (\RuntimeException $e) {
+            return null;
+        } catch (\InvalidArgumentException $e) {
+            return null;
+        }
+
+        $additions = $this->discoverInheritedArgs($docblock, $additions);
+
+        /** @var list<string> $additionStrings */
+        $additionStrings = array_map(
+            static function (WordPressTag $tag): string {
+                $lines = $tag->format();
+
+                if (count($lines) === 0) {
+                    return '';
+                }
+
+                return ' * ' . implode("\n * ", $lines);
+            },
+            $additions
+        );
+
+        $additionStrings = array_filter($additionStrings);
+
+        if (count($additionStrings) === 0) {
+            return null;
+        }
+
+        $newDocComment = sprintf(
+            "%s\n%s\n */",
+            substr($docCommentText, 0, -4),
+            implode("\n", $additionStrings)
+        );
+
+        return new Doc($newDocComment, $docComment->getStartLine(), $docComment->getStartFilePos());
+    }
+
+    /**
+     * @param array<int, \PhpStubs\WordPress\Core\WordPressTag> $additions
+     * @return array<int, \PhpStubs\WordPress\Core\WordPressTag>
+     */
+    private function discoverInheritedArgs(DocBlock $docblock, array $additions): array
+    {
+        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Param> $params */
+        $params = $docblock->getTagsByName('param');
+
+        $phpStanParams = array_filter(
+            $additions,
+            static function (WordPressTag $addition): bool {
+                return $addition->tag === '@phpstan-param';
+            }
+        );
+
+        foreach ($params as $param) {
+            $inherited = $this->getInheritedTagsForParam($param);
+
+            if (count($inherited) === 0) {
+                continue;
+            }
+
+            foreach ($phpStanParams as $addition) {
+                foreach ($inherited as $inherit) {
+                    if ($addition->name !== $inherit->name) {
+                        continue;
+                    }
+
+                    $addition->children = array_merge($addition->children, $inherit->children);
+                    continue 3;
+                }
+            }
+
+            $additions = array_merge($additions, $inherited);
+        }
+
+        return $additions;
+    }
+
+    /**
+     * @return array<int, \PhpStubs\WordPress\Core\WordPressTag>
+     */
+    private function getInheritedTagsForParam(Param $param): array
+    {
+        $type = $param->getType();
+
+        if (! $type instanceof Type) {
+            return [];
+        }
+
+        $typeName = self::getTypeNameFromType($type);
+
+        if ($typeName === null) {
+            return [];
+        }
+
+        $paramDescription = $param->getDescription();
+
+        if (! $paramDescription instanceof Description) {
+            return [];
+        }
+
+        list($description) = explode("\n\n", $paramDescription->__toString());
+
+        if (strpos($description, '()') === false) {
+            return [];
+        }
+
+        $description = str_replace("\n", ' ', $description);
+        $description = str_replace('@see \\', '@see ', $description);
+        $additions = [];
+
+        foreach ($this->additionalTags as $symbolName => $tags) {
+            $search = sprintf(
+                'see %s()',
+                $symbolName
+            );
+
+            if (stripos($description, $search) === false) {
+                continue;
+            }
+
+            $match = self::getMatchingInheritedTag($param, $tags, $symbolName);
+
+            if (! ($match instanceof WordPressTag)) {
+                continue;
+            }
+
+            $additions[] = $match;
+        }
+
+        return $additions;
+    }
+
+    /**
+     * @param array<int, \PhpStubs\WordPress\Core\WordPressTag> $tags
+     */
+    private static function getMatchingInheritedTag(Param $param, array $tags, string $symbolName): ?WordPressTag
+    {
+        $paramName = $param->getVariableName();
+        $matchNames = [
+            $paramName,
+            'args',
+            'options',
+            'query',
+        ];
+        $matchingTags = array_filter(
+            $tags,
+            static function (WordPressTag $tag) use ($matchNames): bool {
+                return in_array($tag->name, $matchNames, true);
+            }
+        );
+
+        foreach ($matchingTags as $tag) {
+            $addTag = clone $tag;
+            $addTag->name = $paramName;
+            $addTag->description = sprintf(
+                'See %s()',
+                $symbolName
+            );
+
+            return $addTag;
+        }
+
+        return null;
+    }
+
+    /**
+     * @return list<string>
+     */
+    private function getAdditionalTagsFromMap(string $symbolName): array
+    {
+        if (! isset($this->functionMap)) {
+            $this->functionMap = require dirname(__DIR__) . '/functionMap.php';
+        }
+
+        if (! isset($this->functionMap[$symbolName])) {
+            return [];
+        }
+
+        $parameters = $this->functionMap[$symbolName];
+        $returnType = array_shift($parameters);
+        $additions = [];
+
+        foreach ($parameters as $paramName => $paramType) {
+            if (strpos($paramName, '@') === 0) {
+                $format = ( $paramType === '' ) ? '%s' : '%s %s';
+                $additions[] = sprintf(
+                    $format,
+                    $paramName,
+                    $paramType
+                );
+                continue;
+            }
+
+            $additions[] = sprintf(
+                '@phpstan-param %s $%s',
+                $paramType,
+                $paramName
+            );
+        }
+
+        if ($returnType !== null) {
+            $additions[] = sprintf(
+                '@phpstan-return %s',
+                $returnType
+            );
+        }
+
+        return $additions;
+    }
+
+    private function addStringTags(string $name, Doc $docComment): ?Doc
+    {
+        if (! isset($this->additionalTagStrings[$name])) {
+            return null;
+        }
+
+        $additions = $this->additionalTagStrings[$name];
+
+        $docCommentText = $docComment->getText();
+        $newDocComment = sprintf(
+            "%s\n * %s\n */",
+            substr($docCommentText, 0, -4),
+            implode("\n * ", $additions)
+        );
+
+        return new Doc($newDocComment, $docComment->getStartLine(), $docComment->getStartFilePos());
+    }
+
+    private function getAdditionFromParam(Param $tag): ?WordPressTag
+    {
+        $tagDescription = $tag->getDescription();
+        $tagVariableName = $tag->getVariableName();
+        $tagVariableType = $tag->getType();
+
+        // Skip if information we need is missing.
+        if (! $tagDescription instanceof Description || ! $tagVariableName || ! $tagVariableType instanceof Type) {
+            return null;
+        }
+
+        $tagDescriptionType = self::getTypeNameFromDescription($tagDescription, $tagVariableType);
+
+        if ($tagDescriptionType !== null) {
+            $tag = new WordPressTag();
+            $tag->tag = '@phpstan-param';
+            $tag->type = $tagDescriptionType;
+            $tag->name = $tagVariableName;
+
+            return $tag;
+        }
+
+        $elements = self::getElementsFromDescription($tagDescription, true);
+
+        if (count($elements) === 0) {
+            return null;
+        }
+
+        $tagVariableType = self::getTypeNameFromType($tagVariableType);
+
+        if ($tagVariableType === null) {
+            return null;
+        }
+
+        // It's common for an args parameter to accept a query var string or array with `string|array`.
+        // Remove the accepted string type for these so we get the strongest typing we can manage.
+        $tagVariableType = str_replace(['|string', 'string|'], '', $tagVariableType);
+
+        $tag = new WordPressTag();
+        $tag->tag = '@phpstan-param';
+        $tag->type = $tagVariableType;
+        $tag->name = $tagVariableName;
+        $tag->children = $elements;
+
+        return $tag;
+    }
+
+    private function getAdditionFromReturn(Return_ $tag): ?WordPressTag
+    {
+        $tagDescription = $tag->getDescription();
+        $tagVariableType = $tag->getType();
+
+        // Skip if information we need is missing.
+        if (! $tagDescription instanceof Description || ! $tagVariableType instanceof Type) {
+            return null;
+        }
+
+        $tagDescriptionType = self::getTypeNameFromDescription($tagDescription, $tagVariableType);
+
+        if ($tagDescriptionType !== null) {
+            $tag = new WordPressTag();
+            $tag->tag = '@phpstan-return';
+            $tag->type = $tagDescriptionType;
+
+            return $tag;
+        }
+
+        $elements = self::getElementsFromDescription($tagDescription, false);
+
+        if (count($elements) === 0) {
+            return null;
+        }
+
+        $tagVariableType = self::getTypeNameFromType($tagVariableType);
+
+        if ($tagVariableType === null) {
+            return null;
+        }
+
+        $tag = new WordPressTag();
+        $tag->tag = '@phpstan-return';
+        $tag->type = $tagVariableType;
+        $tag->children = $elements;
+
+        return $tag;
+    }
+
+    private static function getAdditionFromVar(Var_ $tag): ?WordPressTag
+    {
+        $tagDescription = $tag->getDescription();
+        $tagVariableType = $tag->getType();
+
+        // Skip if information we need is missing.
+        if (! $tagDescription instanceof Description || ! $tagVariableType instanceof Type) {
+            return null;
+        }
+
+        $elements = self::getElementsFromDescription($tagDescription, false);
+
+        if (count($elements) === 0) {
+            return null;
+        }
+
+        $tagVariableType = self::getTypeNameFromType($tagVariableType);
+
+        if ($tagVariableType === null) {
+            return null;
+        }
+
+        $tag = new WordPressTag();
+        $tag->tag = '@phpstan-var';
+        $tag->type = $tagVariableType;
+        $tag->children = $elements;
+
+        return $tag;
+    }
+
+    private static function getTypeNameFromDescription(Description $tagVariableDescription, Type $tagVariableType): ?string
+    {
+        if (! ($tagVariableType instanceof \phpDocumentor\Reflection\Types\String_)) {
+            return null;
+        }
+
+        return self::getTypeNameFromDescriptionString($tagVariableDescription->__toString());
+    }
+
+    private static function getTypeNameFromDescriptionString(?string $tagDescription = null): ?string
+    {
+        if ($tagDescription === null) {
+            return null;
+        }
+
+        $fullDescription = str_replace("\n", ' ', $tagDescription);
+
+        /**
+         * This matches phrases that contain a list of two or more single-quoted strings, with the last
+         * item separated by 'or' or 'and'. The Oxford comma is optional. For example:
+         *
+         * - Either 'am', 'pm', 'AM', or 'PM'
+         * - Accepts 'comment' or 'term'
+         * - Either 'plugin' or 'theme'
+         * - Accepts 'big', or 'little'.
+         * - One of 'default', 'theme', or 'custom'
+         * - Either 'network-active', 'active' or 'inactive'
+         * - : 'top' or 'bottom'
+         */
+        $matched = preg_match("#(?>returns|either|one of|accepts|values are|:) ('.+'),? (?>or|and) '([^']+)'#i", $fullDescription, $matches);
+
+        if (! $matched) {
+            return null;
+        }
+
+        list(, $items, $final) = $matches;
+
+        // Pluck out phrases between single quotes, so messy sentences are handled:
+        preg_match_all("#'([^']+)'#", $items, $matches);
+
+        list(,$accepted) = $matches;
+
+        // Append the final item:
+        $accepted[] = $final;
+
+        return "'" . implode("'|'", $accepted) . "'";
+    }
+
+    private static function getTypeNameFromType(Type $tagVariableType): ?string
+    {
+        return self::getTypeNameFromString($tagVariableType->__toString());
+    }
+
+    private static function getTypeNameFromString(string $tagVariable): ?string
+    {
+        // PHPStan doesn't support typed array shapes (`int[]{...}`) so replace
+        // typed arrays such as `int[]` with `array`.
+        $tagVariableType = preg_replace('#[a-zA-Z0-9_]+\[\]#', 'array', $tagVariable);
+
+        if ($tagVariableType === null) {
+            return null;
+        }
+
+        $tagVariableType = str_replace(
+            [
+            'stdClass',
+            '\\object',
+            ],
+            'object',
+            $tagVariableType
+        );
+
+        $supportedTypes = [
+            'object',
+            'array',
+        ];
+
+        foreach ($supportedTypes as $supportedType) {
+            if (strpos($tagVariableType, "{$supportedType}|") === false) {
+                continue;
+            }
+            // Move the type that uses the hash notation to the end of union types so the shape works.
+            $tagVariableType = str_replace("{$supportedType}|", '', $tagVariableType) . "|{$supportedType}";
+        }
+
+        return $tagVariableType;
+    }
+
+    /**
+     * @return list<\PhpStubs\WordPress\Core\WordPressArg>
+     */
+    private static function getElementsFromDescription(Description $tagDescription, bool $optional): array
+    {
+        $text = $tagDescription->__toString();
+
+        // Skip if the description doesn't contain at least one correctly
+        // formatted `@type`, which indicates an array hash.
+        if (strpos($text, '    @type ') === false) {
+            return [];
+        }
+
+        return self::getTypesAtLevel($text, $optional, 1);
+    }
+
+    /**
+     * @return list<\PhpStubs\WordPress\Core\WordPressArg>
+     */
+    private static function getTypesAtLevel(string $text, bool $optional, int $level): array
+    {
+        // Populate `$types` with the value of each top level `@type`.
+        $spaces = str_repeat(' ', ($level * 4));
+        $types = preg_split("/\R+{$spaces}@type /", $text);
+
+        if ($types === false) {
+            return [];
+        }
+
+        unset($types[0]);
+        $elements = [];
+
+        foreach ($types as $typeTag) {
+            $parts = preg_split('#\s+#', trim($typeTag), 3);
+
+            if ($parts === false || count($parts) < 2) {
+                return [];
+            }
+
+            list($type, $name) = $parts;
+
+            $optionalArg = $optional;
+            $nameTrimmed = ltrim($name, '$');
+
+            if (is_numeric($nameTrimmed)) {
+                $optionalArg = false;
+            } elseif ($optional && ($level > 1)) {
+                $optionalArg = self::isOptional($parts[2]);
+            }
+
+            if (strpos($name, '...$') !== false) {
+                $name = null;
+            } elseif (strpos($name, '$') !== 0) {
+                return [];
+            } else {
+                $name = $nameTrimmed;
+            }
+
+            $arg = new WordPressArg();
+            $arg->type = $type;
+            $arg->optional = $optionalArg;
+            $arg->name = $name;
+
+            $nextLevel = $level + 1;
+            $subTypes = self::getTypesAtLevel($typeTag, $optional, $nextLevel);
+
+            if (count($subTypes) > 0) {
+                $type = self::getTypeNameFromString($type);
+
+                if ($type !== null) {
+                    $arg->type = $type;
+                }
+                $arg->children = $subTypes;
+            }
+
+            $elements[] = $arg;
+        }
+
+        return $elements;
+    }
+
+    private static function isOptional(string $description): bool
+    {
+        return (stripos($description, 'Optional') !== false)
+            || (stripos($description, 'Default ') !== false)
+            || (stripos($description, 'Default: ') !== false)
+            || (stripos($description, 'Defaults to ') !== false);
+    }
+
+    private function voidOrNever(Node $node): string
+    {
+        if (! ($node instanceof Function_) && ! ($node instanceof ClassMethod)) {
+            return '';
+        }
+
+        if (! isset($node->stmts) || count($node->stmts) === 0) {
+            // Interfaces and abstract methods.
+            return '';
+        }
+
+        $return = $this->nodeFinder->findInstanceOf($node, Stmt_Return::class);
+
+        // If there is a return statement, it's not return type never.
+        if (count($return) !== 0) {
+            // If there is at least one return statement that is not void,
+            // it's not return type void.
+            if (
+                $this->nodeFinder->findFirst(
+                    $return,
+                    static function (Node $node): bool {
+                        return isset($node->expr);
+                    }
+                ) instanceof Node
+            ) {
+                return '';
+            }
+            // If there is no return statement that is not void,
+            // it's return type void.
+            return 'void';
+        }
+
+        // Check for never return type.
+        foreach ($node->stmts as $stmt) {
+            if (! ($stmt instanceof Expression)) {
+                continue;
+            }
+            // If a first level statement is exit/die, it's return type never.
+            if ($stmt->expr instanceof Exit_) {
+                if ($stmt->expr->expr instanceof String_) {
+                    if (strpos($stmt->expr->expr->value, 'must be overridden') !== false) {
+                        return '';
+                    }
+                }
+                return 'never';
+            }
+            if (! ($stmt->expr instanceof FuncCall) || ! ($stmt->expr->name instanceof Name)) {
+                continue;
+            }
+            $name = $stmt->expr->name;
+            // If a first level statement is a call to wp_send_json(_success/error),
+            // it's return type never.
+            if (strpos($name->toString(), 'wp_send_json') === 0) {
+                return 'never';
+            }
+            // Skip all functions but wp_die().
+            if (strpos($name->toString(), 'wp_die') !== 0) {
+                continue;
+            }
+            $args = $stmt->expr->getArgs();
+            // If wp_die is called without 3rd parameter, it's return type never.
+            if (count($args) < 3) {
+                return 'never';
+            }
+            // If wp_die is called with 3rd parameter, we need additional checks.
+            try {
+                $arg = (new ConstExprEvaluator())->evaluateSilently($args[2]->value);
+            } catch (\PhpParser\ConstExprEvaluationException $e) {
+                // If we don't know the value of the 3rd parameter, we can't be sure.
+                continue;
+            }
+
+            if (is_int($arg)) {
+                return 'never';
+            }
+            if (is_array($arg)) {
+                if (! isset($arg['exit']) || (bool)$arg['exit'] === true) {
+                    return 'never';
+                }
+            }
+
+            continue;
+        }
+        return '';
+    }
+}
diff --git a/src/WithChildren.php b/src/WithChildren.php
new file mode 100644
index 0000000..0aacb6f
--- /dev/null
+++ b/src/WithChildren.php
@@ -0,0 +1,42 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpStubs\WordPress\Core;
+
+abstract class WithChildren
+{
+    /** @var list<\PhpStubs\WordPress\Core\WordPressArg> */
+    public $children = [];
+
+    public function isArrayShape(): bool
+    {
+        foreach ($this->children as $child) {
+            if ($child->name === null) {
+                return false;
+            }
+        }
+
+        return $this->hasChildren();
+    }
+
+    public function isMixedShape(): bool
+    {
+        $hasStaticKey = false;
+
+        foreach ($this->children as $child) {
+            if ($child->name !== null) {
+                $hasStaticKey = true;
+            } elseif ($hasStaticKey) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
+    public function hasChildren(): bool
+    {
+        return count($this->children) > 0;
+    }
+}
diff --git a/src/WordPressArg.php b/src/WordPressArg.php
new file mode 100644
index 0000000..3f0dac9
--- /dev/null
+++ b/src/WordPressArg.php
@@ -0,0 +1,98 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpStubs\WordPress\Core;
+
+final class WordPressArg extends WithChildren
+{
+    /** @var string */
+    public $type;
+
+    /** @var bool */
+    public $optional = false;
+
+    /** @var ?string */
+    public $name = null;
+
+    /** @return list<string> */
+    public function format(int $level = 1): array
+    {
+        $strings = [];
+        $padding = str_repeat(' ', ($level * 2));
+
+        if ($this->isMixedShape()) {
+            return [];
+        }
+
+        if (count($this->children) === 0) {
+            return $this->formatChildless($strings, $padding);
+        }
+
+        $childStrings = [];
+
+        foreach ($this->children as $child) {
+            $childStrings = array_merge($childStrings, $child->format($level + 1));
+        }
+
+        if (count($childStrings) === 0) {
+            return [];
+        }
+
+        if ($this->isArrayShape()) {
+            if ($this->name === null) {
+                return array_merge($strings, $childStrings);
+            }
+            return $this->formatArrayShape($strings, $childStrings, $padding);
+        }
+
+        return $this->formatArrayWithoutShape($strings, $childStrings, $padding);
+    }
+
+    /**
+     * @param list<string> $strings
+     * @return list<string>
+     */
+    private function formatChildless(array $strings, string $padding): array
+    {
+        $strings[] = $this->formatLine('%s%s%s: %s,', $padding);
+        return $strings;
+    }
+
+    /**
+     * @param list<string> $strings
+     * @param list<string> $childStrings
+     * @return list<string>
+     */
+    private function formatArrayShape(array $strings, array $childStrings, string $padding): array
+    {
+        $strings[] = $this->formatLine('%s%s%s: %s{', $padding);
+        $strings = array_merge($strings, $childStrings);
+        $strings[] = sprintf('%s},', $padding);
+        return $strings;
+    }
+
+    /**
+     * @param list<string> $strings
+     * @param list<string> $childStrings
+     * @return list<string>
+     */
+    private function formatArrayWithoutShape(array $strings, array $childStrings, string $padding): array
+    {
+        $strings[] = $this->formatLine('%s%s%s: array<array-key, %s{', $padding);
+        $strings = array_merge($strings, $childStrings);
+        $strings[] = sprintf('%s}>,', $padding);
+        return $strings;
+    }
+
+    private function formatLine(string $format, string $padding): string
+    {
+        return sprintf(
+            $format,
+            $padding,
+            $this->name,
+            ($this->optional) ? '?' : '',
+            $this->type
+        );
+    }
+}
diff --git a/src/WordPressTag.php b/src/WordPressTag.php
new file mode 100644
index 0000000..eb86380
--- /dev/null
+++ b/src/WordPressTag.php
@@ -0,0 +1,96 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpStubs\WordPress\Core;
+
+final class WordPressTag extends WithChildren
+{
+    /** @var string */
+    public $tag;
+
+    /** @var string */
+    public $type;
+
+    /** @var ?string */
+    public $name = null;
+
+    /** @var ?string */
+    public $description = null;
+
+    /**
+     * @return list<string>
+     */
+    public function format(): array
+    {
+        if (! $this->hasChildren()) {
+            return [
+                sprintf(
+                    '%s %s%s',
+                    $this->tag,
+                    $this->type,
+                    $this->name !== null ? " \${$this->name}" : ''
+                ),
+            ];
+        }
+
+        if ($this->isMixedShape()) {
+            return [];
+        }
+
+        $strings = [];
+        $childStrings = [];
+        $level = 1;
+
+        if (! $this->isArrayShape()) {
+            $level = 0;
+        }
+
+        foreach ($this->children as $child) {
+            $childStrings = array_merge($childStrings, $child->format($level));
+        }
+
+        if (count($childStrings) === 0) {
+            return [];
+        }
+
+        $name = $this->name !== null ? " \${$this->name}" : '';
+
+        if ($this->isArrayShape()) {
+            $strings[] = sprintf(
+                '%s %s{',
+                $this->tag,
+                $this->type
+            );
+        } else {
+            if (count($this->children) <= 0 || count($this->children[0]->children) <= 0) {
+                $strings[] = sprintf(
+                    '%s array<int|string, %s>%s',
+                    $this->tag,
+                    $this->type,
+                    $name
+                );
+                return $strings;
+            }
+
+            $strings[] = sprintf(
+                '%s %s<int|string, array{',
+                $this->tag,
+                $this->type
+            );
+        }
+
+        $strings = array_merge($strings, $childStrings);
+        $description = '';
+
+        if ($this->description !== null) {
+            $description = " {$this->description}";
+        }
+
+        $strings[] = $this->isArrayShape()
+            ? sprintf('}%s%s', $name, $description)
+            : sprintf('}>%s%s', $name, $description);
+
+        return $strings;
+    }
+}
diff --git a/visitor.php b/visitor.php
index ba5d55e..8e41764 100644
--- a/visitor.php
+++ b/visitor.php
@@ -2,1072 +2,6 @@
 
 declare(strict_types=1);
 
-use phpDocumentor\Reflection\DocBlock;
-use phpDocumentor\Reflection\DocBlock\Description;
-use phpDocumentor\Reflection\DocBlock\Tags\Param;
-use phpDocumentor\Reflection\DocBlock\Tags\Return_;
-use phpDocumentor\Reflection\DocBlock\Tags\Var_;
-use phpDocumentor\Reflection\Type;
-use phpDocumentor\Reflection\Types\Never_;
-use phpDocumentor\Reflection\Types\Void_;
-use PhpParser\Comment\Doc;
-use PhpParser\ConstExprEvaluator;
-use PhpParser\Node;
-use PhpParser\NodeFinder;
-use PhpParser\Node\Identifier;
-use PhpParser\Node\Name;
-use PhpParser\Node\Expr\Exit_;
-use PhpParser\Node\Expr\FuncCall;
-use PhpParser\Node\Scalar\String_;
-use PhpParser\Node\Stmt\Class_;
-use PhpParser\Node\Stmt\ClassMethod;
-use PhpParser\Node\Stmt\Expression;
-use PhpParser\Node\Stmt\Function_;
-use PhpParser\Node\Stmt\Property;
-use PhpParser\Node\Stmt\Return_ as Stmt_Return;
-use StubsGenerator\NodeVisitor;
+require sprintf('%s/vendor/autoload.php', __DIR__);
 
-abstract class WithChildren
-{
-    /** @var list<\WordPressArg> */
-    public $children = [];
-
-    public function isArrayShape(): bool
-    {
-        foreach ($this->children as $child) {
-            if ($child->name === null) {
-                return false;
-            }
-        }
-
-        return $this->hasChildren();
-    }
-
-    public function isMixedShape(): bool
-    {
-        $hasStaticKey = false;
-
-        foreach ($this->children as $child) {
-            if ($child->name !== null) {
-                $hasStaticKey = true;
-            } elseif ($hasStaticKey) {
-                return true;
-            }
-        }
-
-        return false;
-    }
-
-    public function hasChildren(): bool
-    {
-        return count($this->children) > 0;
-    }
-}
-
-final class WordPressTag extends WithChildren
-{
-    /** @var string */
-    public $tag;
-
-    /** @var string */
-    public $type;
-
-    /** @var ?string */
-    public $name = null;
-
-    /** @var ?string */
-    public $description = null;
-
-    /**
-     * @return list<string>
-     */
-    public function format(): array
-    {
-        if (! $this->hasChildren()) {
-            return [
-                sprintf(
-                    '%s %s%s',
-                    $this->tag,
-                    $this->type,
-                    $this->name !== null ? " \$$this->name" : ''
-                ),
-            ];
-        }
-
-        if ($this->isMixedShape()) {
-            return [];
-        }
-
-        $strings = [];
-        $childStrings = [];
-        $level = 1;
-
-        if (! $this->isArrayShape()) {
-            $level = 0;
-        }
-
-        foreach ($this->children as $child) {
-            $childStrings = array_merge($childStrings, $child->format($level));
-        }
-
-        if (count($childStrings) === 0) {
-            return [];
-        }
-
-        $name = $this->name !== null ? " \$$this->name" : '';
-
-        if ($this->isArrayShape()) {
-            $strings[] = sprintf(
-                '%s %s{',
-                $this->tag,
-                $this->type
-            );
-        } else {
-            if (count($this->children) > 0 && count($this->children[0]->children) > 0) {
-                $strings[] = sprintf(
-                    '%s %s<int|string, array{',
-                    $this->tag,
-                    $this->type
-                );
-            } else {
-                $strings[] = sprintf(
-                    '%s array<int|string, %s>%s',
-                    $this->tag,
-                    $this->type,
-                    $name
-                );
-                return $strings;
-            }
-        }
-
-        $strings = array_merge($strings, $childStrings);
-        $description = '';
-
-        if ($this->description !== null) {
-            $description = " $this->description";
-        }
-
-        $strings[] = $this->isArrayShape()
-            ? sprintf('}%s%s', $name, $description)
-            : sprintf('}>%s%s', $name, $description);
-
-        return $strings;
-    }
-}
-
-final class WordPressArg extends WithChildren
-{
-    /** @var string */
-    public $type;
-
-    /** @var bool */
-    public $optional = false;
-
-    /** @var ?string */
-    public $name = null;
-
-    /** @return list<string> */
-    public function format(int $level = 1): array
-    {
-        $strings = [];
-        $padding = str_repeat(' ', ($level * 2));
-
-        if ($this->isMixedShape()) {
-            return [];
-        }
-
-        if (count($this->children) === 0) {
-            return $this->formatChildless($strings, $padding);
-        }
-
-        $childStrings = [];
-
-        foreach ($this->children as $child) {
-            $childStrings = array_merge($childStrings, $child->format($level + 1));
-        }
-
-        if (count($childStrings) === 0) {
-            return [];
-        }
-
-        if ($this->isArrayShape()) {
-            if ($this->name === null) {
-                return array_merge($strings, $childStrings);
-            }
-            return $this->formatArrayShape($strings, $childStrings, $padding);
-        }
-
-        return $this->formatArrayWithoutShape($strings, $childStrings, $padding);
-    }
-
-    /**
-     * @param list<string> $strings
-     * @return list<string>
-     */
-    private function formatChildless(array $strings, string $padding): array
-    {
-        $strings[] = $this->formatLine('%s%s%s: %s,', $padding);
-        return $strings;
-    }
-
-    /**
-     * @param list<string> $strings
-     * @param list<string> $childStrings
-     * @return list<string>
-     */
-    private function formatArrayShape(array $strings, array $childStrings, string $padding): array
-    {
-        $strings[] = $this->formatLine('%s%s%s: %s{', $padding);
-        $strings = array_merge($strings, $childStrings);
-        $strings[] = sprintf('%s},', $padding);
-        return $strings;
-    }
-
-    /**
-     * @param list<string> $strings
-     * @param list<string> $childStrings
-     * @return list<string>
-     */
-    private function formatArrayWithoutShape(array $strings, array $childStrings, string $padding): array
-    {
-        $strings[] = $this->formatLine('%s%s%s: array<array-key, %s{', $padding);
-        $strings = array_merge($strings, $childStrings);
-        $strings[] = sprintf('%s}>,', $padding);
-        return $strings;
-    }
-
-    private function formatLine(string $format, string $padding): string
-    {
-        return sprintf(
-            $format,
-            $padding,
-            $this->name,
-            ($this->optional) ? '?' : '',
-            $this->type
-        );
-    }
-}
-
-return new class extends NodeVisitor {
-    private \phpDocumentor\Reflection\DocBlockFactory $docBlockFactory;
-
-    /** @var ?array<string,array<int|string,string>> */
-    private $functionMap = null;
-
-    /** @var array<string, array<int, \WordPressTag>> */
-    private $additionalTags = [];
-
-    /** @var array<string, array<int, string>> */
-    private $additionalTagStrings = [];
-
-    private \PhpParser\NodeFinder $nodeFinder;
-
-    public function __construct()
-    {
-        $this->docBlockFactory = \phpDocumentor\Reflection\DocBlockFactory::createInstance();
-        $this->nodeFinder = new NodeFinder();
-    }
-
-    public function enterNode(Node $node)
-    {
-        $voidOrNever = $this->voidOrNever($node);
-
-        parent::enterNode($node);
-
-        if (!($node instanceof Function_) && !($node instanceof ClassMethod) && !($node instanceof Property) && !($node instanceof Class_)) {
-            return null;
-        }
-
-        $docComment = $node->getDocComment();
-
-        if (!($docComment instanceof Doc)) {
-            return null;
-        }
-
-        $symbolName = self::getNodeName($node);
-
-        if ($node instanceof ClassMethod) {
-            /** @var \PhpParser\Node\Stmt\ClassLike $parent */
-            $parent = $this->stack[count($this->stack) - 2];
-
-            if (isset($parent->name)) {
-                $symbolName = sprintf(
-                    '%1$s::%2$s',
-                    $parent->name->name,
-                    $node->name->name
-                );
-            }
-        }
-        $node->setAttribute('fullSymbolName', $symbolName);
-
-        $additions = $this->generateAdditionalTagsFromDoc($docComment);
-        if (count($additions) > 0) {
-            $this->additionalTags[$symbolName] = $additions;
-        }
-
-        $additions = $this->getAdditionalTagsFromMap($symbolName);
-        if (count($additions) > 0) {
-            $this->additionalTagStrings[$symbolName] = $additions;
-        }
-
-        if ($voidOrNever !== '') {
-            $addition = sprintf(
-                '@phpstan-return %s',
-                $voidOrNever === 'never'
-                    ? (new Never_())->__toString()
-                    : (new Void_())->__toString()
-            );
-            if (
-                !isset($this->additionalTagStrings[$symbolName])
-                || !in_array($addition, $this->additionalTagStrings[$symbolName], true)
-            ) {
-                $this->additionalTagStrings[$symbolName][] = $addition;
-            }
-        }
-
-        return null;
-    }
-
-    private static function getNodeName(Node $node): string
-    {
-        if ((($node instanceof Function_) || ($node instanceof ClassMethod) || ($node instanceof Class_)) && $node->name instanceof Identifier) {
-            return $node->name->name;
-        }
-
-        if ($node instanceof Property) {
-            return sprintf(
-                'property_%s',
-                uniqid()
-            );
-        }
-
-        return '';
-    }
-
-    /**
-     * @return array<\PhpParser\Node>
-     */
-    public function getStubStmts(): array
-    {
-        $stmts = parent::getStubStmts();
-
-        foreach ($stmts as $stmt) {
-            $this->postProcessNode($stmt);
-        }
-        return $stmts;
-    }
-
-    private function postProcessNode(Node $node): void
-    {
-        if (isset($node->stmts) && is_array($node->stmts)) {
-            foreach ($node->stmts as $stmt) {
-                $this->postProcessNode($stmt);
-            }
-        }
-
-        if (! ($node instanceof Function_) && ! ($node instanceof ClassMethod) && ! ($node instanceof Property) && ! ($node instanceof Class_)) {
-            return;
-        }
-
-        $name = $node->getAttribute('fullSymbolName');
-
-        if ($name === null) {
-            return;
-        }
-
-        $docComment = $node->getDocComment();
-
-        if (!($docComment instanceof Doc)) {
-            return;
-        }
-
-        $newDocComment = $this->addTags($name, $docComment);
-
-        if ($newDocComment instanceof Doc) {
-            $node->setDocComment($newDocComment);
-        }
-
-        if (! isset($this->additionalTagStrings[$name])) {
-            return;
-        }
-
-        $docComment = $node->getDocComment();
-
-        if (!($docComment instanceof Doc)) {
-            return;
-        }
-
-        $newDocComment = $this->addStringTags($name, $docComment);
-
-        if ($newDocComment instanceof Doc) {
-            $node->setDocComment($newDocComment);
-        }
-    }
-
-    /**
-     * @return array<int, \WordPressTag>
-     */
-    private function generateAdditionalTagsFromDoc(Doc $docComment): array
-    {
-        $docCommentText = $docComment->getText();
-
-        try {
-            $docblock = $this->docBlockFactory->create($docCommentText);
-        } catch (\RuntimeException $e) {
-            return [];
-        } catch (\InvalidArgumentException $e) {
-            return [];
-        }
-
-        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Param> $params*/
-        $params = $docblock->getTagsByName('param');
-
-        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Return_> $returns */
-        $returns = $docblock->getTagsByName('return');
-
-        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Var_> $vars */
-        $vars = $docblock->getTagsByName('var');
-
-        /** @var list<\WordPressTag> $additions */
-        $additions = [];
-
-        foreach ($params as $param) {
-            $addition = self::getAdditionFromParam($param);
-
-            if ($addition instanceof WordPressTag) {
-                $additions[] = $addition;
-            }
-        }
-
-        foreach ($returns as $return) {
-            $addition = self::getAdditionFromReturn($return);
-
-            if ($addition instanceof WordPressTag) {
-                $additions[] = $addition;
-            }
-        }
-
-        foreach ($vars as $var) {
-            $addition = self::getAdditionFromVar($var);
-
-            if ($addition instanceof WordPressTag) {
-                $additions[] = $addition;
-            }
-        }
-
-        return $additions;
-    }
-
-    private function addTags(string $name, Doc $docComment): ?Doc
-    {
-        $additions = $this->additionalTags[$name] ?? [];
-        $docCommentText = $docComment->getText();
-
-        try {
-            $docblock = $this->docBlockFactory->create($docCommentText);
-        } catch (\RuntimeException $e) {
-            return null;
-        } catch (\InvalidArgumentException $e) {
-            return null;
-        }
-
-        $additions = $this->discoverInheritedArgs($docblock, $additions);
-
-        /** @var list<string> $additionStrings */
-        $additionStrings = array_map(
-            static function (WordPressTag $tag): string {
-                $lines = $tag->format();
-
-                if (count($lines) === 0) {
-                    return '';
-                }
-
-                return ' * ' . implode("\n * ", $lines);
-            },
-            $additions
-        );
-
-        $additionStrings = array_filter($additionStrings);
-
-        if (count($additionStrings) === 0) {
-            return null;
-        }
-
-        $newDocComment = sprintf(
-            "%s\n%s\n */",
-            substr($docCommentText, 0, -4),
-            implode("\n", $additionStrings)
-        );
-
-        return new Doc($newDocComment, $docComment->getStartLine(), $docComment->getStartFilePos());
-    }
-
-    /**
-     * @param array<int, \WordPressTag> $additions
-     * @return array<int, \WordPressTag>
-     */
-    private function discoverInheritedArgs(DocBlock $docblock, array $additions): array
-    {
-        /** @var list<\phpDocumentor\Reflection\DocBlock\Tags\Param> $params */
-        $params = $docblock->getTagsByName('param');
-
-        $phpStanParams = array_filter(
-            $additions,
-            static function (WordPressTag $addition): bool {
-                return $addition->tag === '@phpstan-param';
-            }
-        );
-
-        foreach ($params as $param) {
-            $inherited = $this->getInheritedTagsForParam($param);
-
-            if (count($inherited) === 0) {
-                continue;
-            }
-
-            foreach ($phpStanParams as $addition) {
-                foreach ($inherited as $inherit) {
-                    if ($addition->name !== $inherit->name) {
-                        continue;
-                    }
-
-                    $addition->children = array_merge($addition->children, $inherit->children);
-                    continue 3;
-                }
-            }
-
-            $additions = array_merge($additions, $inherited);
-        }
-
-        return $additions;
-    }
-
-    /**
-     * @return array<int, \WordPressTag>
-     */
-    private function getInheritedTagsForParam(Param $param): array
-    {
-        $type = $param->getType();
-
-        if (!$type instanceof Type) {
-            return [];
-        }
-
-        $typeName = self::getTypeNameFromType($type);
-
-        if ($typeName === null) {
-            return [];
-        }
-
-        $paramDescription = $param->getDescription();
-
-        if (!$paramDescription instanceof Description) {
-            return [];
-        }
-
-        list($description) = explode("\n\n", $paramDescription->__toString());
-
-        if (strpos($description, '()') === false) {
-            return [];
-        }
-
-        $description = str_replace("\n", ' ', $description);
-        $description = str_replace('@see \\', '@see ', $description);
-        $additions = [];
-
-        foreach ($this->additionalTags as $symbolName => $tags) {
-            $search = sprintf(
-                'see %s()',
-                $symbolName
-            );
-
-            if (stripos($description, $search) === false) {
-                continue;
-            }
-
-            $match = self::getMatchingInheritedTag($param, $tags, $symbolName);
-
-            if ($match instanceof WordPressTag) {
-                $additions[] = $match;
-            }
-        }
-
-        return $additions;
-    }
-
-    /**
-     * @param array<int, \WordPressTag> $tags
-     */
-    private static function getMatchingInheritedTag(Param $param, array $tags, string $symbolName): ?WordPressTag
-    {
-        $paramName = $param->getVariableName();
-        $matchNames = [
-            $paramName,
-            'args',
-            'options',
-            'query',
-        ];
-        $matchingTags = array_filter(
-            $tags,
-            static function (WordPressTag $tag) use ($matchNames): bool {
-                return in_array($tag->name, $matchNames, true);
-            }
-        );
-
-        foreach ($matchingTags as $tag) {
-            $addTag = clone $tag;
-            $addTag->name = $paramName;
-            $addTag->description = sprintf(
-                'See %s()',
-                $symbolName
-            );
-
-            return $addTag;
-        }
-
-        return null;
-    }
-
-    /**
-     * @return list<string>
-     */
-    private function getAdditionalTagsFromMap(string $symbolName): array
-    {
-        if (! isset($this->functionMap)) {
-            $this->functionMap = require __DIR__ . '/functionMap.php';
-        }
-
-        if (! isset($this->functionMap[$symbolName])) {
-            return [];
-        }
-
-        $parameters = $this->functionMap[$symbolName];
-        $returnType = array_shift($parameters);
-        $additions = [];
-
-        foreach ($parameters as $paramName => $paramType) {
-            if (strpos($paramName, '@') === 0) {
-                $format = ( $paramType === '' ) ? '%s' : '%s %s';
-                $additions[] = sprintf(
-                    $format,
-                    $paramName,
-                    $paramType
-                );
-                continue;
-            }
-
-            $additions[] = sprintf(
-                '@phpstan-param %s $%s',
-                $paramType,
-                $paramName
-            );
-        }
-
-        if ($returnType !== null) {
-            $additions[] = sprintf(
-                '@phpstan-return %s',
-                $returnType
-            );
-        }
-
-        return $additions;
-    }
-
-    private function addStringTags(string $name, Doc $docComment): ?Doc
-    {
-        if (!isset($this->additionalTagStrings[$name])) {
-            return null;
-        }
-
-        $additions = $this->additionalTagStrings[$name];
-
-        $docCommentText = $docComment->getText();
-        $newDocComment = sprintf(
-            "%s\n * %s\n */",
-            substr($docCommentText, 0, -4),
-            implode("\n * ", $additions)
-        );
-
-        return new Doc($newDocComment, $docComment->getStartLine(), $docComment->getStartFilePos());
-    }
-
-    private function getAdditionFromParam(Param $tag): ?WordPressTag
-    {
-        $tagDescription = $tag->getDescription();
-        $tagVariableName = $tag->getVariableName();
-        $tagVariableType = $tag->getType();
-
-        // Skip if information we need is missing.
-        if (!$tagDescription instanceof Description || !$tagVariableName || !$tagVariableType instanceof Type) {
-            return null;
-        }
-
-        $tagDescriptionType = self::getTypeNameFromDescription($tagDescription, $tagVariableType);
-
-        if ($tagDescriptionType !== null) {
-            $tag = new WordPressTag();
-            $tag->tag = '@phpstan-param';
-            $tag->type = $tagDescriptionType;
-            $tag->name = $tagVariableName;
-
-            return $tag;
-        }
-
-        $elements = self::getElementsFromDescription($tagDescription, true);
-
-        if (count($elements) === 0) {
-            return null;
-        }
-
-        $tagVariableType = self::getTypeNameFromType($tagVariableType);
-
-        if ($tagVariableType === null) {
-            return null;
-        }
-
-        // It's common for an args parameter to accept a query var string or array with `string|array`.
-        // Remove the accepted string type for these so we get the strongest typing we can manage.
-        $tagVariableType = str_replace(['|string', 'string|'], '', $tagVariableType);
-
-        $tag = new WordPressTag();
-        $tag->tag = '@phpstan-param';
-        $tag->type = $tagVariableType;
-        $tag->name = $tagVariableName;
-        $tag->children = $elements;
-
-        return $tag;
-    }
-
-    private function getAdditionFromReturn(Return_ $tag): ?WordPressTag
-    {
-        $tagDescription = $tag->getDescription();
-        $tagVariableType = $tag->getType();
-
-        // Skip if information we need is missing.
-        if (!$tagDescription instanceof Description || !$tagVariableType instanceof Type) {
-            return null;
-        }
-
-        $tagDescriptionType = self::getTypeNameFromDescription($tagDescription, $tagVariableType);
-
-        if ($tagDescriptionType !== null) {
-            $tag = new WordPressTag();
-            $tag->tag = '@phpstan-return';
-            $tag->type = $tagDescriptionType;
-
-            return $tag;
-        }
-
-        $elements = self::getElementsFromDescription($tagDescription, false);
-
-        if (count($elements) === 0) {
-            return null;
-        }
-
-        $tagVariableType = self::getTypeNameFromType($tagVariableType);
-
-        if ($tagVariableType === null) {
-            return null;
-        }
-
-        $tag = new WordPressTag();
-        $tag->tag = '@phpstan-return';
-        $tag->type = $tagVariableType;
-        $tag->children = $elements;
-
-        return $tag;
-    }
-
-    private static function getAdditionFromVar(Var_ $tag): ?WordPressTag
-    {
-        $tagDescription = $tag->getDescription();
-        $tagVariableType = $tag->getType();
-
-        // Skip if information we need is missing.
-        if (!$tagDescription instanceof Description || !$tagVariableType instanceof Type) {
-            return null;
-        }
-
-        $elements = self::getElementsFromDescription($tagDescription, false);
-
-        if (count($elements) === 0) {
-            return null;
-        }
-
-        $tagVariableType = self::getTypeNameFromType($tagVariableType);
-
-        if ($tagVariableType === null) {
-            return null;
-        }
-
-        $tag = new WordPressTag();
-        $tag->tag = '@phpstan-var';
-        $tag->type = $tagVariableType;
-        $tag->children = $elements;
-
-        return $tag;
-    }
-
-    private static function getTypeNameFromDescription(Description $tagVariableDescription, Type $tagVariableType): ?string
-    {
-        if (!($tagVariableType instanceof \phpDocumentor\Reflection\Types\String_)) {
-            return null;
-        }
-
-        return self::getTypeNameFromDescriptionString($tagVariableDescription->__toString());
-    }
-
-    private static function getTypeNameFromDescriptionString(?string $tagDescription = null): ?string
-    {
-        if ($tagDescription === null) {
-            return null;
-        }
-
-        $fullDescription = str_replace("\n", ' ', $tagDescription);
-
-        /**
-         * This matches phrases that contain a list of two or more single-quoted strings, with the last
-         * item separated by 'or' or 'and'. The Oxford comma is optional. For example:
-         *
-         * - Either 'am', 'pm', 'AM', or 'PM'
-         * - Accepts 'comment' or 'term'
-         * - Either 'plugin' or 'theme'
-         * - Accepts 'big', or 'little'.
-         * - One of 'default', 'theme', or 'custom'
-         * - Either 'network-active', 'active' or 'inactive'
-         * - : 'top' or 'bottom'
-         */
-        $matched = preg_match("#(?>returns|either|one of|accepts|values are|:) ('.+'),? (?>or|and) '([^']+)'#i", $fullDescription, $matches);
-
-        if (! $matched) {
-            return null;
-        }
-
-        list(, $items, $final) = $matches;
-
-        // Pluck out phrases between single quotes, so messy sentences are handled:
-        preg_match_all("#'([^']+)'#", $items, $matches);
-
-        list(,$accepted) = $matches;
-
-        // Append the final item:
-        $accepted[] = $final;
-
-        return "'" . implode("'|'", $accepted) . "'";
-    }
-
-    private static function getTypeNameFromType(Type $tagVariableType): ?string
-    {
-        return self::getTypeNameFromString($tagVariableType->__toString());
-    }
-
-    private static function getTypeNameFromString(string $tagVariable): ?string
-    {
-        // PHPStan doesn't support typed array shapes (`int[]{...}`) so replace
-        // typed arrays such as `int[]` with `array`.
-        $tagVariableType = preg_replace('#[a-zA-Z0-9_]+\[\]#', 'array', $tagVariable);
-
-        if ($tagVariableType === null) {
-            return null;
-        }
-
-        $tagVariableType = str_replace(
-            [
-            'stdClass',
-            '\\object',
-            ],
-            'object',
-            $tagVariableType
-        );
-
-        $supportedTypes = [
-            'object',
-            'array',
-        ];
-
-        foreach ($supportedTypes as $supportedType) {
-            if (strpos($tagVariableType, "{$supportedType}|") === false) {
-                continue;
-            }
-            // Move the type that uses the hash notation to the end of union types so the shape works.
-            $tagVariableType = str_replace("{$supportedType}|", '', $tagVariableType) . "|{$supportedType}";
-        }
-
-        return $tagVariableType;
-    }
-
-    /**
-     * @return list<\WordPressArg>
-     */
-    private static function getElementsFromDescription(Description $tagDescription, bool $optional): array
-    {
-        $text = $tagDescription->__toString();
-
-        // Skip if the description doesn't contain at least one correctly
-        // formatted `@type`, which indicates an array hash.
-        if (strpos($text, '    @type ') === false) {
-            return [];
-        }
-
-        return self::getTypesAtLevel($text, $optional, 1);
-    }
-
-    /**
-     * @return list<\WordPressArg>
-     */
-    private static function getTypesAtLevel(string $text, bool $optional, int $level): array
-    {
-        // Populate `$types` with the value of each top level `@type`.
-        $spaces = str_repeat(' ', ($level * 4));
-        $types = preg_split("/\R+{$spaces}@type /", $text);
-
-        if ($types === false) {
-            return [];
-        }
-
-        unset($types[0]);
-        $elements = [];
-
-        foreach ($types as $typeTag) {
-            $parts = preg_split('#\s+#', trim($typeTag), 3);
-
-            if ($parts === false || count($parts) < 2) {
-                return [];
-            }
-
-            list($type, $name) = $parts;
-
-            $optionalArg = $optional;
-            $nameTrimmed = ltrim($name, '$');
-
-            if (is_numeric($nameTrimmed)) {
-                $optionalArg = false;
-            } elseif ($optional && ($level > 1)) {
-                $optionalArg = self::isOptional($parts[2]);
-            }
-
-            if (strpos($name, '...$') !== false) {
-                $name = null;
-            } elseif (strpos($name, '$') !== 0) {
-                return [];
-            } else {
-                $name = $nameTrimmed;
-            }
-
-            $arg = new WordPressArg();
-            $arg->type = $type;
-            $arg->optional = $optionalArg;
-            $arg->name = $name;
-
-            $nextLevel = $level + 1;
-            $subTypes = self::getTypesAtLevel($typeTag, $optional, $nextLevel);
-
-            if (count($subTypes) > 0) {
-                $type = self::getTypeNameFromString($type);
-
-                if ($type !== null) {
-                    $arg->type = $type;
-                }
-                $arg->children = $subTypes;
-            }
-
-            $elements[] = $arg;
-        }
-
-        return $elements;
-    }
-
-    private static function isOptional(string $description): bool
-    {
-        return (stripos($description, 'Optional') !== false)
-            || (stripos($description, 'Default ') !== false)
-            || (stripos($description, 'Default: ') !== false)
-            || (stripos($description, 'Defaults to ') !== false);
-    }
-
-    private function voidOrNever(Node $node): string
-    {
-        if (!($node instanceof Function_) && !($node instanceof ClassMethod)) {
-            return '';
-        }
-
-        if (!isset($node->stmts) || count($node->stmts) === 0) {
-            // Interfaces and abstract methods.
-            return '';
-        }
-
-        $return = $this->nodeFinder->findInstanceOf($node, Stmt_Return::class);
-
-        // If there is a return statement, it's not return type never.
-        if (count($return) !== 0) {
-            // If there is at least one return statement that is not void,
-            // it's not return type void.
-            if (
-                $this->nodeFinder->findFirst(
-                    $return,
-                    static function (Node $node): bool {
-                        return isset($node->expr);
-                    }
-                ) instanceof Node
-            ) {
-                return '';
-            }
-            // If there is no return statement that is not void,
-            // it's return type void.
-            return 'void';
-        }
-
-        // Check for never return type.
-        foreach ($node->stmts as $stmt) {
-            if (!($stmt instanceof Expression)) {
-                continue;
-            }
-            // If a first level statement is exit/die, it's return type never.
-            if ($stmt->expr instanceof Exit_) {
-                if ($stmt->expr->expr instanceof String_) {
-                    if (strpos($stmt->expr->expr->value, 'must be overridden') !== false) {
-                        return '';
-                    }
-                }
-                return 'never';
-            }
-            if (!($stmt->expr instanceof FuncCall) || !($stmt->expr->name instanceof Name)) {
-                continue;
-            }
-            $name = $stmt->expr->name;
-            // If a first level statement is a call to wp_send_json(_success/error),
-            // it's return type never.
-            if (strpos($name->toString(), 'wp_send_json') === 0) {
-                return 'never';
-            }
-            // Skip all functions but wp_die().
-            if (strpos($name->toString(), 'wp_die') !== 0) {
-                continue;
-            }
-            $args = $stmt->expr->getArgs();
-            // If wp_die is called without 3rd parameter, it's return type never.
-            if (count($args) < 3) {
-                return 'never';
-            }
-            // If wp_die is called with 3rd parameter, we need additional checks.
-            try {
-                $arg = (new ConstExprEvaluator())->evaluateSilently($args[2]->value);
-            } catch (\PhpParser\ConstExprEvaluationException $e) {
-                // If we don't know the value of the 3rd parameter, we can't be sure.
-                continue;
-            }
-
-            if (is_int($arg)) {
-                return 'never';
-            }
-            if (is_array($arg)) {
-                if (!isset($arg['exit']) || (bool)$arg['exit'] === true) {
-                    return 'never';
-                }
-            }
-
-            continue;
-        }
-        return '';
-    }
-};
+return new PhpStubs\WordPress\Core\Visitor();