From bc27d8ea90a55a9195bfd1c4c0be28f089e2dbbe Mon Sep 17 00:00:00 2001
From: Andrew Nicols <andrew@nicols.co.uk>
Date: Tue, 19 Mar 2024 15:54:03 +0800
Subject: [PATCH] Create new Util method for counting global artifacts

---
 moodle/Tests/Util/TokenUtilTest.php | 58 +++++++++++++++++++++++++++++
 moodle/Util/Docblocks.php           |  5 ---
 moodle/Util/TokenUtil.php           | 29 +++++++++++++++
 3 files changed, 87 insertions(+), 5 deletions(-)

diff --git a/moodle/Tests/Util/TokenUtilTest.php b/moodle/Tests/Util/TokenUtilTest.php
index d32e537..9f44094 100644
--- a/moodle/Tests/Util/TokenUtilTest.php
+++ b/moodle/Tests/Util/TokenUtilTest.php
@@ -103,4 +103,62 @@ public static function objectPropertiesProvider(): array {
 
         return $cases;
     }
+
+    /**
+     * @dataProvider countGlobalScopesInFileProvider
+     */
+    public function testCountGlobalScopesInFile(
+        string $content,
+        int $expectedCount
+    ): void {
+        $config = new Config([]);
+        $ruleset = new Ruleset($config);
+
+        $phpcsFile = new DummyFile($content, $ruleset, $config);
+        $phpcsFile->process();
+
+        $this->assertEquals($expectedCount, TokenUtil::countGlobalScopesInFile($phpcsFile));
+    }
+
+    public static function countGlobalScopesInFileProvider(): array {
+        $cases = [
+            'No global scopes' => [
+                '<?php $a = 1;',
+                0,
+            ],
+            'One global scope' => [
+                '<?php class Example {}',
+                1,
+            ],
+            'Two global scopes' => [
+                '<?php class Example {} class AnotherExample {}',
+                2,
+            ],
+            'Class method is not global' => [
+                '<?php class Example { public function exampleMethod() {} }',
+                1,
+            ],
+            'Global method counts' => [
+                '<?php function exampleFunction() {}',
+                1,
+            ],
+            'Interfaces count' => [
+                '<?php interface ExampleInterface {}',
+                1,
+            ],
+            'Traits count' => [
+                '<?php trait ExampleTrait {}',
+                1,
+            ],
+        ];
+
+        if (version_compare(PHP_VERSION, '8.1.0') >= 0) {
+            $cases['Enums count'] = [
+                '<?php enum ExampleEnum {}',
+                1,
+            ];
+        }
+
+        return $cases;
+    }
 }
diff --git a/moodle/Util/Docblocks.php b/moodle/Util/Docblocks.php
index 4ff3b99..e20af98 100644
--- a/moodle/Util/Docblocks.php
+++ b/moodle/Util/Docblocks.php
@@ -17,12 +17,7 @@
 
 namespace MoodleHQ\MoodleCS\moodle\Util;
 
-use PHP_CodeSniffer\Config;
-use PHP_CodeSniffer\Exceptions\DeepExitException;
-use PHP_CodeSniffer\Files\DummyFile;
 use PHP_CodeSniffer\Files\File;
-use PHP_CodeSniffer\Ruleset;
-use PHP_CodeSniffer\Util\Tokens;
 
 /**
  * Utilities related to PHP DocBlocks.
diff --git a/moodle/Util/TokenUtil.php b/moodle/Util/TokenUtil.php
index 6fe38ad..07c13ce 100644
--- a/moodle/Util/TokenUtil.php
+++ b/moodle/Util/TokenUtil.php
@@ -18,6 +18,7 @@
 namespace MoodleHQ\MoodleCS\moodle\Util;
 
 use PHP_CodeSniffer\Files\File;
+use PHP_CodeSniffer\Util\Tokens;
 use PHPCSUtils\Utils\ObjectDeclarations;
 
 class TokenUtil
@@ -58,4 +59,32 @@ public static function getObjectName(
 
         return ObjectDeclarations::getName($phpcsFile, $stackPtr);
     }
+
+    /**
+     * Count the number of global scopes in a file.
+     *
+     * @param File $phpcsFile
+     * @return int
+     */
+    public static function countGlobalScopesInFile(
+        File $phpcsFile
+    ): int {
+        $tokens = $phpcsFile->getTokens();
+        $artifactCount = 0;
+        $find = Tokens::$ooScopeTokens;
+        $find[] = T_FUNCTION;
+
+        $typePtr = 0;
+        while ($typePtr = $phpcsFile->findNext($find, $typePtr + 1)) {
+            $token = $tokens[$typePtr];
+            if ($token['code'] === T_FUNCTION && !empty($token['conditions'])) {
+                // Skip methods of classes, traits and interfaces.
+                continue;
+            }
+
+            $artifactCount++;
+        }
+
+        return $artifactCount;
+    }
 }