amechama/sharkey
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(server): make sure getFileInfo doesn't fail if detectSensitivity ever fails (#9020)

Browse Source
This commit is contained in:
Acid Chicken (硫酸鶏) 2022-07-19 17:09:21 +09:00 committed by GitHub
parent e35f599b6d
commit a3c2dbbfb5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
packages/backend/src/misc/get-file-info.ts
View File

@ -101,13 +101,17 @@ export async function getFileInfo(path: string, opts: {
let porn = false; let porn = false;
if (!opts.skipSensitiveDetection) { if (!opts.skipSensitiveDetection) {
[sensitive, porn] = await detectSensitivity( await detectSensitivity(
path, path,
type.mime, type.mime,
opts.sensitiveThreshold ?? 0.5, opts.sensitiveThreshold ?? 0.5,
opts.sensitiveThresholdForPorn ?? 0.75, opts.sensitiveThresholdForPorn ?? 0.75,
opts.enableSensitiveMediaDetectionForVideos ?? false, opts.enableSensitiveMediaDetectionForVideos ?? false,
); ).then(value => {
[sensitive, porn] = value;
}, error => {
warnings.push(`detectSensitivity failed: ${error}`);
});
} }
return { return {