joplin/ReactNativeClient/lib/shim-init-node.js

const fs = require("fs-extra");
const { shim } = require("lib/shim.js");
const { GeolocationNode } = require("lib/geolocation-node.js");
const { FileApiDriverLocal } = require("lib/file-api-driver-local.js");
const { time } = require("lib/time-utils.js");
const { setLocale, defaultLocale, closestSupportedLocale } = require("lib/locale.js");
const { FsDriverNode } = require("lib/fs-driver-node.js");

function shimInit() {
	shim.fsDriver = () => {
		throw new Error("Not implemented");
	};
	shim.FileApiDriverLocal = FileApiDriverLocal;
	shim.Geolocation = GeolocationNode;
	shim.FormData = require("form-data");
	shim.sjclModule = require("lib/vendor/sjcl.js");

	shim.fsDriver = () => {
		if (!shim.fsDriver_) shim.fsDriver_ = new FsDriverNode();
		return shim.fsDriver_;
	};

	shim.randomBytes = async count => {
		const buffer = require("crypto").randomBytes(count);
		return Array.from(buffer);
	};

	shim.detectAndSetLocale = function(Setting) {
		let locale = process.env.LANG;
		if (!locale) locale = defaultLocale();
		locale = locale.split(".");
		locale = locale[0];
		locale = closestSupportedLocale(locale);
		Setting.setValue("locale", locale);
		setLocale(locale);
		return locale;
	};

	const resizeImage_ = async function(filePath, targetPath) {
		const sharp = require("sharp");
		const Resource = require("lib/models/Resource.js");

		return new Promise((resolve, reject) => {
			sharp(filePath)
				.resize(Resource.IMAGE_MAX_DIMENSION, Resource.IMAGE_MAX_DIMENSION)
				.max()
				.withoutEnlargement()
				.toFile(targetPath, (err, info) => {
					if (err) {
						reject(err);
					} else {
						resolve(info);
					}
				});
		});
	};

	shim.attachFileToNote = async function(note, filePath) {
		const Resource = require("lib/models/Resource.js");
		const { uuid } = require("lib/uuid.js");
		const { basename, fileExtension, safeFileExtension } = require("lib/path-utils.js");
		const mime = require("mime/lite");
		const Note = require("lib/models/Note.js");

		if (!await fs.pathExists(filePath)) throw new Error(_("Cannot access %s", filePath));

		let resource = Resource.new();
		resource.id = uuid.create();
		resource.mime = mime.getType(filePath);
		resource.title = basename(filePath);
		resource.file_extension = safeFileExtension(fileExtension(filePath));

		if (!resource.mime) resource.mime = "application/octet-stream";

		let targetPath = Resource.fullPath(resource);

		if (resource.mime == "image/jpeg" || resource.mime == "image/jpg" || resource.mime == "image/png") {
			const result = await resizeImage_(filePath, targetPath);
		} else {
			await fs.copy(filePath, targetPath, { overwrite: true });
		}

		await Resource.save(resource, { isNew: true });

		const newBody = [];
		if (note.body) newBody.push(note.body);
		newBody.push(Resource.markdownTag(resource));

		const newNote = Object.assign({}, note, {
			body: newBody.join("\n\n"),
		});
		return await Note.save(newNote);
	};

	const nodeFetch = require("node-fetch");

	shim.readLocalFileBase64 = path => {
		const data = fs.readFileSync(path);
		return new Buffer(data).toString("base64");
	};

	shim.fetch = async function(url, options = null) {
		return shim.fetchWithRetry(() => {
			return nodeFetch(url, options);
		}, options);
	};

	shim.fetchBlob = async function(url, options) {
		if (!options || !options.path) throw new Error("fetchBlob: target file path is missing");
		if (!options.method) options.method = "GET";
		//if (!('maxRetry' in options)) options.maxRetry = 5;

		const urlParse = require("url").parse;

		url = urlParse(url.trim());
		const http = url.protocol.toLowerCase() == "http:" ? require("follow-redirects").http : require("follow-redirects").https;
		const headers = options.headers ? options.headers : {};
		const method = options.method ? options.method : "GET";
		if (method != "GET") throw new Error("Only GET is supported");
		const filePath = options.path;

		function makeResponse(response) {
			return {
				ok: response.statusCode < 400,
				path: filePath,
				text: () => {
					return response.statusMessage;
				},
				json: () => {
					return { message: response.statusCode + ": " + response.statusMessage };
				},
				status: response.statusCode,
				headers: response.headers,
			};
		}

		const requestOptions = {
			protocol: url.protocol,
			host: url.host,
			port: url.port,
			method: method,
			path: url.path + (url.query ? "?" + url.query : ""),
			headers: headers,
		};

		const doFetchOperation = async () => {
			return new Promise((resolve, reject) => {
				try {
					// Note: relative paths aren't supported
					const file = fs.createWriteStream(filePath);

					const request = http.get(requestOptions, function(response) {
						response.pipe(file);

						file.on("finish", function() {
							file.close(() => {
								resolve(makeResponse(response));
							});
						});
					});

					request.on("error", function(error) {
						fs.unlink(filePath);
						reject(error);
					});
				} catch (error) {
					fs.unlink(filePath);
					reject(error);
				}
			});
		};

		return shim.fetchWithRetry(doFetchOperation, options);
	};

	shim.uploadBlob = async function(url, options) {
		if (!options || !options.path) throw new Error("uploadBlob: source file path is missing");
		const content = await fs.readFile(options.path);
		options = Object.assign({}, options, {
			body: content,
		});
		return shim.fetch(url, options);
	};

	shim.stringByteLength = function(string) {
		return Buffer.byteLength(string, "utf-8");
	};
}

module.exports = { shimInit };