fix: template functions miss debug info issue

Author: Jesse Chen

src/compiler.ts

@@ -9830,7 +9830,7 @@ export class Compiler extends DiagnosticEmitter {
     let targetFunction = this.currentFlow.targetFunction;
     let source = range.source;
     if (source.debugInfoIndex < 0) source.debugInfoIndex = this.module.addDebugInfoFile(source.normalizedPath);
-    range.debugInfoRef = expr;
+    range.debugInfo.set(targetFunction.internalName, expr);
     targetFunction.debugLocations.push(range);
   }
 

src/diagnostics.ts

@@ -11,6 +11,7 @@ import {
   DiagnosticCode,
   diagnosticCodeToString
 } from "./diagnosticMessages.generated";
+import { ExpressionRef } from "./module";
 
 import {
   isLineBreak,
@@ -45,7 +46,7 @@ export const enum DiagnosticCategory {
 export class Range {
 
   source!: Source;
-  debugInfoRef: usize = 0;
+  debugInfo: Map<string, ExpressionRef> = new Map<string, ExpressionRef>();
 
   constructor(public start: i32, public end: i32) {}
 

src/program.ts

@@ -118,7 +118,8 @@ import {
   Module,
   FunctionRef,
   MemorySegment,
-  getFunctionName
+  getFunctionName,
+  ExpressionRef
 } from "./module";
 
 import {
@@ -3916,9 +3917,11 @@ export class Function extends TypedElement {
       for (let i = 0, k = debugLocations.length; i < k; ++i) {
         let range = debugLocations[i];
         let source = range.source;
+        const debugInfo = range.debugInfo.get(this.internalName);
+        assert(debugInfo);
         module.setDebugLocation(
           ref,
-          range.debugInfoRef,
+          debugInfo as ExpressionRef,
           source.debugInfoIndex,
           source.lineAt(range.start),
           source.columnAt() - 1 // source maps are 0-based