UNPKG

@mieweb/wikigdrive

Version:

Google Drive to MarkDown synchronization

87 lines (86 loc) 2.98 kB
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 []; } }