Assign .program to the builder BEFORE calling afterProgram (#1011)

src/ProgramBuilder.spec.ts

@@ -8,7 +8,7 @@ import { standardizePath as s, util } from './util';
 import { Logger, LogLevel } from './Logger';
 import * as diagnosticUtils from './diagnosticUtils';
 import type { BscFile, BsDiagnostic } from '.';
-import { Range } from '.';
+import { Deferred, Range } from '.';
 import { DiagnosticSeverity } from 'vscode-languageserver';
 import { BrsFile } from './files/BrsFile';
 import { expectZeroDiagnostics } from './testHelpers.spec';
@@ -41,6 +41,21 @@ describe('ProgramBuilder', () => {
         builder.dispose();
     });
 
+    it('includes .program in the afterProgramCreate event', async () => {
+        builder = new ProgramBuilder();
+        const deferred = new Deferred<Program>();
+        builder.plugins.add({
+            name: 'test',
+            afterProgramCreate: () => {
+                deferred.resolve(builder.program);
+            }
+        });
+        builder['createProgram']();
+        expect(
+            await deferred.promise
+        ).to.exist;
+    });
+
     describe('loadAllFilesAST', () => {
         it('loads .bs, .brs, .xml files', async () => {
             sinon.stub(util, 'getFilePaths').returns(Promise.resolve([{

src/ProgramBuilder.ts

@@ -124,7 +124,7 @@ export class ProgramBuilder {
         }
         this.logger.logLevel = this.options.logLevel as LogLevel;
 
-        this.program = this.createProgram();
+        this.createProgram();
 
         //parse every file in the entire project
         await this.loadAllFilesAST();
@@ -139,10 +139,11 @@ export class ProgramBuilder {
     }
 
     protected createProgram() {
-        const program = new Program(this.options, this.logger, this.plugins);
+        this.program = new Program(this.options, this.logger, this.plugins);
 
-        this.plugins.emit('afterProgramCreate', program);
-        return program;
+        this.plugins.emit('afterProgramCreate', this.program);
+
+        return this.program;
     }
 
     protected loadPlugins() {