more robust recursive folder check
Some checks failed
Lint / pnpm_install (pull_request) Has been cancelled
Lint / lint (backend) (pull_request) Has been cancelled
Lint / lint (frontend) (pull_request) Has been cancelled
Lint / lint (frontend-embed) (pull_request) Has been cancelled
Lint / lint (frontend-shared) (pull_request) Has been cancelled
Lint / lint (misskey-bubble-game) (pull_request) Has been cancelled
Lint / lint (misskey-js) (pull_request) Has been cancelled
Lint / lint (misskey-reversi) (pull_request) Has been cancelled
Lint / lint (sw) (pull_request) Has been cancelled
Lint / typecheck (backend) (pull_request) Has been cancelled
Lint / typecheck (misskey-js) (pull_request) Has been cancelled
Lint / typecheck (sw) (pull_request) Has been cancelled
Publish Docker image / Build (pull_request) Has been cancelled
Test (backend) / e2e (20.16.0) (pull_request) Has been cancelled
Test (backend) / unit (20.16.0) (pull_request) Has been cancelled
Test (production install and build) / production (20.16.0) (pull_request) Has been cancelled

Signed-off-by: eternal-flame-AD <yume@yumechi.jp>
This commit is contained in:
ゆめ 2024-11-10 17:23:26 -06:00
parent 7a106a390d
commit 437e4e0b29
No known key found for this signature in database
3 changed files with 15 additions and 9 deletions

View file

@ -30,7 +30,8 @@ export class DriveFolderEntityService {
public async pack(
src: MiDriveFolder['id'] | MiDriveFolder,
options?: {
detail: boolean
detail: boolean,
maxDepth: number,
},
): Promise<Packed<'DriveFolder'>> {
const opts = Object.assign({
@ -55,7 +56,8 @@ export class DriveFolderEntityService {
...(folder.parentId ? {
parent: this.pack(folder.parentId, {
detail: true,
detail: (options?.maxDepth ? options.maxDepth > 0 : true),
maxDepth: options?.maxDepth || 32,
}),
} : {}),
} : {}),

View file

@ -32,7 +32,7 @@ export const meta = {
},
recursiveNesting: {
message: 'It can not be structured like nesting folders recursively.',
message: 'Folders are linked recursively or too deeply.',
code: 'RECURSIVE_NESTING',
id: 'dbeb024837894013aed44279f9199740',
},
@ -94,7 +94,10 @@ export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-
}
// Check if the circular reference will occur
const checkCircle = async (folderId: string): Promise<boolean> => {
const checkCircle = async (folderId: string, limit: number = 32): Promise<boolean> => {
if (limit <= 0) {
return false;
}
const folder2 = await this.driveFoldersRepository.findOneByOrFail({
id: folderId,
});
@ -102,7 +105,7 @@ export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-
if (folder2.id === folder.id) {
return true;
} else if (folder2.parentId) {
return await checkCircle(folder2.parentId);
return await checkCircle(folder2.parentId, limit - 1);
} else {
return false;
}

View file

@ -927,14 +927,15 @@ describe('Endpoints', () => {
const folderC = (await api('drive/folders/create', {
name: 'test',
}, alice)).body;
await api('drive/folders/update', {
assert.ok(folderA.id && folderB.id && folderC.id);
assert.strictEqual((await api('drive/folders/update', {
folderId: folderB.id,
parentId: folderA.id,
}, alice);
await api('drive/folders/update', {
}, alice)).status, 200);
assert.strictEqual((await api('drive/folders/update', {
folderId: folderC.id,
parentId: folderB.id,
}, alice);
}, alice)).status, 200);
const res = await api('drive/folders/update', {
folderId: folderA.id,