diff --git a/.config/example.yml b/.config/example.yml index 44d798bc8..f75224bf7 100644 --- a/.config/example.yml +++ b/.config/example.yml @@ -140,3 +140,6 @@ autoAdmin: true #proxySmtp: http://127.0.0.1:3128 # use HTTP/1.1 CONNECT #proxySmtp: socks4://127.0.0.1:1080 # use SOCKS4 #proxySmtp: socks5://127.0.0.1:1080 # use SOCKS5 + +# Media Proxy +#mediaProxy: http://127.0.0.1:3000 diff --git a/src/config/types.ts b/src/config/types.ts index e766d1f14..2bf94af74 100644 --- a/src/config/types.ts +++ b/src/config/types.ts @@ -52,6 +52,8 @@ export type Source = { host: string; port: number; }; + + mediaProxy?: string; }; /** diff --git a/src/models/repositories/drive-file.ts b/src/models/repositories/drive-file.ts index 702f195e9..a20d39304 100644 --- a/src/models/repositories/drive-file.ts +++ b/src/models/repositories/drive-file.ts @@ -6,6 +6,7 @@ import { toPuny } from '../../misc/convert-host'; import { ensure } from '../../prelude/ensure'; import { awaitAll } from '../../prelude/await-all'; import { SchemaType } from '../../misc/schema'; +import config from '../../config'; export type PackedDriveFile = SchemaType; @@ -22,7 +23,11 @@ export class DriveFileRepository extends Repository { } public getPublicUrl(file: DriveFile, thumbnail = false): string | null { - return thumbnail ? (file.thumbnailUrl || file.webpublicUrl || null) : (file.webpublicUrl || file.url); + let url = thumbnail ? (file.thumbnailUrl || file.webpublicUrl || null) : (file.webpublicUrl || file.url); + if (file.src !== null && file.userHost !== null && config.mediaProxy !== null) { + url = `${config.mediaProxy}/${thumbnail ? 'thumbnail' : ''}?url=${file.src}`; + } + return url; } public async clacDriveUsageOf(user: User['id'] | User): Promise {