Add .ReplyAlbum and .RespondAlbum for Message Type.

AmarnathCJD · Aug 11, 2024 · 60ceea0 · 60ceea0
1 parent 4c9cd93
commit 60ceea0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/telegram/messages.go b/telegram/messages.go
@@ -481,7 +481,7 @@ func (c *Client) sendAlbum(Peer InputPeer, Album []*InputSingleMedia, sendAs Inp
 			m = append(m, packMessage(c, update))
 		}
 	} else {
-		return nil, errors.New("no response")
+		return nil, errors.New("no response for sendAlbum")
 	}
 	return m, nil
 }

diff --git a/telegram/newmessage.go b/telegram/newmessage.go
@@ -551,6 +551,7 @@ func (m *NewMessage) ReplyMedia(Media interface{}, Opts ...MediaOptions) (*NewMe
 	} else {
 		Opts[0].ReplyID = m.ID
 	}
+
 	resp, err := m.Client.SendMedia(m.ChatID(), Media, &Opts[0])
 	if err != nil {
 		return nil, err
@@ -560,6 +561,14 @@ func (m *NewMessage) ReplyMedia(Media interface{}, Opts ...MediaOptions) (*NewMe
 	return &response, err
 }
 
+func (m *NewMessage) ReplyAlbum(Album any, Opts ...*MediaOptions) ([]*NewMessage, error) {
+	if len(Opts) == 0 {
+		Opts = append(Opts, &MediaOptions{})
+	}
+	Opts[0].ReplyID = m.ID
+	return m.Client.SendAlbum(m.ChatID(), Album, Opts...)
+}
+
 func (m *NewMessage) RespondMedia(Media interface{}, Opts ...MediaOptions) (*NewMessage, error) {
 	if len(Opts) == 0 {
 		Opts = append(Opts, MediaOptions{})
@@ -573,6 +582,13 @@ func (m *NewMessage) RespondMedia(Media interface{}, Opts ...MediaOptions) (*New
 	return &response, err
 }
 
+func (m *NewMessage) RespondAlbum(Album any, Opts ...*MediaOptions) ([]*NewMessage, error) {
+	if len(Opts) == 0 {
+		Opts = append(Opts, &MediaOptions{})
+	}
+	return m.Client.SendAlbum(m.ChatID(), Album, Opts...)
+}
+
 // Delete deletes the message
 func (m *NewMessage) Delete() (*MessagesAffectedMessages, error) {
 	return m.Client.DeleteMessages(m.Peer, []int32{m.ID})