Simplify interceptor code and rename variables for clarity.

This commit is contained in:
Sander Declerck 2025-11-12 14:03:33 +01:00
parent 2cf23d5109
commit ad6d9bcdd5
No known key found for this signature in database
4 changed files with 69 additions and 96 deletions

View file

@ -1,5 +1,5 @@
import { isMalwarePackage } from "../../scanning/audit/index.js";
import { createInterceptorBuilder } from "./interceptorBuilder.js";
import { interceptRequests } from "./interceptorBuilder.js";
const knownPipRegistries = [
"files.pythonhosted.org",
@ -27,19 +27,15 @@ export function pipInterceptorForUrl(url) {
* @returns {import("./interceptorBuilder.js").Interceptor | undefined}
*/
function buildPipInterceptor(registry) {
const builder = createInterceptorBuilder();
builder.onRequest(async (req) => {
return interceptRequests(async (reqContext) => {
const { packageName, version } = parsePipPackageFromUrl(
req.targetUrl,
reqContext.targetUrl,
registry
);
if (await isMalwarePackage(packageName, version)) {
req.blockMalware(packageName, version, req.targetUrl);
reqContext.blockMalware(packageName, version);
}
});
return builder.build();
}
/**