Framework Rewrite

framework/UnionType.js

@@ -0,0 +1,37 @@
+const ArgumentType = require('./ArgumentType');
+
+module.exports = class ArgumentUnionType extends ArgumentType {
+	constructor(client, id) {
+		super(client, id);
+
+		this.types = [];
+		const typeIDs = id.split('|');
+		for (const typeID of typeIDs) {
+			const type = client.registry.types.get(typeID);
+			if (!type) throw new Error(`Argument type "${typeID}" is not registered.`);
+			this.types.push(type);
+		}
+	}
+
+	async validate(val, msg, arg) {
+		let results = this.types.map(type => !type.isEmpty(val, msg, arg) && type.validate(val, msg, arg));
+		results = await Promise.all(results);
+		if (results.some(valid => valid && typeof valid !== 'string')) return true;
+		const errors = results.filter(valid => typeof valid === 'string');
+		if (errors.length > 0) return errors.join('\n');
+		return false;
+	}
+
+	async parse(val, msg, arg) {
+		let results = this.types.map(type => !type.isEmpty(val, msg, arg) && type.validate(val, msg, arg));
+		results = await Promise.all(results);
+		for (let i = 0; i < results.length; i++) {
+			if (results[i] && typeof results[i] !== 'string') return this.types[i].parse(val, msg, arg);
+		}
+		throw new Error(`Couldn't parse value "${val}" with union type ${this.id}.`);
+	}
+
+	isEmpty(val, msg, arg) {
+		return !this.types.some(type => !type.isEmpty(val, msg, arg));
+	}
+};