monkeydo/monkey/MonkeyMaster.mjs

// Task manager for Monkeydo dedicated workers (monkeys)

import * as Comlink from "https://unpkg.com/comlink/dist/esm/comlink.mjs";

export default class MonkeyMaster {
	constructor() {
		this.comlink = null;

		this.ready = false;
		// Tasks will be queued here on runtime if the worker isn't ready
		this.queue = {
			_flags: [],
			set flag(flag) {
				this._flags.push(flag);
			},
			// Attempt to send all queued flags
			sendAllFlags: () => {
				// Copy flags and clear queue
				const flags = [...this.queue._flags];
				this.queue._flags = [];

				flags.forEach(flag => this.setFlag(flag));
			}
		};
	}

	// Import worker relative to this module
	getWorkerPath() {
		const name = "Monkey.js";
		const url = new URL(import.meta.url);

		// Replace pathname of this file with worker
		const path = url.pathname.split("/");
		path[path.length - 1] = name;

		url.pathname = path.join("/");
		return url.toString();
	}

	async init() {
		// Spawn and wrap dedicated worker with Comlink
		const worker = new Worker(this.getWorkerPath());
		const Monkey = Comlink.wrap(worker);
		this.comlink = await new Monkey();

		// Wait for comlink to initialize proxy and send queued flags
		return await new Promise((resolve,reject) => {
			if(!this.comlink) reject("Failed to open proxy to worker");

			this.ready = true;
			this.queue.sendAllFlags();
			resolve();
		});
	}

	// Return a flag array index by name
	flagStringToIndex(flag) {
		const flags = [
			"MANIFEST_LOADED",
			"PLAYING"
		];

		// Translate string to index
		if(typeof flag === "string" || flag < 0) {
			const key = flags.indexOf(flag.toUpperCase());
			if(key < 0) return;
		}

		// Check key is in bounds
		if(flag < 0 || flags > flags.length - 1) {
			throw new Error(`Array key '${flag}' out of range`);
		}
		return flag;
	}

	async getFlag(flag) {
		const key = this.flagStringToIndex(flag);
		return await this.comlink.flag(key);
	}

	// Set or queue worker runtime flag
	async setFlag(flag,value) {
		const key = this.flagStringToIndex(flag);
		if(!this.ready) {
			this.queue.flag = [key,value];
			return;
		}

		// Tell worker to update flag by key
		const update = await this.comlink.flag(key,value);
		if(!update) {
			this.queue.flag = [key,value];
		}
		return update;
	}

	async loadManifest(manifest) {
		if(!this.ready) await this.init();
		try {
			const url = new URL(manifest);
			this.comlink.fetchManifest(url.toString());
		}
		catch {
			this.comlink.loadManifest(manifest);
		}
		return true;
	}
}