factorio-clusterio/packages/lib/src/config/classes.ts

// Configuration classes
import { Type, Static } from "@sinclair/typebox";
import events from "events";

import isDeepStrictEqual from "../is_deep_strict_equal";
import { basicType } from "../helpers";


export type ConfigLocation = "controller" | "host" | "instance" | "control";

/**
 * Invalid Access exception
 *
 * Exception class for when an access attempt is made to a field that
 * does not permitt this access is done.
 *
 * @extends Error
 */
export class InvalidAccess extends Error { };

/**
 * Invalid Value exception
 *
 * Exception class for when invalid values are attempted to be set on a
 * field.
 *
 * @extends Error
 */
export class InvalidValue extends Error { };

/**
 * Invalid Field exception
 *
 * Exception class for when a group or field that does not exist is
 * attempted to be accessed.
 *
 * @extends Error
 */
export class InvalidField extends Error { };

/**
 * Split the given string on the first instance of separator
 *
 * Splits `string` on the first instance of `separator` and returns an
 * array consisting of the string up to the separator and the string
 * after the separator.  Returns an array with the string and an empty
 * string if the separator is not present.
 *
 * @param separator - Separator to split string by.
 * @param string - String to split
 * @returns string split on separator.
 * @internal
 */
function splitOn(separator: string, string: string) {
	let index = string.indexOf(separator);
	if (index === -1) {
		return [string, ""];
	}
	return [string.slice(0, index), string.slice(index + separator.length)];
}

export type SerializedConfig = ReturnType<Config["serialize"]>;
type SerializedGroup = ReturnType<ConfigGroup["serialize"]>;

/**
 * Collection of config groups
 *
 * Represents a collection of ConfigGroup instances that can be
 * managed as one.
 *
 * This is an EventEmitter with the following events:
 * - fieldChanged -
 *     invoked after a field has been updated in one of the config groups
 *     belonging to this config.  Takes three parameters: group, field,
 *     prev.
 *
 * @extends events.EventEmitter
 */
export class Config extends events.EventEmitter {
	declare static _finalized: boolean;
	declare static _initialized: boolean;
	declare static _groups: Map<string, typeof ConfigGroup>;

	_initialized = false;
	_groups = new Map<string, ConfigGroup>();
	_unknownGroups = new Map<string, SerializedGroup>();

	/** Set to true when a field in the config is changed. */
	dirty = false;

	/**
	 * Create a new instance of the given config
	 *
	 * @param location -
	 *     Location to evaluate access for this instance from
	 */
	constructor(
		public location: ConfigLocation,
	) {
		super();

		if (!(this.constructor as typeof Config)._finalized) {
			throw new Error(`Cannot instantiate incomplete Config class ${this.constructor.name}`);
		}

		if (typeof location !== "string") {
			throw new Error("location must be a string");
		}
	}

	async init() {
		for (let [name, GroupClass] of (this.constructor as typeof Config)._groups) {
			if (!this._groups.has(name)) {
				let group = new GroupClass(this);
				await group.init();
				this._groups.set(name, group);
			}
		}
		this._initialized = true;
	}

	_validate(serializedConfig: SerializedConfig) {
		if (basicType(serializedConfig) !== "object") {
			throw new Error(`Expected object, not ${basicType(serializedConfig)} for config`);
		}

		if (basicType(serializedConfig.groups) !== "array") {
			throw new Error(`Expected groups to be an array, not ${basicType(serializedConfig.groups)}`);
		}
	}

	/**
	 * Load config from a serialized object
	 *
	 * @param serializedConfig - Serialized config to load.
	 * @param location - Location used for access control.
	 */
	async load(
		serializedConfig: SerializedConfig,
		location = this.location
	) {
		this._validate(serializedConfig);
		for (let serializedGroup of serializedConfig.groups) {
			let GroupClass = (this.constructor as typeof Config)._groups.get(serializedGroup.name);
			if (!GroupClass) {
				this._unknownGroups.set(serializedGroup.name, serializedGroup);
				continue;
			}

			let group = new GroupClass(this);
			await group.load(serializedGroup, location);
			this._groups.set(group.name, group);
		}

		await this.init();
	}

	/**
	 * Serialize the config to a plain JavaScript object
	 *
	 * @param location - Location used for access control.
	 * @returns JSON serializable representation of the config.
	 */
	serialize(location = this.location) {
		let groups: SerializedGroup[] = [...this._unknownGroups.values()];
		for (let group of this._groups.values()) {
			groups.push(group.serialize(location));
		}

		return { groups };
	}

	/**
	 * Update the configuration based on a serialized config
	 *
	 * Updates all groups in the config with groups in the serialized
	 * config passed.
	 *
	 * @param serializedConfig - Output from .serialize().
	 * @param notify - Invoke fieldChanged events if true.
	 * @param location - Location used for access control.
	 */
	update(
		serializedConfig: SerializedConfig,
		notify: boolean,
		location = this.location
	) {
		this._validate(serializedConfig);
		for (let serializedGroup of serializedConfig.groups) {
			let group = this._groups.get(serializedGroup.name);
			if (group) {
				group.update(serializedGroup, notify, location);
			} else {
				this._unknownGroups.set(serializedGroup.name, serializedGroup);
			}
		}
	}

	/**
	 * Check if field can be accessed from the given location
	 *
	 * @param name - Name of the field to check.
	 * @param location - Location used for access control.
	 * @returns true if field is accessible
	 */
	canAccess(name: string, location = this.location) {
		let [group, field] = splitOn(".", name);
		return this.group(group).canAccess(field, location);
	}

	/**
	 * Get the value for a config field
	 *
	 * @param name - Name of field to get.
	 * @param location - Location used for access control.
	 * @returns the value for the field.
	 */
	get(name: string, location = this.location) {
		let [group, field] = splitOn(".", name);
		return this.group(group).get(field, location);
	}

	/**
	 * Set the value for a config field
	 *
	 * @param name - Name of field to set.
	 * @param value - the value to set on field
	 * @param location - Location used for access control.
	 */
	set(name: string, value: unknown, location = this.location) {
		let [group, field] = splitOn(".", name);
		this.group(group).set(field, value, location);
	}

	/**
	 * Set property of an object value for a config field
	 *
	 * @param name - Name of field to set property on.
	 * @param prop - Name of property to set on field.
	 * @param value - the value to set the property to.
	 * @param location - Location used for access control.
	 */
	setProp(name: string, prop: string, value: unknown, location = this.location) {
		let [group, field] = splitOn(".", name);
		this.group(group).setProp(field, prop, value, location);
	}

	/**
	 * Get the config group instance with the given name
	 *
	 * @param name - Name of group to get.
	 * @returns config group.
	 */
	group(name: string) {
		if (!this._initialized) {
			throw new Error(`${this.constructor.name} instance is uninitialized`);
		}
		let group = this._groups.get(name);
		if (!group) {
			throw new InvalidField(`No config group named '${name}'`);
		}
		return group;
	}

	static _initSubclass() {
		// These properties are initialized here in order to ensure that
		// they end up on the sub-class.
		if (!this._initialized) {
			this._initialized = true;
			this._finalized = false;
			this._groups = new Map();
		}
	}

	/**
	 * Mapping of group name to group class
	 *
	 * Returns the internal mapping of group names to group classes that
	 * mave been registered with this config.  Note that mutating this
	 * object will lead to unexpected behaviour.
	 *
	 * @returns config groups
	 */
	static get groups() {
		this._initSubclass();
		return this._groups;
	}

	/**
	 * Locks this Config class from adding more groups.
	 */
	static finalize() {
		this._initSubclass();
		this._finalized = true;
	}

	/**
	 * Register a ConfigGroup with this Config
	 *
	 * Adds the ConfigGroup to this config as a group that will get
	 * initialized, loaded and serialized with this config class.
	 *
	 * @param group - class to register.
	 */
	static registerGroup(group: typeof ConfigGroup) {
		this._initSubclass();

		if (this._finalized) {
			throw new Error("Cannot register group on finalized config");
		}

		if (!group._finalized) {
			throw new Error("Group must be finalized before it can be registered");
		}

		if (this._groups.has(group.groupName)) {
			throw new Error(`${group.groupName} has already been registered`);
		}

		this._groups.set(group.groupName, group);
	}
}

// TODO: remove after config refactor
export class RawConfig {
	constructor(
		public serializedConfig: SerializedConfig
	) { }

	static jsonSchema = Type.Object({
		"serializedConfig": Type.Object({
			"groups": Type.Array(
				Type.Object({
					"name": Type.String(),
					"fields": Type.Record(
						Type.String(),
						Type.Unknown(),
					),
				}),
			),
		}),
	});

	static fromJSON(json: Static<typeof this.jsonSchema>) {
		return new this(json.serializedConfig);
	}
}


type FieldType = "boolean" | "string" | "number" | "object";
export interface FieldDefinition {
	type: FieldType;
	name: string;
	fullName: string;
	title?: string;
	description?: string;
	optional: boolean;
	initial_value?: unknown | (() => unknown);
	enum?: unknown[];
	restartRequired: boolean;
	restartRequiredProps?: string[];
	access: ConfigLocation[];
}

/**
 * Collection of related config entries
 *
 * Type checks and stores a collection of related config entries.  Attempt
 * to set invalid values for fields will result in an error thrown while
 * loading a config with invalid values will result in those being replaced
 * by the innitial value for those fields.
 */
export class ConfigGroup {
	declare static _finalized: boolean;
	declare static _initialized: boolean;
	declare static _definitions: Map<string, FieldDefinition>;
	declare static groupName: string;
	declare static defaultAccess: ConfigLocation[];

	_fields = new Map<string, unknown>();
	_unknownFields = new Map<string, unknown>();


	/**
	 * Creates a new config group.
	 *
	 * After the creation of the new config group you have to call
	 * either .init() or .load() on it in order to fully initialize it.
	 *
	 * @param config -
	 *     Parent config for this group instance.
	 */
	constructor(
		public config: Config
	) {
		if (!(this.constructor as typeof ConfigGroup)._finalized) {
			throw new Error(`Cannot instantiate incomplete ConfigGroup class ${this.constructor.name}`);
		}
	}

	/**
	 * Check access for a field
	 *
	 * @param def - definition for for the field to check.
	 * @param remote - location of remote access to check for.
	 * @param error - throw on failure to pass check.
	 * @throws {InvalidAccess}
	 *     if access is not granted and error is true.
	 * @returns true if access is granted
	 * @private
	 */
	_checkAccess(def: FieldDefinition, remote: ConfigLocation, error: boolean) {
		for (let loc of [remote, this.config.location]) {
			if (!def.access.includes(loc)) {
				if (error) {
					throw new InvalidAccess(`Field '${def.name}' is not accessible from ${loc}`);
				} else {
					return false;
				}
			}
		}
		return true;
	}

	/**
	 * Initialize new config group
	 *
	 * Computes and assigns the initial values for all fields of the
	 * config group.
	 */
	async init() {
		for (let [name, def] of (this.constructor as typeof ConfigGroup)._definitions) {
			if (this._fields.has(name)) {
				continue;
			}
			if (!def.access.includes(this.config.location)) {
				continue;
			}

			let value = null;
			if (typeof def.initial_value === "function") {
				value = await def.initial_value();

			} else if (def.initial_value !== undefined) {
				value = def.initial_value;
			}

			this._fields.set(name, value);
		}
	}

	/**
	 * Load from serialized group
	 *
	 * Loads all the values from the passed serialized version of the
	 * group and then initializes any possible missing fields from it to
	 * their default values.
	 *
	 * @param serializedGroup -
	 *     Result from a previous call to .serialize().
	 * @param location - Location used for access control.
	 */
	async load(serializedGroup: SerializedGroup, location = this.config.location) {
		this.update(serializedGroup, false, location);
		await this.init();
	}

	/**
	 * Name of the group
	 *
	 * Shortcut for `group.constructor.groupName`
	 */
	get name() {
		return (this.constructor as typeof ConfigGroup).groupName;
	}

	/**
	 * Check if field can be accessed from the given location
	 *
	 * @param name - Name of field to check.
	 * @param location - Location used for access control.
	 * @returns true if field is accessible
	 */
	canAccess(name: string, location = this.config.location) {
		let def = (this.constructor as typeof ConfigGroup)._definitions.get(name);
		if (!def) {
			throw new InvalidField(`No field named '${name}'`);
		}
		return this._checkAccess(def, location, false);
	}

	/**
	 * Get value for field
	 *
	 * @param name - Name of field to get.
	 * @param location - Location used for access control.
	 * @returns Value stored for the field.
	 */
	get(name: string, location = this.config.location) {
		let def = (this.constructor as typeof ConfigGroup)._definitions.get(name);
		if (!def) {
			throw new InvalidField(`No field named '${name}'`);
		}
		this._checkAccess(def, location, true);

		return this._fields.get(name);
	}

	/**
	 * Set value of field
	 *
	 * The field must be defined for the config group and the value is
	 * type checked against the field definition.
	 *
	 * @param name - Name of field to set.
	 * @param value - Value to set for field.
	 * @param location - Location used for access control.
	 * @throws {InvalidField} if field is not defined.
	 * @throws {InvalidValue} if value is not allowed for the field.
	 */
	set(name: string, value: unknown, location = this.config.location) {
		let def = (this.constructor as typeof ConfigGroup)._definitions.get(name);
		if (!def) {
			throw new InvalidField(`No field named '${name}'`);
		}
		this._checkAccess(def, location, true);

		// Empty strings are treates as null
		if (value === "") {
			value = null;
		}

		if (def.enum && !def.enum.includes(value)) {
			throw new InvalidValue(`Expected one of [${def.enum.join(", ")}], not ${value}`);
		}

		if (typeof value === "string") {
			if (def.type === "boolean") {
				if (value === "true") {
					value = true;
				} else if (value === "false") {
					value = false;
				}

			} else if (def.type === "number") {
				let numberRegExp = /^[+-]?(Infinity|\d+\.(\d+)?([eE][+-]?\d+)?|\.?\d+([eE][+-]?\d+)?)$/;
				if (numberRegExp.test(value.trim())) {
					value = Number.parseFloat(value);
				}

			} else if (def.type === "object") {
				try {
					value = JSON.parse(value);
				} catch (err: any) {
					throw new InvalidValue(`Error parsing value for ${name}: ${err.message}`);
				}
			}
		}

		if (value === null) {
			if (!def.optional) {
				throw new InvalidValue(`Field ${name} cannot be null`);
			}

		} else if (basicType(value) !== def.type) {
			throw new InvalidValue(`Expected type of ${name} to be ${def.type}, not ${basicType(value)}`);
		}

		let prev = this._fields.get(name);
		this._fields.set(name, value);
		if (this.config && !isDeepStrictEqual(value, prev)) {
			this.config.dirty = true;
			this.config.emit("fieldChanged", this, name, prev);
		}
	}

	/**
	 * Set property of object field
	 *
	 * Update value of stored object field by setting the specified property
	 * on it.  The field must be defined as an object for the config group.
	 *
	 * @param name - Name of field to set property on.
	 * @param prop - Name of property to set on field.
	 * @param value - the value to set the property to.
	 * @param location - Location used for access control.
	 * @throws {InvalidField} if field is not defined.
	 * @throws {InvalidValue} if field is not an object.
	 */
	setProp(name: string, prop: string, value: unknown, location = this.config.location) {
		let def = (this.constructor as typeof ConfigGroup)._definitions.get(name);
		if (!def) {
			throw new InvalidField(`No field named '${name}'`);
		}
		this._checkAccess(def, location, true);

		if (def.type !== "object") {
			throw new InvalidValue(`Cannot set property on non-object field '${name}'`);
		}

		let prev = this._fields.get(name) as Record<string, unknown>;
		let updated = {...prev || {}};

		if (value !== undefined) {
			updated[prop] = value;
		} else {
			delete updated[prop];
		}
		this._fields.set(name, updated);
		if (this.config && !isDeepStrictEqual(updated, prev)) {
			this.config.dirty = true;
			this.config.emit("fieldChanged", this, name, prev);
		}
	}

	/**
	 * Update a config group from a serialized group
	 *
	 * Overwrites the values of all the fields in this config group that
	 * has a valid value in the passed serialized group with that value.
	 *
	 * @param serializedGroup -
	 *     Result from a previous call to .serialize().
	 * @param notify - Invoke fieldChanged events if true.
	 * @param location - Location used for access control.
	 */
	update(
		serializedGroup: SerializedGroup,
		notify: boolean,
		location = this.config.location
	) {
		if (basicType(serializedGroup) !== "object") {
			throw new Error(`Expected object, not ${basicType(serializedGroup)} for ConfigGroup`);
		}

		if (serializedGroup.name !== this.name) {
			throw new Error(`Expected group name ${this.name}, not ${serializedGroup.name}`);
		}

		if (basicType(serializedGroup.fields) !== "object") {
			throw new Error(`Expected fields to be an object, not ${basicType(serializedGroup.fields)}`);
		}

		for (let [name, value] of Object.entries(serializedGroup.fields)) {
			let def = (this.constructor as typeof ConfigGroup)._definitions.get(name);
			if (!def) {
				this._unknownFields.set(name, value);
				continue;
			}

			if (!this._checkAccess(def, location, false)) {
				continue;
			}

			if (def.optional) {
				if (value !== null && basicType(value) !== def.type) {
					continue;
				}

			} else if (basicType(value) !== def.type) {
				continue;
			}

			if (def.enum && !def.enum.includes(value)) {
				continue;
			}

			let prev = this._fields.get(name);
			this._fields.set(name, value);
			if (notify && this.config && !isDeepStrictEqual(value, prev)) {
				this.config.dirty = true;
				this.config.emit("fieldChanged", this, name, prev);
			}
		}
	}

	/**
	 * Serialize the config to a plain JavaScript object
	 *
	 * @param location - Location used for access control.
	 * @returns JSON serializable representation of the group.
	 */
	serialize(location: ConfigLocation = this.config.location) {
		let fields: Record<string, unknown> = {};
		for (let [name, value] of this._fields) {
			let def = (this.constructor as typeof ConfigGroup)._definitions.get(name)!;
			if (!this._checkAccess(def, location, false)) {
				continue;
			}

			fields[name] = value;
		}

		for (let [name, value] of this._unknownFields) {
			fields[name] = value;
		}

		return {
			name: this.name,
			fields,
		};
	}

	static _initSubclass() {
		// These properties are initialized here in order to ensure that
		// they end up on the sub-class.
		if (!this._initialized) {
			this._initialized = true;
			this._definitions = new Map();
		}
	}

	/**
	 * Mapping of config name to field meta data
	 *
	 * Returns the internal meta data for the fields of this class.
	 * Note that mutating this object will lead to unexpected behaviour.
	 */
	static get definitions() {
		this._initSubclass();
		return this._definitions;
	}

	/**
	 * Define a new configuration field
	 *
	 * @param item - Field definition.
	 * @param item.access -
	 *     Array of strings declaring where this config entry can be read
	 *     and modified from.
	 * @param item.type -
	 *     string declaring the type of the config value, supports
	 *     boolean, string, number and object.
	 * @param item.name -
	 *     Name of the configuration field.  Should follow the
	 *     lower_case_underscore style.
	 * @param item.title -
	 *     Text used to identify the config field in user interfaces.
	 * @param item.description -
	 *     Text used to describe the config field in user interfaces.
	 * @param item.enum -
	 *     Array of all values that are acceptible.  Useful for making
	 *     enumerated values.
	 * @param item.restartRequired -
	 *     True if a restart of the entity this config is attached to is
	 *     required for changes to take effect.  Informative only, defaults
	 *     to false.
	 * @param item.restartRequiredProps -
	 *     Properties to invert the value of restartRequired for if present.
	 * @param item.optional -
	 *     True if this field can be set to null.  Defaults to false.
	 * @param item.initial_value -
	 *     Value this config field should take in a newly initialized
	 *     config.  This can also be an async function returning the
	 *     value to use.
	 */
	// eslint-disable-next-line complexity
	static define(item: {
		access?: ConfigLocation[],
		type: FieldType,
		name: string,
		title?: string,
		description?: string,
		enum?: unknown[],
		restartRequired?: boolean,
		restartRequiredProps?: string[],
		optional?: boolean,
		initial_value?: unknown | (() => unknown),
	}) {
		if (this._finalized) {
			throw new Error(`Cannot define field for ConfigGroup class ${this.name} after it has been finalized`);
		}

		if (basicType(this.groupName) !== "string") {
			throw new Error(`Expected ConfigGroup class ${this.name} to have the groupName property set to a string`);
		}

		this._initSubclass();

		let itemKeys = Object.keys(item);

		// Check for required properties
		["name", "type"].forEach(prop => {
			if (!itemKeys.includes(prop)) {
				throw new Error(`${prop} is required when defining an field`);
			}
		});

		// Validate properties
		{
			const {
				type, access, enum: enum_, name, title, description, restartRequired,
				restartRequiredProps, optional, initial_value, ...rest
			} = item;

			let validTypes = ["boolean", "string", "number", "object"];
			if (type !== undefined) {
				if (!validTypes.includes(type)) {
					throw new Error(`${type} is not a valid type`);
				}
			}

			if (access !== undefined) {
				if (basicType(access) !== "array") {
					throw new Error("access must be an array");
				}
			}

			if (enum_ !== undefined) {
				if (basicType(enum_) !== "array") {
					throw new Error("enum must be an array");
				}
			}

			if (name !== undefined) {
				if (typeof name !== "string") {
					throw new Error("name must be a string");
				}
			}

			if (title !== undefined) {
				if (typeof title !== "string") {
					throw new Error("title must be a string");
				}
			}

			if (description !== undefined) {
				if (typeof description !== "string") {
					throw new Error("description must be a string");
				}
			}

			if (restartRequired !== undefined) {
				if (typeof restartRequired !== "boolean") {
					throw new Error("restartRequired must be a boolean");
				}
			}

			if (restartRequiredProps !== undefined) {
				if (basicType(restartRequiredProps) !== "array") {
					throw new Error("restartRequiredProps must be a array");
				}
			}

			if (optional !== undefined) {
				if (typeof optional !== "boolean") {
					throw new Error("optional must be a boolean");
				}
			}

			if (initial_value !== undefined) {
				if (basicType(initial_value) !== item.type && basicType(initial_value) !== "function") {
					throw new Error("initial_value must match the type or be a function");
				}
			}

			for (let key of Object.keys(rest)) {
				throw new Error(`Unknown property ${key}`);
			}
		}

		if (this._definitions.has(item.name)) {
			throw new Error(`Config field ${item.name} has already been defined`);
		}

		let computed: FieldDefinition = {
			access: this.defaultAccess,
			optional: false,
			fullName: `${this.groupName}.${item.name}`,
			restartRequired: false,
			...item,
		};

		if (item.restartRequiredProps && item.type !== "object") {
			throw new Error("restartRequiredProps is only allowed if type is object");
		}

		if (!computed.access) {
			throw new Error(`access not set and no defaultAccess defined for ${this.name}`);
		}

		if (!computed.optional && computed.initial_value === undefined) {
			throw new Error(`Non-optional field ${item.name} needs an initial_value`);
		}

		this._definitions.set(item.name, computed);
	}

	/**
	 * Locks this ConfigGroup class from adding more definitons.
	 */
	static finalize() {
		if (basicType(this.groupName) !== "string") {
			throw new Error(`Expected ConfigGroup class ${this.name} to have the groupName property set to a string`);
		}

		this._initSubclass();
		this._finalized = true;
	}
}

/**
 * Config group for plugins
 *
 * Grouping that includes per plugin fields clusterio uses to manage
 * plugins.  This is currently only the enabled field.  It is otherwise
 * identical to the ConfigGroup class.
 *
 * @extends ConfigGroup
 */
export class PluginConfigGroup extends ConfigGroup {
	static _initSubclass() {
		if (!this._initialized) {
			super._initSubclass();
			this.define({
				name: "load_plugin",
				title: "Load Plugin",
				restartRequired: true,
				type: "boolean",
				initial_value: true,
			});
		}
	}
}