Merge pull request #7 from WindowsSov8forUs/master

add: 添加扩展元素构造函数
satori-protocol-go · Jun 19, 2024 · 1c41123 · 1c41123
2 parents 63a4f1a + e541770
commit 1c41123
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pkg/message/message_element_extend.go b/pkg/message/message_element_extend.go
@@ -39,4 +39,17 @@ func (e *MessageElementExtend) Parse(n *html.Node) (MessageElement, error) {
 	return result, nil
 }
 
+func NewMessageElementExtend(tag string, attrs map[string]string, children ...MessageElement) MessageElement {
+	result := &MessageElementExtend{
+		tag: tag,
+	}
+	for key, value := range attrs {
+		result.ExtendAttributes = result.AddAttribute(key, value)
+	}
+	result.ChildrenMessageElement = &ChildrenMessageElement{
+		children: children,
+	}
+	return result
+}
+
 var ExtendParser = &MessageElementExtend{}