From 956efaafb30eeefeddf00d15090c3557ffd382fa Mon Sep 17 00:00:00 2001
From: Christopher Dwyer-Perkins <chris@inverted-solutions.com>
Date: Wed, 15 Jan 2025 13:07:54 -0400
Subject: [PATCH] Updated the promises prefix and FileFactory to auto import it

---
 bsc-plugin/package-lock.json              |  6 +++---
 bsc-plugin/package.json                   |  2 +-
 bsc-plugin/src/lib/rooibos/FileFactory.ts | 18 ++++++++++++++----
 3 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/bsc-plugin/package-lock.json b/bsc-plugin/package-lock.json
index a9e752a9..4bd2faab 100644
--- a/bsc-plugin/package-lock.json
+++ b/bsc-plugin/package-lock.json
@@ -12,7 +12,7 @@
             "dependencies": {
                 "roku-debug": "^0.21.10",
                 "roku-deploy": "^3.12.1",
-                "rooibospromises": "npm:@rokucommunity/promises@^0.5.0",
+                "rooibos_promises": "npm:@rokucommunity/promises@^0.5.0",
                 "source-map": "^0.7.3",
                 "undent": "^0.1.0",
                 "vscode-languageserver": "~6.1.1",
@@ -8902,7 +8902,7 @@
                 "node": ">=6 <7 || >=8"
             }
         },
-        "node_modules/rooibospromises": {
+        "node_modules/rooibos_promises": {
             "name": "@rokucommunity/promises",
             "version": "0.5.0",
             "resolved": "https://registry.npmjs.org/@rokucommunity/promises/-/promises-0.5.0.tgz",
@@ -16831,7 +16831,7 @@
                 }
             }
         },
-        "rooibospromises": {
+        "rooibos_promises": {
             "version": "npm:@rokucommunity/promises@0.5.0",
             "resolved": "https://registry.npmjs.org/@rokucommunity/promises/-/promises-0.5.0.tgz",
             "integrity": "sha512-Xl/8z4+WMeSHCG418Y9O49r/PQphwJ7efVmqlLoOvEg3EQ3ET//VHDQJa0YucuuTQurHfrLjXDj4foluhdHRSg=="
diff --git a/bsc-plugin/package.json b/bsc-plugin/package.json
index a0d1a148..8c08a7cb 100644
--- a/bsc-plugin/package.json
+++ b/bsc-plugin/package.json
@@ -36,7 +36,7 @@
     "dependencies": {
         "roku-debug": "^0.21.10",
         "roku-deploy": "^3.12.1",
-        "rooibospromises": "npm:@rokucommunity/promises@^0.5.0",
+        "rooibos_promises": "npm:@rokucommunity/promises@^0.5.0",
         "source-map": "^0.7.3",
         "undent": "^0.1.0",
         "vscode-languageserver": "~6.1.1",
diff --git a/bsc-plugin/src/lib/rooibos/FileFactory.ts b/bsc-plugin/src/lib/rooibos/FileFactory.ts
index 71ddeaec..cd98282f 100644
--- a/bsc-plugin/src/lib/rooibos/FileFactory.ts
+++ b/bsc-plugin/src/lib/rooibos/FileFactory.ts
@@ -24,7 +24,7 @@ export class FileFactory {
         this.coverageComponentBrsTemplate = fs.readFileSync(path.join(this.frameworkSourcePath, '/source/rooibos/CodeCoverage.brs'), 'utf8');
     }
 
-    public addedSourceFrameworkFilePaths: string[] = [];
+    public sourceFilesToAutoImport: string[] = [];
     public addedFrameworkFiles: BscFile[] = [];
 
     public addFrameworkFiles(program: Program) {
@@ -43,10 +43,10 @@ export class FileFactory {
         });
 
         for (let filePath of globedFiles) {
-            if (/^source[/\\]rooibos[/\\]/g.test(filePath)) {
+            if (this.shouldAddFileToImportList(filePath)) {
                 // Save a list of all source files added to the program
                 // to be imported by node test components
-                this.addedSourceFrameworkFilePaths.push(filePath);
+                this.sourceFilesToAutoImport.push(filePath);
             }
             let sourcePath = path.resolve(this.frameworkSourcePath, filePath);
             let fileContents = fs.readFileSync(sourcePath, 'utf8').toString();
@@ -67,7 +67,7 @@ export class FileFactory {
 
     public createTestXML(name: string, baseName: string, suite?: TestSuite): string {
         let scriptImports = [];
-        for (let filePath of this.addedSourceFrameworkFilePaths) {
+        for (let filePath of this.sourceFilesToAutoImport) {
             scriptImports.push(`<script type="text/brighterscript" uri="pkg:/${filePath}" />`);
         }
 
@@ -117,6 +117,16 @@ export class FileFactory {
         return result !== undefined;
     }
 
+    private shouldAddFileToImportList(destFilePath): boolean {
+        const pathDetails = path.parse(destFilePath);
+        if (pathDetails.dir === 'source' || pathDetails.dir.startsWith('source\\') || pathDetails.dir.startsWith('source/')) {
+            if (pathDetails.ext === '.brs' || (pathDetails.ext === '.bs' && !pathDetails.name.endsWith('.d'))) {
+                return true;
+            }
+        }
+        return false;
+    }
+
     public addFile(program, projectPath: string, contents: string) {
         try {
             const file = program.setFile({