joplin/ReactNativeClient/lib/file-api-driver-memory.js

const { time } = require('lib/time-utils.js');
const fs = require('fs-extra');

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('ascii');
	}

	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);
	}

	setTimestamp(path, timestampMs) {
		let item = this.itemByPath(path);
		if (!item) return Promise.reject(new Error('File not found: ' + path));
		item.updated_time = timestampMs;
		return Promise.resolve();
	}

	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;
	}

	mkdir(path) {
		let index = this.itemIndexByPath(path);
		if (index >= 0) return Promise.resolve();
		this.items_.push(this.newItem(path, true));
		return Promise.resolve();
	}

	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();
		}
		return Promise.resolve();
	}

	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);
		}
		return Promise.resolve();
	}

	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;
		return Promise.resolve();
	}

	format() {
		this.items_ = [];
		return Promise.resolve();
	}

	async delta(path, options = null) {
		let limit = 3;

		let output = {
			hasMore: false,
			context: {},
			items: [],
		};

		let context = options ? options.context : null;
		let fromTime = 0;

		if (context) fromTime = context.fromTime;

		let sortedItems = this.items_.slice().concat(this.deletedItems_);
		sortedItems.sort((a, b) => {
			if (a.updated_time < b.updated_time) return -1;
			if (a.updated_time > b.updated_time) return +1;
			return 0;
		});

		let hasMore = false;
		let items = [];
		let maxTime = 0;
		for (let i = 0; i < sortedItems.length; i++) {
			let item = sortedItems[i];
			if (item.updated_time >= fromTime) {
				item = Object.assign({}, item);
				item.path = item.path.substr(path.length + 1);
				items.push(item);
				if (item.updated_time > maxTime) maxTime = item.updated_time;
			}

			if (items.length >= limit) {
				hasMore = true;
				break;
			}
		}

		output.items = items;
		output.hasMore = hasMore;
		output.context = { fromTime: maxTime };

		return output;
	}

}

module.exports = { FileApiDriverMemory };