From 60e6ef073319fd55e913aa7f62aa0d8010dea4f5 Mon Sep 17 00:00:00 2001
From: Nathaniel Sabanski <sabanski.n@gmail.com>
Date: Wed, 12 Jun 2024 15:04:59 -0700
Subject: [PATCH] Improvements to attribute()

---
 surreal.js | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/surreal.js b/surreal.js
index 569bdc3..2c9c11a 100644
--- a/surreal.js
+++ b/surreal.js
@@ -180,28 +180,28 @@ let $ = { // Convenience for internals.
 	},
 	// Add or remove attributes from element(s)
 	attribute(e, name, value=undefined) {
-		// Get. This one is special. Format: "name", "value"
+		// Get. (Format: "name", "value") Special: Ends call chain.
 		if (typeof name === 'string' && value === undefined) {
 			if ($.isNodeList(e)) return [] // Not supported for Get. For many elements, wrap attribute() in any(...).run(...) or any(...).forEach(...)
 			if ($.isNode(e)) return e.getAttribute(name)
-			return null // No value.
+			return null // No value. Ends call chain.
 		}
 		// Remove.
 		if (typeof name === 'string' && value === null) {
 			if ($.isNodeList(e)) e.forEach(_ => { $.attribute(_, name, value) })
-			else e.removeAttribute(name)
+			if ($.isNode(e)) e.removeAttribute(name)
 			return e
 		}
 		// Add / Set.
 		if (typeof name === 'string') {
 			if ($.isNodeList(e)) e.forEach(_ => { $.attribute(_, name, value) })
-			else e.setAttribute(name, value)
+			if ($.isNode(e)) e.setAttribute(name, value)
 			return e
 		}
 		// Format: { "name": "value", "blah": true }
 		if (typeof name === 'object') {
 			if ($.isNodeList(e)) e.forEach(_ => { Object.entries(name).forEach(([key, val]) => { $.attribute(_, key, val) }) })
-			else if ($.isNode(e)) Object.entries(name).forEach(([key, val]) => { $.attribute(e, key, val) })
+			if ($.isNode(e)) Object.entries(name).forEach(([key, val]) => { $.attribute(e, key, val) })
 			return e
 		}
 		return e