Skip to content

Commit ca77237

Browse files
committed
fix(extractor): use setTypeMapEntry (first-wins) in C++/CUDA declaration handlers
1 parent 7509e4f commit ca77237

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

src/extractors/cpp.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import type {
55
TreeSitterNode,
66
TreeSitterTree,
77
} from '../types.js';
8-
import { extractModifierVisibility, findChild, nodeEndLine } from './helpers.js';
8+
import { extractModifierVisibility, findChild, nodeEndLine, setTypeMapEntry } from './helpers.js';
99

1010
/**
1111
* Extract symbols from C++ files.
@@ -232,7 +232,7 @@ function handleCppDeclaration(node: TreeSitterNode, ctx: ExtractorOutput): void
232232
if (!nameNode) continue;
233233
const varName = unwrapCppDeclaratorName(nameNode);
234234
if (varName) {
235-
ctx.typeMap.set(varName, { type: typeName, confidence: 0.9 });
235+
setTypeMapEntry(ctx.typeMap, varName, typeName, 0.9);
236236
}
237237
}
238238
}

src/extractors/cuda.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import type {
55
TreeSitterNode,
66
TreeSitterTree,
77
} from '../types.js';
8-
import { extractModifierVisibility, findChild, nodeEndLine } from './helpers.js';
8+
import { extractModifierVisibility, findChild, nodeEndLine, setTypeMapEntry } from './helpers.js';
99

1010
/**
1111
* Extract symbols from CUDA files.
@@ -232,7 +232,7 @@ function handleCudaDeclaration(node: TreeSitterNode, ctx: ExtractorOutput): void
232232
if (!nameNode) continue;
233233
const varName = extractCudaFieldName(nameNode);
234234
if (varName) {
235-
ctx.typeMap.set(varName, { type: typeName, confidence: 0.9 });
235+
setTypeMapEntry(ctx.typeMap, varName, typeName, 0.9);
236236
}
237237
}
238238
}

0 commit comments

Comments
 (0)