Implement unarchiving of NSURL (closes #1)

Author: dgelessus

src/typedstream/types/foundation.py

@@ -167,6 +167,29 @@ def _init_from_unarchiver_(self, unarchiver: archiving.Unarchiver, class_version
 			raise ValueError(f"Unsupported version: {class_version}")
 
 
+@archiving.archived_class
+class NSURL(NSObject):
+	relative_to: "typing.Optional[NSURL]"
+	value: str
+	
+	def _init_from_unarchiver_(self, unarchiver: archiving.Unarchiver, class_version: int) -> None:
+		if class_version != 0:
+			raise ValueError(f"Unsupported version: {class_version}")
+		
+		is_relative = unarchiver.decode_value_of_type(b"c")
+		if is_relative == 0:
+			self.relative_to = None
+		elif is_relative == 1:
+			self.relative_to = unarchiver.decode_value_of_type(NSURL)
+		else:
+			raise ValueError(f"Unexpected value for boolean: {is_relative}")
+		
+		self.value = unarchiver.decode_value_of_type(NSString).value
+	
+	def __repr__(self) -> str:
+		return f"{type(self).__name__}({self.value!r})"
+
+
 @archiving.archived_class
 class NSValue(NSObject, advanced_repr.AsMultilineStringBase):
 	detect_backreferences = False

tests/test_typedstream.py

@@ -78,5 +78,21 @@ def test_read_file_unarchive(self) -> None:
 					unarchiver.decode_all()
 
 
+class FoundationUnarchiveTests(unittest.TestCase):
+	def test_unarchive_nsurl_absolute(self) -> None:
+		url = typedstream.unarchive_from_data(b"\x04\x0bstreamtyped\x81\xe8\x03\x84\x01@\x84\x84\x84\x05NSURL\x00\x84\x84\x08NSObject\x00\x85\x84\x01c\x00\x92\x84\x84\x84\x08NSString\x01\x94\x84\x01+\x1ehttps://example.com/index.html\x86\x86")
+		self.assertEqual(type(url), typedstream.types.foundation.NSURL)
+		self.assertIs(url.relative_to, None)
+		self.assertEqual(url.value, "https://example.com/index.html")
+	
+	def test_unarchive_nsurl_relative(self) -> None:
+		url = typedstream.unarchive_from_data(b"\x04\x0bstreamtyped\x81\xe8\x03\x84\x01@\x84\x84\x84\x05NSURL\x00\x84\x84\x08NSObject\x00\x85\x84\x01c\x01\x92\x84\x93\x95\x00\x92\x84\x84\x84\x08NSString\x01\x94\x84\x01+\x14https://example.com/\x86\x86\x92\x84\x97\x97\nindex.html\x86\x86")
+		self.assertEqual(type(url), typedstream.types.foundation.NSURL)
+		self.assertIsInstance(url.relative_to, typedstream.types.foundation.NSURL)
+		self.assertIs(url.relative_to.relative_to, None)
+		self.assertEqual(url.relative_to.value, "https://example.com/")
+		self.assertEqual(url.value, "index.html")
+
+
 if __name__ == "__main__":
 	unittest.main()