const { time } = require('lib/time-utils.js'); const fs = require('fs-extra'); const { basicDelta } = require('lib/file-api'); class FileApiDriverMemory { constructor() { this.items_ = []; this.deletedItems_ = []; } encodeContent_(content) { if (content instanceof Buffer) { return content.toString('base64'); } else { return Buffer.from(content).toString('base64'); } } decodeContent_(content) { return Buffer.from(content, 'base64').toString('utf-8'); } itemIndexByPath(path) { for (let i = 0; i < this.items_.length; i++) { if (this.items_[i].path == path) return i; } return -1; } itemByPath(path) { let index = this.itemIndexByPath(path); return index < 0 ? null : this.items_[index]; } newItem(path, isDir = false) { let now = time.unixMs(); return { path: path, isDir: isDir, updated_time: now, // In milliseconds!! // created_time: now, // In milliseconds!! content: '', }; } stat(path) { let item = this.itemByPath(path); return Promise.resolve(item ? Object.assign({}, item) : null); } async setTimestamp(path, timestampMs) { let item = this.itemByPath(path); if (!item) return Promise.reject(new Error('File not found: ' + path)); item.updated_time = timestampMs; } async list(path, options) { let output = []; for (let i = 0; i < this.items_.length; i++) { let item = this.items_[i]; if (item.path == path) continue; if (item.path.indexOf(path + '/') === 0) { let s = item.path.substr(path.length + 1); if (s.split('/').length === 1) { let it = Object.assign({}, item); it.path = it.path.substr(path.length + 1); output.push(it); } } } return Promise.resolve({ items: output, hasMore: false, context: null, }); } async get(path, options) { let item = this.itemByPath(path); if (!item) return Promise.resolve(null); if (item.isDir) return Promise.reject(new Error(path + ' is a directory, not a file')); let output = null; if (options.target === 'file') { await fs.writeFile(options.path, Buffer.from(item.content, 'base64')); } else { const content = this.decodeContent_(item.content); output = Promise.resolve(content); } return output; } async mkdir(path) { let index = this.itemIndexByPath(path); if (index >= 0) return; this.items_.push(this.newItem(path, true)); } async put(path, content, options = null) { if (!options) options = {}; if (options.source === 'file') content = await fs.readFile(options.path); let index = this.itemIndexByPath(path); if (index < 0) { let item = this.newItem(path, false); item.content = this.encodeContent_(content); this.items_.push(item); } else { this.items_[index].content = this.encodeContent_(content); this.items_[index].updated_time = time.unix(); } } async delete(path) { let index = this.itemIndexByPath(path); if (index >= 0) { let item = Object.assign({}, this.items_[index]); item.isDeleted = true; item.updated_time = time.unixMs(); this.deletedItems_.push(item); this.items_.splice(index, 1); } } async move(oldPath, newPath) { let sourceItem = this.itemByPath(oldPath); if (!sourceItem) return Promise.reject(new Error('Path not found: ' + oldPath)); this.delete(newPath); // Overwrite if newPath already exists sourceItem.path = newPath; } async format() { this.items_ = []; } async delta(path, options = null) { const getStatFn = async path => { let output = this.items_.slice(); for (let i = 0; i < output.length; i++) { const item = Object.assign({}, output[i]); item.path = item.path.substr(path.length + 1); output[i] = item; } return output; }; const output = await basicDelta(path, getStatFn, options); return output; } async clearRoot() { this.items_ = []; } } module.exports = { FileApiDriverMemory };