summaryrefslogtreecommitdiff
path: root/src/models/drive-folder.ts
blob: def519fade1d9a8f38f33d66a13593817f4b5678 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import * as mongo from 'mongodb';
const deepcopy = require('deepcopy');
import db from '../db/mongodb';
import DriveFile from './drive-file';

const DriveFolder = db.get<IDriveFolder>('driveFolders');
export default DriveFolder;

export type IDriveFolder = {
	_id: mongo.ObjectID;
	createdAt: Date;
	name: string;
	userId: mongo.ObjectID;
	parentId: mongo.ObjectID;
};

export function isValidFolderName(name: string): boolean {
	return (
		(name.trim().length > 0) &&
		(name.length <= 200)
	);
}

/**
 * DriveFolderを物理削除します
 */
export async function deleteDriveFolder(driveFolder: string | mongo.ObjectID | IDriveFolder) {
	let d: IDriveFolder;

	// Populate
	if (mongo.ObjectID.prototype.isPrototypeOf(driveFolder)) {
		d = await DriveFolder.findOne({
			_id: driveFolder
		});
	} else if (typeof driveFolder === 'string') {
		d = await DriveFolder.findOne({
			_id: new mongo.ObjectID(driveFolder)
		});
	} else {
		d = driveFolder as IDriveFolder;
	}

	if (d == null) return;

	// このDriveFolderに格納されているDriveFileがあればすべてルートに移動
	await DriveFile.update({
		'metadata.folderId': d._id
	}, {
		$set: {
			'metadata.folderId': null
		}
	});

	// このDriveFolderに格納されているDriveFolderがあればすべてルートに移動
	await DriveFolder.update({
		parentId: d._id
	}, {
		$set: {
			parentId: null
		}
	});

	// このDriveFolderを削除
	await DriveFolder.remove({
		_id: d._id
	});
}

/**
 * Pack a drive folder for API response
 */
export const pack = (
	folder: any,
	options?: {
		detail: boolean
	}
) => new Promise<any>(async (resolve, reject) => {
	const opts = Object.assign({
		detail: false
	}, options);

	let _folder: any;

	// Populate the folder if 'folder' is ID
	if (mongo.ObjectID.prototype.isPrototypeOf(folder)) {
		_folder = await DriveFolder.findOne({ _id: folder });
	} else if (typeof folder === 'string') {
		_folder = await DriveFolder.findOne({ _id: new mongo.ObjectID(folder) });
	} else {
		_folder = deepcopy(folder);
	}

	// Rename _id to id
	_folder.id = _folder._id;
	delete _folder._id;

	if (opts.detail) {
		const childFoldersCount = await DriveFolder.count({
			parentId: _folder.id
		});

		const childFilesCount = await DriveFile.count({
			'metadata.folderId': _folder.id
		});

		_folder.foldersCount = childFoldersCount;
		_folder.filesCount = childFilesCount;
	}

	if (opts.detail && _folder.parentId) {
		// Populate parent folder
		_folder.parent = await pack(_folder.parentId, {
			detail: true
		});
	}

	resolve(_folder);
});