From e27b8310f674e3c57b8bb550e49093f8c69c106f Mon Sep 17 00:00:00 2001 From: Skyler Grey Date: Wed, 8 Mar 2023 19:42:03 +0000 Subject: [PATCH] Fix issues with images not being scanned --- src/reflex/scanners.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/reflex/scanners.ts b/src/reflex/scanners.ts index 23337ac..cce8b84 100644 --- a/src/reflex/scanners.ts +++ b/src/reflex/scanners.ts @@ -32,7 +32,7 @@ const clamscanner = await new ClamScan().init({ export async function testNSFW(link: string): Promise { const [fileStream, hash] = await streamAttachment(link); const alreadyHaveCheck = await client.database.scanCache.read(hash); - if (alreadyHaveCheck?.nsfw) return { nsfw: alreadyHaveCheck.nsfw }; + if (alreadyHaveCheck?.nsfw !== undefined) return { nsfw: alreadyHaveCheck.nsfw }; const image = tf.tensor3d(new Uint8Array(fileStream)); @@ -48,7 +48,7 @@ export async function testNSFW(link: string): Promise { export async function testMalware(link: string): Promise { const [fileName, hash] = await saveAttachment(link); const alreadyHaveCheck = await client.database.scanCache.read(hash); - if (alreadyHaveCheck?.malware) return { safe: alreadyHaveCheck.malware }; + if (alreadyHaveCheck?.malware !== undefined) return { safe: alreadyHaveCheck.malware }; let malware; try { malware = (await clamscanner.scanFile(fileName)).isInfected; @@ -61,7 +61,7 @@ export async function testMalware(link: string): Promise { export async function testLink(link: string): Promise<{ safe: boolean; tags: string[] }> { const alreadyHaveCheck = await client.database.scanCache.read(link); - if (alreadyHaveCheck?.bad_link) return { safe: alreadyHaveCheck.bad_link, tags: alreadyHaveCheck.tags ?? [] }; + if (alreadyHaveCheck?.bad_link !== undefined) return { safe: alreadyHaveCheck.bad_link, tags: alreadyHaveCheck.tags ?? [] }; const scanned: { safe?: boolean; tags?: string[] } = await fetch("https://unscan.p.rapidapi.com/link", { method: "POST", headers: {