refactoring

Resolve #7779

packages/backend/src/server/api/endpoints/drive/folders/update.ts

@@ -0,0 +1,123 @@
+import $ from 'cafy';
+import { ID } from '@/misc/cafy-id';
+import { publishDriveStream } from '@/services/stream';
+import define from '../../../define';
+import { ApiError } from '../../../error';
+import { DriveFolders } from '@/models/index';
+
+export const meta = {
+	tags: ['drive'],
+
+	requireCredential: true as const,
+
+	kind: 'write:drive',
+
+	params: {
+		folderId: {
+			validator: $.type(ID),
+		},
+
+		name: {
+			validator: $.optional.str.pipe(DriveFolders.validateFolderName),
+		},
+
+		parentId: {
+			validator: $.optional.nullable.type(ID),
+		}
+	},
+
+	errors: {
+		noSuchFolder: {
+			message: 'No such folder.',
+			code: 'NO_SUCH_FOLDER',
+			id: 'f7974dac-2c0d-4a27-926e-23583b28e98e'
+		},
+
+		noSuchParentFolder: {
+			message: 'No such parent folder.',
+			code: 'NO_SUCH_PARENT_FOLDER',
+			id: 'ce104e3a-faaf-49d5-b459-10ff0cbbcaa1'
+		},
+
+		recursiveNesting: {
+			message: 'It can not be structured like nesting folders recursively.',
+			code: 'NO_SUCH_PARENT_FOLDER',
+			id: 'ce104e3a-faaf-49d5-b459-10ff0cbbcaa1'
+		},
+	},
+
+	res: {
+		type: 'object' as const,
+		optional: false as const, nullable: false as const,
+		ref: 'DriveFolder'
+	}
+};
+
+export default define(meta, async (ps, user) => {
+	// Fetch folder
+	const folder = await DriveFolders.findOne({
+		id: ps.folderId,
+		userId: user.id
+	});
+
+	if (folder == null) {
+		throw new ApiError(meta.errors.noSuchFolder);
+	}
+
+	if (ps.name) folder.name = ps.name;
+
+	if (ps.parentId !== undefined) {
+		if (ps.parentId === folder.id) {
+			throw new ApiError(meta.errors.recursiveNesting);
+		} else if (ps.parentId === null) {
+			folder.parentId = null;
+		} else {
+			// Get parent folder
+			const parent = await DriveFolders.findOne({
+				id: ps.parentId,
+				userId: user.id
+			});
+
+			if (parent == null) {
+				throw new ApiError(meta.errors.noSuchParentFolder);
+			}
+
+			// Check if the circular reference will occur
+			async function checkCircle(folderId: any): Promise<boolean> {
+				// Fetch folder
+				const folder2 = await DriveFolders.findOne({
+					id: folderId
+				});
+
+				if (folder2!.id === folder!.id) {
+					return true;
+				} else if (folder2!.parentId) {
+					return await checkCircle(folder2!.parentId);
+				} else {
+					return false;
+				}
+			}
+
+			if (parent.parentId !== null) {
+				if (await checkCircle(parent.parentId)) {
+					throw new ApiError(meta.errors.recursiveNesting);
+				}
+			}
+
+			folder.parentId = parent.id;
+		}
+	}
+
+	// Update
+	DriveFolders.update(folder.id, {
+		name: folder.name,
+		parentId: folder.parentId
+	});
+
+	const folderObj = await DriveFolders.pack(folder);
+
+	// Publish folderUpdated event
+	publishDriveStream(user.id, 'folderUpdated', folderObj);
+
+	return folderObj;
+});