1
0

resolution.test.ts 147 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978
  1. /**
  2. * Resolution Module Tests
  3. *
  4. * Tests for Phase 3: Reference Resolution
  5. */
  6. import { describe, it, expect, beforeEach, afterEach } from 'vitest';
  7. import * as fs from 'fs';
  8. import * as path from 'path';
  9. import * as os from 'os';
  10. import { CodeGraph } from '../src';
  11. import { Node, UnresolvedReference } from '../src/types';
  12. import { ReferenceResolver, createResolver, ResolutionContext } from '../src/resolution';
  13. import { matchReference, resolveMethodOnType, matchByQualifiedName, preferCallSiteFile } from '../src/resolution/name-matcher';
  14. import { resolveImportPath, extractImportMappings, resolveJvmImport, loadCppIncludeDirs, clearCppIncludeDirCache, isPhpIncludePathRef } from '../src/resolution/import-resolver';
  15. import type { UnresolvedRef } from '../src/resolution/types';
  16. import { detectFrameworks, getAllFrameworkResolvers } from '../src/resolution/frameworks';
  17. import { QueryBuilder } from '../src/db/queries';
  18. import { DatabaseConnection } from '../src/db';
  19. describe('Resolution Module', () => {
  20. let tempDir: string;
  21. let cg: CodeGraph;
  22. beforeEach(() => {
  23. // Create temp directory
  24. tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-resolution-test-'));
  25. });
  26. afterEach(() => {
  27. // Clean up
  28. if (cg) {
  29. cg.destroy();
  30. } else if (fs.existsSync(tempDir)) {
  31. fs.rmSync(tempDir, { recursive: true });
  32. }
  33. });
  34. describe('Name Matcher', () => {
  35. it('should match exact name references', () => {
  36. // Create a mock context
  37. const mockNodes: Node[] = [
  38. {
  39. id: 'func:test.ts:myFunction:10',
  40. kind: 'function',
  41. name: 'myFunction',
  42. qualifiedName: 'test.ts::myFunction',
  43. filePath: 'test.ts',
  44. language: 'typescript',
  45. startLine: 10,
  46. endLine: 20,
  47. startColumn: 0,
  48. endColumn: 0,
  49. updatedAt: Date.now(),
  50. },
  51. ];
  52. const context: ResolutionContext = {
  53. getNodesInFile: () => mockNodes,
  54. getNodesByName: (name) => mockNodes.filter((n) => n.name === name),
  55. getNodesByQualifiedName: () => [],
  56. getNodesByKind: () => [],
  57. fileExists: () => true,
  58. readFile: () => null,
  59. getProjectRoot: () => '/test',
  60. getAllFiles: () => ['test.ts'],
  61. };
  62. const ref = {
  63. fromNodeId: 'caller:main.ts:caller:5',
  64. referenceName: 'myFunction',
  65. referenceKind: 'calls' as const,
  66. line: 5,
  67. column: 10,
  68. filePath: 'main.ts',
  69. language: 'typescript' as const,
  70. };
  71. const result = matchReference(ref, context);
  72. expect(result).not.toBeNull();
  73. expect(result?.targetNodeId).toBe('func:test.ts:myFunction:10');
  74. expect(result?.resolvedBy).toBe('exact-match');
  75. });
  76. it('should prefer same-module candidates over cross-module matches', () => {
  77. // Simulates a Python monorepo where multiple apps define navigate()
  78. const candidateA: Node = {
  79. id: 'func:apps/app_a/src/server.py:navigate:10',
  80. kind: 'function',
  81. name: 'navigate',
  82. qualifiedName: 'apps/app_a/src/server.py::navigate',
  83. filePath: 'apps/app_a/src/server.py',
  84. language: 'python',
  85. startLine: 10,
  86. endLine: 20,
  87. startColumn: 0,
  88. endColumn: 0,
  89. updatedAt: Date.now(),
  90. };
  91. const candidateB: Node = {
  92. id: 'func:apps/app_b/src/server.py:navigate:15',
  93. kind: 'function',
  94. name: 'navigate',
  95. qualifiedName: 'apps/app_b/src/server.py::navigate',
  96. filePath: 'apps/app_b/src/server.py',
  97. language: 'python',
  98. startLine: 15,
  99. endLine: 25,
  100. startColumn: 0,
  101. endColumn: 0,
  102. updatedAt: Date.now(),
  103. };
  104. const context: ResolutionContext = {
  105. getNodesInFile: () => [],
  106. getNodesByName: (name) => name === 'navigate' ? [candidateA, candidateB] : [],
  107. getNodesByQualifiedName: () => [],
  108. getNodesByKind: () => [],
  109. fileExists: () => true,
  110. readFile: () => null,
  111. getProjectRoot: () => '/test',
  112. getAllFiles: () => [],
  113. getNodesByLowerName: () => [],
  114. getImportMappings: () => [],
  115. };
  116. // Reference from app_a should resolve to app_a's navigate, not app_b's
  117. const ref = {
  118. fromNodeId: 'func:apps/app_a/src/handler.py:handler:5',
  119. referenceName: 'navigate',
  120. referenceKind: 'calls' as const,
  121. line: 5,
  122. column: 10,
  123. filePath: 'apps/app_a/src/handler.py',
  124. language: 'python' as const,
  125. };
  126. const result = matchReference(ref, context);
  127. expect(result).not.toBeNull();
  128. expect(result?.targetNodeId).toBe('func:apps/app_a/src/server.py:navigate:10');
  129. expect(result?.resolvedBy).toBe('exact-match');
  130. });
  131. it('should lower confidence for cross-module exact matches', () => {
  132. // Only one candidate but in a completely different module
  133. const candidates: Node[] = [
  134. {
  135. id: 'func:apps/app_b/src/server.py:navigate:10',
  136. kind: 'function',
  137. name: 'navigate',
  138. qualifiedName: 'apps/app_b/src/server.py::navigate',
  139. filePath: 'apps/app_b/src/server.py',
  140. language: 'python',
  141. startLine: 10,
  142. endLine: 20,
  143. startColumn: 0,
  144. endColumn: 0,
  145. updatedAt: Date.now(),
  146. },
  147. {
  148. id: 'func:apps/app_c/src/server.py:navigate:10',
  149. kind: 'function',
  150. name: 'navigate',
  151. qualifiedName: 'apps/app_c/src/server.py::navigate',
  152. filePath: 'apps/app_c/src/server.py',
  153. language: 'python',
  154. startLine: 10,
  155. endLine: 20,
  156. startColumn: 0,
  157. endColumn: 0,
  158. updatedAt: Date.now(),
  159. },
  160. ];
  161. const context: ResolutionContext = {
  162. getNodesInFile: () => [],
  163. getNodesByName: (name) => name === 'navigate' ? candidates : [],
  164. getNodesByQualifiedName: () => [],
  165. getNodesByKind: () => [],
  166. fileExists: () => true,
  167. readFile: () => null,
  168. getProjectRoot: () => '/test',
  169. getAllFiles: () => [],
  170. getNodesByLowerName: () => [],
  171. getImportMappings: () => [],
  172. };
  173. // Reference from app_a — neither candidate is in the same module
  174. const ref = {
  175. fromNodeId: 'func:apps/app_a/src/handler.py:handler:5',
  176. referenceName: 'navigate',
  177. referenceKind: 'calls' as const,
  178. line: 5,
  179. column: 10,
  180. filePath: 'apps/app_a/src/handler.py',
  181. language: 'python' as const,
  182. };
  183. const result = matchReference(ref, context);
  184. // Should still resolve but with low confidence
  185. expect(result).not.toBeNull();
  186. expect(result?.confidence).toBeLessThanOrEqual(0.4);
  187. });
  188. it('should match qualified name references', () => {
  189. const mockClassNode: Node = {
  190. id: 'class:user.ts:User:5',
  191. kind: 'class',
  192. name: 'User',
  193. qualifiedName: 'user.ts::User',
  194. filePath: 'user.ts',
  195. language: 'typescript',
  196. startLine: 5,
  197. endLine: 30,
  198. startColumn: 0,
  199. endColumn: 0,
  200. updatedAt: Date.now(),
  201. };
  202. const mockMethodNode: Node = {
  203. id: 'method:user.ts:User.save:15',
  204. kind: 'method',
  205. name: 'save',
  206. qualifiedName: 'user.ts::User::save',
  207. filePath: 'user.ts',
  208. language: 'typescript',
  209. startLine: 15,
  210. endLine: 25,
  211. startColumn: 0,
  212. endColumn: 0,
  213. updatedAt: Date.now(),
  214. };
  215. const context: ResolutionContext = {
  216. getNodesInFile: (fp) => fp === 'user.ts' ? [mockClassNode, mockMethodNode] : [],
  217. getNodesByName: (name) => {
  218. if (name === 'User') return [mockClassNode];
  219. if (name === 'save') return [mockMethodNode];
  220. return [];
  221. },
  222. getNodesByQualifiedName: (qn) => {
  223. if (qn === 'user.ts::User::save') return [mockMethodNode];
  224. return [];
  225. },
  226. getNodesByKind: () => [],
  227. fileExists: () => true,
  228. readFile: () => null,
  229. getProjectRoot: () => '/test',
  230. getAllFiles: () => ['user.ts'],
  231. };
  232. const ref = {
  233. fromNodeId: 'caller:main.ts:main:5',
  234. referenceName: 'User.save',
  235. referenceKind: 'calls' as const,
  236. line: 5,
  237. column: 10,
  238. filePath: 'main.ts',
  239. language: 'typescript' as const,
  240. };
  241. const result = matchReference(ref, context);
  242. expect(result).not.toBeNull();
  243. expect(result?.targetNodeId).toBe('method:user.ts:User.save:15');
  244. });
  245. });
  246. describe('Ubiquitous-name ceiling (#999)', () => {
  247. // A vendored theme/SDK re-declares the same method name across thousands of
  248. // files (Metronic's `init`/`update`/… on every widget). The fuzzy strategies
  249. // used to score every same-named candidate per ref — O(K) per ref, O(K²)
  250. // total — which pinned a core for 15-28 min at "Resolving refs … 94%". Above
  251. // the ceiling they must DECLINE instead, since no proximity/word-overlap
  252. // score can pick the one true target among thousands anyway.
  253. const CEILING = 500;
  254. // A spy context: counts how many nodes the strategy actually inspects, so we
  255. // can assert the cap short-circuits BEFORE the O(K) scoring (not just that it
  256. // returns null).
  257. const makeManyMethods = (n: number, name: string): Node[] =>
  258. Array.from({ length: n }, (_, i) => ({
  259. id: `method:widget${i}.js:Widget${i}.${name}:1`,
  260. kind: 'method' as const,
  261. name,
  262. qualifiedName: `widget${i}.js::Widget${i}::${name}`,
  263. filePath: `static/theme/widget${i}.js`,
  264. language: 'javascript' as const,
  265. startLine: 1,
  266. endLine: 5,
  267. startColumn: 0,
  268. endColumn: 0,
  269. updatedAt: Date.now(),
  270. }));
  271. const spyContext = (nodes: Node[]): { ctx: ResolutionContext; lookups: () => number } => {
  272. let scanned = 0;
  273. const ctx: ResolutionContext = {
  274. getNodesInFile: () => [],
  275. getNodesByName: (name) => {
  276. const hit = nodes.filter((n) => n.name === name);
  277. scanned += hit.length;
  278. return hit;
  279. },
  280. getNodesByQualifiedName: () => [],
  281. getNodesByKind: () => [],
  282. fileExists: () => true,
  283. readFile: () => null,
  284. getProjectRoot: () => '/test',
  285. getAllFiles: () => [],
  286. getNodesByLowerName: () => [],
  287. getImportMappings: () => [],
  288. };
  289. return { ctx, lookups: () => scanned };
  290. };
  291. it('declines a method call (`obj.init`) above the ceiling instead of scoring K candidates', () => {
  292. const { ctx } = spyContext(makeManyMethods(CEILING + 1, 'init'));
  293. const ref = {
  294. fromNodeId: 'method:caller.js:caller:1',
  295. referenceName: 'widget.init',
  296. referenceKind: 'calls' as const,
  297. line: 2,
  298. column: 4,
  299. filePath: 'static/theme/caller.js',
  300. language: 'javascript' as const,
  301. };
  302. expect(matchReference(ref, ctx)).toBeNull();
  303. });
  304. it('declines a bare exact-name ref above the ceiling', () => {
  305. const { ctx } = spyContext(makeManyMethods(CEILING + 1, 'render'));
  306. const ref = {
  307. fromNodeId: 'method:caller.js:caller:1',
  308. referenceName: 'render',
  309. referenceKind: 'calls' as const,
  310. line: 2,
  311. column: 4,
  312. filePath: 'static/theme/caller.js',
  313. language: 'javascript' as const,
  314. };
  315. expect(matchReference(ref, ctx)).toBeNull();
  316. });
  317. it('still resolves a SAME-FILE definition when one exists (precise path unaffected)', () => {
  318. // Strategy 1 (class-name) and same-file matching are precise — a ubiquitous
  319. // name with an unambiguous local target still resolves.
  320. const nodes = makeManyMethods(CEILING + 1, 'init');
  321. const local: Node = {
  322. id: 'class:static/theme/caller.js:Widgetly:1',
  323. kind: 'class',
  324. name: 'Widgetly',
  325. qualifiedName: 'static/theme/caller.js::Widgetly',
  326. filePath: 'static/theme/caller.js',
  327. language: 'javascript',
  328. startLine: 1, endLine: 9, startColumn: 0, endColumn: 0, updatedAt: Date.now(),
  329. };
  330. const localMethod: Node = {
  331. id: 'method:static/theme/caller.js:Widgetly.init:2',
  332. kind: 'method',
  333. name: 'init',
  334. qualifiedName: 'static/theme/caller.js::Widgetly::init',
  335. filePath: 'static/theme/caller.js',
  336. language: 'javascript',
  337. startLine: 2, endLine: 4, startColumn: 0, endColumn: 0, updatedAt: Date.now(),
  338. };
  339. const all = [...nodes, local, localMethod];
  340. const ctx: ResolutionContext = {
  341. getNodesInFile: (fp) => all.filter((n) => n.filePath === fp),
  342. getNodesByName: (name) => all.filter((n) => n.name === name),
  343. getNodesByQualifiedName: () => [],
  344. getNodesByKind: () => [],
  345. fileExists: () => true,
  346. readFile: () => null,
  347. getProjectRoot: () => '/test',
  348. getAllFiles: () => [],
  349. getNodesByLowerName: () => [],
  350. getImportMappings: () => [],
  351. };
  352. // `Widgetly.init` names the class explicitly → Strategy 1 resolves it.
  353. const ref = {
  354. fromNodeId: 'method:static/theme/caller.js:caller:6',
  355. referenceName: 'Widgetly.init',
  356. referenceKind: 'calls' as const,
  357. line: 6,
  358. column: 4,
  359. filePath: 'static/theme/caller.js',
  360. language: 'javascript' as const,
  361. };
  362. const result = matchReference(ref, ctx);
  363. expect(result?.targetNodeId).toBe('method:static/theme/caller.js:Widgetly.init:2');
  364. });
  365. it('still scores normally JUST below the ceiling (no behavior change for normal repos)', () => {
  366. // Real repos top out near ~40 same-named methods; this proves a sub-ceiling
  367. // collision still resolves via proximity, so the cap is invisible to them.
  368. const nodes = makeManyMethods(CEILING - 1, 'update');
  369. // Make ONE candidate share the caller's directory so proximity picks it.
  370. nodes[0] = {
  371. ...nodes[0]!,
  372. id: 'method:static/theme/app/Widget0.update:1',
  373. qualifiedName: 'static/theme/app/widget.js::Widget0::update',
  374. filePath: 'static/theme/app/widget.js',
  375. };
  376. const ctx: ResolutionContext = {
  377. getNodesInFile: () => [],
  378. getNodesByName: (name) => nodes.filter((n) => n.name === name),
  379. getNodesByQualifiedName: () => [],
  380. getNodesByKind: () => [],
  381. fileExists: () => true,
  382. readFile: () => null,
  383. getProjectRoot: () => '/test',
  384. getAllFiles: () => [],
  385. getNodesByLowerName: () => [],
  386. getImportMappings: () => [],
  387. };
  388. const ref = {
  389. fromNodeId: 'method:static/theme/app/caller.js:caller:1',
  390. referenceName: 'update',
  391. referenceKind: 'calls' as const,
  392. line: 2,
  393. column: 4,
  394. filePath: 'static/theme/app/caller.js',
  395. language: 'javascript' as const,
  396. };
  397. // Below the ceiling the fuzzy path runs and resolves SOMETHING (not capped).
  398. expect(matchReference(ref, ctx)).not.toBeNull();
  399. });
  400. });
  401. describe('Import Resolver', () => {
  402. it('should resolve relative import paths', () => {
  403. const context: ResolutionContext = {
  404. getNodesInFile: () => [],
  405. getNodesByName: () => [],
  406. getNodesByQualifiedName: () => [],
  407. getNodesByKind: () => [],
  408. fileExists: (p) => p === 'src/components/utils.ts' || p === 'src/components/utils/index.ts',
  409. readFile: () => null,
  410. getProjectRoot: () => '',
  411. getAllFiles: () => ['src/components/utils.ts', 'src/components/utils/index.ts'],
  412. };
  413. const result = resolveImportPath(
  414. './utils',
  415. 'src/components/Button.ts',
  416. 'typescript',
  417. context
  418. );
  419. expect(result).toBe('src/components/utils.ts');
  420. });
  421. it('should resolve parent directory imports', () => {
  422. const context: ResolutionContext = {
  423. getNodesInFile: () => [],
  424. getNodesByName: () => [],
  425. getNodesByQualifiedName: () => [],
  426. getNodesByKind: () => [],
  427. fileExists: (p) => p === 'src/helpers.ts' || p === 'src/helpers/index.ts',
  428. readFile: () => null,
  429. getProjectRoot: () => '',
  430. getAllFiles: () => ['src/helpers.ts', 'src/helpers/index.ts'],
  431. };
  432. const result = resolveImportPath(
  433. '../helpers',
  434. 'src/components/Button.ts',
  435. 'typescript',
  436. context
  437. );
  438. expect(result).toBe('src/helpers.ts');
  439. });
  440. it('should extract JS/TS import mappings', () => {
  441. const content = `
  442. import { foo } from './foo';
  443. import bar from '../bar';
  444. import * as utils from './utils';
  445. import { baz, qux } from './baz';
  446. `;
  447. const mappings = extractImportMappings(
  448. 'src/index.ts',
  449. content,
  450. 'typescript'
  451. );
  452. expect(mappings.length).toBeGreaterThan(0);
  453. expect(mappings.some((m) => m.localName === 'foo')).toBe(true);
  454. expect(mappings.some((m) => m.localName === 'bar')).toBe(true);
  455. });
  456. it('should extract Python import mappings', () => {
  457. const content = `
  458. from utils import helper
  459. from .models import User
  460. import os
  461. from ..services import auth_service
  462. `;
  463. const mappings = extractImportMappings(
  464. 'src/main.py',
  465. content,
  466. 'python'
  467. );
  468. expect(mappings.length).toBeGreaterThan(0);
  469. expect(mappings.some((m) => m.localName === 'helper')).toBe(true);
  470. expect(mappings.some((m) => m.localName === 'User')).toBe(true);
  471. });
  472. });
  473. describe('JVM FQN Import Resolution', () => {
  474. // Build a ResolutionContext stub whose getNodesByQualifiedName answers
  475. // from a fixed table — the only context method resolveJvmImport touches.
  476. const makeContext = (byQName: Record<string, Node[]>): ResolutionContext => ({
  477. getNodesInFile: () => [],
  478. getNodesByName: () => [],
  479. getNodesByQualifiedName: (q) => byQName[q] ?? [],
  480. getNodesByKind: () => [],
  481. fileExists: () => false,
  482. readFile: () => null,
  483. getProjectRoot: () => '',
  484. getAllFiles: () => [],
  485. });
  486. const node = (id: string, name: string, qualifiedName: string, kind: Node['kind'] = 'class', language: Node['language'] = 'kotlin'): Node => ({
  487. id, kind, name, qualifiedName,
  488. filePath: 'Models.kt', language,
  489. startLine: 1, endLine: 1, startColumn: 0, endColumn: 0,
  490. updatedAt: 0,
  491. });
  492. const importRef = (referenceName: string, language: Node['language'] = 'kotlin'): UnresolvedRef => ({
  493. fromNodeId: 'caller',
  494. referenceName,
  495. referenceKind: 'imports',
  496. line: 1, column: 0,
  497. filePath: 'Caller.kt',
  498. language,
  499. });
  500. it('resolves a Kotlin class import by FQN regardless of filename', () => {
  501. const target = node('n1', 'Bar', 'com.example.foo::Bar');
  502. const ctx = makeContext({ 'com.example.foo::Bar': [target] });
  503. const result = resolveJvmImport(importRef('com.example.foo.Bar'), ctx);
  504. expect(result?.targetNodeId).toBe('n1');
  505. expect(result?.resolvedBy).toBe('import');
  506. });
  507. it('resolves a Kotlin top-level function import by FQN', () => {
  508. const util = node('n2', 'util', 'com.example.foo::util', 'function');
  509. const ctx = makeContext({ 'com.example.foo::util': [util] });
  510. const result = resolveJvmImport(importRef('com.example.foo.util'), ctx);
  511. expect(result?.targetNodeId).toBe('n2');
  512. });
  513. it('resolves a Java import by FQN', () => {
  514. const target = node('n3', 'Bar', 'com.example.foo::Bar', 'class', 'java');
  515. const ctx = makeContext({ 'com.example.foo::Bar': [target] });
  516. const result = resolveJvmImport(importRef('com.example.foo.Bar', 'java'), ctx);
  517. expect(result?.targetNodeId).toBe('n3');
  518. });
  519. it('resolves cross-language: Kotlin importing a Java class', () => {
  520. // The Kotlin file declares `import com.example.JavaBar` — the target is
  521. // a Java class node. JVM interop means the resolver doesn't care about
  522. // the source language of the target, only that the FQN matches.
  523. const target = node('n4', 'JavaBar', 'com.example::JavaBar', 'class', 'java');
  524. const ctx = makeContext({ 'com.example::JavaBar': [target] });
  525. const result = resolveJvmImport(importRef('com.example.JavaBar'), ctx);
  526. expect(result?.targetNodeId).toBe('n4');
  527. });
  528. it('disambiguates a name collision across packages', () => {
  529. // Two classes named `Bar` in different packages. Each import resolves
  530. // to the one whose FQN matches — not to "whichever was found first".
  531. const barA = node('n5a', 'Bar', 'com.example.alpha::Bar');
  532. const barB = node('n5b', 'Bar', 'com.example.beta::Bar');
  533. const ctx = makeContext({
  534. 'com.example.alpha::Bar': [barA],
  535. 'com.example.beta::Bar': [barB],
  536. });
  537. expect(resolveJvmImport(importRef('com.example.alpha.Bar'), ctx)?.targetNodeId).toBe('n5a');
  538. expect(resolveJvmImport(importRef('com.example.beta.Bar'), ctx)?.targetNodeId).toBe('n5b');
  539. });
  540. it('returns null for wildcard imports', () => {
  541. const ctx = makeContext({});
  542. expect(resolveJvmImport(importRef('com.example.foo.*'), ctx)).toBeNull();
  543. });
  544. it('returns null for unqualified names', () => {
  545. // A single-segment name has no package; nothing to look up by FQN.
  546. const ctx = makeContext({ 'Bar': [node('n6', 'Bar', 'Bar')] });
  547. expect(resolveJvmImport(importRef('Bar'), ctx)).toBeNull();
  548. });
  549. it('returns null for non-JVM languages', () => {
  550. const target = node('n7', 'Bar', 'com.example::Bar');
  551. const ctx = makeContext({ 'com.example::Bar': [target] });
  552. expect(resolveJvmImport(importRef('com.example.Bar', 'typescript'), ctx)).toBeNull();
  553. });
  554. it('returns null for non-imports reference kinds', () => {
  555. // The resolver intentionally only acts on `imports` refs; ordinary
  556. // `calls`/`extends` refs fall through to the framework + name-matcher
  557. // strategies.
  558. const target = node('n8', 'Bar', 'com.example::Bar');
  559. const ctx = makeContext({ 'com.example::Bar': [target] });
  560. const ref: UnresolvedRef = {
  561. fromNodeId: 'caller', referenceName: 'com.example.Bar',
  562. referenceKind: 'calls', line: 1, column: 0,
  563. filePath: 'Caller.kt', language: 'kotlin',
  564. };
  565. expect(resolveJvmImport(ref, ctx)).toBeNull();
  566. });
  567. it('returns null when the FQN is not in the index', () => {
  568. const ctx = makeContext({});
  569. expect(resolveJvmImport(importRef('com.example.Unknown'), ctx)).toBeNull();
  570. });
  571. });
  572. describe('Framework Detection', () => {
  573. it('should detect React framework', () => {
  574. const context: ResolutionContext = {
  575. getNodesInFile: () => [],
  576. getNodesByName: () => [],
  577. getNodesByQualifiedName: () => [],
  578. getNodesByKind: () => [],
  579. fileExists: () => false,
  580. readFile: (p) => {
  581. if (p === 'package.json') {
  582. return JSON.stringify({
  583. dependencies: { react: '^18.0.0' },
  584. });
  585. }
  586. return null;
  587. },
  588. getProjectRoot: () => '/test',
  589. getAllFiles: () => ['package.json', 'src/App.tsx'],
  590. };
  591. const frameworks = detectFrameworks(context);
  592. expect(frameworks.some((f) => f.name === 'react')).toBe(true);
  593. });
  594. it('should detect Express framework', () => {
  595. const context: ResolutionContext = {
  596. getNodesInFile: () => [],
  597. getNodesByName: () => [],
  598. getNodesByQualifiedName: () => [],
  599. getNodesByKind: () => [],
  600. fileExists: () => false,
  601. readFile: (p) => {
  602. if (p === 'package.json') {
  603. return JSON.stringify({
  604. dependencies: { express: '^4.18.0' },
  605. });
  606. }
  607. return null;
  608. },
  609. getProjectRoot: () => '/test',
  610. getAllFiles: () => ['package.json', 'src/app.js'],
  611. };
  612. const frameworks = detectFrameworks(context);
  613. expect(frameworks.some((f) => f.name === 'express')).toBe(true);
  614. });
  615. it('should detect Laravel framework', () => {
  616. const context: ResolutionContext = {
  617. getNodesInFile: () => [],
  618. getNodesByName: () => [],
  619. getNodesByQualifiedName: () => [],
  620. getNodesByKind: () => [],
  621. fileExists: (p) => p === 'artisan',
  622. readFile: () => null,
  623. getProjectRoot: () => '/test',
  624. getAllFiles: () => ['artisan', 'app/Http/Kernel.php'],
  625. };
  626. const frameworks = detectFrameworks(context);
  627. expect(frameworks.some((f) => f.name === 'laravel')).toBe(true);
  628. });
  629. it('should return all framework resolvers', () => {
  630. const resolvers = getAllFrameworkResolvers();
  631. expect(resolvers.length).toBeGreaterThan(0);
  632. expect(resolvers.some((r) => r.name === 'react')).toBe(true);
  633. expect(resolvers.some((r) => r.name === 'express')).toBe(true);
  634. expect(resolvers.some((r) => r.name === 'laravel')).toBe(true);
  635. });
  636. });
  637. describe('React Framework Resolver', () => {
  638. it('should resolve React component references', () => {
  639. const mockNodes: Node[] = [
  640. {
  641. id: 'component:src/Button.tsx:Button:5',
  642. kind: 'component',
  643. name: 'Button',
  644. qualifiedName: 'src/Button.tsx::Button',
  645. filePath: 'src/Button.tsx',
  646. language: 'tsx',
  647. startLine: 5,
  648. endLine: 20,
  649. startColumn: 0,
  650. endColumn: 0,
  651. updatedAt: Date.now(),
  652. },
  653. ];
  654. const context: ResolutionContext = {
  655. getNodesInFile: (fp) => (fp === 'src/Button.tsx' ? mockNodes : []),
  656. getNodesByName: () => mockNodes,
  657. getNodesByQualifiedName: () => [],
  658. getNodesByKind: () => [],
  659. fileExists: () => false,
  660. readFile: (p) => {
  661. if (p === 'package.json') {
  662. return JSON.stringify({ dependencies: { react: '^18.0.0' } });
  663. }
  664. return null;
  665. },
  666. getProjectRoot: () => '/test',
  667. getAllFiles: () => ['package.json', 'src/Button.tsx', 'src/App.tsx'],
  668. };
  669. const frameworks = detectFrameworks(context);
  670. const reactResolver = frameworks.find((f) => f.name === 'react');
  671. expect(reactResolver).toBeDefined();
  672. const ref = {
  673. fromNodeId: 'component:src/App.tsx:App:1',
  674. referenceName: 'Button',
  675. referenceKind: 'renders' as const,
  676. line: 10,
  677. column: 5,
  678. filePath: 'src/App.tsx',
  679. // Refs extracted from .tsx files carry language 'tsx' — component
  680. // resolution is gated to JSX-capable refs (#764: PascalCase TYPE refs
  681. // from plain .ts files were resolving to arbitrary same-named classes).
  682. language: 'tsx' as const,
  683. };
  684. const result = reactResolver!.resolve(ref, context);
  685. expect(result).not.toBeNull();
  686. expect(result?.targetNodeId).toBe('component:src/Button.tsx:Button:5');
  687. // The same PascalCase name referenced from a plain .ts file is a TYPE
  688. // reference, not a component usage — component resolution must decline
  689. // and leave it to proximity-aware name matching (#764: a .ts GraphQL
  690. // types file's own `Account` alias was losing to an arbitrary same-named
  691. // class in another monorepo package).
  692. const tsRef = { ...ref, filePath: 'src/models.ts', language: 'typescript' as const };
  693. expect(reactResolver!.resolve(tsRef, context)).toBeNull();
  694. });
  695. it('should resolve custom hook references', () => {
  696. const mockNodes: Node[] = [
  697. {
  698. id: 'hook:src/hooks/useAuth.ts:useAuth:1',
  699. kind: 'function',
  700. name: 'useAuth',
  701. qualifiedName: 'src/hooks/useAuth.ts::useAuth',
  702. filePath: 'src/hooks/useAuth.ts',
  703. language: 'typescript',
  704. startLine: 1,
  705. endLine: 20,
  706. startColumn: 0,
  707. endColumn: 0,
  708. updatedAt: Date.now(),
  709. },
  710. ];
  711. const context: ResolutionContext = {
  712. getNodesInFile: (fp) => (fp.includes('useAuth') ? mockNodes : []),
  713. getNodesByName: () => mockNodes,
  714. getNodesByQualifiedName: () => [],
  715. getNodesByKind: () => [],
  716. fileExists: () => false,
  717. readFile: (p) => {
  718. if (p === 'package.json') {
  719. return JSON.stringify({ dependencies: { react: '^18.0.0' } });
  720. }
  721. return null;
  722. },
  723. getProjectRoot: () => '/test',
  724. getAllFiles: () => ['package.json', 'src/hooks/useAuth.ts'],
  725. };
  726. const frameworks = detectFrameworks(context);
  727. const reactResolver = frameworks.find((f) => f.name === 'react');
  728. const ref = {
  729. fromNodeId: 'component:src/App.tsx:App:1',
  730. referenceName: 'useAuth',
  731. referenceKind: 'calls' as const,
  732. line: 5,
  733. column: 10,
  734. filePath: 'src/App.tsx',
  735. language: 'typescript' as const,
  736. };
  737. const result = reactResolver!.resolve(ref, context);
  738. expect(result).not.toBeNull();
  739. expect(result?.targetNodeId).toBe('hook:src/hooks/useAuth.ts:useAuth:1');
  740. });
  741. });
  742. describe('Integration Tests', () => {
  743. it('should create resolver from CodeGraph instance', async () => {
  744. // Create a simple TypeScript project
  745. fs.writeFileSync(
  746. path.join(tempDir, 'package.json'),
  747. JSON.stringify({ name: 'test', dependencies: { react: '^18.0.0' } })
  748. );
  749. const srcDir = path.join(tempDir, 'src');
  750. fs.mkdirSync(srcDir);
  751. // Create utility file
  752. fs.writeFileSync(
  753. path.join(srcDir, 'utils.ts'),
  754. `export function formatDate(date: Date): string {
  755. return date.toISOString();
  756. }
  757. export function parseDate(str: string): Date {
  758. return new Date(str);
  759. }`
  760. );
  761. // Create main file that uses utils
  762. fs.writeFileSync(
  763. path.join(srcDir, 'main.ts'),
  764. `import { formatDate, parseDate } from './utils';
  765. function processDate(input: string): string {
  766. const date = parseDate(input);
  767. return formatDate(date);
  768. }`
  769. );
  770. // Initialize and index
  771. cg = await CodeGraph.init(tempDir, { index: true });
  772. // Check that resolver detected React framework
  773. const frameworks = cg.getDetectedFrameworks();
  774. expect(frameworks).toContain('react');
  775. // Get stats to verify indexing worked
  776. const stats = cg.getStats();
  777. expect(stats.fileCount).toBe(2);
  778. expect(stats.nodeCount).toBeGreaterThan(0);
  779. });
  780. it('should resolve references after indexing', async () => {
  781. // Create a project with references
  782. const srcDir = path.join(tempDir, 'src');
  783. fs.mkdirSync(srcDir, { recursive: true });
  784. fs.writeFileSync(
  785. path.join(srcDir, 'helper.ts'),
  786. `export function helperFunction(): void {
  787. console.log('helper');
  788. }`
  789. );
  790. fs.writeFileSync(
  791. path.join(srcDir, 'main.ts'),
  792. `import { helperFunction } from './helper';
  793. function main(): void {
  794. helperFunction();
  795. }`
  796. );
  797. cg = await CodeGraph.init(tempDir, { index: true });
  798. // Run reference resolution
  799. const result = cg.resolveReferences();
  800. // Should have attempted resolution
  801. expect(result.stats.total).toBeGreaterThanOrEqual(0);
  802. });
  803. it('promotes calls→instantiates when target resolves to a class (Python)', async () => {
  804. // Python has no `new` keyword — `Foo()` is the standard
  805. // instantiation syntax. Extraction can't tell that apart from
  806. // a function call without symbol info, so it emits a `calls`
  807. // ref. Resolution promotes it to `instantiates` once the
  808. // target is known to be a class.
  809. const srcDir = path.join(tempDir, 'src');
  810. fs.mkdirSync(srcDir, { recursive: true });
  811. fs.writeFileSync(
  812. path.join(srcDir, 'app.py'),
  813. `class UserService:
  814. def __init__(self):
  815. self.db = None
  816. def bootstrap():
  817. return UserService()
  818. `
  819. );
  820. cg = await CodeGraph.init(tempDir, { index: true });
  821. cg.resolveReferences();
  822. const bootstrap = cg
  823. .getNodesByKind('function')
  824. .find((n) => n.name === 'bootstrap');
  825. expect(bootstrap).toBeDefined();
  826. const outgoing = cg.getOutgoingEdges(bootstrap!.id);
  827. const instantiates = outgoing.find((e) => e.kind === 'instantiates');
  828. expect(instantiates).toBeDefined();
  829. // Same edge must NOT also appear as a `calls` edge — promotion
  830. // replaces the kind, doesn't duplicate.
  831. const callsToUserService = outgoing.filter(
  832. (e) => e.kind === 'calls' && e.target === instantiates!.target
  833. );
  834. expect(callsToUserService).toHaveLength(0);
  835. });
  836. it('records instantiates for C++ stack/brace construction, targeting the class (#1035)', async () => {
  837. // `Calculator calc(0)` (direct-init) and `Widget w{1, 2}` (brace-init)
  838. // carry the constructor args directly on the declarator — there's no
  839. // call/new node — so they recorded no `instantiates` edge, while heap
  840. // `new Calculator(0)` did. Both stack forms now do.
  841. fs.writeFileSync(
  842. path.join(tempDir, 'm.cpp'),
  843. `class Calculator { public: Calculator(int seed) {} int add(int a, int b){ return a+b; } };
  844. class Widget { public: Widget(int a, int b) {} };
  845. int runStack(int a, int b) { Calculator calc(0); return calc.add(a, b); }
  846. int runBrace() { Widget w{1, 2}; return 0; }
  847. int runHeap(int a, int b) { Calculator* c = new Calculator(0); return c->add(a, b); }
  848. void noise() { int x(5); int y{6}; Calculator deferred; }
  849. `
  850. );
  851. cg = await CodeGraph.init(tempDir, { index: true });
  852. const fn = (name: string) => cg.getNodesByKind('function').find((n) => n.name === name)!;
  853. const instTargets = (name: string) =>
  854. cg
  855. .getOutgoingEdges(fn(name).id)
  856. .filter((e) => e.kind === 'instantiates')
  857. .map((e) => cg.getNode(e.target)!);
  858. // Direct-init (the issue) and brace-init both instantiate, targeting the
  859. // CLASS node — not the same-named constructor method.
  860. const stack = instTargets('runStack');
  861. expect(stack.map((n) => `${n.kind}:${n.name}`)).toContain('class:Calculator');
  862. expect(instTargets('runBrace').map((n) => `${n.kind}:${n.name}`)).toContain('class:Widget');
  863. // Heap still works (regression guard).
  864. expect(instTargets('runHeap').map((n) => `${n.kind}:${n.name}`)).toContain('class:Calculator');
  865. // Primitives (`int x(0)`/`int y{6}`) and bare default construction
  866. // (`Calculator deferred;`) must NOT mint an instantiates edge.
  867. expect(instTargets('noise')).toHaveLength(0);
  868. });
  869. it('resolves a cross-file static method call to the method, not the class (#825)', async () => {
  870. // `Foo.bar()` where `Foo` is an imported class must link to the static
  871. // method `Foo::bar`, NOT to the class `Foo`. Previously the import
  872. // resolver dropped the `.bar` member and resolved to `Foo`, which the
  873. // calls→instantiates promotion then turned into `run instantiates Foo`,
  874. // leaving the static method with zero callers and a hollow impact radius.
  875. fs.writeFileSync(
  876. path.join(tempDir, 'helpers.ts'),
  877. `export class Foo {\n static bar(x: number) { return x + 1; }\n}\n`
  878. );
  879. fs.writeFileSync(
  880. path.join(tempDir, 'caller.ts'),
  881. `import { Foo } from './helpers';\nexport function run() { return Foo.bar(41); }\n`
  882. );
  883. cg = await CodeGraph.init(tempDir, { index: true });
  884. cg.resolveReferences();
  885. const bar = cg.getNodesByKind('method').find((n) => n.name === 'bar');
  886. const foo = cg.getNodesByKind('class').find((n) => n.name === 'Foo');
  887. const run = cg.getNodesByKind('function').find((n) => n.name === 'run');
  888. expect(bar).toBeDefined();
  889. expect(foo).toBeDefined();
  890. expect(run).toBeDefined();
  891. // `run` is reported as a caller of the static method `Foo.bar`.
  892. const barCallers = cg.getCallers(bar!.id).map((c) => c.node.name);
  893. expect(barCallers).toContain('run');
  894. // And the call is NOT mis-promoted to `run instantiates Foo`.
  895. const outgoing = cg.getOutgoingEdges(run!.id);
  896. expect(
  897. outgoing.filter((e) => e.kind === 'instantiates' && e.target === foo!.id)
  898. ).toHaveLength(0);
  899. // The real edge is a `calls` edge to the method.
  900. expect(
  901. outgoing.some((e) => e.kind === 'calls' && e.target === bar!.id)
  902. ).toBe(true);
  903. });
  904. it('resolves Go cross-package qualified calls via go.mod module path (#388)', async () => {
  905. // Pre-#388, every `pkga.FuncX(...)` call in a Go monorepo was flagged
  906. // external (isExternalImport returned true for any non-`/internal/`
  907. // import without `.`-prefix) and resolution fell through to name-match
  908. // with path proximity — recall on cross-package callers was ~<1%.
  909. fs.writeFileSync(
  910. path.join(tempDir, 'go.mod'),
  911. 'module github.com/example/myproject\n\ngo 1.21\n'
  912. );
  913. const pkgaDir = path.join(tempDir, 'pkga');
  914. const pkgbDir = path.join(tempDir, 'pkgb');
  915. const pkgcDir = path.join(tempDir, 'pkgc');
  916. fs.mkdirSync(pkgaDir);
  917. fs.mkdirSync(pkgbDir);
  918. fs.mkdirSync(pkgcDir);
  919. // Same-name exported function in two packages — only the imported one
  920. // should resolve. Exercises disambiguation, not just connectivity.
  921. fs.writeFileSync(
  922. path.join(pkgaDir, 'conv.go'),
  923. 'package pkga\nfunc Convert(x int) int { return x * 2 }\n'
  924. );
  925. fs.writeFileSync(
  926. path.join(pkgbDir, 'conv.go'),
  927. 'package pkgb\nfunc Convert(x int) int { return x + 1 }\n'
  928. );
  929. fs.writeFileSync(
  930. path.join(pkgcDir, 'use.go'),
  931. `package pkgc
  932. import "github.com/example/myproject/pkga"
  933. func UsePkga() {
  934. pkga.Convert(5)
  935. }
  936. `
  937. );
  938. cg = await CodeGraph.init(tempDir, { index: true });
  939. const usePkga = cg.getNodesByKind('function').filter((n) => n.name ==='UsePkga')[0];
  940. expect(usePkga).toBeDefined();
  941. const outgoing = cg.getOutgoingEdges(usePkga!.id);
  942. const callEdges = outgoing.filter((e) => e.kind === 'calls');
  943. expect(callEdges).toHaveLength(1);
  944. const target = cg.getNode(callEdges[0]!.target);
  945. expect(target?.name).toBe('Convert');
  946. // Critical: the resolver must pick the imported pkga's Convert,
  947. // not pkgb's. With the broken (pre-fix) resolver this lands on
  948. // whichever Convert happens to be cheaper under path proximity.
  949. expect(target?.filePath.replace(/\\/g, '/')).toBe('pkga/conv.go');
  950. });
  951. it('resolves Go aliased imports across packages (#388)', async () => {
  952. fs.writeFileSync(
  953. path.join(tempDir, 'go.mod'),
  954. 'module github.com/example/myproject\n\ngo 1.21\n'
  955. );
  956. fs.mkdirSync(path.join(tempDir, 'pkgb'));
  957. fs.mkdirSync(path.join(tempDir, 'pkgd'));
  958. fs.writeFileSync(
  959. path.join(tempDir, 'pkgb', 'lib.go'),
  960. 'package pkgb\nfunc Compute(x int) int { return x }\n'
  961. );
  962. fs.writeFileSync(
  963. path.join(tempDir, 'pkgd', 'use.go'),
  964. `package pkgd
  965. import (
  966. "fmt"
  967. alias "github.com/example/myproject/pkgb"
  968. )
  969. func UseAliased() {
  970. fmt.Println("hi")
  971. alias.Compute(3)
  972. }
  973. `
  974. );
  975. cg = await CodeGraph.init(tempDir, { index: true });
  976. const useAliased = cg.getNodesByKind('function').filter((n) => n.name ==='UseAliased')[0];
  977. expect(useAliased).toBeDefined();
  978. const calls = cg.getOutgoingEdges(useAliased!.id).filter((e) => e.kind === 'calls');
  979. // fmt.Println is stdlib — must stay external. alias.Compute must resolve.
  980. expect(calls).toHaveLength(1);
  981. const target = cg.getNode(calls[0]!.target);
  982. expect(target?.name).toBe('Compute');
  983. expect(target?.filePath.replace(/\\/g, '/')).toBe('pkgb/lib.go');
  984. });
  985. it('resolves Python module-attribute calls after `from pkg import module` (#578)', async () => {
  986. // Pre-#578, a `module.func()` call where `module` was bound via
  987. // `from pkg import module` dropped its `calls` edge. The file→file import
  988. // edge resolved (resolveModuleImportToFile falls back to a dotted-module
  989. // file lookup for absolute package paths), but resolvePythonModuleMember
  990. // had no such fallback — resolveImportPath returns null for an absolute
  991. // package path like `pkg.module`, so the member never resolved and
  992. // callers/callees/impact on the target came back empty. Same root-cause
  993. // class as the Go cross-package qualified call (#388).
  994. fs.mkdirSync(path.join(tempDir, 'pkg'));
  995. fs.writeFileSync(path.join(tempDir, 'pkg', '__init__.py'), '');
  996. fs.writeFileSync(
  997. path.join(tempDir, 'pkg', 'module.py'),
  998. 'def func():\n return 1\n'
  999. );
  1000. fs.writeFileSync(
  1001. path.join(tempDir, 'main.py'),
  1002. `from pkg import module
  1003. import os
  1004. def caller():
  1005. return module.func()
  1006. def external_caller():
  1007. return os.getcwd()
  1008. `
  1009. );
  1010. cg = await CodeGraph.init(tempDir, { index: true });
  1011. const caller = cg.getNodesByKind('function').filter((n) => n.name === 'caller')[0];
  1012. expect(caller).toBeDefined();
  1013. const calls = cg.getOutgoingEdges(caller!.id).filter((e) => e.kind === 'calls');
  1014. // module.func() must resolve to the real function in the submodule file.
  1015. expect(calls).toHaveLength(1);
  1016. const target = cg.getNode(calls[0]!.target);
  1017. expect(target?.name).toBe('func');
  1018. expect(target?.filePath.replace(/\\/g, '/')).toBe('pkg/module.py');
  1019. // The flip side of the fix: an attribute call through a *stdlib* module
  1020. // (`os.getcwd()`) must still create no edge — the fallback only matches
  1021. // real in-repo module files.
  1022. const externalCaller = cg.getNodesByKind('function').filter((n) => n.name === 'external_caller')[0];
  1023. expect(externalCaller).toBeDefined();
  1024. const externalCalls = cg.getOutgoingEdges(externalCaller!.id).filter((e) => e.kind === 'calls');
  1025. expect(externalCalls).toHaveLength(0);
  1026. });
  1027. it('attaches Go methods to their receiver type across files (#583, cross-file half)', async () => {
  1028. // In Go a type's methods are commonly declared in a different file from the
  1029. // `type` declaration (`type Box` in box.go, `func (b *Box) Get()` in
  1030. // box_methods.go). Extraction only attaches the struct→method `contains`
  1031. // edge when the type is in the SAME file (the owner lookup is file-scoped),
  1032. // so a cross-file method was orphaned from its struct — breaking member
  1033. // outlines and any callers/callees/impact traversal through `contains`. A
  1034. // resolution-phase pass now links them within the package (= directory).
  1035. fs.writeFileSync(
  1036. path.join(tempDir, 'box.go'),
  1037. 'package main\n\ntype Box struct{ v int }\n'
  1038. );
  1039. fs.writeFileSync(
  1040. path.join(tempDir, 'box_methods.go'),
  1041. 'package main\n\nfunc (b *Box) Get() int { return b.v }\nfunc (b *Box) Set(x int) { b.v = x }\n'
  1042. );
  1043. // Generic receiver declared cross-file too — exercises #583 half A
  1044. // (generic `*Stack[T]` receiver parsing) and half B (cross-file) together.
  1045. fs.writeFileSync(
  1046. path.join(tempDir, 'stack.go'),
  1047. 'package main\n\ntype Stack[T any] struct {\n\titems []T\n}\n'
  1048. );
  1049. fs.writeFileSync(
  1050. path.join(tempDir, 'stack_push.go'),
  1051. 'package main\n\nfunc (s *Stack[T]) Push(v T) { s.items = append(s.items, v) }\n'
  1052. );
  1053. // A same-named type in another package must NOT capture this package's
  1054. // methods — the link is scoped to the receiver type's own directory.
  1055. fs.mkdirSync(path.join(tempDir, 'other'));
  1056. fs.writeFileSync(
  1057. path.join(tempDir, 'other', 'box.go'),
  1058. 'package other\n\ntype Box struct{ w int }\n'
  1059. );
  1060. cg = await CodeGraph.init(tempDir, { index: true });
  1061. const methodsOf = (typeName: string, file: string): string[] => {
  1062. const node = cg
  1063. .getNodesByKind('struct')
  1064. .find((n) => n.name === typeName && n.filePath.replace(/\\/g, '/') === file);
  1065. expect(node, `${typeName} @ ${file}`).toBeDefined();
  1066. return cg
  1067. .getOutgoingEdges(node!.id)
  1068. .filter((e) => e.kind === 'contains')
  1069. .map((e) => cg.getNode(e.target))
  1070. .filter((n) => !!n && n.kind === 'method')
  1071. .map((n) => n!.name)
  1072. .sort();
  1073. };
  1074. // Cross-file (non-generic) methods now attach to their struct.
  1075. expect(methodsOf('Box', 'box.go')).toEqual(['Get', 'Set']);
  1076. // Generic + cross-file.
  1077. expect(methodsOf('Stack', 'stack.go')).toEqual(['Push']);
  1078. // Cross-package isolation: other/Box defines no methods of its own.
  1079. expect(methodsOf('Box', 'other/box.go')).toEqual([]);
  1080. });
  1081. it('TS type_alias object-shape members resolve method calls (#359)', async () => {
  1082. // Pre-#359, `recorder.stop()` (recorder: RecorderHandle) attached
  1083. // to `StdioMcpClient.stop` in a sibling directory via path-proximity
  1084. // because the type_alias had no `stop` node — only the unrelated
  1085. // class did. Now type_alias produces member nodes (property/method),
  1086. // so the camelCase receiver↔type word overlap pulls the call to
  1087. // `RecorderHandle::stop` instead of the look-alike class.
  1088. fs.mkdirSync(path.join(tempDir, 'voice'));
  1089. fs.mkdirSync(path.join(tempDir, 'codegraph'));
  1090. fs.writeFileSync(
  1091. path.join(tempDir, 'voice', 'recorder.ts'),
  1092. `export type RecorderHandle = {
  1093. wavPath: string;
  1094. stop: () => Promise<{ ok: true }>;
  1095. };
  1096. `
  1097. );
  1098. fs.writeFileSync(
  1099. path.join(tempDir, 'voice', 'controller.ts'),
  1100. `import type { RecorderHandle } from "./recorder";
  1101. export async function finaliseRecording(recorder: RecorderHandle) {
  1102. return await recorder.stop();
  1103. }
  1104. `
  1105. );
  1106. fs.writeFileSync(
  1107. path.join(tempDir, 'codegraph', 'stdio-client.ts'),
  1108. `export class StdioMcpClient {
  1109. private stopped = false;
  1110. async stop(): Promise<void> { this.stopped = true; }
  1111. }
  1112. `
  1113. );
  1114. cg = await CodeGraph.init(tempDir, { index: true });
  1115. const handleStop = cg
  1116. .getNodesByKind('method')
  1117. .find((n) => n.qualifiedName === 'RecorderHandle::stop');
  1118. expect(handleStop).toBeDefined();
  1119. const clientStop = cg
  1120. .getNodesByKind('method')
  1121. .find((n) => n.qualifiedName === 'StdioMcpClient::stop');
  1122. expect(clientStop).toBeDefined();
  1123. const handleCallers = cg.getIncomingEdges(handleStop!.id).filter((e) => e.kind === 'calls');
  1124. const clientCallers = cg.getIncomingEdges(clientStop!.id).filter((e) => e.kind === 'calls');
  1125. expect(handleCallers.length).toBeGreaterThanOrEqual(1);
  1126. // The class method must have NO callers — voice/'s call must NOT
  1127. // mis-attribute. A non-empty list would mean the false-positive
  1128. // path is still firing.
  1129. expect(clientCallers).toHaveLength(0);
  1130. // Function-typed property surfaces as a `method` node, not `property`,
  1131. // because `stop()` semantics at the call site are method semantics.
  1132. expect(handleStop!.kind).toBe('method');
  1133. });
  1134. it('Java import disambiguates same-name classes across modules (#314)', async () => {
  1135. // Pre-#314 the import resolver had no Java branch at all, so a
  1136. // multi-module Maven repo where `dao/converter/FooConverter` and
  1137. // `service/converter/FooConverter` both export a `convert` method
  1138. // resolved by file-path proximity — picking whichever class was
  1139. // closer to the caller, which is wrong any time the caller lives
  1140. // in an equidistant cross-cutting module.
  1141. const daoDir = path.join(tempDir, 'dao/src/main/java/com/example/dao/converter');
  1142. const serviceDir = path.join(tempDir, 'service/src/main/java/com/example/service/converter');
  1143. const webDir = path.join(tempDir, 'web/src/main/java/com/example/web');
  1144. fs.mkdirSync(daoDir, { recursive: true });
  1145. fs.mkdirSync(serviceDir, { recursive: true });
  1146. fs.mkdirSync(webDir, { recursive: true });
  1147. fs.writeFileSync(
  1148. path.join(daoDir, 'FooConverter.java'),
  1149. `package com.example.dao.converter;
  1150. public class FooConverter { public String convert(String x) { return "dao:" + x; } }
  1151. `
  1152. );
  1153. fs.writeFileSync(
  1154. path.join(serviceDir, 'FooConverter.java'),
  1155. `package com.example.service.converter;
  1156. public class FooConverter { public String convert(String x) { return "svc:" + x; } }
  1157. `
  1158. );
  1159. // The caller imports the SERVICE version — even though dao is
  1160. // alphabetically/lexically first in the candidate list, the
  1161. // import must trump that order.
  1162. fs.writeFileSync(
  1163. path.join(webDir, 'Handler.java'),
  1164. `package com.example.web;
  1165. import com.example.service.converter.FooConverter;
  1166. public class Handler {
  1167. private FooConverter fooConverter;
  1168. public String use() { return fooConverter.convert("input"); }
  1169. }
  1170. `
  1171. );
  1172. cg = await CodeGraph.init(tempDir, { index: true });
  1173. const use = cg
  1174. .getNodesByKind('method')
  1175. .find((n) => n.qualifiedName === 'com.example.web::Handler::use');
  1176. expect(use).toBeDefined();
  1177. const calls = cg.getOutgoingEdges(use!.id).filter((e) => e.kind === 'calls');
  1178. expect(calls.length).toBeGreaterThanOrEqual(1);
  1179. const target = cg.getNode(calls[0]!.target);
  1180. expect(target?.name).toBe('convert');
  1181. expect(target?.filePath.replace(/\\/g, '/')).toBe(
  1182. 'service/src/main/java/com/example/service/converter/FooConverter.java'
  1183. );
  1184. });
  1185. it('C# extracts references from method/property/field types (#381)', async () => {
  1186. // Pre-#381, every C# project produced ZERO `references` edges:
  1187. // csharp.ts was missing returnField, and the type-leaf walker
  1188. // only recognized TS/Java's `type_identifier` nodes — C# uses
  1189. // `identifier`/`predefined_type`/`qualified_name`/`generic_name`.
  1190. const srcDir = path.join(tempDir, 'src');
  1191. fs.mkdirSync(srcDir, { recursive: true });
  1192. fs.writeFileSync(
  1193. path.join(srcDir, 'Dtos.cs'),
  1194. `namespace MyApp;
  1195. public class SessionInfoDto { public string Id { get; set; } = ""; }
  1196. public class UserDto { public string Name { get; set; } = ""; }
  1197. `
  1198. );
  1199. fs.writeFileSync(
  1200. path.join(srcDir, 'Service.cs'),
  1201. `using System.Threading.Tasks;
  1202. namespace MyApp;
  1203. public class DataExporter
  1204. {
  1205. public SessionInfoDto Build(UserDto user, SessionInfoDto session) { return session; }
  1206. public Task<SessionInfoDto> BuildAsync(UserDto user) { return Task.FromResult(new SessionInfoDto()); }
  1207. public SessionInfoDto Latest { get; set; } = new();
  1208. private UserDto _cached;
  1209. }
  1210. `
  1211. );
  1212. cg = await CodeGraph.init(tempDir, { index: true });
  1213. const sessionDto = cg
  1214. .getNodesByKind('class')
  1215. .find((n) => n.name === 'SessionInfoDto');
  1216. const userDto = cg
  1217. .getNodesByKind('class')
  1218. .find((n) => n.name === 'UserDto');
  1219. expect(sessionDto).toBeDefined();
  1220. expect(userDto).toBeDefined();
  1221. const sessionIncoming = cg
  1222. .getIncomingEdges(sessionDto!.id)
  1223. .filter((e) => e.kind === 'references');
  1224. const userIncoming = cg
  1225. .getIncomingEdges(userDto!.id)
  1226. .filter((e) => e.kind === 'references');
  1227. // SessionInfoDto: Build return, Build param, BuildAsync return (inside Task<>), Latest property.
  1228. // UserDto: Build param, BuildAsync param, _cached field.
  1229. expect(sessionIncoming.length).toBeGreaterThanOrEqual(4);
  1230. expect(userIncoming.length).toBeGreaterThanOrEqual(3);
  1231. });
  1232. it('C# primary-constructor parameters record their type dependencies (#237)', async () => {
  1233. // C# 12 primary constructors declare a type's injected dependencies inline
  1234. // (`class Svc(IRepo repo, [FromKeyedServices("k")] ICache cache)`). Each
  1235. // ctor parameter's type is recorded as a `references` edge from the class,
  1236. // so a DI-registered contract reached only through a primary ctor is no
  1237. // longer reported as having no dependents.
  1238. fs.mkdirSync(path.join(tempDir, 'src'), { recursive: true });
  1239. fs.writeFileSync(
  1240. path.join(tempDir, 'src', 'Contracts.cs'),
  1241. `namespace App;
  1242. public interface IRepo { }
  1243. public class ICache { }
  1244. `
  1245. );
  1246. fs.writeFileSync(
  1247. path.join(tempDir, 'src', 'OrderService.cs'),
  1248. `namespace App;
  1249. public sealed class OrderService(IRepo repo, [FromKeyedServices("primary")] ICache cache)
  1250. {
  1251. public void Run() { }
  1252. }
  1253. `
  1254. );
  1255. cg = await CodeGraph.init(tempDir, { index: true });
  1256. const svc = cg.getNodesByKind('class').find((n) => n.name === 'OrderService');
  1257. expect(svc).toBeDefined();
  1258. // The class itself must index (it used to vanish under the old grammar).
  1259. const out = cg.getOutgoingEdges(svc!.id).filter((e) => e.kind === 'references');
  1260. const depNames = out.map((e) => cg.getNode(e.target)?.name);
  1261. expect(depNames).toContain('IRepo');
  1262. expect(depNames).toContain('ICache'); // the keyed-DI ([FromKeyedServices]) dependency
  1263. });
  1264. it('Go: leaves stdlib calls (fmt.Println, etc.) external', async () => {
  1265. fs.writeFileSync(
  1266. path.join(tempDir, 'go.mod'),
  1267. 'module github.com/example/myproject\n\ngo 1.21\n'
  1268. );
  1269. fs.writeFileSync(
  1270. path.join(tempDir, 'main.go'),
  1271. `package main
  1272. import "fmt"
  1273. func main() {
  1274. fmt.Println("hi")
  1275. }
  1276. `
  1277. );
  1278. cg = await CodeGraph.init(tempDir, { index: true });
  1279. const mainFn = cg.getNodesByKind('function').filter((n) => n.name ==='main')[0];
  1280. const calls = cg.getOutgoingEdges(mainFn!.id).filter((e) => e.kind === 'calls');
  1281. // No spurious in-project edge — fmt.* must stay unresolved/external.
  1282. expect(calls).toHaveLength(0);
  1283. });
  1284. });
  1285. describe('Same-name method disambiguation (#1079)', () => {
  1286. // resolveMethodOnType picks among several methods that share a
  1287. // `Type::method` qualifiedName. The precedence is:
  1288. // 1. preferredFqn (Java/Kotlin import — target is intentionally in
  1289. // ANOTHER file, #314),
  1290. // 2. the call site's OWN file (language-agnostic, #1079),
  1291. // 3. matches[0] (first-indexed) as a last resort.
  1292. const methodNode = (
  1293. id: string,
  1294. filePath: string,
  1295. language: Node['language'] = 'cpp',
  1296. qualifiedName = 'Logger::log',
  1297. name = 'log',
  1298. ): Node => ({
  1299. id, kind: 'method', name, qualifiedName, filePath, language,
  1300. startLine: 1, endLine: 1, startColumn: 0, endColumn: 0, updatedAt: 0,
  1301. });
  1302. const callRef = (filePath: string, language: Node['language'] = 'cpp'): UnresolvedRef => ({
  1303. fromNodeId: 'caller', referenceName: 'lg.log', referenceKind: 'calls',
  1304. line: 2, column: 0, filePath, language,
  1305. });
  1306. const ctxFor = (candidates: Node[]): ResolutionContext => ({
  1307. getNodesInFile: () => [],
  1308. getNodesByName: (name) => candidates.filter((c) => c.name === name),
  1309. getNodesByQualifiedName: () => [],
  1310. getNodesByKind: () => [],
  1311. fileExists: () => false,
  1312. readFile: () => null,
  1313. getProjectRoot: () => '',
  1314. getAllFiles: () => [],
  1315. });
  1316. it('prefers the definition in the call site\'s own file (#1079)', () => {
  1317. // matches[0] is the a/ definition; the call comes from b/, so it must
  1318. // resolve to b/ — not collapse onto the first-indexed match.
  1319. const logA = methodNode('m:a', 'a/svc.cpp');
  1320. const logB = methodNode('m:b', 'b/svc.cpp');
  1321. const result = resolveMethodOnType(
  1322. 'Logger', 'log', callRef('b/svc.cpp'), ctxFor([logA, logB]), 0.9, 'instance-method',
  1323. );
  1324. expect(result?.targetNodeId).toBe('m:b');
  1325. });
  1326. it('lets an import FQN pin a cross-file target over the same-file preference (#314)', () => {
  1327. // Java: two `Bar::doIt` in different packages. The import FQN pins the
  1328. // alpha package; even though the call site lives in beta's file, the FQN
  1329. // must win — the same-file preference runs only AFTER preferredFqn.
  1330. const alpha = methodNode('m:alpha', 'com/example/alpha/Bar.java', 'java', 'Bar::doIt', 'doIt');
  1331. const beta = methodNode('m:beta', 'com/example/beta/Bar.java', 'java', 'Bar::doIt', 'doIt');
  1332. const result = resolveMethodOnType(
  1333. 'Bar', 'doIt', callRef('com/example/beta/Bar.java', 'java'),
  1334. ctxFor([alpha, beta]), 0.9, 'instance-method', 'com.example.alpha.Bar',
  1335. );
  1336. expect(result?.targetNodeId).toBe('m:alpha');
  1337. });
  1338. it('falls back to the first match when nothing disambiguates', () => {
  1339. // Call site is a third file: no FQN, no same-file candidate → matches[0].
  1340. const logA = methodNode('m:a', 'a/svc.cpp');
  1341. const logB = methodNode('m:b', 'b/svc.cpp');
  1342. const result = resolveMethodOnType(
  1343. 'Logger', 'log', callRef('c/other.cpp'), ctxFor([logA, logB]), 0.9, 'instance-method',
  1344. );
  1345. expect(result?.targetNodeId).toBe('m:a');
  1346. });
  1347. it('resolves C++ calls end-to-end to same-named classes in different files (#1079)', async () => {
  1348. // The exact repro from the issue: two files, each with its own
  1349. // `Logger::log`. Before the fix both callers pointed at the first def.
  1350. fs.mkdirSync(path.join(tempDir, 'a'), { recursive: true });
  1351. fs.mkdirSync(path.join(tempDir, 'b'), { recursive: true });
  1352. fs.writeFileSync(
  1353. path.join(tempDir, 'a', 'svc.cpp'),
  1354. `class Logger { public: void log() { int a = 1; } };\nvoid useA() { Logger lg; lg.log(); }\n`,
  1355. );
  1356. fs.writeFileSync(
  1357. path.join(tempDir, 'b', 'svc.cpp'),
  1358. `class Logger { public: void log() { int b = 2; } };\nvoid useB() { Logger lg; lg.log(); }\n`,
  1359. );
  1360. cg = await CodeGraph.init(tempDir, { index: true });
  1361. cg.resolveReferences();
  1362. const logInDir = (dir: string) =>
  1363. cg.getNodesByKind('method').find(
  1364. (n) => n.name === 'log' && n.filePath.replace(/\\/g, '/').endsWith(`${dir}/svc.cpp`),
  1365. )!;
  1366. const callTargets = (fnName: string) =>
  1367. cg
  1368. .getOutgoingEdges(cg.getNodesByKind('function').find((n) => n.name === fnName)!.id)
  1369. .filter((e) => e.kind === 'calls')
  1370. .map((e) => e.target);
  1371. const logA = logInDir('a');
  1372. const logB = logInDir('b');
  1373. expect(logA).toBeDefined();
  1374. expect(logB).toBeDefined();
  1375. expect(logA.id).not.toBe(logB.id);
  1376. // Each caller resolves to the Logger::log in its OWN file.
  1377. expect(callTargets('useA')).toContain(logA.id);
  1378. expect(callTargets('useB')).toContain(logB.id);
  1379. });
  1380. it('preferCallSiteFile puts same-file candidates first and is otherwise a no-op', () => {
  1381. const a = methodNode('m:a', 'a/svc.cpp');
  1382. const b = methodNode('m:b', 'b/svc.cpp');
  1383. // Same-file first; the rest keep their original order (stable).
  1384. expect(preferCallSiteFile([a, b], 'b/svc.cpp').map((n) => n.id)).toEqual(['m:b', 'm:a']);
  1385. expect(preferCallSiteFile([a, b], 'a/svc.cpp').map((n) => n.id)).toEqual(['m:a', 'm:b']);
  1386. // No same-file match → unchanged; <2 candidates → returned as-is.
  1387. expect(preferCallSiteFile([a, b], 'c/other.cpp').map((n) => n.id)).toEqual(['m:a', 'm:b']);
  1388. expect(preferCallSiteFile([a], 'z/none.cpp')).toHaveLength(1);
  1389. });
  1390. it('matchByQualifiedName prefers the same-file target when a qualified name is ambiguous (#1079)', () => {
  1391. // Two `Logger::log` definitions; an explicit `Logger::log()` call from b/
  1392. // must resolve to b/'s definition, not the first-indexed one.
  1393. const a = methodNode('m:a', 'a/svc.cpp');
  1394. const b = methodNode('m:b', 'b/svc.cpp');
  1395. const ctx: ResolutionContext = {
  1396. getNodesInFile: () => [],
  1397. getNodesByName: (name) => [a, b].filter((n) => n.name === name),
  1398. getNodesByQualifiedName: (q) => (q === 'Logger::log' ? [a, b] : []),
  1399. getNodesByKind: () => [],
  1400. fileExists: () => false,
  1401. readFile: () => null,
  1402. getProjectRoot: () => '',
  1403. getAllFiles: () => [],
  1404. };
  1405. const ref: UnresolvedRef = {
  1406. fromNodeId: 'caller', referenceName: 'Logger::log', referenceKind: 'calls',
  1407. line: 2, column: 0, filePath: 'b/svc.cpp', language: 'cpp',
  1408. };
  1409. expect(matchByQualifiedName(ref, ctx)?.targetNodeId).toBe('m:b');
  1410. });
  1411. it('resolves a static/class-receiver call to the class in the caller\'s file (#1079)', async () => {
  1412. // `Logger.log()` — the receiver is the class NAME, so this routes through
  1413. // the class-name-receiver strategy (not the C++ instance path). It was
  1414. // file-blind across languages; verified here on TypeScript.
  1415. fs.mkdirSync(path.join(tempDir, 'a'), { recursive: true });
  1416. fs.mkdirSync(path.join(tempDir, 'b'), { recursive: true });
  1417. fs.writeFileSync(
  1418. path.join(tempDir, 'a', 'svc.ts'),
  1419. `class Logger { static log() { return 1; } }\nexport function useA() { return Logger.log(); }\n`,
  1420. );
  1421. fs.writeFileSync(
  1422. path.join(tempDir, 'b', 'svc.ts'),
  1423. `class Logger { static log() { return 2; } }\nexport function useB() { return Logger.log(); }\n`,
  1424. );
  1425. cg = await CodeGraph.init(tempDir, { index: true });
  1426. cg.resolveReferences();
  1427. const logInDir = (dir: string) =>
  1428. cg.getNodesByKind('method').find(
  1429. (n) => n.name === 'log' && n.filePath.replace(/\\/g, '/').endsWith(`${dir}/svc.ts`),
  1430. )!;
  1431. const callTargets = (fnName: string) =>
  1432. cg
  1433. .getOutgoingEdges(cg.getNodesByKind('function').find((n) => n.name === fnName)!.id)
  1434. .filter((e) => e.kind === 'calls')
  1435. .map((e) => e.target);
  1436. const logA = logInDir('a');
  1437. const logB = logInDir('b');
  1438. expect(logA?.id).not.toBe(logB?.id);
  1439. expect(callTargets('useA')).toContain(logA.id);
  1440. expect(callTargets('useB')).toContain(logB.id);
  1441. });
  1442. it('resolves an explicitly-qualified call to the definition in the caller\'s file (#1079)', async () => {
  1443. // `Logger::log()` with two `Logger::log` definitions routes through the
  1444. // qualified-name strategy, whose partial match previously picked the first.
  1445. fs.mkdirSync(path.join(tempDir, 'a'), { recursive: true });
  1446. fs.mkdirSync(path.join(tempDir, 'b'), { recursive: true });
  1447. fs.writeFileSync(
  1448. path.join(tempDir, 'a', 'svc.cpp'),
  1449. `class Logger { public: static void log() { int a = 1; } };\nvoid useA() { Logger::log(); }\n`,
  1450. );
  1451. fs.writeFileSync(
  1452. path.join(tempDir, 'b', 'svc.cpp'),
  1453. `class Logger { public: static void log() { int b = 2; } };\nvoid useB() { Logger::log(); }\n`,
  1454. );
  1455. cg = await CodeGraph.init(tempDir, { index: true });
  1456. cg.resolveReferences();
  1457. const logInDir = (dir: string) =>
  1458. cg.getNodesByKind('method').find(
  1459. (n) => n.name === 'log' && n.filePath.replace(/\\/g, '/').endsWith(`${dir}/svc.cpp`),
  1460. )!;
  1461. const callTargets = (fnName: string) =>
  1462. cg
  1463. .getOutgoingEdges(cg.getNodesByKind('function').find((n) => n.name === fnName)!.id)
  1464. .filter((e) => e.kind === 'calls')
  1465. .map((e) => e.target);
  1466. const logA = logInDir('a');
  1467. const logB = logInDir('b');
  1468. expect(logA?.id).not.toBe(logB?.id);
  1469. expect(callTargets('useA')).toContain(logA.id);
  1470. expect(callTargets('useB')).toContain(logB.id);
  1471. });
  1472. });
  1473. describe('Local-variable receiver-type inference (#1108)', () => {
  1474. // `lg.log()` where `lg` is a local whose type is inferred from its
  1475. // declaration/initializer. Before this, only C++ resolved these; every
  1476. // other language produced no method edge. Each case is one file with a
  1477. // single Logger + a caller using a local-variable receiver — a correct
  1478. // resolution makes the caller a caller of `log`.
  1479. const cases: Array<{ lang: string; file: string; src: string }> = [
  1480. { lang: 'TypeScript (= new T)', file: 'svc.ts',
  1481. src: `class Logger { log() { return 1; } }\nexport function use() { const lg = new Logger(); return lg.log(); }\n` },
  1482. { lang: 'JavaScript (= new T)', file: 'svc.js',
  1483. src: `class Logger { log() { return 1; } }\nexport function use() { const lg = new Logger(); return lg.log(); }\n` },
  1484. { lang: 'Python (= T())', file: 'svc.py',
  1485. src: `class Logger:\n def log(self):\n return 1\ndef use():\n lg = Logger()\n return lg.log()\n` },
  1486. { lang: 'Java (T x = new T)', file: 'Svc.java',
  1487. src: `class Logger { void log() { int a = 1; } }\nclass Use { void run() { Logger lg = new Logger(); lg.log(); } }\n` },
  1488. { lang: 'C# (var x = new T)', file: 'Svc.cs',
  1489. src: `class Logger { void Log() { int a = 1; } }\nclass Use { void Run() { var lg = new Logger(); lg.Log(); } }\n` },
  1490. { lang: 'Kotlin (val x = T())', file: 'Svc.kt',
  1491. src: `class Logger { fun log(): Int { return 1 } }\nfun use(): Int { val lg = Logger(); return lg.log() }\n` },
  1492. { lang: 'Swift (let x = T())', file: 'svc.swift',
  1493. src: `class Logger { func log() -> Int { return 1 } }\nfunc use() -> Int { let lg = Logger(); return lg.log() }\n` },
  1494. { lang: 'Go (x := T{})', file: 'svc.go',
  1495. src: `package a\ntype Logger struct{}\nfunc (l Logger) Log() int { return 1 }\nfunc Use() int { lg := Logger{}; return lg.Log() }\n` },
  1496. { lang: 'Rust (let x = T{})', file: 'svc.rs',
  1497. src: `pub struct Logger { n: i32 }\nimpl Logger { pub fn log(&self) -> i32 { self.n } }\npub fn use_it() -> i32 { let lg = Logger { n: 1 }; lg.log() }\n` },
  1498. { lang: 'Dart (var x = T())', file: 'svc.dart',
  1499. src: `class Logger { int log() { return 1; } }\nint use() { var lg = Logger(); return lg.log(); }\n` },
  1500. { lang: 'PHP ($x = new T)', file: 'svc.php',
  1501. src: `<?php\nclass Logger { function log() { return 1; } }\nfunction useIt() { $lg = new Logger(); return $lg->log(); }\n` },
  1502. { lang: 'Scala (val x = new T)', file: 'Svc.scala',
  1503. src: `class Logger { def log(): Int = 1 }\nobject A { def use(): Int = { val lg = new Logger(); lg.log() } }\n` },
  1504. ];
  1505. for (const c of cases) {
  1506. it(`resolves a local-variable method call — ${c.lang}`, async () => {
  1507. fs.writeFileSync(path.join(tempDir, c.file), c.src);
  1508. cg = await CodeGraph.init(tempDir, { index: true });
  1509. cg.resolveReferences();
  1510. const logMethod = cg
  1511. .getNodesByKind('method')
  1512. .find((n) => n.name.toLowerCase() === 'log');
  1513. expect(logMethod, `${c.lang}: log method should be indexed`).toBeDefined();
  1514. // The enclosing caller resolves through the local variable to `log`.
  1515. const callers = cg.getCallers(logMethod!.id).map((x) => x.node.name);
  1516. expect(
  1517. callers.length,
  1518. `${c.lang}: log should have a caller (got [${callers.join(', ')}])`,
  1519. ).toBeGreaterThan(0);
  1520. });
  1521. }
  1522. });
  1523. describe('Name Matcher: kind bias for new ref kinds', () => {
  1524. const baseContext = (candidates: Node[]): ResolutionContext => ({
  1525. getNodesInFile: () => [],
  1526. getNodesByName: (name) => candidates.filter((c) => c.name === name),
  1527. getNodesByQualifiedName: () => [],
  1528. getNodesByKind: () => [],
  1529. fileExists: () => true,
  1530. readFile: () => null,
  1531. getProjectRoot: () => '/test',
  1532. getAllFiles: () => [],
  1533. getNodesByLowerName: () => [],
  1534. getImportMappings: () => [],
  1535. });
  1536. it('prefers a class candidate over a function for `instantiates` refs', () => {
  1537. // A class and a function share a name across the codebase.
  1538. // Without the kind bias, the function (which gets the +25 `calls`
  1539. // bonus historically applied to all candidates of that kind) would
  1540. // win. Now the instantiates branch reverses it.
  1541. const fn: Node = {
  1542. id: 'func:utils.ts:Logger:5', kind: 'function', name: 'Logger',
  1543. qualifiedName: 'utils.ts::Logger', filePath: 'utils.ts', language: 'typescript',
  1544. startLine: 5, endLine: 7, startColumn: 0, endColumn: 0, updatedAt: Date.now(),
  1545. };
  1546. const cls: Node = {
  1547. id: 'class:logger.ts:Logger:10', kind: 'class', name: 'Logger',
  1548. qualifiedName: 'logger.ts::Logger', filePath: 'logger.ts', language: 'typescript',
  1549. startLine: 10, endLine: 30, startColumn: 0, endColumn: 0, updatedAt: Date.now(),
  1550. };
  1551. const ref = {
  1552. fromNodeId: 'func:main.ts:bootstrap:1',
  1553. referenceName: 'Logger',
  1554. referenceKind: 'instantiates' as const,
  1555. line: 5, column: 0, filePath: 'main.ts', language: 'typescript' as const,
  1556. };
  1557. const result = matchReference(ref, baseContext([fn, cls]));
  1558. expect(result?.targetNodeId).toBe('class:logger.ts:Logger:10');
  1559. });
  1560. it('prefers a function candidate over a non-function for `decorates` refs', () => {
  1561. const variable: Node = {
  1562. id: 'var:config.ts:Inject:5', kind: 'variable', name: 'Inject',
  1563. qualifiedName: 'config.ts::Inject', filePath: 'config.ts', language: 'typescript',
  1564. startLine: 5, endLine: 5, startColumn: 0, endColumn: 0, updatedAt: Date.now(),
  1565. };
  1566. const decorator: Node = {
  1567. id: 'func:di.ts:Inject:10', kind: 'function', name: 'Inject',
  1568. qualifiedName: 'di.ts::Inject', filePath: 'di.ts', language: 'typescript',
  1569. startLine: 10, endLine: 20, startColumn: 0, endColumn: 0, updatedAt: Date.now(),
  1570. };
  1571. const ref = {
  1572. fromNodeId: 'class:svc.ts:UserService:1',
  1573. referenceName: 'Inject',
  1574. referenceKind: 'decorates' as const,
  1575. line: 5, column: 0, filePath: 'svc.ts', language: 'typescript' as const,
  1576. };
  1577. const result = matchReference(ref, baseContext([variable, decorator]));
  1578. expect(result?.targetNodeId).toBe('func:di.ts:Inject:10');
  1579. });
  1580. });
  1581. describe('tsconfig path aliases', () => {
  1582. it('resolves an aliased import to the alias-mapped file (not a same-named file elsewhere)', async () => {
  1583. // Two same-named exports in different directories. Without alias
  1584. // resolution, name-matcher would pick whichever it finds first;
  1585. // with alias resolution, the import path uniquely picks one.
  1586. fs.mkdirSync(path.join(tempDir, 'src/utils'), { recursive: true });
  1587. fs.mkdirSync(path.join(tempDir, 'src/legacy'), { recursive: true });
  1588. fs.writeFileSync(
  1589. path.join(tempDir, 'src/utils/format.ts'),
  1590. `export function pickMe(): number { return 1; }\n`
  1591. );
  1592. fs.writeFileSync(
  1593. path.join(tempDir, 'src/legacy/format.ts'),
  1594. `export function pickMe(): number { return 99; }\n`
  1595. );
  1596. fs.writeFileSync(
  1597. path.join(tempDir, 'src/main.ts'),
  1598. `import { pickMe } from '@utils/format';\nexport function go(): number { return pickMe(); }\n`
  1599. );
  1600. fs.writeFileSync(
  1601. path.join(tempDir, 'tsconfig.json'),
  1602. JSON.stringify({
  1603. compilerOptions: {
  1604. baseUrl: './src',
  1605. paths: { '@utils/*': ['utils/*'] },
  1606. },
  1607. })
  1608. );
  1609. cg = await CodeGraph.init(tempDir, { index: true });
  1610. cg.resolveReferences();
  1611. // The two pickMe nodes live in different files. The aliased
  1612. // import should attach the call edge to the @utils-mapped one,
  1613. // not the legacy duplicate.
  1614. const all = cg.getNodesByKind('function').filter((n) => n.name === 'pickMe');
  1615. const utilsNode = all.find((n) => n.filePath === 'src/utils/format.ts');
  1616. const legacyNode = all.find((n) => n.filePath === 'src/legacy/format.ts');
  1617. expect(utilsNode).toBeDefined();
  1618. expect(legacyNode).toBeDefined();
  1619. const utilsCallers = cg.getCallers(utilsNode!.id);
  1620. const legacyCallers = cg.getCallers(legacyNode!.id);
  1621. expect(utilsCallers.length).toBeGreaterThan(0);
  1622. expect(utilsCallers.some((c) => c.node.filePath === 'src/main.ts')).toBe(true);
  1623. // The legacy node should NOT have a caller from src/main.ts —
  1624. // the alias correctly picked the utils version.
  1625. expect(legacyCallers.some((c) => c.node.filePath === 'src/main.ts')).toBe(false);
  1626. });
  1627. it('falls back gracefully when tsconfig is absent', async () => {
  1628. fs.mkdirSync(path.join(tempDir, 'src'), { recursive: true });
  1629. fs.writeFileSync(
  1630. path.join(tempDir, 'src/a.ts'),
  1631. `export function aFn(): void {}\n`
  1632. );
  1633. fs.writeFileSync(
  1634. path.join(tempDir, 'src/b.ts'),
  1635. `import { aFn } from './a';\nexport function bFn(): void { aFn(); }\n`
  1636. );
  1637. cg = await CodeGraph.init(tempDir, { index: true });
  1638. // No tsconfig present — index should still complete and the
  1639. // relative-import-based call edge should be created.
  1640. const aFn = cg.getNodesByKind('function').find((n) => n.name === 'aFn');
  1641. expect(aFn).toBeDefined();
  1642. const callers = cg.getCallers(aFn!.id);
  1643. expect(callers.some((c) => c.node.filePath === 'src/b.ts')).toBe(true);
  1644. });
  1645. });
  1646. describe('re-export chain following', () => {
  1647. it('chases a 3-hop barrel chain (wildcard → named → declaration)', async () => {
  1648. // main.ts → all.ts (wildcard) → index.ts (named) → auth.ts (declaration).
  1649. // Without chain following, `signIn` resolves to nothing because
  1650. // none of the barrel files declare it directly.
  1651. fs.mkdirSync(path.join(tempDir, 'src/services'), { recursive: true });
  1652. fs.writeFileSync(
  1653. path.join(tempDir, 'src/services/auth.ts'),
  1654. `export function signIn(): void {}\n`
  1655. );
  1656. fs.writeFileSync(
  1657. path.join(tempDir, 'src/services/index.ts'),
  1658. `export { signIn } from './auth';\n`
  1659. );
  1660. fs.writeFileSync(
  1661. path.join(tempDir, 'src/all.ts'),
  1662. `export * from './services/index';\n`
  1663. );
  1664. fs.writeFileSync(
  1665. path.join(tempDir, 'src/main.ts'),
  1666. `import { signIn } from './all';\nexport function go(): void { signIn(); }\n`
  1667. );
  1668. cg = await CodeGraph.init(tempDir, { index: true });
  1669. cg.resolveReferences();
  1670. const signInNode = cg
  1671. .getNodesByKind('function')
  1672. .find((n) => n.name === 'signIn' && n.filePath === 'src/services/auth.ts');
  1673. expect(signInNode).toBeDefined();
  1674. const callers = cg.getCallers(signInNode!.id);
  1675. expect(callers.some((c) => c.node.filePath === 'src/main.ts')).toBe(true);
  1676. });
  1677. it('follows a renamed named re-export (export { foo as bar } from ...)', async () => {
  1678. // The chase has to look up `foo` in the upstream module even
  1679. // though the importer asked for `bar` — exercises the rename
  1680. // branch of findExportedSymbol.
  1681. fs.mkdirSync(path.join(tempDir, 'src'), { recursive: true });
  1682. fs.writeFileSync(
  1683. path.join(tempDir, 'src/auth.ts'),
  1684. `export function signIn(): void {}\n`
  1685. );
  1686. fs.writeFileSync(
  1687. path.join(tempDir, 'src/index.ts'),
  1688. `export { signIn as login } from './auth';\n`
  1689. );
  1690. fs.writeFileSync(
  1691. path.join(tempDir, 'src/main.ts'),
  1692. `import { login } from './index';\nexport function go(): void { login(); }\n`
  1693. );
  1694. cg = await CodeGraph.init(tempDir, { index: true });
  1695. cg.resolveReferences();
  1696. const signInNode = cg
  1697. .getNodesByKind('function')
  1698. .find((n) => n.name === 'signIn' && n.filePath === 'src/auth.ts');
  1699. expect(signInNode).toBeDefined();
  1700. const callers = cg.getCallers(signInNode!.id);
  1701. expect(callers.some((c) => c.node.filePath === 'src/main.ts')).toBe(true);
  1702. });
  1703. it('follows a default re-export of a .svelte component (export { default as Foo } from ./RealButton.svelte) (#629)', async () => {
  1704. // The ubiquitous Svelte/React component-barrel form. The leaf is a
  1705. // .svelte component (extracted as kind 'component', the default
  1706. // export). The re-export ALIAS (`Foo`) deliberately differs from the
  1707. // component's real name (`RealButton`) so the name-matcher fallback
  1708. // can't coincidentally connect them — the only path to the edge is
  1709. // the import-chase, which must match a `component` (not just
  1710. // function/class) for the default export. Otherwise the
  1711. // consumer↔component edge is never created and `callers` returns a
  1712. // false 0.
  1713. fs.mkdirSync(path.join(tempDir, 'src/lib'), { recursive: true });
  1714. fs.writeFileSync(
  1715. path.join(tempDir, 'src/lib/RealButton.svelte'),
  1716. `<script lang="ts">\n export let label: string = '';\n</script>\n\n<button>{label}</button>\n`
  1717. );
  1718. fs.writeFileSync(
  1719. path.join(tempDir, 'src/lib/index.ts'),
  1720. `export { default as Foo } from './RealButton.svelte';\n`
  1721. );
  1722. fs.writeFileSync(
  1723. path.join(tempDir, 'src/Bar.svelte'),
  1724. `<script lang="ts">\n import { Foo } from './lib';\n</script>\n\n<Foo />\n`
  1725. );
  1726. cg = await CodeGraph.init(tempDir, { index: true });
  1727. cg.resolveReferences();
  1728. const fooNode = cg
  1729. .getNodesByKind('component')
  1730. .find((n) => n.name === 'RealButton' && n.filePath === 'src/lib/RealButton.svelte');
  1731. expect(fooNode).toBeDefined();
  1732. const callers = cg.getCallers(fooNode!.id);
  1733. expect(callers.some((c) => c.node.filePath === 'src/Bar.svelte')).toBe(true);
  1734. });
  1735. it('links an .astro page to the component and TS util it uses (#768)', async () => {
  1736. // The canonical Astro shape: a page imports a layout/component in
  1737. // frontmatter and uses it as a template tag; the component's template
  1738. // calls an imported .ts util. Both hops must produce graph edges or
  1739. // an Astro project is invisible to callers/impact.
  1740. fs.mkdirSync(path.join(tempDir, 'src/components'), { recursive: true });
  1741. fs.mkdirSync(path.join(tempDir, 'src/utils'), { recursive: true });
  1742. fs.mkdirSync(path.join(tempDir, 'src/pages'), { recursive: true });
  1743. fs.writeFileSync(
  1744. path.join(tempDir, 'src/utils/format.ts'),
  1745. `export function formatDate(d: Date): string { return d.toISOString(); }\n`
  1746. );
  1747. fs.writeFileSync(
  1748. path.join(tempDir, 'src/components/PostCard.astro'),
  1749. `---\nimport { formatDate } from '../utils/format';\nconst { date } = Astro.props;\n---\n<time>{formatDate(date)}</time>\n`
  1750. );
  1751. fs.writeFileSync(
  1752. path.join(tempDir, 'src/pages/index.astro'),
  1753. `---\nimport PostCard from '../components/PostCard.astro';\n---\n<PostCard date={new Date()} />\n`
  1754. );
  1755. cg = await CodeGraph.init(tempDir, { index: true });
  1756. cg.resolveReferences();
  1757. // Hop 1: page → component (template tag through the frontmatter import)
  1758. const cardNode = cg
  1759. .getNodesByKind('component')
  1760. .find((n) => n.name === 'PostCard' && n.filePath === 'src/components/PostCard.astro');
  1761. expect(cardNode).toBeDefined();
  1762. const cardCallers = cg.getCallers(cardNode!.id);
  1763. expect(cardCallers.some((c) => c.node.filePath === 'src/pages/index.astro')).toBe(true);
  1764. // Hop 2: component template call → .ts util
  1765. const fmtNode = cg
  1766. .getNodesByKind('function')
  1767. .find((n) => n.name === 'formatDate' && n.filePath === 'src/utils/format.ts');
  1768. expect(fmtNode).toBeDefined();
  1769. const fmtCallers = cg.getCallers(fmtNode!.id);
  1770. expect(fmtCallers.some((c) => c.node.filePath === 'src/components/PostCard.astro')).toBe(true);
  1771. });
  1772. it('resolves a bare directory import (import { x } from "." / "./") to index.ts (#629)', async () => {
  1773. // `import { helper } from '.'` (or './') must map to the
  1774. // directory's index.ts before the re-export chase can run. The
  1775. // barrel renames `realHelper` → `helper` so the name-matcher can't
  1776. // mask a path-resolution failure: only the bare-dir resolution +
  1777. // rename chase can connect the edge.
  1778. fs.mkdirSync(path.join(tempDir, 'src'), { recursive: true });
  1779. fs.writeFileSync(
  1780. path.join(tempDir, 'src/util.ts'),
  1781. `export function realHelper(): void {}\n`
  1782. );
  1783. fs.writeFileSync(
  1784. path.join(tempDir, 'src/index.ts'),
  1785. `export { realHelper as helper } from './util';\n`
  1786. );
  1787. fs.writeFileSync(
  1788. path.join(tempDir, 'src/main.ts'),
  1789. `import { helper } from '.';\nexport function go(): void { helper(); }\n`
  1790. );
  1791. fs.writeFileSync(
  1792. path.join(tempDir, 'src/main2.ts'),
  1793. `import { helper } from './';\nexport function go2(): void { helper(); }\n`
  1794. );
  1795. cg = await CodeGraph.init(tempDir, { index: true });
  1796. cg.resolveReferences();
  1797. const helperNode = cg
  1798. .getNodesByKind('function')
  1799. .find((n) => n.name === 'realHelper' && n.filePath === 'src/util.ts');
  1800. expect(helperNode).toBeDefined();
  1801. const callers = cg.getCallers(helperNode!.id);
  1802. expect(callers.some((c) => c.node.filePath === 'src/main.ts')).toBe(true);
  1803. expect(callers.some((c) => c.node.filePath === 'src/main2.ts')).toBe(true);
  1804. });
  1805. it('resolves a workspace package-subpath barrel (@scope/pkg/sub) to its index (#629)', async () => {
  1806. // bun/npm/pnpm workspace: `@scope/ui/widgets` → the `ui` package's
  1807. // `widgets/` subdir index, which re-exports a .svelte component.
  1808. // Alias `Thing` ≠ component `Widget` defeats the name-matcher, so
  1809. // only workspace-package resolution can connect the edge.
  1810. fs.mkdirSync(path.join(tempDir, 'packages/ui/widgets'), { recursive: true });
  1811. fs.writeFileSync(
  1812. path.join(tempDir, 'package.json'),
  1813. JSON.stringify({ name: 'root', private: true, workspaces: ['packages/*'] }, null, 2)
  1814. );
  1815. fs.writeFileSync(
  1816. path.join(tempDir, 'packages/ui/package.json'),
  1817. JSON.stringify({ name: '@scope/ui', version: '1.0.0' }, null, 2)
  1818. );
  1819. fs.writeFileSync(
  1820. path.join(tempDir, 'packages/ui/widgets/Widget.svelte'),
  1821. `<script lang="ts">\n export let label: string = '';\n</script>\n\n<button>{label}</button>\n`
  1822. );
  1823. fs.writeFileSync(
  1824. path.join(tempDir, 'packages/ui/widgets/index.ts'),
  1825. `export { default as Thing } from './Widget.svelte';\n`
  1826. );
  1827. fs.mkdirSync(path.join(tempDir, 'app'), { recursive: true });
  1828. fs.writeFileSync(
  1829. path.join(tempDir, 'app/App.svelte'),
  1830. `<script lang="ts">\n import { Thing } from '@scope/ui/widgets';\n</script>\n\n<Thing />\n`
  1831. );
  1832. cg = await CodeGraph.init(tempDir, { index: true });
  1833. cg.resolveReferences();
  1834. const buttonNode = cg
  1835. .getNodesByKind('component')
  1836. .find((n) => n.name === 'Widget' && n.filePath === 'packages/ui/widgets/Widget.svelte');
  1837. expect(buttonNode).toBeDefined();
  1838. const callers = cg.getCallers(buttonNode!.id);
  1839. expect(callers.some((c) => c.node.filePath === 'app/App.svelte')).toBe(true);
  1840. });
  1841. it('resolves a barrel import from a Vue SFC <script> block (#629)', async () => {
  1842. // The same import-resolution gaps (no SFC import mappings, no SFC
  1843. // extension list, barrel parsed in the consumer's language) broke
  1844. // Vue SFCs too. Guards the resolver-side generalization to `.vue`.
  1845. // The barrel renames `realRun` → `run` so only the import-chase (not
  1846. // the name-matcher) can connect the call.
  1847. fs.mkdirSync(path.join(tempDir, 'src'), { recursive: true });
  1848. fs.writeFileSync(
  1849. path.join(tempDir, 'src/util.ts'),
  1850. `export function realRun(): void {}\n`
  1851. );
  1852. fs.writeFileSync(
  1853. path.join(tempDir, 'src/index.ts'),
  1854. `export { realRun as run } from './util';\n`
  1855. );
  1856. fs.writeFileSync(
  1857. path.join(tempDir, 'src/App.vue'),
  1858. `<script lang="ts">\nimport { run } from './';\nexport default { mounted() { run(); } };\n</script>\n<template><div/></template>\n`
  1859. );
  1860. cg = await CodeGraph.init(tempDir, { index: true });
  1861. cg.resolveReferences();
  1862. const runNode = cg
  1863. .getNodesByKind('function')
  1864. .find((n) => n.name === 'realRun' && n.filePath === 'src/util.ts');
  1865. expect(runNode).toBeDefined();
  1866. const callers = cg.getCallers(runNode!.id);
  1867. expect(callers.some((c) => c.node.filePath === 'src/App.vue')).toBe(true);
  1868. });
  1869. it('follows a Vue component used in a <template> through a default re-export barrel (#629)', async () => {
  1870. // End-to-end Vue analogue of the Svelte case: the leaf is a `.vue`
  1871. // component re-exported under an alias (`Thing`) that differs from its
  1872. // real name (`Widget`), and the consumer uses it ONLY in markup
  1873. // (`<Thing />`). Requires both the new template-tag extraction AND the
  1874. // barrel default-export chase to connect the edge.
  1875. fs.mkdirSync(path.join(tempDir, 'src/lib'), { recursive: true });
  1876. fs.writeFileSync(
  1877. path.join(tempDir, 'src/lib/Widget.vue'),
  1878. `<script setup lang="ts">\ndefineProps<{ label?: string }>();\n</script>\n<template><button>x</button></template>\n`
  1879. );
  1880. fs.writeFileSync(
  1881. path.join(tempDir, 'src/lib/index.ts'),
  1882. `export { default as Thing } from './Widget.vue';\n`
  1883. );
  1884. fs.writeFileSync(
  1885. path.join(tempDir, 'src/App.vue'),
  1886. `<script setup lang="ts">\nimport { Thing } from './lib';\n</script>\n<template>\n <Thing />\n</template>\n`
  1887. );
  1888. cg = await CodeGraph.init(tempDir, { index: true });
  1889. cg.resolveReferences();
  1890. const widgetNode = cg
  1891. .getNodesByKind('component')
  1892. .find((n) => n.name === 'Widget' && n.filePath === 'src/lib/Widget.vue');
  1893. expect(widgetNode).toBeDefined();
  1894. const callers = cg.getCallers(widgetNode!.id);
  1895. expect(callers.some((c) => c.node.filePath === 'src/App.vue')).toBe(true);
  1896. });
  1897. });
  1898. describe('C/C++ Import Resolution', () => {
  1899. afterEach(() => {
  1900. clearCppIncludeDirCache();
  1901. });
  1902. it('should resolve C include to header in same directory', () => {
  1903. const context: ResolutionContext = {
  1904. getNodesInFile: () => [],
  1905. getNodesByName: () => [],
  1906. getNodesByQualifiedName: () => [],
  1907. getNodesByKind: () => [],
  1908. fileExists: (p) => p === 'utils.h',
  1909. readFile: () => null,
  1910. getProjectRoot: () => '',
  1911. getAllFiles: () => ['utils.h', 'main.c'],
  1912. };
  1913. const result = resolveImportPath(
  1914. 'utils.h',
  1915. 'main.c',
  1916. 'c',
  1917. context
  1918. );
  1919. expect(result).toBe('utils.h');
  1920. });
  1921. it('should resolve C++ include with .hpp extension', () => {
  1922. const context: ResolutionContext = {
  1923. getNodesInFile: () => [],
  1924. getNodesByName: () => [],
  1925. getNodesByQualifiedName: () => [],
  1926. getNodesByKind: () => [],
  1927. fileExists: (p) => p === 'include/myclass.hpp',
  1928. readFile: () => null,
  1929. getProjectRoot: () => '',
  1930. getAllFiles: () => ['include/myclass.hpp', 'src/main.cpp'],
  1931. getCppIncludeDirs: () => ['include'],
  1932. };
  1933. const result = resolveImportPath(
  1934. 'myclass.hpp',
  1935. 'src/main.cpp',
  1936. 'cpp',
  1937. context
  1938. );
  1939. expect(result).toBe('include/myclass.hpp');
  1940. });
  1941. it('should resolve include with subdirectory path', () => {
  1942. const context: ResolutionContext = {
  1943. getNodesInFile: () => [],
  1944. getNodesByName: () => [],
  1945. getNodesByQualifiedName: () => [],
  1946. getNodesByKind: () => [],
  1947. fileExists: (p) => p === 'utils/helpers.h',
  1948. readFile: () => null,
  1949. getProjectRoot: () => '',
  1950. getAllFiles: () => ['utils/helpers.h', 'main.c'],
  1951. };
  1952. const result = resolveImportPath(
  1953. 'utils/helpers.h',
  1954. 'main.c',
  1955. 'c',
  1956. context
  1957. );
  1958. expect(result).toBe('utils/helpers.h');
  1959. });
  1960. it('should resolve include via include directories', () => {
  1961. const context: ResolutionContext = {
  1962. getNodesInFile: () => [],
  1963. getNodesByName: () => [],
  1964. getNodesByQualifiedName: () => [],
  1965. getNodesByKind: () => [],
  1966. fileExists: (p) => p === 'include/myheader.h',
  1967. readFile: () => null,
  1968. getProjectRoot: () => '',
  1969. getAllFiles: () => ['include/myheader.h', 'src/main.cpp'],
  1970. getCppIncludeDirs: () => ['include'],
  1971. };
  1972. const result = resolveImportPath(
  1973. 'myheader.h',
  1974. 'src/main.cpp',
  1975. 'cpp',
  1976. context
  1977. );
  1978. expect(result).toBe('include/myheader.h');
  1979. });
  1980. it('should resolve include trying multiple extensions', () => {
  1981. const context: ResolutionContext = {
  1982. getNodesInFile: () => [],
  1983. getNodesByName: () => [],
  1984. getNodesByQualifiedName: () => [],
  1985. getNodesByKind: () => [],
  1986. // myclass.h does not exist, but myclass.hpp does
  1987. fileExists: (p) => p === 'include/myclass.hpp',
  1988. readFile: () => null,
  1989. getProjectRoot: () => '',
  1990. getAllFiles: () => ['include/myclass.hpp', 'src/main.cpp'],
  1991. getCppIncludeDirs: () => ['include'],
  1992. };
  1993. const result = resolveImportPath(
  1994. 'myclass',
  1995. 'src/main.cpp',
  1996. 'cpp',
  1997. context
  1998. );
  1999. expect(result).toBe('include/myclass.hpp');
  2000. });
  2001. it('should return null for system headers', () => {
  2002. const context: ResolutionContext = {
  2003. getNodesInFile: () => [],
  2004. getNodesByName: () => [],
  2005. getNodesByQualifiedName: () => [],
  2006. getNodesByKind: () => [],
  2007. fileExists: () => true,
  2008. readFile: () => null,
  2009. getProjectRoot: () => '',
  2010. getAllFiles: () => [],
  2011. };
  2012. // C standard library header
  2013. expect(resolveImportPath('stdio.h', 'main.c', 'c', context)).toBeNull();
  2014. // C++ standard library header
  2015. expect(resolveImportPath('vector', 'main.cpp', 'cpp', context)).toBeNull();
  2016. // C++ C-wrapper header
  2017. expect(resolveImportPath('cstdio', 'main.cpp', 'cpp', context)).toBeNull();
  2018. });
  2019. it('should return null for single-component third-party paths that cannot be resolved', () => {
  2020. const context: ResolutionContext = {
  2021. getNodesInFile: () => [],
  2022. getNodesByName: () => [],
  2023. getNodesByQualifiedName: () => [],
  2024. getNodesByKind: () => [],
  2025. fileExists: () => false,
  2026. readFile: () => null,
  2027. getProjectRoot: () => '',
  2028. getAllFiles: () => [],
  2029. getCppIncludeDirs: () => [],
  2030. };
  2031. // Third-party bare header without path — not resolvable, returns null
  2032. const result = resolveImportPath(
  2033. 'openssl/ssl.h',
  2034. 'main.cpp',
  2035. 'cpp',
  2036. context
  2037. );
  2038. expect(result).toBeNull();
  2039. });
  2040. it('should not filter project headers with path separators', () => {
  2041. const context: ResolutionContext = {
  2042. getNodesInFile: () => [],
  2043. getNodesByName: () => [],
  2044. getNodesByQualifiedName: () => [],
  2045. getNodesByKind: () => [],
  2046. fileExists: (p) => p === 'mylib/utils.h',
  2047. readFile: () => null,
  2048. getProjectRoot: () => '',
  2049. getAllFiles: () => ['mylib/utils.h'],
  2050. };
  2051. // Path with separator should NOT be filtered as external
  2052. const result = resolveImportPath(
  2053. 'mylib/utils.h',
  2054. 'main.c',
  2055. 'c',
  2056. context
  2057. );
  2058. expect(result).toBe('mylib/utils.h');
  2059. });
  2060. it('should extract C/C++ import mappings from #include directives', () => {
  2061. const code = `#include <iostream>
  2062. #include "myheader.h"
  2063. #include "utils/helpers.hpp"`;
  2064. const mappings = extractImportMappings('main.cpp', code, 'cpp');
  2065. expect(mappings.length).toBe(3);
  2066. expect(mappings[0]).toEqual({
  2067. localName: 'iostream',
  2068. exportedName: '*',
  2069. source: 'iostream',
  2070. isDefault: false,
  2071. isNamespace: true,
  2072. });
  2073. expect(mappings[1]).toEqual({
  2074. localName: 'myheader',
  2075. exportedName: '*',
  2076. source: 'myheader.h',
  2077. isDefault: false,
  2078. isNamespace: true,
  2079. });
  2080. expect(mappings[2]).toEqual({
  2081. localName: 'helpers',
  2082. exportedName: '*',
  2083. source: 'utils/helpers.hpp',
  2084. isDefault: false,
  2085. isNamespace: true,
  2086. });
  2087. });
  2088. it('should discover include directories from compile_commands.json', () => {
  2089. // Create a temp project with compile_commands.json
  2090. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-cpp-test-'));
  2091. try {
  2092. const compileDb = [
  2093. {
  2094. directory: tempProject,
  2095. command: 'g++ -Iinclude -Isrc/lib -isystem /usr/include -c src/main.cpp',
  2096. file: 'src/main.cpp',
  2097. },
  2098. ];
  2099. fs.writeFileSync(
  2100. path.join(tempProject, 'compile_commands.json'),
  2101. JSON.stringify(compileDb)
  2102. );
  2103. // Create the include dirs so they exist
  2104. fs.mkdirSync(path.join(tempProject, 'include'), { recursive: true });
  2105. fs.mkdirSync(path.join(tempProject, 'src', 'lib'), { recursive: true });
  2106. clearCppIncludeDirCache();
  2107. const dirs = loadCppIncludeDirs(tempProject);
  2108. // Should find include and src/lib (relative to project root)
  2109. // /usr/include is absolute and outside project, should be excluded
  2110. expect(dirs).toContain('include');
  2111. expect(dirs).toContain('src/lib');
  2112. expect(dirs.some(d => d.includes('usr'))).toBe(false);
  2113. } finally {
  2114. fs.rmSync(tempProject, { recursive: true });
  2115. }
  2116. });
  2117. it('should fall back to heuristic include dirs when no compile_commands.json', () => {
  2118. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-cpp-test-'));
  2119. try {
  2120. // Create include/ and src/ directories with headers
  2121. fs.mkdirSync(path.join(tempProject, 'include'), { recursive: true });
  2122. fs.writeFileSync(path.join(tempProject, 'include', 'types.h'), '');
  2123. fs.mkdirSync(path.join(tempProject, 'src'), { recursive: true });
  2124. fs.writeFileSync(path.join(tempProject, 'src', 'main.cpp'), '');
  2125. // Create a directory without headers — should not be included
  2126. fs.mkdirSync(path.join(tempProject, 'docs'), { recursive: true });
  2127. clearCppIncludeDirCache();
  2128. const dirs = loadCppIncludeDirs(tempProject);
  2129. expect(dirs).toContain('include');
  2130. expect(dirs).toContain('src');
  2131. expect(dirs).not.toContain('docs');
  2132. } finally {
  2133. fs.rmSync(tempProject, { recursive: true });
  2134. }
  2135. });
  2136. // Documents the cross-language `.h` behavior. Objective-C and C++ share
  2137. // the `.h` extension, so in a mixed iOS-style project an Obj-C header
  2138. // dir gets claimed as a C/C++ include dir too. That's intentional — a
  2139. // C++ file legitimately can `#include "Foo.h"` against an Obj-C header
  2140. // (Obj-C++ / .mm callers), and false-positive inclusion is far cheaper
  2141. // than missing real resolutions. The test pins this so a later
  2142. // "exclude objc dirs" refactor breaks loudly and reviewers see the
  2143. // trade-off explicitly.
  2144. it('heuristic claims any top-level dir containing .h files, including Obj-C', () => {
  2145. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-cpp-test-'));
  2146. try {
  2147. // C++ side: an `cppmod` dir with a .hpp (C++-only extension)
  2148. fs.mkdirSync(path.join(tempProject, 'cppmod'), { recursive: true });
  2149. fs.writeFileSync(path.join(tempProject, 'cppmod', 'shared.hpp'), '');
  2150. // Obj-C side: an `iosmod` dir with .h + .m (no .cpp/.hpp).
  2151. fs.mkdirSync(path.join(tempProject, 'iosmod'), { recursive: true });
  2152. fs.writeFileSync(path.join(tempProject, 'iosmod', 'View.h'), '');
  2153. fs.writeFileSync(path.join(tempProject, 'iosmod', 'View.m'), '');
  2154. clearCppIncludeDirCache();
  2155. const dirs = loadCppIncludeDirs(tempProject);
  2156. // Both included — Obj-C dirs are intentionally allowed.
  2157. expect(dirs).toContain('cppmod');
  2158. expect(dirs).toContain('iosmod');
  2159. } finally {
  2160. fs.rmSync(tempProject, { recursive: true });
  2161. }
  2162. });
  2163. // End-to-end: ensure `#include "X.h"` produces a file→file `imports` edge
  2164. // in the actual indexing pipeline (not just a phantom file→import-node
  2165. // edge). This pins the include-dir resolution path so the headline PR
  2166. // feature can't silently regress to a no-op in the indexing flow.
  2167. it('connects #include to the real header file via include-dir scan (end-to-end)', async () => {
  2168. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-cpp-e2e-'));
  2169. try {
  2170. fs.mkdirSync(path.join(tempProject, 'include'), { recursive: true });
  2171. fs.mkdirSync(path.join(tempProject, 'src'), { recursive: true });
  2172. fs.writeFileSync(
  2173. path.join(tempProject, 'include', 'utils.h'),
  2174. `#ifndef UTILS_H\n#define UTILS_H\nint add(int, int);\n#endif\n`
  2175. );
  2176. fs.writeFileSync(
  2177. path.join(tempProject, 'src', 'main.cpp'),
  2178. `#include "utils.h"\n#include <vector>\nint main(){ return add(1,2); }\n`
  2179. );
  2180. clearCppIncludeDirCache();
  2181. cg = await CodeGraph.init(tempProject, { index: true });
  2182. // Sanity: file nodes exist for the header and the cpp.
  2183. const allFiles = cg.getStats();
  2184. expect(allFiles.fileCount).toBe(2);
  2185. // The `#include "utils.h"` edge should target the real
  2186. // `include/utils.h` file node — not a floating `import` node
  2187. // living inside main.cpp.
  2188. const db = DatabaseConnection.open(path.join(tempProject, '.codegraph', 'codegraph.db'));
  2189. const rows = db.getDb().prepare(`
  2190. select dst.kind as dstKind, dst.file_path as dstPath
  2191. from edges e
  2192. join nodes src on e.source = src.id
  2193. join nodes dst on e.target = dst.id
  2194. where e.kind = 'imports'
  2195. and src.kind = 'file'
  2196. and src.file_path = 'src/main.cpp'
  2197. `).all() as Array<{ dstKind: string; dstPath: string }>;
  2198. const resolvedToHeader = rows.find(
  2199. (r) => r.dstKind === 'file' && r.dstPath === 'include/utils.h'
  2200. );
  2201. expect(resolvedToHeader, 'main.cpp → include/utils.h imports edge missing').toBeDefined();
  2202. // `<vector>` should NOT produce a file edge — it's a stdlib header.
  2203. const stdlibFile = rows.find(
  2204. (r) => r.dstKind === 'file' && r.dstPath && r.dstPath.endsWith('vector')
  2205. );
  2206. expect(stdlibFile).toBeUndefined();
  2207. } finally {
  2208. fs.rmSync(tempProject, { recursive: true, force: true });
  2209. }
  2210. });
  2211. });
  2212. describe('C++ templated base-class inheritance (#1043)', () => {
  2213. // A class deriving from a TEMPLATE — `class D : public Base<int>` (or a CRTP
  2214. // `class W : public CRTPBase<W>`, or a qualified `class Q : public ns::Tpl<int>`)
  2215. // recorded its base as the full instantiation text (`Base<int>`), which never
  2216. // name-matched the template, indexed as the bare node `Base`. The `<…>` args
  2217. // are now stripped so the `extends` edge resolves end-to-end.
  2218. it('resolves an extends edge to a templated base (plain, CRTP, struct, multi-base)', async () => {
  2219. fs.writeFileSync(
  2220. path.join(tempDir, 'lib.hpp'),
  2221. `#pragma once
  2222. template<typename T> class Base { public: void foo(); };
  2223. template<typename Derived> class CRTPBase {};
  2224. class Plain {};
  2225. class Widget : public Base<int> {}; // plain template base
  2226. class App : public CRTPBase<App> {}; // CRTP (curiously-recurring)
  2227. struct Node : public Base<double> {}; // struct inheriting a template
  2228. class Both : public Base<char>, public Plain {}; // templated + plain in one clause
  2229. `
  2230. );
  2231. cg = await CodeGraph.init(tempDir, { index: true });
  2232. const db = DatabaseConnection.open(path.join(tempDir, '.codegraph', 'codegraph.db'));
  2233. const edges = db
  2234. .getDb()
  2235. .prepare(
  2236. `select src.name as fromName, dst.name as toName
  2237. from edges e
  2238. join nodes src on e.source = src.id
  2239. join nodes dst on e.target = dst.id
  2240. where e.kind = 'extends'`
  2241. )
  2242. .all() as Array<{ fromName: string; toName: string }>;
  2243. const has = (from: string, to: string) =>
  2244. edges.some((r) => r.fromName === from && r.toName === to);
  2245. // Every templated base now resolves to the bare template node.
  2246. expect(has('Widget', 'Base'), 'Widget : Base<int>').toBe(true);
  2247. expect(has('App', 'CRTPBase'), 'App : CRTPBase<App> (CRTP)').toBe(true);
  2248. expect(has('Node', 'Base'), 'struct Node : Base<double>').toBe(true);
  2249. // A mixed clause resolves BOTH the templated and the plain base.
  2250. expect(has('Both', 'Base'), 'Both : Base<char>').toBe(true);
  2251. expect(has('Both', 'Plain'), 'Both : Plain (non-templated, regression guard)').toBe(true);
  2252. });
  2253. });
  2254. describe('PHP Include Resolution', () => {
  2255. it('isPhpIncludePathRef distinguishes include paths from namespace use (#660)', () => {
  2256. const mk = (name: string, over: Partial<UnresolvedRef> = {}): UnresolvedRef => ({
  2257. fromNodeId: 'f', referenceName: name, referenceKind: 'imports',
  2258. line: 1, column: 0, filePath: 'x.php', language: 'php', ...over,
  2259. });
  2260. // include paths: contain a slash or a file extension
  2261. expect(isPhpIncludePathRef(mk('lib.php'))).toBe(true);
  2262. expect(isPhpIncludePathRef(mk('inc/db.php'))).toBe(true);
  2263. expect(isPhpIncludePathRef(mk('../config.php'))).toBe(true);
  2264. // namespace use symbols: a bare class (Closure) or FQN — never a path,
  2265. // so they must NOT be treated as includes (would mis-connect to a
  2266. // same-named Closure.php / Bar.php file).
  2267. expect(isPhpIncludePathRef(mk('Closure'))).toBe(false);
  2268. expect(isPhpIncludePathRef(mk('PDO'))).toBe(false);
  2269. expect(isPhpIncludePathRef(mk('App\\Foo\\Bar'))).toBe(false);
  2270. // scoped to PHP imports only
  2271. expect(isPhpIncludePathRef(mk('lib.php', { language: 'c' }))).toBe(false);
  2272. expect(isPhpIncludePathRef(mk('lib.php', { referenceKind: 'calls' }))).toBe(false);
  2273. });
  2274. it('resolves require_once to a file→file imports edge (#660)', async () => {
  2275. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-php-e2e-'));
  2276. try {
  2277. fs.mkdirSync(path.join(tempProject, 'src'), { recursive: true });
  2278. fs.writeFileSync(
  2279. path.join(tempProject, 'src', 'lib.php'),
  2280. `<?php\nfunction greet() { return "hi"; }\n`
  2281. );
  2282. fs.writeFileSync(
  2283. path.join(tempProject, 'src', 'page.php'),
  2284. `<?php\nrequire_once("lib.php");\necho greet();\n`
  2285. );
  2286. cg = await CodeGraph.init(tempProject, { index: true });
  2287. // reporter's repro: page.php's `require_once("lib.php")` must resolve
  2288. // to the real src/lib.php file node — a file→file `imports` edge, so
  2289. // callers(lib.php) now includes page.php.
  2290. const db = DatabaseConnection.open(path.join(tempProject, '.codegraph', 'codegraph.db'));
  2291. const rows = db.getDb().prepare(`
  2292. select dst.kind as dstKind, dst.file_path as dstPath
  2293. from edges e
  2294. join nodes src on e.source = src.id
  2295. join nodes dst on e.target = dst.id
  2296. where e.kind = 'imports'
  2297. and src.kind = 'file'
  2298. and src.file_path = 'src/page.php'
  2299. `).all() as Array<{ dstKind: string; dstPath: string }>;
  2300. const resolved = rows.find(
  2301. (r) => r.dstKind === 'file' && r.dstPath === 'src/lib.php'
  2302. );
  2303. expect(resolved, 'page.php → src/lib.php imports edge missing').toBeDefined();
  2304. } finally {
  2305. fs.rmSync(tempProject, { recursive: true, force: true });
  2306. }
  2307. });
  2308. it('resolves a subdirectory include path to the correct file (#660)', async () => {
  2309. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-php-subdir-'));
  2310. try {
  2311. fs.mkdirSync(path.join(tempProject, 'inc'), { recursive: true });
  2312. fs.writeFileSync(
  2313. path.join(tempProject, 'inc', 'db.php'),
  2314. `<?php\nfunction query() { return 1; }\n`
  2315. );
  2316. fs.writeFileSync(
  2317. path.join(tempProject, 'index.php'),
  2318. `<?php\nrequire "inc/db.php";\nquery();\n`
  2319. );
  2320. cg = await CodeGraph.init(tempProject, { index: true });
  2321. const db = DatabaseConnection.open(path.join(tempProject, '.codegraph', 'codegraph.db'));
  2322. const rows = db.getDb().prepare(`
  2323. select dst.kind as dstKind, dst.file_path as dstPath
  2324. from edges e
  2325. join nodes src on e.source = src.id
  2326. join nodes dst on e.target = dst.id
  2327. where e.kind = 'imports'
  2328. and src.kind = 'file'
  2329. and src.file_path = 'index.php'
  2330. `).all() as Array<{ dstKind: string; dstPath: string }>;
  2331. expect(
  2332. rows.find((r) => r.dstKind === 'file' && r.dstPath === 'inc/db.php'),
  2333. 'index.php → inc/db.php imports edge missing'
  2334. ).toBeDefined();
  2335. } finally {
  2336. fs.rmSync(tempProject, { recursive: true, force: true });
  2337. }
  2338. });
  2339. it('does not mis-connect an unresolvable include to a same-named file elsewhere (#660)', async () => {
  2340. const tempProject = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-php-misresolve-'));
  2341. try {
  2342. // app/page.php's `require "inc/db.php"` resolves relative to app/, where
  2343. // inc/db.php does NOT exist. A same-named lib/inc/db.php exists elsewhere
  2344. // but is unrelated — no edge should be created (a wrong edge is worse
  2345. // than a missing one).
  2346. fs.mkdirSync(path.join(tempProject, 'app'), { recursive: true });
  2347. fs.mkdirSync(path.join(tempProject, 'lib', 'inc'), { recursive: true });
  2348. fs.writeFileSync(
  2349. path.join(tempProject, 'lib', 'inc', 'db.php'),
  2350. `<?php\nfunction unrelated() {}\n`
  2351. );
  2352. fs.writeFileSync(
  2353. path.join(tempProject, 'app', 'page.php'),
  2354. `<?php\nrequire "inc/db.php";\n`
  2355. );
  2356. cg = await CodeGraph.init(tempProject, { index: true });
  2357. const db = DatabaseConnection.open(path.join(tempProject, '.codegraph', 'codegraph.db'));
  2358. const rows = db.getDb().prepare(`
  2359. select dst.kind as dstKind, dst.file_path as dstPath
  2360. from edges e
  2361. join nodes src on e.source = src.id
  2362. join nodes dst on e.target = dst.id
  2363. where e.kind = 'imports'
  2364. and src.kind = 'file'
  2365. and src.file_path = 'app/page.php'
  2366. `).all() as Array<{ dstKind: string; dstPath: string }>;
  2367. expect(
  2368. rows.find((r) => r.dstKind === 'file' && r.dstPath === 'lib/inc/db.php'),
  2369. 'app/page.php must NOT mis-connect to unrelated lib/inc/db.php'
  2370. ).toBeUndefined();
  2371. } finally {
  2372. fs.rmSync(tempProject, { recursive: true, force: true });
  2373. }
  2374. });
  2375. });
  2376. describe('C++ chained-call receiver resolution (#645)', () => {
  2377. async function indexCpp(files: Record<string, string>): Promise<void> {
  2378. for (const [name, content] of Object.entries(files)) {
  2379. fs.writeFileSync(path.join(tempDir, name), content);
  2380. }
  2381. cg = await CodeGraph.init(tempDir, { index: true });
  2382. }
  2383. function callerNamesOf(qualifiedName: string): string[] {
  2384. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2385. if (!target) return [];
  2386. const names = cg
  2387. .getIncomingEdges(target.id)
  2388. .filter((e) => e.kind === 'calls')
  2389. .map((e) => cg.getNode(e.source)?.name)
  2390. .filter((n): n is string => !!n);
  2391. return [...new Set(names)].sort();
  2392. }
  2393. it('resolves singleton chains and auto locals to the right class, never the first-sorted one', async () => {
  2394. // Two classes share writeLog; Logger sorts first so it wins any name-only
  2395. // tie. All three call forms target Metrics.
  2396. await indexCpp({
  2397. 'logger.hpp': `#pragma once
  2398. #include <string>
  2399. class Logger { public: static Logger& instance(); void writeLog(const std::string&); };
  2400. class Metrics { public: static Metrics& instance(); void writeLog(const std::string&); };
  2401. `,
  2402. 'impl.cpp': `#include "logger.hpp"
  2403. Logger& Logger::instance() { static Logger l; return l; }
  2404. Metrics& Metrics::instance() { static Metrics m; return m; }
  2405. void Logger::writeLog(const std::string&) {}
  2406. void Metrics::writeLog(const std::string&) {}
  2407. `,
  2408. 'app.cpp': `#include "logger.hpp"
  2409. void a() { Metrics::instance().writeLog("x"); } // chained singleton
  2410. void b() { auto& m = Metrics::instance(); m.writeLog("x"); } // stored in auto
  2411. void c() { Metrics& m = Metrics::instance(); m.writeLog("x"); } // explicit type
  2412. `,
  2413. });
  2414. expect(callerNamesOf('Metrics::writeLog')).toEqual(['a', 'b', 'c']);
  2415. expect(callerNamesOf('Logger::writeLog')).toEqual([]);
  2416. });
  2417. it('resolves factories, free-function factories, and member chains via the inner call return type', async () => {
  2418. await indexCpp({
  2419. 'types.hpp': `#pragma once
  2420. #include <memory>
  2421. struct Widget { void draw(); };
  2422. struct Session { void run(); };
  2423. struct View { void render(); };
  2424. class WidgetFactory { public: static Widget create(); };
  2425. class Manager { public: View view(); };
  2426. Session* openSession();
  2427. // Decoy that sorts first and has all three methods — must never win.
  2428. struct Aaa { void draw(); void run(); void render(); };
  2429. `,
  2430. 'impl.cpp': `#include "types.hpp"
  2431. void Widget::draw() {}
  2432. void Session::run() {}
  2433. void View::render() {}
  2434. void Aaa::draw() {}
  2435. void Aaa::run() {}
  2436. void Aaa::render() {}
  2437. Widget WidgetFactory::create() { return Widget(); }
  2438. View Manager::view() { return View(); }
  2439. Session* openSession() { return nullptr; }
  2440. `,
  2441. 'app.cpp': `#include "types.hpp"
  2442. void factory() { WidgetFactory::create().draw(); } // -> Widget::draw
  2443. void freefunc() { openSession()->run(); } // -> Session::run
  2444. void member() { Manager mgr; mgr.view().render(); } // -> View::render
  2445. void makeUnique() { auto w = std::make_unique<Widget>(); w->draw(); } // -> Widget::draw
  2446. `,
  2447. });
  2448. expect(callerNamesOf('Widget::draw')).toEqual(['factory', 'makeUnique']);
  2449. expect(callerNamesOf('Session::run')).toEqual(['freefunc']);
  2450. expect(callerNamesOf('View::render')).toEqual(['member']);
  2451. // The first-sorted decoy never captures any of them.
  2452. expect(callerNamesOf('Aaa::draw')).toEqual([]);
  2453. expect(callerNamesOf('Aaa::run')).toEqual([]);
  2454. expect(callerNamesOf('Aaa::render')).toEqual([]);
  2455. });
  2456. it('creates NO edge when the inferred type lacks the method (silent miss, not a wrong edge)', async () => {
  2457. await indexCpp({
  2458. 'types.hpp': `#pragma once
  2459. struct Widget { void draw(); };
  2460. struct Other { void onlyOther(); };
  2461. class WidgetFactory { public: static Widget create(); };
  2462. `,
  2463. 'impl.cpp': `#include "types.hpp"
  2464. void Widget::draw() {}
  2465. void Other::onlyOther() {}
  2466. Widget WidgetFactory::create() { return Widget(); }
  2467. `,
  2468. 'app.cpp': `#include "types.hpp"
  2469. // Widget has no onlyOther() — must produce NO edge, never a wrong one to Other.
  2470. void wrong() { WidgetFactory::create().onlyOther(); }
  2471. `,
  2472. });
  2473. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  2474. });
  2475. });
  2476. describe('PHP chained static-factory call resolution (#608)', () => {
  2477. function callerNamesOf(qualifiedName: string): string[] {
  2478. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2479. if (!target) return [];
  2480. const names = cg
  2481. .getIncomingEdges(target.id)
  2482. .filter((e) => e.kind === 'calls')
  2483. .map((e) => cg.getNode(e.source)?.name)
  2484. .filter((n): n is string => !!n);
  2485. return [...new Set(names)].sort();
  2486. }
  2487. it('resolves Cls::for($x)->method() via the factory\'s `: self` return (#608)', async () => {
  2488. fs.writeFileSync(
  2489. path.join(tempDir, 'ApiClient.php'),
  2490. `<?php\nclass ApiClient {\n public static function for(string $c): self { return new self; }\n public function createOrder(array $p): array { return []; }\n}\n`
  2491. );
  2492. fs.writeFileSync(
  2493. path.join(tempDir, 'DispatchOrder.php'),
  2494. `<?php\nclass DispatchOrder {\n public function handle(): void {\n ApiClient::for('cred')->createOrder([]);\n }\n}\n`
  2495. );
  2496. cg = await CodeGraph.init(tempDir, { index: true });
  2497. // The chained call's edge attaches to the factory result's method.
  2498. expect(callerNamesOf('ApiClient::createOrder')).toContain('handle');
  2499. });
  2500. it('creates NO edge when the factory result lacks the method (#608)', async () => {
  2501. fs.writeFileSync(
  2502. path.join(tempDir, 'lib.php'),
  2503. `<?php\nclass ApiClient { public static function for(string $c): self { return new self; } }\nclass Other { public function onlyOther(): void {} }\nclass Caller { public function go(): void { ApiClient::for('x')->onlyOther(); } }\n`
  2504. );
  2505. cg = await CodeGraph.init(tempDir, { index: true });
  2506. // ApiClient has no onlyOther — must not mis-attach to the same-named Other::onlyOther.
  2507. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  2508. });
  2509. });
  2510. describe('Java chained static-factory call resolution (#645/#608 mechanism)', () => {
  2511. function callerNamesOf(qualifiedName: string): string[] {
  2512. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2513. if (!target) return [];
  2514. const names = cg
  2515. .getIncomingEdges(target.id)
  2516. .filter((e) => e.kind === 'calls')
  2517. .map((e) => cg.getNode(e.source)?.name)
  2518. .filter((n): n is string => !!n);
  2519. return [...new Set(names)].sort();
  2520. }
  2521. it('resolves Foo.getInstance().bar() via the factory return type, never a same-named decoy', async () => {
  2522. // Aaa sorts first and has a same-named bar() — it must never win the chain.
  2523. fs.writeFileSync(
  2524. path.join(tempDir, 'Main.java'),
  2525. `class Aaa { void bar() {} }
  2526. class Foo {
  2527. static Foo getInstance() { return new Foo(); }
  2528. void bar() {}
  2529. }
  2530. class Caller {
  2531. void run() { Foo.getInstance().bar(); }
  2532. }
  2533. `
  2534. );
  2535. cg = await CodeGraph.init(tempDir, { index: true });
  2536. expect(callerNamesOf('Foo::bar')).toEqual(['run']);
  2537. expect(callerNamesOf('Aaa::bar')).toEqual([]);
  2538. });
  2539. it('resolves a factory chain that passes arguments — Foo.create(cfg).build()', async () => {
  2540. // The factory call carries an argument; the extractor must normalize the
  2541. // receiver to empty parens (`Foo.create().build`) so the chain still splits.
  2542. fs.writeFileSync(
  2543. path.join(tempDir, 'Main.java'),
  2544. `class Config {}
  2545. class Foo {
  2546. static Foo create(Config c) { return new Foo(); }
  2547. void build() {}
  2548. }
  2549. class Caller {
  2550. void run() { Foo.create(new Config()).build(); }
  2551. }
  2552. `
  2553. );
  2554. cg = await CodeGraph.init(tempDir, { index: true });
  2555. expect(callerNamesOf('Foo::build')).toEqual(['run']);
  2556. });
  2557. it('creates NO edge when the factory return type lacks the method (silent miss, not a wrong edge)', async () => {
  2558. fs.writeFileSync(
  2559. path.join(tempDir, 'Main.java'),
  2560. `class Foo {
  2561. static Foo getInstance() { return new Foo(); }
  2562. }
  2563. class Other { void onlyOther() {} }
  2564. class Caller {
  2565. void run() { Foo.getInstance().onlyOther(); }
  2566. }
  2567. `
  2568. );
  2569. cg = await CodeGraph.init(tempDir, { index: true });
  2570. // Foo has no onlyOther() — must not mis-attach to the same-named Other::onlyOther.
  2571. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  2572. });
  2573. });
  2574. describe('Kotlin chained companion-factory call resolution (#645/#608 mechanism)', () => {
  2575. function callerNamesOf(qualifiedName: string): string[] {
  2576. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2577. if (!target) return [];
  2578. const names = cg
  2579. .getIncomingEdges(target.id)
  2580. .filter((e) => e.kind === 'calls')
  2581. .map((e) => cg.getNode(e.source)?.name)
  2582. .filter((n): n is string => !!n);
  2583. return [...new Set(names)].sort();
  2584. }
  2585. it('resolves Foo.getInstance().bar() via the companion return type, never a same-named decoy', async () => {
  2586. // Aaa sorts first and has a same-named bar() — without the chain fix Kotlin
  2587. // dropped the receiver to a bare `bar` and attached to Aaa (a wrong edge).
  2588. fs.writeFileSync(
  2589. path.join(tempDir, 'Main.kt'),
  2590. `class Aaa { fun bar() {} }
  2591. class Foo {
  2592. companion object {
  2593. fun getInstance(): Foo = Foo()
  2594. }
  2595. fun bar() {}
  2596. }
  2597. class Caller {
  2598. fun run() { Foo.getInstance().bar() }
  2599. }
  2600. `
  2601. );
  2602. cg = await CodeGraph.init(tempDir, { index: true });
  2603. expect(callerNamesOf('Foo::bar')).toEqual(['run']);
  2604. expect(callerNamesOf('Aaa::bar')).toEqual([]);
  2605. });
  2606. it('resolves a companion factory chain that passes arguments — Foo.create(cfg).build()', async () => {
  2607. fs.writeFileSync(
  2608. path.join(tempDir, 'Main.kt'),
  2609. `class Config
  2610. class Foo {
  2611. companion object {
  2612. fun create(c: Config): Foo = Foo()
  2613. }
  2614. fun build() {}
  2615. }
  2616. class Caller {
  2617. fun run() { Foo.create(Config()).build() }
  2618. }
  2619. `
  2620. );
  2621. cg = await CodeGraph.init(tempDir, { index: true });
  2622. expect(callerNamesOf('Foo::build')).toEqual(['run']);
  2623. });
  2624. it('creates NO edge when the companion return type lacks the method (silent miss, not a wrong edge)', async () => {
  2625. fs.writeFileSync(
  2626. path.join(tempDir, 'Main.kt'),
  2627. `class Foo {
  2628. companion object {
  2629. fun getInstance(): Foo = Foo()
  2630. }
  2631. }
  2632. class Other { fun onlyOther() {} }
  2633. class Caller {
  2634. fun run() { Foo.getInstance().onlyOther() }
  2635. }
  2636. `
  2637. );
  2638. cg = await CodeGraph.init(tempDir, { index: true });
  2639. // Foo has no onlyOther() — must not mis-attach to the same-named Other::onlyOther.
  2640. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  2641. });
  2642. });
  2643. describe('C# chained static-factory call resolution (#645/#608 mechanism)', () => {
  2644. function callerNamesOf(qualifiedName: string): string[] {
  2645. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2646. if (!target) return [];
  2647. const names = cg
  2648. .getIncomingEdges(target.id)
  2649. .filter((e) => e.kind === 'calls')
  2650. .map((e) => cg.getNode(e.source)?.name)
  2651. .filter((n): n is string => !!n);
  2652. return [...new Set(names)].sort();
  2653. }
  2654. it('resolves Foo.Create().Bar() via the factory return type, never a same-named decoy', async () => {
  2655. // Aaa sorts first and has a same-named Bar() — it must never win the chain.
  2656. fs.writeFileSync(
  2657. path.join(tempDir, 'Main.cs'),
  2658. `class Aaa { void Bar() {} }
  2659. class Foo {
  2660. static Foo Create() { return new Foo(); }
  2661. void Bar() {}
  2662. }
  2663. class Caller {
  2664. void Run() { Foo.Create().Bar(); }
  2665. }
  2666. `
  2667. );
  2668. cg = await CodeGraph.init(tempDir, { index: true });
  2669. expect(callerNamesOf('Foo::Bar')).toEqual(['Run']);
  2670. expect(callerNamesOf('Aaa::Bar')).toEqual([]);
  2671. });
  2672. it('resolves a factory chain that passes arguments — Foo.Make(cfg).Build()', async () => {
  2673. fs.writeFileSync(
  2674. path.join(tempDir, 'Main.cs'),
  2675. `class Config {}
  2676. class Foo {
  2677. static Foo Make(Config c) { return new Foo(); }
  2678. void Build() {}
  2679. }
  2680. class Caller {
  2681. void Run() { Foo.Make(new Config()).Build(); }
  2682. }
  2683. `
  2684. );
  2685. cg = await CodeGraph.init(tempDir, { index: true });
  2686. expect(callerNamesOf('Foo::Build')).toEqual(['Run']);
  2687. });
  2688. it('creates NO edge when the factory return type lacks the method (silent miss, not a wrong edge)', async () => {
  2689. fs.writeFileSync(
  2690. path.join(tempDir, 'Main.cs'),
  2691. `class Foo {
  2692. static Foo Create() { return new Foo(); }
  2693. }
  2694. class Other { void OnlyOther() {} }
  2695. class Caller {
  2696. void Run() { Foo.Create().OnlyOther(); }
  2697. }
  2698. `
  2699. );
  2700. cg = await CodeGraph.init(tempDir, { index: true });
  2701. // Foo has no OnlyOther() — must not mis-attach to the same-named Other::OnlyOther.
  2702. expect(callerNamesOf('Other::OnlyOther')).toEqual([]);
  2703. });
  2704. });
  2705. describe('Swift chained static-factory call resolution (#645/#608 mechanism)', () => {
  2706. function callerNamesOf(qualifiedName: string): string[] {
  2707. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2708. if (!target) return [];
  2709. const names = cg
  2710. .getIncomingEdges(target.id)
  2711. .filter((e) => e.kind === 'calls')
  2712. .map((e) => cg.getNode(e.source)?.name)
  2713. .filter((n): n is string => !!n);
  2714. return [...new Set(names)].sort();
  2715. }
  2716. it('resolves Foo.make().draw() via the factory return type, never a same-named decoy', async () => {
  2717. // Aaa sorts first and has a same-named draw() — without the fix Swift dropped
  2718. // the receiver to a bare `draw` and attached to Aaa (a wrong edge).
  2719. fs.writeFileSync(
  2720. path.join(tempDir, 'Main.swift'),
  2721. `class Aaa { func draw() {} }
  2722. class Foo {
  2723. static func make() -> Foo { return Foo() }
  2724. func draw() {}
  2725. }
  2726. func runCaller() { Foo.make().draw() }
  2727. `
  2728. );
  2729. cg = await CodeGraph.init(tempDir, { index: true });
  2730. expect(callerNamesOf('Foo::draw')).toEqual(['runCaller']);
  2731. expect(callerNamesOf('Aaa::draw')).toEqual([]);
  2732. });
  2733. it('resolves a constructor chain Foo().draw() and an args factory chain Foo.build(c).render()', async () => {
  2734. fs.writeFileSync(
  2735. path.join(tempDir, 'Main.swift'),
  2736. `class Config {}
  2737. class Foo {
  2738. static func build(_ c: Config) -> Foo { return Foo() }
  2739. func draw() {}
  2740. func render() {}
  2741. }
  2742. func runCaller() {
  2743. Foo().draw()
  2744. Foo.build(Config()).render()
  2745. }
  2746. `
  2747. );
  2748. cg = await CodeGraph.init(tempDir, { index: true });
  2749. expect(callerNamesOf('Foo::draw')).toEqual(['runCaller']);
  2750. expect(callerNamesOf('Foo::render')).toEqual(['runCaller']);
  2751. });
  2752. it('creates NO edge when the factory return type lacks the method (silent miss, not a wrong edge)', async () => {
  2753. fs.writeFileSync(
  2754. path.join(tempDir, 'Main.swift'),
  2755. `class Foo {
  2756. static func make() -> Foo { return Foo() }
  2757. }
  2758. class Other { func onlyOther() {} }
  2759. func runCaller() { Foo.make().onlyOther() }
  2760. `
  2761. );
  2762. cg = await CodeGraph.init(tempDir, { index: true });
  2763. // Foo has no onlyOther() — must not mis-attach to the same-named Other::onlyOther.
  2764. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  2765. });
  2766. });
  2767. describe('Chained call resolves a method on a supertype (conformance, #750)', () => {
  2768. function callerNamesOf(qualifiedName: string): string[] {
  2769. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2770. if (!target) return [];
  2771. const names = cg
  2772. .getIncomingEdges(target.id)
  2773. .filter((e) => e.kind === 'calls')
  2774. .map((e) => cg.getNode(e.source)?.name)
  2775. .filter((n): n is string => !!n);
  2776. return [...new Set(names)].sort();
  2777. }
  2778. it('resolves a chained method defined only on a SUPERCLASS the return type extends', async () => {
  2779. // draw() lives on Base; Widget (the factory's return type) has no draw() of
  2780. // its own. Decoy.draw must never win. Needs the conformance second pass.
  2781. fs.writeFileSync(
  2782. path.join(tempDir, 'Main.java'),
  2783. `class Base { void draw() {} }
  2784. class Widget extends Base {}
  2785. class Decoy { void draw() {} }
  2786. class Factory { static Widget create() { return new Widget(); } }
  2787. class Caller {
  2788. void run() { Factory.create().draw(); }
  2789. }
  2790. `
  2791. );
  2792. cg = await CodeGraph.init(tempDir, { index: true });
  2793. expect(callerNamesOf('Base::draw')).toEqual(['run']);
  2794. expect(callerNamesOf('Decoy::draw')).toEqual([]);
  2795. });
  2796. it('resolves a chained method defined on an INTERFACE the return type implements (default method)', async () => {
  2797. fs.writeFileSync(
  2798. path.join(tempDir, 'Main.java'),
  2799. `interface Drawable { default void draw() {} }
  2800. class Widget implements Drawable {}
  2801. class Decoy { void draw() {} }
  2802. class Factory { static Widget create() { return new Widget(); } }
  2803. class Caller {
  2804. void run() { Factory.create().draw(); }
  2805. }
  2806. `
  2807. );
  2808. cg = await CodeGraph.init(tempDir, { index: true });
  2809. expect(callerNamesOf('Drawable::draw')).toEqual(['run']);
  2810. expect(callerNamesOf('Decoy::draw')).toEqual([]);
  2811. });
  2812. it('still creates NO edge when no supertype has the method (safety preserved)', async () => {
  2813. fs.writeFileSync(
  2814. path.join(tempDir, 'Main.java'),
  2815. `class Base {}
  2816. class Widget extends Base {}
  2817. class Other { void onlyOther() {} }
  2818. class Factory { static Widget create() { return new Widget(); } }
  2819. class Caller {
  2820. void run() { Factory.create().onlyOther(); }
  2821. }
  2822. `
  2823. );
  2824. cg = await CodeGraph.init(tempDir, { index: true });
  2825. // Neither Widget nor Base has onlyOther() — must not attach to Other::onlyOther.
  2826. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  2827. });
  2828. });
  2829. describe('Rust chained associated-function call resolution (#645/#608 mechanism)', () => {
  2830. function callerNamesOf(qualifiedName: string): string[] {
  2831. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2832. if (!target) return [];
  2833. const names = cg
  2834. .getIncomingEdges(target.id)
  2835. .filter((e) => e.kind === 'calls')
  2836. .map((e) => cg.getNode(e.source)?.name)
  2837. .filter((n): n is string => !!n);
  2838. return [...new Set(names)].sort();
  2839. }
  2840. it('resolves Foo::new().bar() (and a Self return) via the associated fn, never a same-named decoy', async () => {
  2841. fs.writeFileSync(
  2842. path.join(tempDir, 'main.rs'),
  2843. `struct Aaa { _x: i32 }
  2844. impl Aaa { fn bar(&self) {} }
  2845. struct Foo { _x: i32 }
  2846. impl Foo {
  2847. fn new() -> Foo { Foo { _x: 0 } }
  2848. fn make() -> Self { Foo { _x: 0 } }
  2849. fn bar(&self) {}
  2850. }
  2851. fn caller() {
  2852. Foo::new().bar();
  2853. Foo::make().bar();
  2854. }
  2855. `
  2856. );
  2857. cg = await CodeGraph.init(tempDir, { index: true });
  2858. expect(callerNamesOf('Foo::bar')).toEqual(['caller']);
  2859. expect(callerNamesOf('Aaa::bar')).toEqual([]);
  2860. });
  2861. it('resolves a chain that passes arguments — Foo::with(c).build()', async () => {
  2862. fs.writeFileSync(
  2863. path.join(tempDir, 'main.rs'),
  2864. `struct Config;
  2865. struct Foo { _x: i32 }
  2866. impl Foo {
  2867. fn with(c: Config) -> Foo { Foo { _x: 0 } }
  2868. fn build(&self) {}
  2869. }
  2870. fn caller() { Foo::with(Config).build(); }
  2871. `
  2872. );
  2873. cg = await CodeGraph.init(tempDir, { index: true });
  2874. expect(callerNamesOf('Foo::build')).toEqual(['caller']);
  2875. });
  2876. it('resolves a chained method from a trait the type implements (default method, via conformance)', async () => {
  2877. fs.writeFileSync(
  2878. path.join(tempDir, 'main.rs'),
  2879. `struct Foo { _x: i32 }
  2880. impl Foo { fn new() -> Foo { Foo { _x: 0 } } }
  2881. struct Decoy { _x: i32 }
  2882. impl Decoy { fn draw(&self) {} }
  2883. trait Drawable { fn draw(&self) {} }
  2884. impl Drawable for Foo {}
  2885. fn caller() { Foo::new().draw(); }
  2886. `
  2887. );
  2888. cg = await CodeGraph.init(tempDir, { index: true });
  2889. expect(callerNamesOf('Drawable::draw')).toEqual(['caller']);
  2890. expect(callerNamesOf('Decoy::draw')).toEqual([]);
  2891. });
  2892. it('creates NO edge when neither the type nor a supertype has the method (silent miss)', async () => {
  2893. fs.writeFileSync(
  2894. path.join(tempDir, 'main.rs'),
  2895. `struct Foo { _x: i32 }
  2896. impl Foo { fn new() -> Foo { Foo { _x: 0 } } }
  2897. struct Other { _x: i32 }
  2898. impl Other { fn only_other(&self) {} }
  2899. fn caller() { Foo::new().only_other(); }
  2900. `
  2901. );
  2902. cg = await CodeGraph.init(tempDir, { index: true });
  2903. // Foo has no only_other() — must not mis-attach to the same-named Other::only_other.
  2904. expect(callerNamesOf('Other::only_other')).toEqual([]);
  2905. });
  2906. });
  2907. describe('Go chained factory-function call resolution (#645/#608 mechanism)', () => {
  2908. function callerNamesOf(qualifiedName: string): string[] {
  2909. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  2910. if (!target) return [];
  2911. const names = cg
  2912. .getIncomingEdges(target.id)
  2913. .filter((e) => e.kind === 'calls')
  2914. .map((e) => cg.getNode(e.source)?.name)
  2915. .filter((n): n is string => !!n);
  2916. return [...new Set(names)].sort();
  2917. }
  2918. it('resolves New().Bar() via the factory return type (pointer), never a same-named decoy', async () => {
  2919. fs.writeFileSync(
  2920. path.join(tempDir, 'main.go'),
  2921. `package main
  2922. type Aaa struct{}
  2923. func (a *Aaa) Bar() {}
  2924. type Foo struct{}
  2925. func New() *Foo { return &Foo{} }
  2926. func (f *Foo) Bar() {}
  2927. func caller() { New().Bar() }
  2928. `
  2929. );
  2930. cg = await CodeGraph.init(tempDir, { index: true });
  2931. expect(callerNamesOf('Foo::Bar')).toEqual(['caller']);
  2932. expect(callerNamesOf('Aaa::Bar')).toEqual([]);
  2933. });
  2934. it('resolves an args chain and a multi-return factory — With(c).Build(), (*Foo, error)', async () => {
  2935. fs.writeFileSync(
  2936. path.join(tempDir, 'main.go'),
  2937. `package main
  2938. type Config struct{}
  2939. type Foo struct{}
  2940. func With(c Config) (*Foo, error) { return &Foo{}, nil }
  2941. func (f *Foo) Build() {}
  2942. func caller() { With(Config{}).Build() }
  2943. `
  2944. );
  2945. cg = await CodeGraph.init(tempDir, { index: true });
  2946. expect(callerNamesOf('Foo::Build')).toEqual(['caller']);
  2947. });
  2948. it('resolves a method provided by an embedded struct (via conformance)', async () => {
  2949. fs.writeFileSync(
  2950. path.join(tempDir, 'main.go'),
  2951. `package main
  2952. type Base struct{}
  2953. func (b *Base) Embedded() {}
  2954. type Decoy struct{}
  2955. func (d *Decoy) Embedded() {}
  2956. type Widget struct{ Base }
  2957. func NewWidget() *Widget { return &Widget{} }
  2958. func caller() { NewWidget().Embedded() }
  2959. `
  2960. );
  2961. cg = await CodeGraph.init(tempDir, { index: true });
  2962. expect(callerNamesOf('Base::Embedded')).toEqual(['caller']);
  2963. expect(callerNamesOf('Decoy::Embedded')).toEqual([]);
  2964. });
  2965. it('creates NO edge when neither the type nor an embedded type has the method (silent miss)', async () => {
  2966. fs.writeFileSync(
  2967. path.join(tempDir, 'main.go'),
  2968. `package main
  2969. type Foo struct{}
  2970. func New() *Foo { return &Foo{} }
  2971. type Other struct{}
  2972. func (o *Other) OnlyOther() {}
  2973. func caller() { New().OnlyOther() }
  2974. `
  2975. );
  2976. cg = await CodeGraph.init(tempDir, { index: true });
  2977. // Foo has no OnlyOther() — must not mis-attach to the same-named Other::OnlyOther.
  2978. expect(callerNamesOf('Other::OnlyOther')).toEqual([]);
  2979. });
  2980. it('falls back to bare-name resolution for a VARIABLE-inner chain without exploding the graph', async () => {
  2981. // `engine` is a package-level VARIABLE holding a func value, not a factory
  2982. // FUNCTION — so its return type can't be recovered and the chain falls back
  2983. // to bare-name resolution of the method (restoring the pre-re-encoding edge).
  2984. // Regression for the runaway this fallback originally caused: it resolved
  2985. // with a mutated `original.referenceName` (the bare `ServeHTTP`, not the
  2986. // stored `engine().ServeHTTP`), so the batched resolver's keyed delete
  2987. // no-oped, the offset-0 batch never drained, and edges inserted forever
  2988. // (5M edges / 1.4 GB on a 99-file repo). The fallback now ties the match to
  2989. // the original ref, and a non-progress guard backstops the loop.
  2990. fs.writeFileSync(
  2991. path.join(tempDir, 'main.go'),
  2992. `package main
  2993. type Server struct{}
  2994. func (s *Server) ServeHTTP() {}
  2995. var engine = func() *Server { return &Server{} }
  2996. func caller() { engine().ServeHTTP() }
  2997. `
  2998. );
  2999. cg = await CodeGraph.init(tempDir, { index: true });
  3000. // Recall: the variable-inner chain still finds the method by bare name.
  3001. expect(callerNamesOf('Server::ServeHTTP')).toEqual(['caller']);
  3002. // No runaway: a single call site yields a single edge, not millions.
  3003. const target = cg
  3004. .getNodesByKind('method')
  3005. .find((n) => n.qualifiedName === 'Server::ServeHTTP')!;
  3006. const rawCalls = cg
  3007. .getIncomingEdges(target.id)
  3008. .filter((e) => e.kind === 'calls');
  3009. expect(rawCalls.length).toBeLessThan(5);
  3010. });
  3011. });
  3012. describe('Scala chained static-factory call resolution (#645/#608 mechanism)', () => {
  3013. function callerNamesOf(qualifiedName: string): string[] {
  3014. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  3015. if (!target) return [];
  3016. const names = cg
  3017. .getIncomingEdges(target.id)
  3018. .filter((e) => e.kind === 'calls')
  3019. .map((e) => cg.getNode(e.source)?.name)
  3020. .filter((n): n is string => !!n);
  3021. return [...new Set(names)].sort();
  3022. }
  3023. it('resolves a companion-factory chain Foo.create().doIt() to the return type, never a same-named decoy', async () => {
  3024. fs.writeFileSync(
  3025. path.join(tempDir, 'Main.scala'),
  3026. `object Foo {
  3027. def create(): Bar = new Bar()
  3028. }
  3029. class Bar {
  3030. def doIt(): Unit = {}
  3031. }
  3032. class Decoy {
  3033. def doIt(): Unit = {}
  3034. }
  3035. object Main {
  3036. def run(): Unit = { Foo.create().doIt() }
  3037. }
  3038. `
  3039. );
  3040. cg = await CodeGraph.init(tempDir, { index: true });
  3041. expect(callerNamesOf('Bar::doIt')).toEqual(['run']);
  3042. expect(callerNamesOf('Decoy::doIt')).toEqual([]);
  3043. });
  3044. it('resolves a case-class apply construction Point(x).dist() on the constructed class', async () => {
  3045. fs.writeFileSync(
  3046. path.join(tempDir, 'Main.scala'),
  3047. `class Point(x: Int) {
  3048. def dist(): Int = x
  3049. }
  3050. class Other {
  3051. def dist(): Int = 0
  3052. }
  3053. object Main {
  3054. def run(): Unit = { Point(3).dist() }
  3055. }
  3056. `
  3057. );
  3058. cg = await CodeGraph.init(tempDir, { index: true });
  3059. expect(callerNamesOf('Point::dist')).toEqual(['run']);
  3060. expect(callerNamesOf('Other::dist')).toEqual([]);
  3061. });
  3062. it('resolves a chained method provided by a trait the return type extends (via conformance)', async () => {
  3063. fs.writeFileSync(
  3064. path.join(tempDir, 'Main.scala'),
  3065. `trait Base {
  3066. def shared(): Unit = {}
  3067. }
  3068. class Widget extends Base
  3069. class Decoy {
  3070. def shared(): Unit = {}
  3071. }
  3072. object Factory {
  3073. def make(): Widget = new Widget()
  3074. }
  3075. object Main {
  3076. def run(): Unit = { Factory.make().shared() }
  3077. }
  3078. `
  3079. );
  3080. cg = await CodeGraph.init(tempDir, { index: true });
  3081. expect(callerNamesOf('Base::shared')).toEqual(['run']);
  3082. expect(callerNamesOf('Decoy::shared')).toEqual([]);
  3083. });
  3084. it('creates NO edge when neither the factory return type nor a supertype has the method (silent miss)', async () => {
  3085. fs.writeFileSync(
  3086. path.join(tempDir, 'Main.scala'),
  3087. `object Foo {
  3088. def create(): Bar = new Bar()
  3089. }
  3090. class Bar {
  3091. }
  3092. class Other {
  3093. def onlyOther(): Unit = {}
  3094. }
  3095. object Main {
  3096. def run(): Unit = { Foo.create().onlyOther() }
  3097. }
  3098. `
  3099. );
  3100. cg = await CodeGraph.init(tempDir, { index: true });
  3101. // Bar has no onlyOther() — must not mis-attach to the same-named Other::onlyOther.
  3102. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  3103. });
  3104. });
  3105. describe('Dart chained static-factory / factory-constructor call resolution (#645/#608 mechanism)', () => {
  3106. function callerNamesOf(qualifiedName: string): string[] {
  3107. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  3108. if (!target) return [];
  3109. const names = cg
  3110. .getIncomingEdges(target.id)
  3111. .filter((e) => e.kind === 'calls')
  3112. .map((e) => cg.getNode(e.source)?.name)
  3113. .filter((n): n is string => !!n);
  3114. return [...new Set(names)].sort();
  3115. }
  3116. it('resolves a static-factory chain Foo.makeBar().doIt() to the return type, never a same-named decoy', async () => {
  3117. fs.writeFileSync(
  3118. path.join(tempDir, 'main.dart'),
  3119. `class Foo {
  3120. static Bar makeBar() => Bar();
  3121. }
  3122. class Bar {
  3123. void doIt() {}
  3124. }
  3125. class Decoy {
  3126. void doIt() {}
  3127. }
  3128. void run() {
  3129. Foo.makeBar().doIt();
  3130. }
  3131. `
  3132. );
  3133. cg = await CodeGraph.init(tempDir, { index: true });
  3134. expect(callerNamesOf('Bar::doIt')).toEqual(['run']);
  3135. expect(callerNamesOf('Decoy::doIt')).toEqual([]);
  3136. });
  3137. it('resolves a named factory-constructor chain Foo.create().ship() on the constructed class', async () => {
  3138. fs.writeFileSync(
  3139. path.join(tempDir, 'main.dart'),
  3140. `class Foo {
  3141. Foo._();
  3142. factory Foo.create() => Foo._();
  3143. void ship() {}
  3144. }
  3145. class Decoy {
  3146. void ship() {}
  3147. }
  3148. void run() {
  3149. Foo.create().ship();
  3150. }
  3151. `
  3152. );
  3153. cg = await CodeGraph.init(tempDir, { index: true });
  3154. // The factory constructor `Foo.create` is now a node whose return type is Foo,
  3155. // so `ship` resolves on Foo, not the same-named Decoy.
  3156. expect(callerNamesOf('Foo::ship')).toEqual(['run']);
  3157. expect(callerNamesOf('Decoy::ship')).toEqual([]);
  3158. });
  3159. it('resolves a constructor-receiver chain Bar().doIt() on the constructed class', async () => {
  3160. fs.writeFileSync(
  3161. path.join(tempDir, 'main.dart'),
  3162. `class Bar {
  3163. void doIt() {}
  3164. }
  3165. class Decoy {
  3166. void doIt() {}
  3167. }
  3168. void run() {
  3169. Bar().doIt();
  3170. }
  3171. `
  3172. );
  3173. cg = await CodeGraph.init(tempDir, { index: true });
  3174. expect(callerNamesOf('Bar::doIt')).toEqual(['run']);
  3175. expect(callerNamesOf('Decoy::doIt')).toEqual([]);
  3176. });
  3177. it('resolves a chained method inherited from a superclass the return type extends (via conformance)', async () => {
  3178. fs.writeFileSync(
  3179. path.join(tempDir, 'main.dart'),
  3180. `class Base {
  3181. void render() {}
  3182. }
  3183. class Widget extends Base {
  3184. static Widget make() => Widget();
  3185. }
  3186. class Decoy {
  3187. void render() {}
  3188. }
  3189. void run() {
  3190. Widget.make().render();
  3191. }
  3192. `
  3193. );
  3194. cg = await CodeGraph.init(tempDir, { index: true });
  3195. expect(callerNamesOf('Base::render')).toEqual(['run']);
  3196. expect(callerNamesOf('Decoy::render')).toEqual([]);
  3197. });
  3198. it('creates NO edge when neither the factory return type nor a supertype has the method (silent miss)', async () => {
  3199. fs.writeFileSync(
  3200. path.join(tempDir, 'main.dart'),
  3201. `class Foo {
  3202. static Bar makeBar() => Bar();
  3203. }
  3204. class Bar {
  3205. }
  3206. class Other {
  3207. void onlyOther() {}
  3208. }
  3209. void run() {
  3210. Foo.makeBar().onlyOther();
  3211. }
  3212. `
  3213. );
  3214. cg = await CodeGraph.init(tempDir, { index: true });
  3215. // Bar has no onlyOther() — must not mis-attach to the same-named Other::onlyOther.
  3216. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  3217. });
  3218. it('still extracts a method tree-sitter misparses as a constructor (@override + record return)', async () => {
  3219. // tree-sitter-dart misparses `@override (A, B) reduce()` — the annotation
  3220. // swallows the record return type, so `reduce()` looks like a single-
  3221. // identifier constructor_signature. It must NOT be skipped as an unnamed
  3222. // ctor (its name doesn't match the class); its body call must attribute to
  3223. // `reduce`, not the class.
  3224. fs.writeFileSync(
  3225. path.join(tempDir, 'main.dart'),
  3226. `class Base {}
  3227. class Action extends Base {
  3228. Action({required int x});
  3229. @override
  3230. (int, String) reduce() {
  3231. return (compute(), "y");
  3232. }
  3233. int compute() => 1;
  3234. }
  3235. `
  3236. );
  3237. cg = await CodeGraph.init(tempDir, { index: true });
  3238. // reduce must be a node and its body call must resolve to Action::compute.
  3239. expect(callerNamesOf('Action::compute')).toEqual(['reduce']);
  3240. });
  3241. it('keeps plain construction Foo() as instantiation, not a Foo::Foo method call', async () => {
  3242. // The unnamed constructor is intentionally NOT extracted as a `Foo::Foo`
  3243. // method, so `Foo(...)` resolves to the class (an `instantiates` edge),
  3244. // never hijacked into a call to a phantom constructor method.
  3245. fs.writeFileSync(
  3246. path.join(tempDir, 'main.dart'),
  3247. `class Widget {
  3248. final int x;
  3249. Widget(this.x);
  3250. }
  3251. void run() {
  3252. Widget(3);
  3253. }
  3254. `
  3255. );
  3256. cg = await CodeGraph.init(tempDir, { index: true });
  3257. // No Foo::Foo phantom method node.
  3258. expect(cg.getNodesByKind('method').some((n) => n.qualifiedName === 'Widget::Widget')).toBe(false);
  3259. // The construction resolves to the class as an `instantiates` edge.
  3260. const widget = cg.getNodesByKind('class').find((n) => n.name === 'Widget')!;
  3261. const incoming = cg.getIncomingEdges(widget.id);
  3262. expect(incoming.some((e) => e.kind === 'instantiates')).toBe(true);
  3263. });
  3264. });
  3265. describe('Objective-C chained message-send call resolution (#645/#608 mechanism)', () => {
  3266. function callerNamesOf(qualifiedName: string): string[] {
  3267. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  3268. if (!target) return [];
  3269. const names = cg
  3270. .getIncomingEdges(target.id)
  3271. .filter((e) => e.kind === 'calls')
  3272. .map((e) => cg.getNode(e.source)?.name)
  3273. .filter((n): n is string => !!n);
  3274. return [...new Set(names)].sort();
  3275. }
  3276. it('resolves a chained message send [[Foo create] doIt] via the return type, never a same-named decoy', async () => {
  3277. fs.writeFileSync(
  3278. path.join(tempDir, 'main.m'),
  3279. `@interface Bar : NSObject
  3280. - (void)doIt;
  3281. @end
  3282. @implementation Bar
  3283. - (void)doIt {}
  3284. @end
  3285. @interface Decoy : NSObject
  3286. - (void)doIt;
  3287. @end
  3288. @implementation Decoy
  3289. - (void)doIt {}
  3290. @end
  3291. @interface Foo : NSObject
  3292. + (Bar *)create;
  3293. @end
  3294. @implementation Foo
  3295. + (Bar *)create { return nil; }
  3296. - (void)run { [[Foo create] doIt]; }
  3297. @end
  3298. `
  3299. );
  3300. cg = await CodeGraph.init(tempDir, { index: true });
  3301. expect(callerNamesOf('Bar::doIt')).toEqual(['run']);
  3302. expect(callerNamesOf('Decoy::doIt')).toEqual([]);
  3303. });
  3304. it('resolves a chained message whose method is inherited from a superclass (via conformance)', async () => {
  3305. fs.writeFileSync(
  3306. path.join(tempDir, 'main.m'),
  3307. `@interface Base : NSObject
  3308. - (void)render;
  3309. @end
  3310. @implementation Base
  3311. - (void)render {}
  3312. @end
  3313. @interface Widget : Base
  3314. @end
  3315. @implementation Widget
  3316. @end
  3317. @interface Decoy : NSObject
  3318. - (void)render;
  3319. @end
  3320. @implementation Decoy
  3321. - (void)render {}
  3322. @end
  3323. @interface Factory : NSObject
  3324. + (Widget *)make;
  3325. @end
  3326. @implementation Factory
  3327. + (Widget *)make { return nil; }
  3328. - (void)run { [[Factory make] render]; }
  3329. @end
  3330. `
  3331. );
  3332. cg = await CodeGraph.init(tempDir, { index: true });
  3333. expect(callerNamesOf('Base::render')).toEqual(['run']);
  3334. expect(callerNamesOf('Decoy::render')).toEqual([]);
  3335. });
  3336. it('creates NO edge when the factory return type lacks the method (silent miss)', async () => {
  3337. fs.writeFileSync(
  3338. path.join(tempDir, 'main.m'),
  3339. `@interface Bar : NSObject
  3340. @end
  3341. @implementation Bar
  3342. @end
  3343. @interface Other : NSObject
  3344. - (void)onlyOther;
  3345. @end
  3346. @implementation Other
  3347. - (void)onlyOther {}
  3348. @end
  3349. @interface Foo : NSObject
  3350. + (Bar *)create;
  3351. @end
  3352. @implementation Foo
  3353. + (Bar *)create { return nil; }
  3354. - (void)run { [[Foo create] onlyOther]; }
  3355. @end
  3356. `
  3357. );
  3358. cg = await CodeGraph.init(tempDir, { index: true });
  3359. // Bar has no onlyOther — must not mis-attach to the same-named Other::onlyOther.
  3360. expect(callerNamesOf('Other::onlyOther')).toEqual([]);
  3361. });
  3362. it('resolves a singleton chain [[Cache shared] clearAll] whose factory returns nonnull instancetype', async () => {
  3363. // The factory returns `nonnull instancetype` — the nullability qualifier must
  3364. // be skipped (not captured AS the type), and an instancetype class-message
  3365. // factory returns the receiver class, so clearAll resolves on Cache, never a
  3366. // same-named decoy. (Regression for both: the captured-`nonnull` bug and the
  3367. // ubiquitous `[[X alloc] init]` / singleton pattern.)
  3368. fs.writeFileSync(
  3369. path.join(tempDir, 'main.m'),
  3370. `@interface Cache : NSObject
  3371. + (nonnull instancetype)shared;
  3372. - (void)clearAll;
  3373. @end
  3374. @implementation Cache
  3375. + (nonnull instancetype)shared { return nil; }
  3376. - (void)clearAll {}
  3377. @end
  3378. @interface Decoy : NSObject
  3379. - (void)clearAll;
  3380. @end
  3381. @implementation Decoy
  3382. - (void)clearAll {}
  3383. @end
  3384. @interface Caller : NSObject
  3385. - (void)run;
  3386. @end
  3387. @implementation Caller
  3388. - (void)run { [[Cache shared] clearAll]; }
  3389. @end
  3390. `
  3391. );
  3392. cg = await CodeGraph.init(tempDir, { index: true });
  3393. expect(callerNamesOf('Cache::clearAll')).toEqual(['run']);
  3394. expect(callerNamesOf('Decoy::clearAll')).toEqual([]);
  3395. });
  3396. });
  3397. describe('Pascal/Delphi chained static-factory call resolution (#645/#608 mechanism)', () => {
  3398. function callerNamesOf(qualifiedName: string): string[] {
  3399. const target = cg.getNodesByKind('method').find((n) => n.qualifiedName === qualifiedName);
  3400. if (!target) return [];
  3401. const names = cg
  3402. .getIncomingEdges(target.id)
  3403. .filter((e) => e.kind === 'calls')
  3404. .map((e) => cg.getNode(e.source)?.name)
  3405. .filter((n): n is string => !!n);
  3406. return [...new Set(names)].sort();
  3407. }
  3408. function isCalled(qn: string): boolean {
  3409. const t = cg.getNodesByKind('method').find((n) => n.qualifiedName === qn);
  3410. return !!t && cg.getIncomingEdges(t.id).some((e) => e.kind === 'calls');
  3411. }
  3412. it('resolves a chained factory call TFoo.GetInstance().DoIt() via the return type, never a same-named decoy', async () => {
  3413. fs.writeFileSync(
  3414. path.join(tempDir, 'main.pas'),
  3415. `unit Main;
  3416. interface
  3417. type
  3418. TBar = class
  3419. procedure DoIt;
  3420. end;
  3421. TDecoy = class
  3422. procedure DoIt;
  3423. end;
  3424. TFoo = class
  3425. class function GetInstance: TBar;
  3426. end;
  3427. implementation
  3428. procedure TBar.DoIt; begin end;
  3429. procedure TDecoy.DoIt; begin end;
  3430. class function TFoo.GetInstance: TBar; begin Result := nil; end;
  3431. procedure Run;
  3432. begin
  3433. TFoo.GetInstance().DoIt();
  3434. end;
  3435. end.
  3436. `
  3437. );
  3438. cg = await CodeGraph.init(tempDir, { index: true });
  3439. expect(isCalled('TBar::DoIt')).toBe(true);
  3440. expect(isCalled('TDecoy::DoIt')).toBe(false);
  3441. });
  3442. it('resolves a constructor chain TFoo.Create().Configure() on the constructed class', async () => {
  3443. fs.writeFileSync(
  3444. path.join(tempDir, 'main.pas'),
  3445. `unit Main;
  3446. interface
  3447. type
  3448. TFoo = class
  3449. constructor Create;
  3450. procedure Configure;
  3451. end;
  3452. TDecoy = class
  3453. procedure Configure;
  3454. end;
  3455. implementation
  3456. constructor TFoo.Create; begin end;
  3457. procedure TFoo.Configure; begin end;
  3458. procedure TDecoy.Configure; begin end;
  3459. procedure Run;
  3460. begin
  3461. TFoo.Create().Configure();
  3462. end;
  3463. end.
  3464. `
  3465. );
  3466. cg = await CodeGraph.init(tempDir, { index: true });
  3467. // A constructor returns its own class (no `: TBar` annotation), so Configure
  3468. // resolves on TFoo, not the same-named decoy.
  3469. expect(isCalled('TFoo::Configure')).toBe(true);
  3470. expect(isCalled('TDecoy::Configure')).toBe(false);
  3471. });
  3472. it('resolves a typecast chain TFoo(x).DoIt() on the cast type', async () => {
  3473. fs.writeFileSync(
  3474. path.join(tempDir, 'main.pas'),
  3475. `unit Main;
  3476. interface
  3477. type
  3478. TFoo = class
  3479. procedure DoIt;
  3480. end;
  3481. TDecoy = class
  3482. procedure DoIt;
  3483. end;
  3484. implementation
  3485. procedure TFoo.DoIt; begin end;
  3486. procedure TDecoy.DoIt; begin end;
  3487. procedure Run(obj: TObject);
  3488. begin
  3489. TFoo(obj).DoIt();
  3490. end;
  3491. end.
  3492. `
  3493. );
  3494. cg = await CodeGraph.init(tempDir, { index: true });
  3495. expect(isCalled('TFoo::DoIt')).toBe(true);
  3496. expect(isCalled('TDecoy::DoIt')).toBe(false);
  3497. });
  3498. it('creates NO edge when the factory return type lacks the method (silent miss)', async () => {
  3499. fs.writeFileSync(
  3500. path.join(tempDir, 'main.pas'),
  3501. `unit Main;
  3502. interface
  3503. type
  3504. TBar = class
  3505. end;
  3506. TOther = class
  3507. procedure OnlyOther;
  3508. end;
  3509. TFoo = class
  3510. class function GetInstance: TBar;
  3511. end;
  3512. implementation
  3513. procedure TOther.OnlyOther; begin end;
  3514. class function TFoo.GetInstance: TBar; begin Result := nil; end;
  3515. procedure Run;
  3516. begin
  3517. TFoo.GetInstance().OnlyOther();
  3518. end;
  3519. end.
  3520. `
  3521. );
  3522. cg = await CodeGraph.init(tempDir, { index: true });
  3523. // TBar has no OnlyOther — must not mis-attach to the same-named TOther::OnlyOther.
  3524. expect(isCalled('TOther::OnlyOther')).toBe(false);
  3525. });
  3526. it('extracts paren-less method calls (Pascal lets a no-arg method drop its parens)', async () => {
  3527. fs.writeFileSync(
  3528. path.join(tempDir, 'main.pas'),
  3529. `unit Main;
  3530. interface
  3531. type
  3532. TFoo = class
  3533. procedure DoThing;
  3534. procedure Reset;
  3535. end;
  3536. implementation
  3537. procedure TFoo.DoThing; begin end;
  3538. procedure TFoo.Reset; begin end;
  3539. procedure Run(f: TFoo);
  3540. begin
  3541. f.DoThing;
  3542. f.Reset;
  3543. end;
  3544. end.
  3545. `
  3546. );
  3547. cg = await CodeGraph.init(tempDir, { index: true });
  3548. expect(isCalled('TFoo::DoThing')).toBe(true);
  3549. expect(isCalled('TFoo::Reset')).toBe(true);
  3550. });
  3551. it('resolves a PAREN-LESS chained factory call TFoo.GetInstance.DoIt via the return type', async () => {
  3552. fs.writeFileSync(
  3553. path.join(tempDir, 'main.pas'),
  3554. `unit Main;
  3555. interface
  3556. type
  3557. TBar = class
  3558. procedure DoIt;
  3559. end;
  3560. TDecoy = class
  3561. procedure DoIt;
  3562. end;
  3563. TFoo = class
  3564. class function GetInstance: TBar;
  3565. end;
  3566. implementation
  3567. procedure TBar.DoIt; begin end;
  3568. procedure TDecoy.DoIt; begin end;
  3569. class function TFoo.GetInstance: TBar; begin Result := nil; end;
  3570. procedure Run;
  3571. begin
  3572. TFoo.GetInstance.DoIt;
  3573. end;
  3574. end.
  3575. `
  3576. );
  3577. cg = await CodeGraph.init(tempDir, { index: true });
  3578. expect(isCalled('TBar::DoIt')).toBe(true);
  3579. expect(isCalled('TDecoy::DoIt')).toBe(false);
  3580. });
  3581. it('does NOT turn a property write/read into a call edge (only statement-level dots are calls)', async () => {
  3582. fs.writeFileSync(
  3583. path.join(tempDir, 'main.pas'),
  3584. `unit Main;
  3585. interface
  3586. type
  3587. TFoo = class
  3588. function GetValue: Integer;
  3589. procedure SetValue(v: Integer);
  3590. property Value: Integer read GetValue write SetValue;
  3591. end;
  3592. implementation
  3593. function TFoo.GetValue: Integer; begin Result := 0; end;
  3594. procedure TFoo.SetValue(v: Integer); begin end;
  3595. procedure Run(f: TFoo);
  3596. var x: Integer;
  3597. begin
  3598. f.Value := 5;
  3599. x := f.Value;
  3600. end;
  3601. end.
  3602. `
  3603. );
  3604. cg = await CodeGraph.init(tempDir, { index: true });
  3605. // A property read/write is a bare dot in assignment position, not a statement,
  3606. // so it must not be mis-extracted as a call to the property's getter/setter.
  3607. expect(isCalled('TFoo::GetValue')).toBe(false);
  3608. expect(isCalled('TFoo::SetValue')).toBe(false);
  3609. });
  3610. it('attributes an implementation-only free procedure\'s calls to the procedure, not the file', async () => {
  3611. fs.writeFileSync(
  3612. path.join(tempDir, 'main.pas'),
  3613. `unit Main;
  3614. interface
  3615. type
  3616. TTgt = class
  3617. procedure Hit;
  3618. end;
  3619. TFoo = class
  3620. procedure DoStuff;
  3621. end;
  3622. implementation
  3623. procedure TTgt.Hit; begin end;
  3624. procedure TFoo.DoStuff; var t: TTgt; begin t.Hit; end;
  3625. procedure Helper; var t: TTgt; begin t.Hit; end;
  3626. `
  3627. );
  3628. cg = await CodeGraph.init(tempDir, { index: true });
  3629. // `Helper` is implementation-only (no interface decl, not a method), but its
  3630. // body's call must attribute to `Helper`, not the file/module — alongside the
  3631. // method `DoStuff`.
  3632. expect(callerNamesOf('TTgt::Hit')).toEqual(['DoStuff', 'Helper']);
  3633. });
  3634. });
  3635. });