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)
   }

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)