From e869ba746f14bf14686d03331f37618b3c7b4af9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AD=90=E9=93=AD?= <zi__mm@163.com>
Date: Mon, 11 Nov 2024 18:23:26 +0800
Subject: [PATCH] CardType

---
 plugins/zltx_v2/internal/transform.go    | 17 +++++------
 plugins/zltx_v2/internal/vo/card_type.go | 38 ++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 9 deletions(-)
 create mode 100644 plugins/zltx_v2/internal/vo/card_type.go

diff --git a/plugins/zltx_v2/internal/transform.go b/plugins/zltx_v2/internal/transform.go
index 70c0770..a07c69c 100644
--- a/plugins/zltx_v2/internal/transform.go
+++ b/plugins/zltx_v2/internal/transform.go
@@ -105,11 +105,9 @@ func queryResp(appKey string, request *proto.QueryRequest, resp *api.OrderQueryR
 }
 
 func getQueryCard(appKey string, card *api.Cards) (*Card, error) {
-	// - 1：卡密+卡密：no和pwd都不为空
-	// - 2：卡密，pwd不为空
-	// - 3：链接，url不为空
 	car := &Card{}
-	if card.CardType == 1 {
+	t := vo.CardType(card.CardType)
+	if t.IsPwdNo() {
 		pwd, err := decryptAES(card.Pwd, appKey)
 		if err != nil {
 			return nil, err
@@ -120,13 +118,13 @@ func getQueryCard(appKey string, card *api.Cards) (*Card, error) {
 		}
 		car.Password = pwd
 		car.Number = no
-	} else if card.CardType == 2 {
+	} else if t.IsPwd() {
 		pwd, err := decryptAES(card.Pwd, appKey)
 		if err != nil {
 			return nil, err
 		}
 		car.Password = pwd
-	} else if card.CardType == 3 {
+	} else if t.IsUrl() {
 		// 无
 	}
 	return car, nil
@@ -163,7 +161,8 @@ func getNotifyCard(appKey string, cards []notify.Cards) ([]byte, error) {
 	}
 	card := cards[0]
 	car := &Card{}
-	if card.CardType == 1 {
+	t := vo.CardType(card.CardType)
+	if t.IsPwdNo() {
 		pwd, err := decryptAES(card.Pwd, appKey)
 		if err != nil {
 			return nil, err
@@ -174,13 +173,13 @@ func getNotifyCard(appKey string, cards []notify.Cards) ([]byte, error) {
 		}
 		car.Password = pwd
 		car.Number = no
-	} else if card.CardType == 2 {
+	} else if t.IsPwd() {
 		pwd, err := decryptAES(card.Pwd, appKey)
 		if err != nil {
 			return nil, err
 		}
 		car.Password = pwd
-	} else if card.CardType == 3 {
+	} else if t.IsUrl() {
 		// 无
 	}
 	return car.ToJson(), nil
diff --git a/plugins/zltx_v2/internal/vo/card_type.go b/plugins/zltx_v2/internal/vo/card_type.go
new file mode 100644
index 0000000..321374d
--- /dev/null
+++ b/plugins/zltx_v2/internal/vo/card_type.go
@@ -0,0 +1,38 @@
+package vo
+
+// CardType
+// - 1：卡密+卡密：no和pwd都不为空
+// - 2：卡密，pwd不为空
+// - 3：链接，url不为空
+type CardType int
+
+const (
+	CardTypePwdNo CardType = iota + 1
+	CardTypePwd
+	CardTypeUrl
+)
+
+var cardTypeTextMap = map[CardType]string{
+	CardTypePwdNo: "卡密+卡密：no和pwd都不为空",
+	CardTypePwd:   "卡密，pwd不为空",
+	CardTypeUrl:   "链接，url不为空",
+}
+
+func (o CardType) GetCardTypeText() string {
+	if text, ok := cardTypeTextMap[o]; ok {
+		return text
+	}
+	return ""
+}
+
+func (o CardType) IsPwdNo() bool {
+	return o == CardTypePwdNo
+}
+
+func (o CardType) IsPwd() bool {
+	return o == CardTypePwd
+}
+
+func (o CardType) IsUrl() bool {
+	return o == CardTypeUrl
+}