snapshot of https://github.com/docmost/docmost.git @ 6205bbeb908fe846f87dd6a2cbf562e24777db38 (2026-09-15)

apps/server/src/core/attachment/attachment.utils.ts

@@ -0,0 +1,81 @@
+import { MultipartFile } from '@fastify/multipart';
+import * as path from 'path';
+import { AttachmentType } from './attachment.constants';
+import { sanitizeFileName } from '../../common/helpers';
+import { getMimeType } from '../../common/helpers';
+
+export interface PreparedFile {
+  buffer?: Buffer;
+  fileName: string;
+  fileSize: number;
+  fileExtension: string;
+  mimeType: string;
+  multiPartFile?: MultipartFile;
+}
+
+export async function prepareFile(
+  filePromise: Promise<MultipartFile>,
+  options: { skipBuffer?: boolean } = {},
+): Promise<PreparedFile> {
+  const file = await filePromise;
+
+  if (!file) {
+    throw new Error('No file provided');
+  }
+
+  try {
+    let buffer: Buffer | undefined;
+    let fileSize = 0;
+
+    if (!options.skipBuffer) {
+      buffer = await file.toBuffer();
+      fileSize = buffer.length;
+    }
+
+    const sanitizedFilename = sanitizeFileName(file.filename);
+    const fileName = sanitizedFilename.slice(0, 255);
+    const fileExtension = path.extname(file.filename).toLowerCase();
+
+    return {
+      buffer,
+      fileName,
+      fileSize,
+      fileExtension,
+      mimeType: getMimeType(file.filename),
+      multiPartFile: file,
+    };
+  } catch (error) {
+    throw error;
+  }
+}
+
+export function validateFileType(
+  fileExtension: string,
+  allowedTypes: string[],
+) {
+  if (!allowedTypes.includes(fileExtension)) {
+    throw new Error('Invalid file type');
+  }
+}
+
+export function getAttachmentFolderPath(
+  type: AttachmentType,
+  workspaceId: string,
+): string {
+  switch (type) {
+    case AttachmentType.Avatar:
+      return `${workspaceId}/avatars`;
+    case AttachmentType.WorkspaceIcon:
+      return `${workspaceId}/workspace-logos`;
+    case AttachmentType.SpaceIcon:
+      return `${workspaceId}/space-logos`;
+    case AttachmentType.File:
+      return `${workspaceId}/files`;
+    case AttachmentType.Chat:
+      return `${workspaceId}/chat-files`;
+    default:
+      return `${workspaceId}/files`;
+  }
+}
+
+export const validAttachmentTypes = Object.values(AttachmentType);