refactor: /proxyをFileServerServiceに統合し、/proxyのurlで/filesが指定されていた場合は直接ファイルを解決するようにする (#9709)
* wip? * clean up * Implement? HttpFetchService * ✌️ * remove node-fetch * fix * refactor * fix * gateway timeout * UndiciFetcherクラスを追加 (仮コミット, ビルドもstartもさせていない) * fix * add logger and fix url preview * fix ip check * enhance logger and error handling * fix * fix * clean up * Use custom fetcher for ApRequest / ApResolver * bypassProxyはproxyBypassHostsに判断を委譲するように * set maxRedirections (default 3, ApRequest/ApResolver: 0) * fix * wip???? * wip * ✌️ * set .node-version * clean up * refactor * clean up * refactor * refactor detectRequestType * rename detectResponseType * ✌️ * fix * wip * clean up * no got * remove got * wip * ✌️ * fix * clean up * remove unnnecessary const * good cleanup * no stream * Revert "no stream" This reverts commit 636f9192fcd2b17e71bbf6b5b106b490e0f66244. * fix * cache-control: max-age=300 to error * refactor cleanup
This commit is contained in:
		
							parent
							
								
									452bd6db25
								
							
						
					
					
						commit
						4735ae6451
					
				
					 7 changed files with 354 additions and 265 deletions
				
			
		| 
						 | 
					@ -1 +1 @@
 | 
				
			||||||
v18.12.1
 | 
					v18.13.0
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -9,6 +9,14 @@ export type IImage = {
 | 
				
			||||||
	type: string;
 | 
						type: string;
 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export type IImageStream = {
 | 
				
			||||||
 | 
						data: Readable;
 | 
				
			||||||
 | 
						ext: string | null;
 | 
				
			||||||
 | 
						type: string;
 | 
				
			||||||
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export type IImageStreamable = IImage | IImageStream;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export const webpDefault: sharp.WebpOptions = {
 | 
					export const webpDefault: sharp.WebpOptions = {
 | 
				
			||||||
	quality: 85,
 | 
						quality: 85,
 | 
				
			||||||
	alphaQuality: 95,
 | 
						alphaQuality: 95,
 | 
				
			||||||
| 
						 | 
					@ -19,6 +27,7 @@ export const webpDefault: sharp.WebpOptions = {
 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import { bindThis } from '@/decorators.js';
 | 
					import { bindThis } from '@/decorators.js';
 | 
				
			||||||
 | 
					import { Readable } from 'node:stream';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@Injectable()
 | 
					@Injectable()
 | 
				
			||||||
export class ImageProcessingService {
 | 
					export class ImageProcessingService {
 | 
				
			||||||
| 
						 | 
					@ -64,7 +73,7 @@ export class ImageProcessingService {
 | 
				
			||||||
	 */
 | 
						 */
 | 
				
			||||||
	@bindThis
 | 
						@bindThis
 | 
				
			||||||
	public async convertToWebp(path: string, width: number, height: number, options: sharp.WebpOptions = webpDefault): Promise<IImage> {
 | 
						public async convertToWebp(path: string, width: number, height: number, options: sharp.WebpOptions = webpDefault): Promise<IImage> {
 | 
				
			||||||
		return this.convertSharpToWebp(await sharp(path), width, height, options);
 | 
							return this.convertSharpToWebp(sharp(path), width, height, options);
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	@bindThis
 | 
						@bindThis
 | 
				
			||||||
| 
						 | 
					@ -85,6 +94,27 @@ export class ImageProcessingService {
 | 
				
			||||||
		};
 | 
							};
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						public convertToWebpStream(path: string, width: number, height: number, options: sharp.WebpOptions = webpDefault): IImageStream {
 | 
				
			||||||
 | 
							return this.convertSharpToWebpStream(sharp(path), width, height, options);
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						public convertSharpToWebpStream(sharp: sharp.Sharp, width: number, height: number, options: sharp.WebpOptions = webpDefault): IImageStream {
 | 
				
			||||||
 | 
							const data = sharp
 | 
				
			||||||
 | 
								.resize(width, height, {
 | 
				
			||||||
 | 
									fit: 'inside',
 | 
				
			||||||
 | 
									withoutEnlargement: true,
 | 
				
			||||||
 | 
								})
 | 
				
			||||||
 | 
								.rotate()
 | 
				
			||||||
 | 
								.webp(options)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							return {
 | 
				
			||||||
 | 
								data,
 | 
				
			||||||
 | 
								ext: 'webp',
 | 
				
			||||||
 | 
								type: 'image/webp',
 | 
				
			||||||
 | 
							};
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
	/**
 | 
						/**
 | 
				
			||||||
	 * Convert to PNG
 | 
						 * Convert to PNG
 | 
				
			||||||
	 *   with resize, remove metadata, resolve orientation, stop animation
 | 
						 *   with resize, remove metadata, resolve orientation, stop animation
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -68,6 +68,7 @@ export default class Logger {
 | 
				
			||||||
		if (envOption.withLogTime) log = chalk.gray(time) + ' ' + log;
 | 
							if (envOption.withLogTime) log = chalk.gray(time) + ' ' + log;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		console.log(important ? chalk.bold(log) : log);
 | 
							console.log(important ? chalk.bold(log) : log);
 | 
				
			||||||
 | 
							if (level === 'error' && data) console.log(data);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if (store) {
 | 
							if (store) {
 | 
				
			||||||
			if (this.syslogClient) {
 | 
								if (this.syslogClient) {
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -5,14 +5,14 @@ import { Inject, Injectable } from '@nestjs/common';
 | 
				
			||||||
import fastifyStatic from '@fastify/static';
 | 
					import fastifyStatic from '@fastify/static';
 | 
				
			||||||
import rename from 'rename';
 | 
					import rename from 'rename';
 | 
				
			||||||
import type { Config } from '@/config.js';
 | 
					import type { Config } from '@/config.js';
 | 
				
			||||||
import type { DriveFilesRepository } from '@/models/index.js';
 | 
					import type { DriveFile, DriveFilesRepository } from '@/models/index.js';
 | 
				
			||||||
import { DI } from '@/di-symbols.js';
 | 
					import { DI } from '@/di-symbols.js';
 | 
				
			||||||
import { createTemp } from '@/misc/create-temp.js';
 | 
					import { createTemp } from '@/misc/create-temp.js';
 | 
				
			||||||
import { FILE_TYPE_BROWSERSAFE } from '@/const.js';
 | 
					import { FILE_TYPE_BROWSERSAFE } from '@/const.js';
 | 
				
			||||||
import { StatusError } from '@/misc/status-error.js';
 | 
					import { StatusError } from '@/misc/status-error.js';
 | 
				
			||||||
import type Logger from '@/logger.js';
 | 
					import type Logger from '@/logger.js';
 | 
				
			||||||
import { DownloadService } from '@/core/DownloadService.js';
 | 
					import { DownloadService } from '@/core/DownloadService.js';
 | 
				
			||||||
import { ImageProcessingService } from '@/core/ImageProcessingService.js';
 | 
					import { IImageStreamable, ImageProcessingService, webpDefault } from '@/core/ImageProcessingService.js';
 | 
				
			||||||
import { VideoProcessingService } from '@/core/VideoProcessingService.js';
 | 
					import { VideoProcessingService } from '@/core/VideoProcessingService.js';
 | 
				
			||||||
import { InternalStorageService } from '@/core/InternalStorageService.js';
 | 
					import { InternalStorageService } from '@/core/InternalStorageService.js';
 | 
				
			||||||
import { contentDisposition } from '@/misc/content-disposition.js';
 | 
					import { contentDisposition } from '@/misc/content-disposition.js';
 | 
				
			||||||
| 
						 | 
					@ -20,6 +20,8 @@ import { FileInfoService } from '@/core/FileInfoService.js';
 | 
				
			||||||
import { LoggerService } from '@/core/LoggerService.js';
 | 
					import { LoggerService } from '@/core/LoggerService.js';
 | 
				
			||||||
import { bindThis } from '@/decorators.js';
 | 
					import { bindThis } from '@/decorators.js';
 | 
				
			||||||
import type { FastifyInstance, FastifyRequest, FastifyReply, FastifyPluginOptions } from 'fastify';
 | 
					import type { FastifyInstance, FastifyRequest, FastifyReply, FastifyPluginOptions } from 'fastify';
 | 
				
			||||||
 | 
					import { isMimeImage } from '@/misc/is-mime-image.js';
 | 
				
			||||||
 | 
					import sharp from 'sharp';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const _filename = fileURLToPath(import.meta.url);
 | 
					const _filename = fileURLToPath(import.meta.url);
 | 
				
			||||||
const _dirname = dirname(_filename);
 | 
					const _dirname = dirname(_filename);
 | 
				
			||||||
| 
						 | 
					@ -57,7 +59,7 @@ export class FileServerService {
 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=300');
 | 
								reply.header('Cache-Control', 'max-age=300');
 | 
				
			||||||
		};
 | 
							};
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	
 | 
					
 | 
				
			||||||
	@bindThis
 | 
						@bindThis
 | 
				
			||||||
	public createServer(fastify: FastifyInstance, options: FastifyPluginOptions, done: (err?: Error) => void) {
 | 
						public createServer(fastify: FastifyInstance, options: FastifyPluginOptions, done: (err?: Error) => void) {
 | 
				
			||||||
		fastify.addHook('onRequest', (request, reply, done) => {
 | 
							fastify.addHook('onRequest', (request, reply, done) => {
 | 
				
			||||||
| 
						 | 
					@ -70,23 +72,309 @@ export class FileServerService {
 | 
				
			||||||
			serve: false,
 | 
								serve: false,
 | 
				
			||||||
		});
 | 
							});
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		fastify.get('/app-default.jpg', (request, reply) => {
 | 
							fastify.get('/files/app-default.jpg', (request, reply) => {
 | 
				
			||||||
			const file = fs.createReadStream(`${_dirname}/assets/dummy.png`);
 | 
								const file = fs.createReadStream(`${_dirname}/assets/dummy.png`);
 | 
				
			||||||
			reply.header('Content-Type', 'image/jpeg');
 | 
								reply.header('Content-Type', 'image/jpeg');
 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
								reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
				
			||||||
			return reply.send(file);
 | 
								return reply.send(file);
 | 
				
			||||||
		});
 | 
							});
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		fastify.get<{ Params: { key: string; } }>('/:key', async (request, reply) => await this.sendDriveFile(request, reply));
 | 
							fastify.get<{ Params: { key: string; } }>('/files/:key', async (request, reply) => {
 | 
				
			||||||
		fastify.get<{ Params: { key: string; } }>('/:key/*', async (request, reply) => await this.sendDriveFile(request, reply));
 | 
								return await this.sendDriveFile(request, reply)
 | 
				
			||||||
 | 
									.catch(err => this.errorHandler(request, reply, err));
 | 
				
			||||||
 | 
							});
 | 
				
			||||||
 | 
							fastify.get<{ Params: { key: string; } }>('/files/:key/*', async (request, reply) => {
 | 
				
			||||||
 | 
								return await this.sendDriveFile(request, reply)
 | 
				
			||||||
 | 
									.catch(err => this.errorHandler(request, reply, err));
 | 
				
			||||||
 | 
							});
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							fastify.get<{
 | 
				
			||||||
 | 
								Params: { url: string; };
 | 
				
			||||||
 | 
								Querystring: { url?: string; };
 | 
				
			||||||
 | 
							}>('/proxy/:url*', async (request, reply) => {
 | 
				
			||||||
 | 
								return await this.proxyHandler(request, reply)
 | 
				
			||||||
 | 
									.catch(err => this.errorHandler(request, reply, err));
 | 
				
			||||||
 | 
							});
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		done();
 | 
							done();
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						private async errorHandler(request: FastifyRequest<{ Params?: { [x: string]: any }; Querystring?: { [x: string]: any }; }>, reply: FastifyReply, err?: any) {
 | 
				
			||||||
 | 
							this.logger.error(`${err}`);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							reply.header('Cache-Control', 'max-age=300');
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (request.query && 'fallback' in request.query) {
 | 
				
			||||||
 | 
								return reply.sendFile('/dummy.png', assets);
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (err instanceof StatusError && (err.statusCode === 302 || err.isClientError)) {
 | 
				
			||||||
 | 
								reply.code(err.statusCode);
 | 
				
			||||||
 | 
								return;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							reply.code(500);
 | 
				
			||||||
 | 
							return;
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	@bindThis
 | 
						@bindThis
 | 
				
			||||||
	private async sendDriveFile(request: FastifyRequest<{ Params: { key: string; } }>, reply: FastifyReply) {
 | 
						private async sendDriveFile(request: FastifyRequest<{ Params: { key: string; } }>, reply: FastifyReply) {
 | 
				
			||||||
		const key = request.params.key;
 | 
							const key = request.params.key;
 | 
				
			||||||
 | 
							const file = await this.getFileFromKey(key).then();
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (file === '404') {
 | 
				
			||||||
 | 
								reply.code(404);
 | 
				
			||||||
 | 
								reply.header('Cache-Control', 'max-age=86400');
 | 
				
			||||||
 | 
								return reply.sendFile('/dummy.png', assets);
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (file === '204') {
 | 
				
			||||||
 | 
								reply.code(204);
 | 
				
			||||||
 | 
								reply.header('Cache-Control', 'max-age=86400');
 | 
				
			||||||
 | 
								return;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							try {
 | 
				
			||||||
 | 
								if (file.state === 'remote') {
 | 
				
			||||||
 | 
									const convertFile = async () => {
 | 
				
			||||||
 | 
										if (file.fileRole === 'thumbnail') {
 | 
				
			||||||
 | 
											if (['image/jpeg', 'image/webp', 'image/avif', 'image/png', 'image/svg+xml'].includes(file.mime)) {
 | 
				
			||||||
 | 
												return this.imageProcessingService.convertToWebpStream(
 | 
				
			||||||
 | 
													file.path,
 | 
				
			||||||
 | 
													498,
 | 
				
			||||||
 | 
													280
 | 
				
			||||||
 | 
												);
 | 
				
			||||||
 | 
											} else if (file.mime.startsWith('video/')) {
 | 
				
			||||||
 | 
												return await this.videoProcessingService.generateVideoThumbnail(file.path);
 | 
				
			||||||
 | 
											}
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
										if (file.fileRole === 'webpublic') {
 | 
				
			||||||
 | 
											if (['image/svg+xml'].includes(file.mime)) {
 | 
				
			||||||
 | 
												return this.imageProcessingService.convertToWebpStream(
 | 
				
			||||||
 | 
													file.path,
 | 
				
			||||||
 | 
													2048,
 | 
				
			||||||
 | 
													2048,
 | 
				
			||||||
 | 
													{ ...webpDefault, lossless: true }
 | 
				
			||||||
 | 
												)
 | 
				
			||||||
 | 
											}
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
										return {
 | 
				
			||||||
 | 
											data: fs.createReadStream(file.path),
 | 
				
			||||||
 | 
											ext: file.ext,
 | 
				
			||||||
 | 
											type: file.mime,
 | 
				
			||||||
 | 
										};
 | 
				
			||||||
 | 
									};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									const image = await convertFile();
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									if ('pipe' in image.data && typeof image.data.pipe === 'function') {
 | 
				
			||||||
 | 
										// image.dataがstreamなら、stream終了後にcleanup
 | 
				
			||||||
 | 
										image.data.on('end', file.cleanup);
 | 
				
			||||||
 | 
										image.data.on('close', file.cleanup);
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										// image.dataがstreamでないなら直ちにcleanup
 | 
				
			||||||
 | 
										file.cleanup();
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									reply.header('Content-Type', FILE_TYPE_BROWSERSAFE.includes(image.type) ? image.type : 'application/octet-stream');
 | 
				
			||||||
 | 
									reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
				
			||||||
 | 
									return image.data;
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if (file.fileRole !== 'original') {
 | 
				
			||||||
 | 
									const filename = rename(file.file.name, {
 | 
				
			||||||
 | 
										suffix: file.fileRole === 'thumbnail' ? '-thumb' : '-web',
 | 
				
			||||||
 | 
										extname: file.ext ? `.${file.ext}` : undefined,
 | 
				
			||||||
 | 
									}).toString();
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									reply.header('Content-Type', FILE_TYPE_BROWSERSAFE.includes(file.mime) ? file.mime : 'application/octet-stream');
 | 
				
			||||||
 | 
									reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
				
			||||||
 | 
									reply.header('Content-Disposition', contentDisposition('inline', filename));
 | 
				
			||||||
 | 
									return fs.createReadStream(file.path);
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									const stream = fs.createReadStream(file.path);
 | 
				
			||||||
 | 
									stream.on('error', this.commonReadableHandlerGenerator(reply));
 | 
				
			||||||
 | 
									reply.header('Content-Type', FILE_TYPE_BROWSERSAFE.includes(file.file.type) ? file.file.type : 'application/octet-stream');
 | 
				
			||||||
 | 
									reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
				
			||||||
 | 
									reply.header('Content-Disposition', contentDisposition('inline', file.file.name));
 | 
				
			||||||
 | 
									return stream;
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							} catch (e) {
 | 
				
			||||||
 | 
								if ('cleanup' in file) file.cleanup();
 | 
				
			||||||
 | 
								throw e;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						private async proxyHandler(request: FastifyRequest<{ Params: { url: string; }; Querystring: { url?: string; }; }>, reply: FastifyReply) {
 | 
				
			||||||
 | 
							const url = 'url' in request.query ? request.query.url : 'https://' + request.params.url;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (typeof url !== 'string') {
 | 
				
			||||||
 | 
								reply.code(400);
 | 
				
			||||||
 | 
								return;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Create temp file
 | 
				
			||||||
 | 
							const file = await this.getStreamAndTypeFromUrl(url);
 | 
				
			||||||
 | 
							if (file === '404') {
 | 
				
			||||||
 | 
								reply.code(404);
 | 
				
			||||||
 | 
								reply.header('Cache-Control', 'max-age=86400');
 | 
				
			||||||
 | 
								return reply.sendFile('/dummy.png', assets);
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (file === '204') {
 | 
				
			||||||
 | 
								reply.code(204);
 | 
				
			||||||
 | 
								reply.header('Cache-Control', 'max-age=86400');
 | 
				
			||||||
 | 
								return;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							try {
 | 
				
			||||||
 | 
								const isConvertibleImage = isMimeImage(file.mime, 'sharp-convertible-image');
 | 
				
			||||||
 | 
								const isAnimationConvertibleImage = isMimeImage(file.mime, 'sharp-animation-convertible-image');
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								let image: IImageStreamable | null = null;
 | 
				
			||||||
 | 
								if ('emoji' in request.query && isConvertibleImage) {
 | 
				
			||||||
 | 
									if (!isAnimationConvertibleImage && !('static' in request.query)) {
 | 
				
			||||||
 | 
										image = {
 | 
				
			||||||
 | 
											data: fs.createReadStream(file.path),
 | 
				
			||||||
 | 
											ext: file.ext,
 | 
				
			||||||
 | 
											type: file.mime,
 | 
				
			||||||
 | 
										};
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										const data = sharp(file.path, { animated: !('static' in request.query) })
 | 
				
			||||||
 | 
												.resize({
 | 
				
			||||||
 | 
													height: 128,
 | 
				
			||||||
 | 
													withoutEnlargement: true,
 | 
				
			||||||
 | 
												})
 | 
				
			||||||
 | 
												.webp(webpDefault);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
										image = {
 | 
				
			||||||
 | 
											data,
 | 
				
			||||||
 | 
											ext: 'webp',
 | 
				
			||||||
 | 
											type: 'image/webp',
 | 
				
			||||||
 | 
										};
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								} else if ('static' in request.query && isConvertibleImage) {
 | 
				
			||||||
 | 
									image = this.imageProcessingService.convertToWebpStream(file.path, 498, 280);
 | 
				
			||||||
 | 
								} else if ('preview' in request.query && isConvertibleImage) {
 | 
				
			||||||
 | 
									image = this.imageProcessingService.convertToWebpStream(file.path, 200, 200);
 | 
				
			||||||
 | 
								} else if ('badge' in request.query) {
 | 
				
			||||||
 | 
									if (!isConvertibleImage) {
 | 
				
			||||||
 | 
										// 画像でないなら404でお茶を濁す
 | 
				
			||||||
 | 
										throw new StatusError('Unexpected mime', 404);
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									const mask = sharp(file.path)
 | 
				
			||||||
 | 
										.resize(96, 96, {
 | 
				
			||||||
 | 
											fit: 'inside',
 | 
				
			||||||
 | 
											withoutEnlargement: false,
 | 
				
			||||||
 | 
										})
 | 
				
			||||||
 | 
										.greyscale()
 | 
				
			||||||
 | 
										.normalise()
 | 
				
			||||||
 | 
										.linear(1.75, -(128 * 1.75) + 128) // 1.75x contrast
 | 
				
			||||||
 | 
										.flatten({ background: '#000' })
 | 
				
			||||||
 | 
										.toColorspace('b-w');
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
									const stats = await mask.clone().stats();
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
									if (stats.entropy < 0.1) {
 | 
				
			||||||
 | 
										// エントロピーがあまりない場合は404にする
 | 
				
			||||||
 | 
										throw new StatusError('Skip to provide badge', 404);
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
									const data = sharp({
 | 
				
			||||||
 | 
										create: { width: 96, height: 96, channels: 4, background: { r: 0, g: 0, b: 0, alpha: 0 } },
 | 
				
			||||||
 | 
									})
 | 
				
			||||||
 | 
										.pipelineColorspace('b-w')
 | 
				
			||||||
 | 
										.boolean(await mask.png().toBuffer(), 'eor');
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
									image = {
 | 
				
			||||||
 | 
										data: await data.png().toBuffer(),
 | 
				
			||||||
 | 
										ext: 'png',
 | 
				
			||||||
 | 
										type: 'image/png',
 | 
				
			||||||
 | 
									};
 | 
				
			||||||
 | 
								} else if (file.mime === 'image/svg+xml') {
 | 
				
			||||||
 | 
									image = this.imageProcessingService.convertToWebpStream(file.path, 2048, 2048);
 | 
				
			||||||
 | 
								} else if (!file.mime.startsWith('image/') || !FILE_TYPE_BROWSERSAFE.includes(file.mime)) {
 | 
				
			||||||
 | 
									throw new StatusError('Rejected type', 403, 'Rejected type');
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if (!image) {
 | 
				
			||||||
 | 
									image = {
 | 
				
			||||||
 | 
										data: fs.createReadStream(file.path),
 | 
				
			||||||
 | 
										ext: file.ext,
 | 
				
			||||||
 | 
										type: file.mime,
 | 
				
			||||||
 | 
									};
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if ('cleanup' in file) {
 | 
				
			||||||
 | 
									if ('pipe' in image.data && typeof image.data.pipe === 'function') {
 | 
				
			||||||
 | 
										// image.dataがstreamなら、stream終了後にcleanup
 | 
				
			||||||
 | 
										image.data.on('end', file.cleanup);
 | 
				
			||||||
 | 
										image.data.on('close', file.cleanup);
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										// image.dataがstreamでないなら直ちにcleanup
 | 
				
			||||||
 | 
										file.cleanup();
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								reply.header('Content-Type', image.type);
 | 
				
			||||||
 | 
								reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
				
			||||||
 | 
								return image.data;
 | 
				
			||||||
 | 
							} catch (e) {
 | 
				
			||||||
 | 
								if ('cleanup' in file) file.cleanup();
 | 
				
			||||||
 | 
								throw e;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						private async getStreamAndTypeFromUrl(url: string): Promise<
 | 
				
			||||||
 | 
							{ state: 'remote'; fileRole?: 'thumbnail' | 'webpublic' | 'original'; file?: DriveFile; mime: string; ext: string | null; path: string; cleanup: () => void; }
 | 
				
			||||||
 | 
							| { state: 'stored_internal'; fileRole: 'thumbnail' | 'webpublic' | 'original'; file: DriveFile; mime: string; ext: string | null; path: string; }
 | 
				
			||||||
 | 
							| '404'
 | 
				
			||||||
 | 
							| '204'
 | 
				
			||||||
 | 
						> {
 | 
				
			||||||
 | 
							if (url.startsWith(`${this.config.url}/files/`)) {
 | 
				
			||||||
 | 
								const key = url.replace(`${this.config.url}/files/`, '').split('/').shift();
 | 
				
			||||||
 | 
								if (!key) throw new StatusError('Invalid File Key', 400, 'Invalid File Key');
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								return await this.getFileFromKey(key);
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							return await this.downloadAndDetectTypeFromUrl(url);
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						private async downloadAndDetectTypeFromUrl(url: string): Promise<
 | 
				
			||||||
 | 
							{ state: 'remote' ; mime: string; ext: string | null; path: string; cleanup: () => void; }
 | 
				
			||||||
 | 
						> {
 | 
				
			||||||
 | 
							const [path, cleanup] = await createTemp();
 | 
				
			||||||
 | 
							try {
 | 
				
			||||||
 | 
								await this.downloadService.downloadUrl(url, path);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								const { mime, ext } = await this.fileInfoService.detectType(path);
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
								return {
 | 
				
			||||||
 | 
									state: 'remote',
 | 
				
			||||||
 | 
									mime, ext,
 | 
				
			||||||
 | 
									path, cleanup,
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							} catch (e) {
 | 
				
			||||||
 | 
								cleanup();
 | 
				
			||||||
 | 
								throw e;
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						@bindThis
 | 
				
			||||||
 | 
						private async getFileFromKey(key: string): Promise<
 | 
				
			||||||
 | 
							{ state: 'remote'; fileRole: 'thumbnail' | 'webpublic' | 'original'; file: DriveFile; mime: string; ext: string | null; path: string; cleanup: () => void; }
 | 
				
			||||||
 | 
							| { state: 'stored_internal'; fileRole: 'thumbnail' | 'webpublic' | 'original'; file: DriveFile; mime: string; ext: string | null; path: string; }
 | 
				
			||||||
 | 
							| '404'
 | 
				
			||||||
 | 
							| '204'
 | 
				
			||||||
 | 
						> {
 | 
				
			||||||
		// Fetch drive file
 | 
							// Fetch drive file
 | 
				
			||||||
		const file = await this.driveFilesRepository.createQueryBuilder('file')
 | 
							const file = await this.driveFilesRepository.createQueryBuilder('file')
 | 
				
			||||||
			.where('file.accessKey = :accessKey', { accessKey: key })
 | 
								.where('file.accessKey = :accessKey', { accessKey: key })
 | 
				
			||||||
| 
						 | 
					@ -94,89 +382,41 @@ export class FileServerService {
 | 
				
			||||||
			.orWhere('file.webpublicAccessKey = :webpublicAccessKey', { webpublicAccessKey: key })
 | 
								.orWhere('file.webpublicAccessKey = :webpublicAccessKey', { webpublicAccessKey: key })
 | 
				
			||||||
			.getOne();
 | 
								.getOne();
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if (file == null) {
 | 
							if (file == null) return '404';
 | 
				
			||||||
			reply.code(404);
 | 
					 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=86400');
 | 
					 | 
				
			||||||
			return reply.sendFile('/dummy.png', assets);
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
		const isThumbnail = file.thumbnailAccessKey === key;
 | 
							const isThumbnail = file.thumbnailAccessKey === key;
 | 
				
			||||||
		const isWebpublic = file.webpublicAccessKey === key;
 | 
							const isWebpublic = file.webpublicAccessKey === key;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if (!file.storedInternal) {
 | 
							if (!file.storedInternal) {
 | 
				
			||||||
			if (file.isLink && file.uri) {	// 期限切れリモートファイル
 | 
								if (!(file.isLink && file.uri)) return '204';
 | 
				
			||||||
				const [path, cleanup] = await createTemp();
 | 
								const result = await this.downloadAndDetectTypeFromUrl(file.uri);
 | 
				
			||||||
 | 
								return {
 | 
				
			||||||
				try {
 | 
									...result,
 | 
				
			||||||
					await this.downloadService.downloadUrl(file.uri, path);
 | 
									fileRole: isThumbnail ? 'thumbnail' : isWebpublic ? 'webpublic' : 'original',
 | 
				
			||||||
 | 
									file,
 | 
				
			||||||
					const { mime, ext } = await this.fileInfoService.detectType(path);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
					const convertFile = async () => {
 | 
					 | 
				
			||||||
						if (isThumbnail) {
 | 
					 | 
				
			||||||
							if (['image/jpeg', 'image/webp', 'image/avif', 'image/png', 'image/svg+xml'].includes(mime)) {
 | 
					 | 
				
			||||||
								return await this.imageProcessingService.convertToWebp(path, 498, 280);
 | 
					 | 
				
			||||||
							} else if (mime.startsWith('video/')) {
 | 
					 | 
				
			||||||
								return await this.videoProcessingService.generateVideoThumbnail(path);
 | 
					 | 
				
			||||||
							}
 | 
					 | 
				
			||||||
						}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
						if (isWebpublic) {
 | 
					 | 
				
			||||||
							if (['image/svg+xml'].includes(mime)) {
 | 
					 | 
				
			||||||
								return await this.imageProcessingService.convertToPng(path, 2048, 2048);
 | 
					 | 
				
			||||||
							}
 | 
					 | 
				
			||||||
						}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
						return {
 | 
					 | 
				
			||||||
							data: fs.readFileSync(path),
 | 
					 | 
				
			||||||
							ext,
 | 
					 | 
				
			||||||
							type: mime,
 | 
					 | 
				
			||||||
						};
 | 
					 | 
				
			||||||
					};
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
					const image = await convertFile();
 | 
					 | 
				
			||||||
					reply.header('Content-Type', FILE_TYPE_BROWSERSAFE.includes(image.type) ? image.type : 'application/octet-stream');
 | 
					 | 
				
			||||||
					reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
					 | 
				
			||||||
					return image.data;
 | 
					 | 
				
			||||||
				} catch (err) {
 | 
					 | 
				
			||||||
					this.logger.error(`${err}`);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
					if (err instanceof StatusError && err.isClientError) {
 | 
					 | 
				
			||||||
						reply.code(err.statusCode);
 | 
					 | 
				
			||||||
						reply.header('Cache-Control', 'max-age=86400');
 | 
					 | 
				
			||||||
					} else {
 | 
					 | 
				
			||||||
						reply.code(500);
 | 
					 | 
				
			||||||
						reply.header('Cache-Control', 'max-age=300');
 | 
					 | 
				
			||||||
					}
 | 
					 | 
				
			||||||
				} finally {
 | 
					 | 
				
			||||||
					cleanup();
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				return;
 | 
					 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					 | 
				
			||||||
			reply.code(204);
 | 
					 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=86400');
 | 
					 | 
				
			||||||
			return;
 | 
					 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if (isThumbnail || isWebpublic) {
 | 
							const path = this.internalStorageService.resolvePath(key);
 | 
				
			||||||
			const { mime, ext } = await this.fileInfoService.detectType(this.internalStorageService.resolvePath(key));
 | 
					 | 
				
			||||||
			const filename = rename(file.name, {
 | 
					 | 
				
			||||||
				suffix: isThumbnail ? '-thumb' : '-web',
 | 
					 | 
				
			||||||
				extname: ext ? `.${ext}` : undefined,
 | 
					 | 
				
			||||||
			}).toString();
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
			reply.header('Content-Type', FILE_TYPE_BROWSERSAFE.includes(mime) ? mime : 'application/octet-stream');
 | 
							if (isThumbnail || isWebpublic) {
 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
								const { mime, ext } = await this.fileInfoService.detectType(path);
 | 
				
			||||||
			reply.header('Content-Disposition', contentDisposition('inline', filename));
 | 
								return {
 | 
				
			||||||
			return this.internalStorageService.read(key);
 | 
									state: 'stored_internal',
 | 
				
			||||||
		} else {
 | 
									fileRole: isThumbnail ? 'thumbnail' : 'webpublic',
 | 
				
			||||||
			const readable = this.internalStorageService.read(file.accessKey!);
 | 
									file,
 | 
				
			||||||
			readable.on('error', this.commonReadableHandlerGenerator(reply));
 | 
									mime, ext,
 | 
				
			||||||
			reply.header('Content-Type', FILE_TYPE_BROWSERSAFE.includes(file.type) ? file.type : 'application/octet-stream');
 | 
									path,
 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
								};
 | 
				
			||||||
			reply.header('Content-Disposition', contentDisposition('inline', file.name));
 | 
							}
 | 
				
			||||||
			return readable;
 | 
					
 | 
				
			||||||
 | 
							return {
 | 
				
			||||||
 | 
								state: 'stored_internal',
 | 
				
			||||||
 | 
								fileRole: 'original',
 | 
				
			||||||
 | 
								file,
 | 
				
			||||||
 | 
								mime: file.type,
 | 
				
			||||||
 | 
								ext: null,
 | 
				
			||||||
 | 
								path,
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -1,177 +0,0 @@
 | 
				
			||||||
import * as fs from 'node:fs';
 | 
					 | 
				
			||||||
import { fileURLToPath } from 'node:url';
 | 
					 | 
				
			||||||
import { dirname } from 'node:path';
 | 
					 | 
				
			||||||
import { Inject, Injectable } from '@nestjs/common';
 | 
					 | 
				
			||||||
import sharp from 'sharp';
 | 
					 | 
				
			||||||
import fastifyStatic from '@fastify/static';
 | 
					 | 
				
			||||||
import { DI } from '@/di-symbols.js';
 | 
					 | 
				
			||||||
import type { Config } from '@/config.js';
 | 
					 | 
				
			||||||
import { isMimeImage } from '@/misc/is-mime-image.js';
 | 
					 | 
				
			||||||
import { createTemp } from '@/misc/create-temp.js';
 | 
					 | 
				
			||||||
import { DownloadService } from '@/core/DownloadService.js';
 | 
					 | 
				
			||||||
import { ImageProcessingService, webpDefault } from '@/core/ImageProcessingService.js';
 | 
					 | 
				
			||||||
import type { IImage } from '@/core/ImageProcessingService.js';
 | 
					 | 
				
			||||||
import { FILE_TYPE_BROWSERSAFE } from '@/const.js';
 | 
					 | 
				
			||||||
import { StatusError } from '@/misc/status-error.js';
 | 
					 | 
				
			||||||
import type Logger from '@/logger.js';
 | 
					 | 
				
			||||||
import { FileInfoService } from '@/core/FileInfoService.js';
 | 
					 | 
				
			||||||
import { LoggerService } from '@/core/LoggerService.js';
 | 
					 | 
				
			||||||
import { bindThis } from '@/decorators.js';
 | 
					 | 
				
			||||||
import type { FastifyInstance, FastifyPluginOptions, FastifyReply, FastifyRequest } from 'fastify';
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const _filename = fileURLToPath(import.meta.url);
 | 
					 | 
				
			||||||
const _dirname = dirname(_filename);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const assets = `${_dirname}/../../src/server/assets/`;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
@Injectable()
 | 
					 | 
				
			||||||
export class MediaProxyServerService {
 | 
					 | 
				
			||||||
	private logger: Logger;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	constructor(
 | 
					 | 
				
			||||||
		@Inject(DI.config)
 | 
					 | 
				
			||||||
		private config: Config,
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		private fileInfoService: FileInfoService,
 | 
					 | 
				
			||||||
		private downloadService: DownloadService,
 | 
					 | 
				
			||||||
		private imageProcessingService: ImageProcessingService,
 | 
					 | 
				
			||||||
		private loggerService: LoggerService,
 | 
					 | 
				
			||||||
	) {
 | 
					 | 
				
			||||||
		this.logger = this.loggerService.getLogger('server', 'gray', false);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		//this.createServer = this.createServer.bind(this);
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	@bindThis
 | 
					 | 
				
			||||||
	public createServer(fastify: FastifyInstance, options: FastifyPluginOptions, done: (err?: Error) => void) {
 | 
					 | 
				
			||||||
		fastify.addHook('onRequest', (request, reply, done) => {
 | 
					 | 
				
			||||||
			reply.header('Content-Security-Policy', 'default-src \'none\'; img-src \'self\'; media-src \'self\'; style-src \'unsafe-inline\'');
 | 
					 | 
				
			||||||
			done();
 | 
					 | 
				
			||||||
		});
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		fastify.register(fastifyStatic, {
 | 
					 | 
				
			||||||
			root: _dirname,
 | 
					 | 
				
			||||||
			serve: false,
 | 
					 | 
				
			||||||
		});
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		fastify.get<{
 | 
					 | 
				
			||||||
			Params: { url: string; };
 | 
					 | 
				
			||||||
			Querystring: { url?: string; };
 | 
					 | 
				
			||||||
		}>('/:url*', async (request, reply) => await this.handler(request, reply));
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		done();
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	@bindThis
 | 
					 | 
				
			||||||
	private async handler(request: FastifyRequest<{ Params: { url: string; }; Querystring: { url?: string; }; }>, reply: FastifyReply) {
 | 
					 | 
				
			||||||
		const url = 'url' in request.query ? request.query.url : 'https://' + request.params.url;
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
		if (typeof url !== 'string') {
 | 
					 | 
				
			||||||
			reply.code(400);
 | 
					 | 
				
			||||||
			return;
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
		// Create temp file
 | 
					 | 
				
			||||||
		const [path, cleanup] = await createTemp();
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
		try {
 | 
					 | 
				
			||||||
			await this.downloadService.downloadUrl(url, path);
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
			const { mime, ext } = await this.fileInfoService.detectType(path);
 | 
					 | 
				
			||||||
			const isConvertibleImage = isMimeImage(mime, 'sharp-convertible-image');
 | 
					 | 
				
			||||||
			const isAnimationConvertibleImage = isMimeImage(mime, 'sharp-animation-convertible-image');
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
			let image: IImage;
 | 
					 | 
				
			||||||
			if ('emoji' in request.query && isConvertibleImage) {
 | 
					 | 
				
			||||||
				if (!isAnimationConvertibleImage && !('static' in request.query)) {
 | 
					 | 
				
			||||||
					image = {
 | 
					 | 
				
			||||||
						data: fs.readFileSync(path),
 | 
					 | 
				
			||||||
						ext,
 | 
					 | 
				
			||||||
						type: mime,
 | 
					 | 
				
			||||||
					};
 | 
					 | 
				
			||||||
				} else {
 | 
					 | 
				
			||||||
					const data = await sharp(path, { animated: !('static' in request.query) })
 | 
					 | 
				
			||||||
					.resize({
 | 
					 | 
				
			||||||
						height: 128,
 | 
					 | 
				
			||||||
						withoutEnlargement: true,
 | 
					 | 
				
			||||||
					})
 | 
					 | 
				
			||||||
					.webp(webpDefault)
 | 
					 | 
				
			||||||
					.toBuffer();
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
					image = {
 | 
					 | 
				
			||||||
						data,
 | 
					 | 
				
			||||||
						ext: 'webp',
 | 
					 | 
				
			||||||
						type: 'image/webp',
 | 
					 | 
				
			||||||
					};
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
			} else if ('static' in request.query && isConvertibleImage) {
 | 
					 | 
				
			||||||
				image = await this.imageProcessingService.convertToWebp(path, 498, 280);
 | 
					 | 
				
			||||||
			} else if ('preview' in request.query && isConvertibleImage) {
 | 
					 | 
				
			||||||
				image = await this.imageProcessingService.convertToWebp(path, 200, 200);
 | 
					 | 
				
			||||||
			} else if ('badge' in request.query) {
 | 
					 | 
				
			||||||
				if (!isConvertibleImage) {
 | 
					 | 
				
			||||||
					// 画像でないなら404でお茶を濁す
 | 
					 | 
				
			||||||
					throw new StatusError('Unexpected mime', 404);
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
				const mask = sharp(path)
 | 
					 | 
				
			||||||
					.resize(96, 96, {
 | 
					 | 
				
			||||||
						fit: 'inside',
 | 
					 | 
				
			||||||
						withoutEnlargement: false,
 | 
					 | 
				
			||||||
					})
 | 
					 | 
				
			||||||
					.greyscale()
 | 
					 | 
				
			||||||
					.normalise()
 | 
					 | 
				
			||||||
					.linear(1.75, -(128 * 1.75) + 128) // 1.75x contrast
 | 
					 | 
				
			||||||
					.flatten({ background: '#000' })
 | 
					 | 
				
			||||||
					.toColorspace('b-w');
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
				const stats = await mask.clone().stats();
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
				if (stats.entropy < 0.1) {
 | 
					 | 
				
			||||||
					// エントロピーがあまりない場合は404にする
 | 
					 | 
				
			||||||
					throw new StatusError('Skip to provide badge', 404);
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
				const data = sharp({
 | 
					 | 
				
			||||||
					create: { width: 96, height: 96, channels: 4, background: { r: 0, g: 0, b: 0, alpha: 0 } },
 | 
					 | 
				
			||||||
				})
 | 
					 | 
				
			||||||
					.pipelineColorspace('b-w')
 | 
					 | 
				
			||||||
					.boolean(await mask.png().toBuffer(), 'eor');
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
				image = {
 | 
					 | 
				
			||||||
					data: await data.png().toBuffer(),
 | 
					 | 
				
			||||||
					ext: 'png',
 | 
					 | 
				
			||||||
					type: 'image/png',
 | 
					 | 
				
			||||||
				};
 | 
					 | 
				
			||||||
			} else if (mime === 'image/svg+xml') {
 | 
					 | 
				
			||||||
				image = await this.imageProcessingService.convertToWebp(path, 2048, 2048, webpDefault);
 | 
					 | 
				
			||||||
			} else if (!mime.startsWith('image/') || !FILE_TYPE_BROWSERSAFE.includes(mime)) {
 | 
					 | 
				
			||||||
				throw new StatusError('Rejected type', 403, 'Rejected type');
 | 
					 | 
				
			||||||
			} else {
 | 
					 | 
				
			||||||
				image = {
 | 
					 | 
				
			||||||
					data: fs.readFileSync(path),
 | 
					 | 
				
			||||||
					ext,
 | 
					 | 
				
			||||||
					type: mime,
 | 
					 | 
				
			||||||
				};
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
			reply.header('Content-Type', image.type);
 | 
					 | 
				
			||||||
			reply.header('Cache-Control', 'max-age=31536000, immutable');
 | 
					 | 
				
			||||||
			return image.data;
 | 
					 | 
				
			||||||
		} catch (err) {
 | 
					 | 
				
			||||||
			this.logger.error(`${err}`);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			if ('fallback' in request.query) {
 | 
					 | 
				
			||||||
				return reply.sendFile('/dummy.png', assets);
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
			if (err instanceof StatusError && (err.statusCode === 302 || err.isClientError)) {
 | 
					 | 
				
			||||||
				reply.code(err.statusCode);
 | 
					 | 
				
			||||||
			} else {
 | 
					 | 
				
			||||||
				reply.code(500);
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		} finally {
 | 
					 | 
				
			||||||
			cleanup();
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
| 
						 | 
					@ -3,7 +3,6 @@ import { EndpointsModule } from '@/server/api/EndpointsModule.js';
 | 
				
			||||||
import { CoreModule } from '@/core/CoreModule.js';
 | 
					import { CoreModule } from '@/core/CoreModule.js';
 | 
				
			||||||
import { ApiCallService } from './api/ApiCallService.js';
 | 
					import { ApiCallService } from './api/ApiCallService.js';
 | 
				
			||||||
import { FileServerService } from './FileServerService.js';
 | 
					import { FileServerService } from './FileServerService.js';
 | 
				
			||||||
import { MediaProxyServerService } from './MediaProxyServerService.js';
 | 
					 | 
				
			||||||
import { NodeinfoServerService } from './NodeinfoServerService.js';
 | 
					import { NodeinfoServerService } from './NodeinfoServerService.js';
 | 
				
			||||||
import { ServerService } from './ServerService.js';
 | 
					import { ServerService } from './ServerService.js';
 | 
				
			||||||
import { WellKnownServerService } from './WellKnownServerService.js';
 | 
					import { WellKnownServerService } from './WellKnownServerService.js';
 | 
				
			||||||
| 
						 | 
					@ -51,7 +50,6 @@ import { UserListChannelService } from './api/stream/channels/user-list.js';
 | 
				
			||||||
		UrlPreviewService,
 | 
							UrlPreviewService,
 | 
				
			||||||
		ActivityPubServerService,
 | 
							ActivityPubServerService,
 | 
				
			||||||
		FileServerService,
 | 
							FileServerService,
 | 
				
			||||||
		MediaProxyServerService,
 | 
					 | 
				
			||||||
		NodeinfoServerService,
 | 
							NodeinfoServerService,
 | 
				
			||||||
		ServerService,
 | 
							ServerService,
 | 
				
			||||||
		WellKnownServerService,
 | 
							WellKnownServerService,
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -20,7 +20,6 @@ import { NodeinfoServerService } from './NodeinfoServerService.js';
 | 
				
			||||||
import { ApiServerService } from './api/ApiServerService.js';
 | 
					import { ApiServerService } from './api/ApiServerService.js';
 | 
				
			||||||
import { StreamingApiServerService } from './api/StreamingApiServerService.js';
 | 
					import { StreamingApiServerService } from './api/StreamingApiServerService.js';
 | 
				
			||||||
import { WellKnownServerService } from './WellKnownServerService.js';
 | 
					import { WellKnownServerService } from './WellKnownServerService.js';
 | 
				
			||||||
import { MediaProxyServerService } from './MediaProxyServerService.js';
 | 
					 | 
				
			||||||
import { FileServerService } from './FileServerService.js';
 | 
					import { FileServerService } from './FileServerService.js';
 | 
				
			||||||
import { ClientServerService } from './web/ClientServerService.js';
 | 
					import { ClientServerService } from './web/ClientServerService.js';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					@ -48,7 +47,6 @@ export class ServerService {
 | 
				
			||||||
		private wellKnownServerService: WellKnownServerService,
 | 
							private wellKnownServerService: WellKnownServerService,
 | 
				
			||||||
		private nodeinfoServerService: NodeinfoServerService,
 | 
							private nodeinfoServerService: NodeinfoServerService,
 | 
				
			||||||
		private fileServerService: FileServerService,
 | 
							private fileServerService: FileServerService,
 | 
				
			||||||
		private mediaProxyServerService: MediaProxyServerService,
 | 
					 | 
				
			||||||
		private clientServerService: ClientServerService,
 | 
							private clientServerService: ClientServerService,
 | 
				
			||||||
		private globalEventService: GlobalEventService,
 | 
							private globalEventService: GlobalEventService,
 | 
				
			||||||
		private loggerService: LoggerService,
 | 
							private loggerService: LoggerService,
 | 
				
			||||||
| 
						 | 
					@ -73,8 +71,7 @@ export class ServerService {
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		fastify.register(this.apiServerService.createServer, { prefix: '/api' });
 | 
							fastify.register(this.apiServerService.createServer, { prefix: '/api' });
 | 
				
			||||||
		fastify.register(this.fileServerService.createServer, { prefix: '/files' });
 | 
							fastify.register(this.fileServerService.createServer);
 | 
				
			||||||
		fastify.register(this.mediaProxyServerService.createServer, { prefix: '/proxy' });
 | 
					 | 
				
			||||||
		fastify.register(this.activityPubServerService.createServer);
 | 
							fastify.register(this.activityPubServerService.createServer);
 | 
				
			||||||
		fastify.register(this.nodeinfoServerService.createServer);
 | 
							fastify.register(this.nodeinfoServerService.createServer);
 | 
				
			||||||
		fastify.register(this.wellKnownServerService.createServer);
 | 
							fastify.register(this.wellKnownServerService.createServer);
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue