Limit number of paths for the view

Author: marcnjaramillo

extensions/ql-vscode/src/cli.ts

@@ -679,42 +679,48 @@ export class CodeQLCliServer implements Disposable {
     const fakeSarifPath = '/Users/marcjaramillo/MLH/jdk11-tainted-path.sarif';
     try {
       const p = parser();
-      console.log('Created parser');
+      //TODO: see if there is a way to create a parser that will return the count of codeFlow
+      // paths first and then parse the SARIF based on whether it exceeds a certain threshhold.
+      // Also, see if it's possible to parse only the first 5 or so codeFlows right up front
+      // rather than parsing everything and then getting rid of the paths we don't want.
       const pipeline = chain([
         fs.createReadStream(fakeSarifPath),
         p,
         pick({filter: 'runs.0.results'})
       ]);
-      console.log('Created pipeline');
+
       const asm = Assembler.connectTo(pipeline);
-      console.log('Created assembler');
-      let chunkCount = 0;
       //TODO: figure out error handling for Assembler
       return await new Promise((resolve) => {
-        p.on('data', (_) => {
-          chunkCount++;
-          if(chunkCount % 10000000 === 0) {
-            console.log('Chunks processed:', chunkCount);
-          }
-        });
-        console.log('Created parser counter');
-        
         asm.on('done', (asm) => {
           const dummyTool : sarif.Tool = {driver: {name: ''}};
+          // Counts number of paths for each result and totals them
+          // so we can let the user know how many paths exist even
+          // when there are too many to display
+          const pathsTotal = asm.current.map((result: sarif.Result) => (
+            result.codeFlows!.length
+          )) as number[];
+
+          const sum = pathsTotal.reduce((prev: number, curr: number) => prev + curr, 0);
+
+          const filteredResults = asm.current.map((result: sarif.Result) => (
+            Object.assign({}, result, {codeFlows: result.codeFlows!.slice(0,5)})
+          ));
+
           const log : sarif.Log = {
             version: '2.1.0', 
             runs: [
               { 
                 tool: dummyTool, 
-                results: asm.current 
+                results: filteredResults
               }
             ]
           };
-          console.log('Number of results:', log.runs[0]!.results?.length);
-          console.log(log.runs[0]!.results![0]);
+          console.log('Number of results: ', log.runs[0]!.results?.length);
+          console.log('Total paths: ', sum);
+          console.log(log);
           resolve(log);
         });
-        console.log('Created assembler done listener');
       });
     } catch (err) {
       throw new Error(`Parsing output of interpretation failed: ${err.stderr || err}`);