feat: 为 MessageElement 接口添加获取子元素方法，并为所有元素添加方法实现

satori-protocol-go · Jun 18, 2024 · 22c6eaa · 22c6eaa
1 parent 2487858
commit 22c6eaa
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/pkg/message/message_element.go b/pkg/message/message_element.go
@@ -8,6 +8,7 @@ type MessageElement interface {
 	Tag() string
 	Stringify() string
 	Alias() []string
+	GetChildren() []MessageElement
 }
 
 type noAliasMessageElement struct {

diff --git a/pkg/message/message_element_basic.go b/pkg/message/message_element_basic.go
@@ -31,6 +31,10 @@ func (e *MessageElementText) Parse(n *html.Node) (MessageElement, error) {
 	return nil, nil
 }
 
+func (e *MessageElementText) GetChildren() []MessageElement {
+	return nil
+}
+
 type MessageElementAt struct {
 	*noAliasMessageElement
 	*ChildrenMessageElement

diff --git a/pkg/message/message_element_typography.go b/pkg/message/message_element_typography.go
@@ -20,6 +20,10 @@ func (e *MessageElmentBr) Parse(n *html.Node) (MessageElement, error) {
 	return &MessageElmentBr{}, nil
 }
 
+func (e *MessageElmentBr) GetChildren() []MessageElement {
+	return nil
+}
+
 type MessageElmentP struct {
 	*noAliasMessageElement
 	*ChildrenMessageElement