fix: not declared errors in source native variant REPL (#1479)

* fix: only add func and var identifiers instead of all identifiers * fix: add function declarations into `previousProgramIdentifiers` --------- Co-authored-by: Martin Henz <[email protected]>
source-academy · Sep 3, 2023 · 0107752 · 0107752
1 parent 563bc0d
commit 0107752
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/runner/fullJSRunner.ts b/src/runner/fullJSRunner.ts
@@ -9,10 +9,15 @@ import { RuntimeSourceError } from '../errors/runtimeSourceError'
 import { hoistAndMergeImports } from '../localImports/transformers/hoistAndMergeImports'
 import { getRequireProvider, RequireProvider } from '../modules/requireProvider'
 import { parse } from '../parser/parser'
-import { evallerReplacer, getBuiltins, transpile } from '../transpiler/transpiler'
+import {
+  evallerReplacer,
+  getBuiltins,
+  getGloballyDeclaredIdentifiers,
+  transpile
+} from '../transpiler/transpiler'
 import type { Context, NativeStorage } from '../types'
 import * as create from '../utils/astCreator'
-import { getIdentifiersInProgram } from '../utils/uniqueIds'
+import { getFunctionDeclarationNamesInProgram } from '../utils/uniqueIds'
 import { toSourceError } from './errors'
 import { appendModulesToContext, resolvedErrorPromise } from './utils'
 
@@ -70,7 +75,10 @@ export async function fullJSRunner(
     ...preludeAndBuiltins,
     evallerReplacer(create.identifier(NATIVE_STORAGE_ID), new Set())
   ])
-  getIdentifiersInProgram(preEvalProgram).forEach(id =>
+  getFunctionDeclarationNamesInProgram(preEvalProgram).forEach(id =>
+    context.nativeStorage.previousProgramsIdentifiers.add(id)
+  )
+  getGloballyDeclaredIdentifiers(preEvalProgram).forEach(id =>
     context.nativeStorage.previousProgramsIdentifiers.add(id)
   )
   const preEvalCode: string = generate(preEvalProgram)

diff --git a/src/transpiler/transpiler.ts b/src/transpiler/transpiler.ts
@@ -12,6 +12,7 @@ import { ModuleDocumentation } from '../modules/moduleTypes'
 import { AllowedDeclarations, Chapter, Context, NativeStorage, Variant } from '../types'
 import * as create from '../utils/astCreator'
 import {
+  getFunctionDeclarationNamesInProgram,
   getIdentifiersInNativeStorage,
   getIdentifiersInProgram,
   getUniqueId
@@ -678,6 +679,9 @@ function transpileToFullJS(
     globalIds.native
   )
 
+  getFunctionDeclarationNamesInProgram(program).forEach(id =>
+    context.nativeStorage.previousProgramsIdentifiers.add(id)
+  )
   getGloballyDeclaredIdentifiers(program).forEach(id =>
     context.nativeStorage.previousProgramsIdentifiers.add(id)
   )