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 636f9192fc
.
* fix
* cache-control: max-age=300 to error
* refactor cleanup
This commit is contained in:
parent
452bd6db25
commit
4735ae6451
|
@ -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…
Reference in New Issue