enhance(server): refactor and tweak emoji proxy
This commit is contained in:
		
							parent
							
								
									6a992b6982
								
							
						
					
					
						commit
						54e3fccd87
					
				
					 3 changed files with 41 additions and 44 deletions
				
			
		
							
								
								
									
										
											BIN
										
									
								
								packages/backend/assets/emoji-unknown.png
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								packages/backend/assets/emoji-unknown.png
									
										
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							| After Width: | Height: | Size: 3.4 KiB | 
|  | @ -1,12 +1,11 @@ | ||||||
| import cluster from 'node:cluster'; | import cluster from 'node:cluster'; | ||||||
| import * as fs from 'node:fs'; | import * as fs from 'node:fs'; | ||||||
| import * as http from 'node:http'; |  | ||||||
| import { Inject, Injectable } from '@nestjs/common'; | import { Inject, Injectable } from '@nestjs/common'; | ||||||
| import Fastify from 'fastify'; | import Fastify from 'fastify'; | ||||||
| import { IsNull } from 'typeorm'; | import { IsNull } from 'typeorm'; | ||||||
| import { GlobalEventService } from '@/core/GlobalEventService.js'; | import { GlobalEventService } from '@/core/GlobalEventService.js'; | ||||||
| import type { Config } from '@/config.js'; | import type { Config } from '@/config.js'; | ||||||
| import type { UserProfilesRepository, UsersRepository } from '@/models/index.js'; | import type { EmojisRepository, UserProfilesRepository, UsersRepository } from '@/models/index.js'; | ||||||
| import { DI } from '@/di-symbols.js'; | import { DI } from '@/di-symbols.js'; | ||||||
| import type Logger from '@/logger.js'; | import type Logger from '@/logger.js'; | ||||||
| import { envOption } from '@/env.js'; | import { envOption } from '@/env.js'; | ||||||
|  | @ -39,6 +38,9 @@ export class ServerService { | ||||||
| 		@Inject(DI.userProfilesRepository) | 		@Inject(DI.userProfilesRepository) | ||||||
| 		private userProfilesRepository: UserProfilesRepository, | 		private userProfilesRepository: UserProfilesRepository, | ||||||
| 
 | 
 | ||||||
|  | 		@Inject(DI.emojisRepository) | ||||||
|  | 		private emojisRepository: EmojisRepository, | ||||||
|  | 
 | ||||||
| 		private userEntityService: UserEntityService, | 		private userEntityService: UserEntityService, | ||||||
| 		private apiServerService: ApiServerService, | 		private apiServerService: ApiServerService, | ||||||
| 		private streamingApiServerService: StreamingApiServerService, | 		private streamingApiServerService: StreamingApiServerService, | ||||||
|  | @ -77,6 +79,43 @@ export class ServerService { | ||||||
| 		fastify.register(this.nodeinfoServerService.createServer); | 		fastify.register(this.nodeinfoServerService.createServer); | ||||||
| 		fastify.register(this.wellKnownServerService.createServer); | 		fastify.register(this.wellKnownServerService.createServer); | ||||||
| 
 | 
 | ||||||
|  | 		fastify.get<{ Params: { path: string }; Querystring: { static?: any; }; }>('/emoji/:path(.*)', async (request, reply) => { | ||||||
|  | 			const path = request.params.path; | ||||||
|  | 
 | ||||||
|  | 			if (!path.match(/^[a-zA-Z0-9\-_@\.]+?\.webp$/)) { | ||||||
|  | 				reply.code(404); | ||||||
|  | 				return; | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			reply.header('Cache-Control', 'public, max-age=86400'); | ||||||
|  | 
 | ||||||
|  | 			const name = path.split('@')[0].replace('.webp', ''); | ||||||
|  | 			const host = path.split('@')[1]?.replace('.webp', ''); | ||||||
|  | 
 | ||||||
|  | 			const emoji = await this.emojisRepository.findOneBy({ | ||||||
|  | 				// `@.` is the spec of ReactionService.decodeReaction
 | ||||||
|  | 				host: (host == null || host === '.') ? IsNull() : host, | ||||||
|  | 				name: name, | ||||||
|  | 			}); | ||||||
|  | 
 | ||||||
|  | 			reply.header('Content-Security-Policy', 'default-src \'none\'; style-src \'unsafe-inline\''); | ||||||
|  | 
 | ||||||
|  | 			if (emoji == null) { | ||||||
|  | 				return await reply.redirect('/static-assets/emoji-unknown.png'); | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			const url = new URL('/proxy/emoji.webp', this.config.url); | ||||||
|  | 			// || emoji.originalUrl してるのは後方互換性のため(publicUrlはstringなので??はだめ)
 | ||||||
|  | 			url.searchParams.set('url', emoji.publicUrl || emoji.originalUrl); | ||||||
|  | 			url.searchParams.set('emoji', '1'); | ||||||
|  | 			if ('static' in request.query) url.searchParams.set('static', '1'); | ||||||
|  | 
 | ||||||
|  | 			return await reply.redirect( | ||||||
|  | 				301, | ||||||
|  | 				url.toString(), | ||||||
|  | 			); | ||||||
|  | 		}); | ||||||
|  | 
 | ||||||
| 		fastify.get<{ Params: { acct: string } }>('/avatar/@:acct', async (request, reply) => { | 		fastify.get<{ Params: { acct: string } }>('/avatar/@:acct', async (request, reply) => { | ||||||
| 			const { username, host } = Acct.parse(request.params.acct); | 			const { username, host } = Acct.parse(request.params.acct); | ||||||
| 			const user = await this.usersRepository.findOne({ | 			const user = await this.usersRepository.findOne({ | ||||||
|  |  | ||||||
|  | @ -1,6 +1,5 @@ | ||||||
| import { dirname } from 'node:path'; | import { dirname } from 'node:path'; | ||||||
| import { fileURLToPath } from 'node:url'; | import { fileURLToPath } from 'node:url'; | ||||||
| import { PathOrFileDescriptor, readFileSync } from 'node:fs'; |  | ||||||
| import { Inject, Injectable } from '@nestjs/common'; | import { Inject, Injectable } from '@nestjs/common'; | ||||||
| import { createBullBoard } from '@bull-board/api'; | import { createBullBoard } from '@bull-board/api'; | ||||||
| import { BullAdapter } from '@bull-board/api/bullAdapter.js'; | import { BullAdapter } from '@bull-board/api/bullAdapter.js'; | ||||||
|  | @ -71,9 +70,6 @@ export class ClientServerService { | ||||||
| 		@Inject(DI.pagesRepository) | 		@Inject(DI.pagesRepository) | ||||||
| 		private pagesRepository: PagesRepository, | 		private pagesRepository: PagesRepository, | ||||||
| 
 | 
 | ||||||
| 		@Inject(DI.emojisRepository) |  | ||||||
| 		private emojisRepository: EmojisRepository, |  | ||||||
| 
 |  | ||||||
| 		@Inject(DI.flashsRepository) | 		@Inject(DI.flashsRepository) | ||||||
| 		private flashsRepository: FlashsRepository, | 		private flashsRepository: FlashsRepository, | ||||||
| 
 | 
 | ||||||
|  | @ -225,44 +221,6 @@ export class ClientServerService { | ||||||
| 			return reply.sendFile('/apple-touch-icon.png', staticAssets); | 			return reply.sendFile('/apple-touch-icon.png', staticAssets); | ||||||
| 		}); | 		}); | ||||||
| 
 | 
 | ||||||
| 		fastify.get<{ Params: { path: string }; Querystring: { static?: any; }; }>('/emoji/:path(.*)', async (request, reply) => { |  | ||||||
| 			const path = request.params.path; |  | ||||||
| 
 |  | ||||||
| 			if (!path.match(/^[a-zA-Z0-9\-_@\.]+?\.webp$/)) { |  | ||||||
| 				reply.code(404); |  | ||||||
| 				return; |  | ||||||
| 			} |  | ||||||
| 
 |  | ||||||
| 			reply.header('Cache-Control', 'public, max-age=86400'); |  | ||||||
| 
 |  | ||||||
| 			const name = path.split('@')[0].replace('.webp', ''); |  | ||||||
| 			const host = path.split('@')[1]?.replace('.webp', ''); |  | ||||||
| 
 |  | ||||||
| 			const emoji = await this.emojisRepository.findOneBy({ |  | ||||||
| 				// `@.` is the spec of ReactionService.decodeReaction
 |  | ||||||
| 				host: (host == null || host === '.') ? IsNull() : host, |  | ||||||
| 				name: name, |  | ||||||
| 			}); |  | ||||||
| 
 |  | ||||||
| 			if (emoji == null) { |  | ||||||
| 				reply.code(404); |  | ||||||
| 				return; |  | ||||||
| 			} |  | ||||||
| 
 |  | ||||||
| 			reply.header('Content-Security-Policy', 'default-src \'none\'; style-src \'unsafe-inline\''); |  | ||||||
| 
 |  | ||||||
| 			const url = new URL('/proxy/emoji.webp', this.config.url); |  | ||||||
| 			// || emoji.originalUrl してるのは後方互換性のため(publicUrlはstringなので??はだめ)
 |  | ||||||
| 			url.searchParams.set('url', emoji.publicUrl || emoji.originalUrl); |  | ||||||
| 			url.searchParams.set('emoji', '1'); |  | ||||||
| 			if ('static' in request.query) url.searchParams.set('static', '1'); |  | ||||||
| 
 |  | ||||||
| 			return await reply.redirect( |  | ||||||
| 				301, |  | ||||||
| 				url.toString(), |  | ||||||
| 			); |  | ||||||
| 		}); |  | ||||||
| 
 |  | ||||||
| 		fastify.get<{ Params: { path: string } }>('/fluent-emoji/:path(.*)', async (request, reply) => { | 		fastify.get<{ Params: { path: string } }>('/fluent-emoji/:path(.*)', async (request, reply) => { | ||||||
| 			const path = request.params.path; | 			const path = request.params.path; | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue