PLG-MuDiCS/control/frontend/src/lib/ts/api_handler.ts

import { notifications } from './stores/notification';
import {
	to_display_status,
	type DisplayStatus,
	type Inode,
	type RequestResponse,
	type ShellCommandResponse,
	type TreeElement
} from './types';
import { dev } from '$app/environment';
import { get_sanitized_file_url } from './utils';

export async function get_screenshot(ip: string): Promise<Blob | null> {
	const options = { method: 'PATCH' };
	const response = await request_display(ip, '/takeScreenshot', options);
	if (!response.ok || !response.blob) return null;
	return response.blob;
}

export async function open_file(ip: string, path_to_file: string): Promise<void> {
	const options = { method: 'PATCH', headers: { 'content-type': 'application/octet-stream' } };
	await request_display(ip, get_sanitized_file_url(path_to_file), options);
}

export async function send_keyboard_input(
	ip: string,
	inputs: { key: string; action: 'press' | 'release' }[]
): Promise<void> {
	const options = {
		method: 'PATCH',
		headers: { 'content-type': 'application/json' },
		body: JSON.stringify({
			inputs
		})
	};
	await request_display(ip, '/keyboardInput', options);
}

export async function show_html(ip: string, html: string) {
	const options = {
		method: 'PATCH',
		headers: { 'content-type': 'application/json' },
		body: JSON.stringify({
			html: html
		})
	};
	await request_display(ip, '/showHTML', options);
}

export async function get_file_data(
	ip: string,
	path: string
): Promise<{ folder_element: Inode; date_created: Date }[] | null> {
	interface FileInfo {
		name: string;
		type: string;
		size: string;
		created: string;
	}
	const command = `cd ".${path}" && find . -maxdepth 1 -mindepth 1 -print0 | while IFS= read -r -d '' f; do
        typ=$(file -b --mime-type -- "$f")
        size=$(stat -c '%s' -- "$f")
        created=$(stat -c '%w' -- "$f")
        [ "$created" = "-" ] && created=$(stat -c '%y' -- "$f")
        jq -n --arg name "$f" --arg type "$typ" --arg size "$size" --arg created "$created" \
            '{name:$name, type:$type, size:($size|tostring), created:$created}' | tr -d '\n'
        echo
        done
    `;
	const raw_response = await run_shell_command(ip, command);
	if (!raw_response.ok || !raw_response.json) return null;
	const json_response = raw_response.json as ShellCommandResponse;
	if (json_response.exitCode === 0 && json_response.stdout.trim() === '') return [];
	if (handle_shell_error(ip, json_response, command, true)) return null;
	if (json_response.stdout.trim() === '') return null;

	const response: FileInfo[] = json_response.stdout
		.trim()
		.split('\n')
		.filter(Boolean)
		.map((line: string) => JSON.parse(line) as FileInfo);

	const folder_element_list: { folder_element: Inode; date_created: Date }[] = [];

	for (const response_element of response) {
		// filter hidden files (start with '.' -> './.config')
		if (response_element.name.charAt(2) === '.') continue;
		const folder_element: Inode = {
			path: path,
			name: response_element.name.slice(2), // remove "./"
			type: response_element.type,
			size: Number(response_element.size),
			thumbnail: null
		};
		folder_element_list.push({ folder_element, date_created: new Date(response_element.created) });
	}
	return folder_element_list;
}

export async function get_file_tree_data(ip: string, path: string): Promise<TreeElement[] | null> {
	const command = `cd ".${path}" && tree -Js`;
	const raw_response = await run_shell_command(ip, command);

	if (!raw_response.ok || !raw_response.json) return null;
	const json_response = raw_response.json as ShellCommandResponse;
	if (handle_shell_error(ip, json_response, command, true)) return null;

	const tree_element: TreeElement | null = JSON.parse(json_response.stdout.trim())[0] || null;

	return tree_element?.contents || null;
}

export async function create_path(
	ip: string,
	path: string,
): Promise<void> {
	const command = `mkdir -p ".${path}"`

	const raw_response = await run_shell_command(ip, command);
	if (!raw_response.ok || !raw_response.json) return;
	const json_response = raw_response.json as ShellCommandResponse;
	handle_shell_error(ip, json_response, command, true);
}

export async function rename_file(
	ip: string,
	path: string,
	old_file_name: string,
	new_file_name: string
): Promise<void> {
	const command: string = `cd ".${path}" && mv "${old_file_name}" "${new_file_name}"`;

	const raw_response = await run_shell_command(ip, command);
	if (!raw_response.ok || !raw_response.json) return;
	const json_response = raw_response.json as ShellCommandResponse;
	handle_shell_error(ip, json_response, command, true);
}

export async function delete_files(
	ip: string,
	current_path: string,
	file_names: string[]
): Promise<void> {
	let command: string = `cd ".${current_path}"`;
	for (const file_name of file_names) {
		command += ` && rm -r "${file_name}"`;
	}
	const raw_response = await run_shell_command(ip, command);
	if (!raw_response.ok || !raw_response.json) return;
	const json_response = raw_response.json as ShellCommandResponse;
	handle_shell_error(ip, json_response, command, true);
}

export async function show_blackscreen(ip: string): Promise<void> {
	const options = {
		method: 'PATCH',
		headers: { 'content-type': 'application/json' },
		body: JSON.stringify({
			html: '<p></p>'
		})
	};
	await request_display(ip, '/showHTML', options);
}

export async function get_thumbnail_blob(ip: string, path_to_file: string): Promise<Blob | null> {
	const raw_response = await request_display(
		ip,
		get_sanitized_file_url(path_to_file, true),
		{ method: 'GET' },
		[415]
	);
	if (!raw_response.ok || !raw_response.blob) return null;
	return raw_response.blob;
}

export async function ping_ip(ip: string): Promise<DisplayStatus> {
	const raw_response = await request_control(`/ping?ip=${ip}`, { method: 'GET' });
	if (!raw_response.ok || !raw_response.json) return null;

	const status = raw_response.json.status;
	if (typeof status === 'string') {
		return to_display_status(status);
	}
	return null;
}

async function request_display(
	ip: string,
	api_route: string,
	options: { method: string; headers?: Record<string, string>; body?: string },
	supress_error_handling_http_codes: number[] = []
): Promise<RequestResponse> {
	const url = `http://${ip}:1323/api${api_route}`;
	return await request(url, options, supress_error_handling_http_codes);
}

async function request_control(
	api_route: string,
	options: { method: string; headers?: Record<string, string>; body?: string }
): Promise<RequestResponse> {
	const url = `http://127.0.0.1:8080/api${api_route}`;
	return await request(url, options);
}

async function request(
	url: string,
	options: { method: string; headers?: Record<string, string>; body?: string },
	supress_error_handling_http_codes: number[] = []
): Promise<RequestResponse> {
	try {
		const cache_buster = `${url.includes('?') ? '&' : '?'}=${Date.now()}`;
		if (dev) {
			console.debug('Sending request: ', url + cache_buster, 'with', options.body ?? 'none');
		}
		const response = await fetch(url + cache_buster, options);
		if (response.ok || supress_error_handling_http_codes.includes(response.status)) {
			const contentType = response.headers.get('content-type') || '';
			let request_response: RequestResponse;
			if (!contentType.includes('application/json')) {
				const blob: Blob = await response.blob();
				request_response = { ok: response.ok, http_code: response.status, blob: blob };
			} else {
				const json: Record<string, unknown> = await response.json();
				request_response = { ok: response.ok, http_code: response.status, json: json };
				if (dev) {
					console.debug(request_response);
				}
			}
			return request_response;
		}

		let error_description = url;
		if (response.headers.get('content-type')?.includes('application/json')) {
			try {
				const json: { description: string } = await response.json();
				error_description += '\n' + json.description;
			} catch {
				error_description += '\nCould not parse error description';
			}
		}
		notifications.push('error', `Fehler ${response.status} bei API-Anfrage`, error_description);
	} catch (error: unknown) {
		if (error instanceof TypeError && /fetch|NetworkError/i.test(error.message)) {
			if (dev) {
				console.warn('Request failed - Is the targeted device online?');
			}
		} else {
			console.error(url, error);
			notifications.push('error', `Fataler Fehler bei API-Anfrage`, `${url}\nFehler: ${error}`);
		}
	}
	return { ok: false };
}

function handle_shell_error(
	ip: string,
	shell_response: ShellCommandResponse,
	shell_command: string,
	command_includs_cd: boolean
): boolean {
	if (shell_response.exitCode !== 0) {
		if (
			command_includs_cd &&
			shell_response.stderr &&
			/bash: line \d+: cd: .+: No such file or directory/.test(shell_response.stderr)
		) {
			if (dev) {
				console.debug('current file_path does not exist on display:', ip);
			}
			return true;
		}
		console.error(shell_response);
		notifications.push(
			'error',
			`Fehler ${shell_response.exitCode} in API-Shell`,
			`${ip}\n${shell_command}\nFehler: ${shell_response.stderr}`
		);
		return true;
	}
	return false;
}

async function run_shell_command(ip: string, command: string): Promise<RequestResponse> {
	const options = {
		method: 'PATCH',
		headers: { 'content-type': 'application/json' },
		body: JSON.stringify({
			command: command
		})
	};
	return await request_display(ip, '/shellCommand', options);
}

export async function shutdown(ip: string): Promise<RequestResponse> {
	return await run_shell_command(ip, 'xfce4-session-logout --halt');
}

export async function startup(mac: string): Promise<RequestResponse> {
	return await request_control(`/wakeOnLan`, {
		method: 'POST',
		headers: { 'content-type': 'application/json' },
		body: JSON.stringify({ mac_address: mac })
	});
}