Make body element easily accessible

ChatSecure · Feb 3, 2016 · 58c0dca · 58c0dca
1 parent 1f3a990
commit 58c0dca
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/Code/HTMLDocument.h b/Code/HTMLDocument.h
@@ -54,6 +54,11 @@ NS_ASSUME_NONNULL_BEGIN
  */
 @property (strong, nonatomic) HTMLElement * __nullable rootElement;
 
+/**
+    The first child element of the root with the tag name `body`. Typically the `<body>` element.
+ */
+@property (readonly, nonatomic) HTMLElement * __nullable bodyElement;
+
 @end
 
 NS_ASSUME_NONNULL_END
diff --git a/Code/HTMLDocument.m b/Code/HTMLDocument.m
@@ -85,6 +85,19 @@ - (void)setRootElement:(HTMLElement * __nullable)rootElement
     }
 }
 
+- (HTMLElement * __nullable)bodyElement
+{
+	for (id child in self.rootElement.children) {
+		if ([child isKindOfClass:[HTMLElement class]]) {
+			HTMLElement *elem = child;
+			if ([elem.tagName isEqualToString:@"body"]) {
+				return elem;
+			}
+		}
+	}
+	return nil;
+}
+
 static id FirstNodeOfType(id <NSFastEnumeration> collection, Class type)
 {
     for (id node in collection) {