Add infrastructure to avoid hacky comment/uncomment testing code

Author: olafurpg

src/CommandLineOptions.ts

@@ -17,6 +17,7 @@ export interface MultiProjectOptions {
   cwd: string
   output: string
   indexedProjects: Set<string>
+  shouldIndexFile?: (filename: string) => boolean
 }
 
 /** Configuration options to index a single TypeScript project. */

src/FileIndexer.ts

@@ -39,10 +39,9 @@ export class FileIndexer {
     this.workingDirectoryRegExp = new RegExp(options.cwd, 'g')
   }
   public index(): void {
-    // Uncomment below if you want to skip certain files for local development.
-    // if (!this.sourceFile.fileName.includes('infer-relation')) {
-    //   return
-    // }
+    if (this.options.shouldIndexFile?.(this.sourceFile.fileName) === false) {
+      return
+    }
     this.emitSourceFileOccurrence()
     this.visit(this.sourceFile)
   }
@@ -172,6 +171,14 @@ export class FileIndexer {
       : sym?.declarations || []
     for (const declaration of declarations) {
       let scipSymbol = this.scipSymbol(declaration)
+      if (scipSymbol.value === 'local 4') {
+        console.log({
+          node: node.getText(),
+          kind: ts.SyntaxKind[node.kind],
+          kind2: ts.SyntaxKind[node.parent.kind],
+          kind3: ts.SyntaxKind[node.parent.parent.kind],
+        })
+      }
 
       if (
         ((ts.isIdentifier(node) && ts.isNewExpression(node.parent)) ||
@@ -399,6 +406,25 @@ export class FileIndexer {
       }
       return this.cached(node, package_)
     }
+
+    if (
+      node?.parent?.parent &&
+      ts.isIdentifier(node) &&
+      ts.isMethodDeclaration(node.parent) &&
+      node.parent.name === node &&
+      ts.isObjectLiteralExpression(node.parent.parent)
+    ) {
+      const tpe = this.assignedType.get(node.parent.parent)
+      const property = tpe?.getProperty(node.getText())
+      console.log({
+        node: node.getText(),
+        tpe: tpe && this.checker.typeToString(tpe),
+      })
+      if (property?.declarations && property.declarations.length > 0) {
+        return this.cached(node, this.scipSymbol(property.declarations[0]))
+      }
+    }
+
     if (
       ts.isPropertyAssignment(node) ||
       ts.isShorthandPropertyAssignment(node)

src/main.test.ts

@@ -15,6 +15,9 @@ function isUpdateSnapshot(): boolean {
   return process.argv.includes('--update-snapshots')
 }
 
+const onlyFile = process.env['ONLY']
+const onlyAbsoluteFilename = onlyFile ? join(process.cwd(), onlyFile) : ''
+
 const snapshotNodeModules = join(process.cwd(), 'snapshots', 'node_modules')
 if (!fs.existsSync(snapshotNodeModules)) {
   throw new Error(
@@ -34,16 +37,14 @@ interface PackageJson {
   packageManager?: string
 }
 for (const snapshotDirectory of snapshotDirectories) {
-  // Uncomment below if you want to skip certain tests for local development.
-  // if (!snapshotDirectory.includes('syntax')) {
-  //   continue
-  // }
   const inputRoot = join(inputDirectory, snapshotDirectory)
   const outputRoot = join(outputDirectory, snapshotDirectory)
   if (!fs.statSync(inputRoot).isDirectory()) {
     continue
   }
-  test(snapshotDirectory, () => {
+  const testFunction =
+    onlyFile && !onlyAbsoluteFilename.startsWith(inputRoot) ? test.skip : test
+  testFunction(snapshotDirectory, () => {
     const packageJsonPath = path.join(inputRoot, 'package.json')
     const packageJson = JSON.parse(
       fs.readFileSync(packageJsonPath).toString()
@@ -61,6 +62,12 @@ for (const snapshotDirectory of snapshotDirectories) {
       progressBar: false,
       indexedProjects: new Set(),
       globalCaches: true,
+      shouldIndexFile: filename => {
+        if (onlyFile) {
+          return filename === onlyAbsoluteFilename
+        }
+        return true
+      },
     })
     if (inferTsconfig) {
       fs.rmSync(tsconfigJsonPath)