refactor: use asynclocalstorage for context management

Author: boblat

src/context-helpers/context-manager.ts

@@ -1,22 +1,20 @@
+import { AsyncLocalStorage } from 'node:async_hooks'
 import type { TestExecutionContext } from '../test-execution-context'
 
-declare global {
-  // eslint-disable-next-line no-var
-  var puyaTsExecutionContext: TestExecutionContext | undefined
-}
+export class ContextManager {
+  private static asyncStore = new AsyncLocalStorage<TestExecutionContext>()
 
-export const ctxMgr = {
-  set instance(ctx: TestExecutionContext) {
-    const instance = global.puyaTsExecutionContext
-    if (instance != undefined) throw new Error('Execution context has already been set')
-    global.puyaTsExecutionContext = ctx
-  },
-  get instance() {
-    const instance = global.puyaTsExecutionContext
-    if (instance == undefined) throw new Error('No execution context has been set')
+  static set instance(ctx: TestExecutionContext) {
+    const instance = this.asyncStore.getStore()
+    if (instance !== undefined) throw new Error('Execution context has already been set')
+    this.asyncStore.enterWith(ctx)
+  }
+  static get instance() {
+    const instance = this.asyncStore.getStore()
+    if (instance === undefined) throw new Error('No execution context has been set')
     return instance
-  },
-  reset() {
-    global.puyaTsExecutionContext = undefined
-  },
+  }
+  static reset() {
+    this.asyncStore.disable()
+  }
 }

src/context-helpers/internal-context.ts

@@ -7,7 +7,7 @@ import type { AccountData, ApplicationData, AssetData } from '../impl/reference'
 import type { VoterData } from '../impl/voter-params'
 import type { TransactionGroup } from '../subcontexts/transaction-context'
 import type { TestExecutionContext } from '../test-execution-context'
-import { ctxMgr } from './context-manager'
+import { ContextManager } from './context-manager'
 
 /**
  * For accessing implementation specific functions, with a convenient single entry
@@ -16,7 +16,7 @@ import { ctxMgr } from './context-manager'
  */
 class InternalContext {
   get value() {
-    return ctxMgr.instance as TestExecutionContext
+    return ContextManager.instance as TestExecutionContext
   }
 
   get defaultSender() {

src/test-execution-context.ts

@@ -1,6 +1,6 @@
 import type { Account as AccountType, BaseContract, bytes, LogicSig, uint64 } from '@algorandfoundation/algorand-typescript'
 import { DEFAULT_TEMPLATE_VAR_PREFIX } from './constants'
-import { ctxMgr } from './context-helpers/context-manager'
+import { ContextManager } from './context-helpers/context-manager'
 import type { DecodedLogs, LogDecoding } from './decode-logs'
 import { Account, AccountCls } from './impl/reference'
 import { ContractContext } from './subcontexts/contract-context'
@@ -34,7 +34,7 @@ export class TestExecutionContext {
    * @param {bytes} [defaultSenderAddress] - The default sender address.
    */
   constructor(defaultSenderAddress?: bytes) {
-    ctxMgr.instance = this
+    ContextManager.instance = this
     this.#contractContext = new ContractContext()
     this.#ledgerContext = new LedgerContext()
     this.#txnContext = new TransactionContext()
@@ -215,7 +215,7 @@ export class TestExecutionContext {
     this.#activeLogicSigArgs = []
     this.#template_vars = {}
     this.#compiledApps = []
-    ctxMgr.reset()
-    ctxMgr.instance = this
+    ContextManager.reset()
+    ContextManager.instance = this
   }
 }