Update Files

leenkx/Sources/leenkx/logicnode/AddTraitNode.hx

@@ -0,0 +1,28 @@
+package leenkx.logicnode;
+
+import iron.object.Object;
+
+class AddTraitNode extends LogicNode {
+
+	public function new(tree: LogicTree) {
+		super(tree);
+	}
+
+	override function run(from: Int) {
+		var object: Object = inputs[1].get();
+		var traitName: String = inputs[2].get();
+
+		assert(Error, object != null, "Object should not be null");
+		assert(Error, traitName != null, "Trait name should not be null");
+
+		var cname = Type.resolveClass(Main.projectPackage + "." + traitName);
+		if (cname == null) cname = Type.resolveClass(Main.projectPackage + ".node." + traitName);
+		assert(Error, cname != null, 'No trait with the name "$traitName" found, make sure that the trait is exported!');
+		assert(Warning, object.getTrait(cname) == null, 'Object already has the trait "$traitName" applied');
+
+		var trait = Type.createInstance(cname, []);
+		object.addTrait(trait);
+
+		runOutput(0);
+	}
+}