From 99de2470aa0b24666f0eda23f087c9f3399fc1dd Mon Sep 17 00:00:00 2001
From: wfjsw <wfjsw@users.noreply.github.com>
Date: Tue, 17 Nov 2020 20:04:27 +0800
Subject: [PATCH] finalize

---
 message/elements.go |  8 ++++++++
 message/message.go  | 12 ++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/message/elements.go b/message/elements.go
index 28449583..4d7076ae 100644
--- a/message/elements.go
+++ b/message/elements.go
@@ -247,6 +247,14 @@ func (e *ImageElement) Type() ElementType {
 	return Image
 }
 
+func (e *GroupFlashImgElement) Type() ElementType {
+	return Image
+}
+
+func (e *FriendFlashImgElement) Type() ElementType {
+	return Image
+}
+
 func (e *FaceElement) Type() ElementType {
 	return Face
 }
diff --git a/message/message.go b/message/message.go
index d7b9df9a..d9d62228 100644
--- a/message/message.go
+++ b/message/message.go
@@ -128,6 +128,9 @@ func (msg *PrivateMessage) ToString() (res string) {
 			res += e.Content
 		case *ImageElement:
 			res += "[Image:" + e.Filename + "]"
+		case *FriendFlashImgElement:
+			// NOTE: ignore other components
+			return "[Image (flash):" + e.Filename + "]"
 		case *FaceElement:
 			res += "[" + e.Name + "]"
 		case *AtElement:
@@ -164,6 +167,9 @@ func (msg *GroupMessage) ToString() (res string) {
 			res += "[" + e.Name + "]"
 		case *GroupImageElement:
 			res += "[Image: " + e.ImageId + "]"
+		case *GroupFlashImgElement:
+			// NOTE: ignore other components
+			return "[Image (flash):" + e.Filename + "]"
 		case *AtElement:
 			res += e.Display
 		case *RedBagElement:
@@ -557,6 +563,12 @@ func ToReadableString(m []IMessageElement) (r string) {
 			r += "/" + e.Name
 		case *GroupImageElement:
 			r += "[图片]"
+		// NOTE: flash pic is singular
+		// To be clarified
+		// case *GroupFlashImgElement:
+		// 	return "[闪照]"
+		// case *FriendFlashImgElement:
+		// 	return "[闪照]"
 		case *AtElement:
 			r += e.Display
 		}