@mieweb/wikigdrive
Version:
Google Drive to MarkDown synchronization
87 lines (86 loc) • 2.98 kB
JavaScript
import { MimeTypes } from '../../model/GoogleFile.js';
import { GoogleFilesScanner } from '../transform/GoogleFilesScanner.js';
export class GoogleTreeProcessor {
constructor(filesService) {
Object.defineProperty(this, "filesService", {
enumerable: true,
configurable: true,
writable: true,
value: filesService
});
Object.defineProperty(this, "driveTree", {
enumerable: true,
configurable: true,
writable: true,
value: []
});
}
async load() {
this.driveTree = await this.filesService.readJson('.tree.json') || [];
}
async save() {
await this.filesService.writeJson('.tree.json', this.driveTree);
}
async regenerateTree() {
this.driveTree = await this.internalRegenerateTree(this.filesService);
}
async internalRegenerateTree(filesService, parentId) {
const scanner = new GoogleFilesScanner();
const files = await scanner.scan(filesService);
const retVal = [];
for (const file of files) {
if (file.mimeType === MimeTypes.FOLDER_MIME) {
const subFileService = await filesService.getSubFileService(file.id);
const item = {
id: file.id,
name: file.name,
mimeType: file.mimeType,
modifiedTime: file.modifiedTime,
version: file.version,
parentId,
children: await this.internalRegenerateTree(subFileService, file.id)
};
retVal.push(item);
}
else {
const item = {
id: file.id,
name: file.name,
mimeType: file.mimeType,
modifiedTime: file.modifiedTime,
version: file.version,
parentId
};
retVal.push(item);
}
}
return retVal;
}
getTree() {
return this.driveTree;
}
async findById(fileId) {
return await this.findInTree(item => item.id === fileId, this.driveTree);
}
async findInTree(callBack, children, curPath = '') {
for (const file of children) {
const part = file['id'];
if (callBack(file)) {
return [file, curPath ? curPath + '/' + part : part];
}
}
for (const file of children) {
if (file.mimeType !== MimeTypes.FOLDER_MIME) {
continue;
}
if (file.children) {
const part = file['id'];
const tuple = await this.findInTree(callBack, file.children, curPath ? curPath + '/' + part : part);
if (tuple?.length > 0) {
return tuple;
}
}
}
return [];
}
}