factorio-clusterio/packages/master/web/util/wslink.jsx

import { Control, ControlConnector } from "./../util/websocket"
import * as errors from "@clusterio/lib/errors"
import * as link from "@clusterio/lib/link"
import config from "@clusterio/lib/config"

let control, controlConnector

// Create WS connection to master. Used by all components through the interface exported here.
(async () => {
	config.finalizeConfigs();
	controlConnector = new ControlConnector(
		localStorage.getItem("master_url"),
		10, // Reconnect delay
		localStorage.getItem("master_token"),
	);
	control = new Control(controlConnector, []);
	try {
		await controlConnector.connect();
	} catch (err) {
		console.log(err)
	}
})()

let listSlaves = createListFunction("listSlaves")
let listInstances = createListFunction("listInstances")
let listUsers = createListFunction("listUsers", "name")
let listRoles = createListFunction("listRoles")
let listPermissions = createListFunction("listPermissions", "name")

let startInstance = async function ({ instance_id, save = null }) {
	return await link.messages.startInstance.send(control, { instance_id, save })
}
let stopInstance = async function ({ instance_id }) {
	return await link.messages.stopInstance.send(control, { instance_id })
}
let createInstance = async function (serialized_config) {
	return await link.messages.createInstance.send(control, { serialized_config });
}
let deleteInstance = async function ({ instance_id }) {
	return await link.messages.deleteInstance.send(control, { instance_id });
}
let assignInstance = async function ({ instance_id, slave_id }) {
	return await link.messages.assignInstanceCommand.send(control, {
		instance_id,
		slave_id,
	});
}
let createSave = async function ({ instance_id }) {
	return await link.messages.createSave.send(control, { instance_id });
}
let getInstanceConfig = async function ({ instance_id }) {
	return await link.messages.getInstanceConfig.send(control, { instance_id });
}
let setInstanceConfigField = async function ({ instance_id, field, value }) {
	return await link.messages.setInstanceConfigField.send(control, {
		instance_id,
		field,
		value: typeof value === "object" ? JSON.stringify(value) : value,
	});
}
let setInstanceConfigProp = async function ({ instance_id, field, prop, value }) {
	return await link.messages.setInstanceConfigProp.send(control, {
		instance_id,
		field,
		prop,
		value, // JSON Object
	});
}
let sendRcon = async function ({ instance_id, command }) {
	return await link.messages.sendRcon.send(control, { instance_id, command })
}
let setInstanceOutputSubscriptions = async function ({ instance_id }) {
	return await link.messages.setInstanceOutputSubscriptions.send(control, { instance_ids: [instance_id] })
}

let createUser = async function ({ name }) {
	return await link.messages.createUser.send(control, { name })
}
let deleteUser = async function ({ name }) {
	return await link.messages.deleteUser.send(control, { name })
}
let setRoles = async function ({ name, roles }) {
	return await link.messages.updateUserRoles.send(control, {
		name,
		roles,
	})
}

let createRole = async function ({ name, description, permissions = [] }) {
	return await link.messages.createRole.send(control, { name, description, permissions })
}
let updateRole = async function ({ id, name, description, permissions }) {
	return await link.messages.updateRole.send(control, {
		id, name, description, permissions
	})
}

export {
	listSlaves,
	listInstances,
	listUsers,
	createUser,
	deleteUser,
	listRoles,
	setRoles,
	createRole,
	updateRole,
	listPermissions,
	startInstance,
	stopInstance,
	createInstance,
	deleteInstance,
	assignInstance,
	createSave,
	getInstanceConfig,
	setInstanceConfigField,
	setInstanceConfigProp,
	sendRcon,
	setInstanceOutputSubscriptions,
}

function createListFunction(name, key) {
	return async () => {
		return (await link.messages[name].send(control)).list.map(addKey(key || "id"));
	}
}
function addKey(key) {
	return function (el) {
		return {
			...el,
			key: el[key]
		}
	}
}