Implemented optional data validation

DataStore2/init.lua

@@ -179,13 +179,43 @@ function DataStore:GetTableAsync(default, ...)
 	end)
 end
 
+function DataStore:SetValidator(validator)
+	assert(
+		type(validator) == "function",
+		"function expected, got " .. typeof(validator)
+	)
+	self.validator = validator
+end
+
+local function assertValidatorWithDefaultError(validator, input, defaultError)
+	local isValid = validator(input)
+	if not isValid then
+		error(defaultError) 
+	end
+end
+
 function DataStore:Set(value, _dontCallOnUpdate)
+	if self.validator then
+		assertValidatorWithDefaultError(
+			self.validator,
+			value,
+			"Invalid data"
+		)
+	end
 	self.value = clone(value)
 	self:_Update(_dontCallOnUpdate)
 end
 
 function DataStore:Update(updateFunc)
-	self.value = updateFunc(self.value)
+	local updateFuncReturn = updateFunc(self.value)
+	if self.validator then
+		assertValidatorWithDefaultError(
+			self.validator,
+			updateFuncReturn,
+			"Invalid data"
+		)
+	end
+	self.value = updateFuncReturn
 	self:_Update()
 end
 

Tests/tests/DataStore2.spec.lua

@@ -79,6 +79,20 @@ return function()
 				expect(DataStore2(nonNilKey, fakePlayer):Get("badDefault")).to.equal("abc")
 			end)
 
+			it("should validate the data", function()
+				local dataStore = DataStore2(UUID(), fakePlayer)
+				local function testValidator(dataToValidate)
+					if dataToValidate == "yepp" then
+						return true
+					end
+					return false
+				end
+				dataStore:SetValidator(testValidator)
+				expect(dataStore.validator).to.be.a("function")
+				expect(dataStore:Set("nope")).to.throw()
+				expect(dataStore:Set("yepp")).to.equal(true)
+			end)
+
 			it("should set", function()
 				local dataStore = DataStore2(UUID(), fakePlayer)
 				dataStore:Set(1)