From c9e4e842d986d846f02008e0c62138c7e931440b Mon Sep 17 00:00:00 2001
From: eatmoreapple <eatmoreorange@gmail.com>
Date: Fri, 9 Dec 2022 22:06:47 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=96=87=E4=BB=B6=E7=B1=BB?=
 =?UTF-8?q?=E5=9E=8B=E5=8C=B9=E9=85=8D=E7=AD=96=E7=95=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 global.go | 13 +++++++------
 parser.go |  5 +++--
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/global.go b/global.go
index 75521d4..2d34590 100644
--- a/global.go
+++ b/global.go
@@ -115,7 +115,8 @@ var (
 )
 
 // ALL 跟search函数搭配
-//      friends.Search(openwechat.ALL, )
+//
+//	friends.Search(openwechat.ALL, )
 const ALL = 0
 
 // 性别
@@ -137,11 +138,11 @@ const (
 
 const TimeFormat = "Mon Jan 02 2006 15:04:05 GMT+0800 (中国标准时间)"
 
-var imageType = map[string]bool{
-	"bmp":  true,
-	"png":  true,
-	"jpeg": true,
-	"jpg":  true,
+var imageType = map[string]struct{}{
+	"bmp":  {},
+	"png":  {},
+	"jpeg": {},
+	"jpg":  {},
 }
 
 var videoType = "mp4"
diff --git a/parser.go b/parser.go
index fe86f92..4c01a6b 100644
--- a/parser.go
+++ b/parser.go
@@ -81,9 +81,10 @@ const (
 // 微信匹配文件类型策略
 func getMessageType(filename string) string {
 	ext := getFileExt(filename)
-	if imageType[ext] {
+	if _, ok := imageType[ext]; ok {
 		return pic
-	} else if ext == videoType {
+	}
+	if ext == videoType {
 		return video
 	}
 	return doc