feat: allow using the generated ID in the web viewer (closes #181)

tycrek · Dec 25, 2022 · fe79d84 · fe79d84
1 parent dcd93b6
commit fe79d84
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/src/routers/resource.ts b/src/routers/resource.ts
@@ -8,7 +8,7 @@ import { Request, Response } from 'express';
 import { deleteS3 } from '../storage';
 import { SkynetDelete, SkynetDownload } from '../skynet';
 import { path, log, getTrueHttp, getTrueDomain, formatBytes, formatTimestamp, getS3url, getDirectUrl, getResourceColor, replaceholder } from '../utils';
-const { diskFilePath, s3enabled, viewDirect, useSia }: Config = fs.readJsonSync(path('config.json'));
+const { diskFilePath, s3enabled, viewDirect, useIdInViewer, idInViewerExtension, useSia }: Config = fs.readJsonSync(path('config.json'));
 const { CODE_UNAUTHORIZED, CODE_NOT_FOUND, }: MagicNumbers = fs.readJsonSync(path('MagicNumbers.json'));
 import { data } from '../data';
 import { users } from '../auth';
@@ -47,7 +47,7 @@ router.get('/', (req: Request, res: Response, next) => data().get(req.ass.resour
 	// Send the view to the client
 	res.render('view', {
 		fileIs: fileData.is,
-		title: escape(fileData.originalname),
+		title: useIdInViewer ? `${resourceId}${idInViewerExtension ? `${fileData.ext}` : ''}` : escape(fileData.originalname),
 		mimetype: fileData.mimetype,
 		uploader: users.find(user => user.unid === fileData.uploader)?.username || 'Unknown',
 		timestamp: formatTimestamp(fileData.timestamp, fileData.timeoffset),

diff --git a/src/setup.js b/src/setup.js
@@ -12,6 +12,8 @@ const config = {
 	spaceReplace: '_',
 	mediaStrict: false,
 	viewDirect: false,
+	useIdInViewer: false,
+	idInViewerExtension: false,
 	dataEngine: '@tycrek/papito',
 	frontendName: 'ass-x',
 	useSia: false,
@@ -168,6 +170,18 @@ function doSetup() {
 				default: config.viewDirect,
 				required: false
 			},
+			useIdInViewer: {
+				description: 'Use the ID in the web viewer instead of the filename',
+				type: 'boolean',
+				default: config.useIdInViewer,
+				required: false
+			},
+			idInViewerExtension: {
+				description: '(Only applies if "useIdInViewer" is true) Include the file extension in the ID in the web viewer',
+				type: 'boolean',
+				default: config.idInViewerExtension,
+				required: false
+			},
 			dataEngine: {
 				description: 'Data engine to use (must match an npm package name. If unsure, leave blank)',
 				type: 'string',

diff --git a/src/types/json.d.ts b/src/types/json.d.ts
@@ -12,6 +12,8 @@ declare module 'ass-json' {
 		gfyIdSize: number
 		mediaStrict: boolean
 		viewDirect: boolean
+		useIdInViewer: boolean
+		idInViewerExtension: boolean
 		dataEngine: string
 		frontendName: string
 		indexFile: string