Better repo handling for --repo

Author: aegilops

src/cli.ts

@@ -58,6 +58,7 @@ async function main() {
         if (!args.enterprise && !args.org && !args.repo) throw new Error("Provide --enterprise, --org or --repo with --sync-sboms");
         if (args.enterprise && args.org) throw new Error("Specify only one of --enterprise or --org");
         if (args.repo && (args.enterprise || args.org)) throw new Error("Specify only one of --enterprise, --org, or --repo");
+        if (args.repo && !(args.repo as string).includes("/")) throw new Error("--repo must be in the format owner/repo");
         if (syncing && !args.sbomCache) throw new Error("--sync-sboms requires --sbom-cache to write updated SBOMs to disk");
       } else {
         const malwareOnly = !!args["sync-malware"] && !args.sbomCache && !args.purl && !args["purl-file"] && !args["match-malware"] && !args.uploadSarif && !args.interactive;

src/sbomCollector.ts

@@ -349,7 +349,11 @@ export class SbomCollector {
         renderBar();
       }));
       await Promise.all(tasks);
-      newSboms = newSboms.filter(s => repoNames.has(s.repo) || repoNames.has(s.repo.split("/")[1]));
+      
+      newSboms = newSboms.filter(s => {
+        const repoToCheck = s.repo.includes("/") ? s.repo.split("/")[1] : s.repo;
+        return repoNames.has(repoToCheck);
+      });
       this.sboms.push(...newSboms);
     }
     if (this.opts.showProgressBar) process.stdout.write("\n");