Qt/QML edition

dependencies/drivelist/lib/lsblk/index.ts

@@ -0,0 +1,93 @@
+/*
+ * Copyright 2018 Balena.io
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { execFile } from 'mz/child_process';
+import { readdir, readlink } from 'mz/fs';
+import { join, resolve } from 'path';
+
+import { Drive } from '..';
+import { parse as parseJSON } from './json';
+import { parse as parsePairs } from './pairs';
+
+const DISK_PATH_DIR = '/dev/disk/by-path/';
+
+let SUPPORTS_JSON = true;
+
+async function getDevicePaths(): Promise<Map<string, string>> {
+	const mapping = new Map();
+	for (const filename of await readdir(DISK_PATH_DIR)) {
+		const linkPath = join(DISK_PATH_DIR, filename);
+		let link: string;
+		try {
+			link = await readlink(linkPath);
+		} catch (error) {
+			continue;
+		}
+		const devicePath = resolve(DISK_PATH_DIR, link);
+		mapping.set(devicePath, linkPath);
+	}
+	return mapping;
+}
+
+async function addDevicePaths(devices: Drive[]): Promise<void> {
+	const devicePaths = await getDevicePaths();
+	for (const device of devices) {
+		device.devicePath = devicePaths.get(device.device) || null;
+	}
+}
+
+async function getOutput(command: string, ...args: string[]) {
+	const [stdout] = await execFile(command, args);
+	return stdout;
+}
+
+async function lsblkJSON(): Promise<Drive[]> {
+	return parseJSON(
+		await getOutput(
+			'lsblk',
+			'--bytes',
+			'--all',
+			'--json',
+			'--paths',
+			'--output-all',
+		),
+	);
+}
+
+async function lsblkPairs(): Promise<Drive[]> {
+	return parsePairs(await getOutput('lsblk', '--bytes', '--all', '--pairs'));
+}
+
+async function $lsblk(): Promise<Drive[]> {
+	if (SUPPORTS_JSON) {
+		try {
+			return await lsblkJSON();
+		} catch (error) {
+			SUPPORTS_JSON = false;
+		}
+	}
+	return await lsblkPairs();
+}
+
+export async function lsblk(): Promise<Drive[]> {
+	const drives = await $lsblk();
+	try {
+		await addDevicePaths(drives);
+	} catch (error) {
+		// Couldn't add device paths
+	}
+	return drives;
+}