添加仅接受付费排队功能

Author: RemKeeper

Config.go

@@ -20,9 +20,11 @@ type 配置文件 struct {
 	PsC              string
 	LineKey          string
 	PPs              string
+	IsOnlyGift       int
+	Ppps             string
 }
 
-func GetConfig() (Roomid string, GuardPrintColor, GiftPrintColor int, GiftLinePrice float64, CommonPrintColor int, Linekey string) {
+func GetConfig() (Roomid string, GuardPrintColor, GiftPrintColor int, GiftLinePrice float64, CommonPrintColor int, Linekey string, IsOnlyGift int) {
 	lineupConfigFile := "./lineupConfig.json"
 	Configinfo, OpenErr := os.Open(lineupConfigFile)
 	if OpenErr != nil {
@@ -40,13 +42,13 @@ func GetConfig() (Roomid string, GuardPrintColor, GiftPrintColor int, GiftLinePr
 			if err != nil {
 				return
 			}
-			return ConfigSetGet.RoomId, ConfigSetGet.GuardPrintColor, ConfigSetGet.GiftPrintColor, ConfigSetGet.GiftLinePrice, ConfigSetGet.CommonPrintColor, ConfigSetGet.LineKey
+			return ConfigSetGet.RoomId, ConfigSetGet.GuardPrintColor, ConfigSetGet.GiftPrintColor, ConfigSetGet.GiftLinePrice, ConfigSetGet.CommonPrintColor, ConfigSetGet.LineKey, ConfigSetGet.IsOnlyGift
 		}
 	}
 	return
 }
 
-func SetConfig(Roomid string, GuardPrintColor, GiftPrintColor int, GiftLinePrice float64, CommonPrintColor int, Linekey string) bool {
+func SetConfig(Roomid string, GuardPrintColor, GiftPrintColor int, GiftLinePrice float64, CommonPrintColor int, Linekey string, IsOnlyGift int) bool {
 	ConfigJsonType := 配置文件{
 		RoomId:           Roomid,
 		Ps:               "上面这个是房间号",
@@ -60,6 +62,8 @@ func SetConfig(Roomid string, GuardPrintColor, GiftPrintColor int, GiftLinePrice
 		PsC:              "上面这个是普通用户打印颜色",
 		LineKey:          Linekey,
 		PPs:              "上面这个是排队关键词",
+		IsOnlyGift:       IsOnlyGift,
+		Ppps:             "上面这个是仅限礼物以及舰长",
 	}
 	ConfigJson, _ := json.MarshalIndent(ConfigJsonType, "", " ")
 	fmt.Println(string(ConfigJson))

main.go

@@ -20,7 +20,9 @@ func main() {
 	var SelectLine int
 	KeyBordString := make(chan string, 1)
 
-	RoomId, GuardPrintColor, GiftPrintColor, GiftLinePrice, CommonPrintColor, Linekey := GetConfig()
+	//IsOnlyGift := 0
+
+	RoomId, GuardPrintColor, GiftPrintColor, GiftLinePrice, CommonPrintColor, Linekey, IsOnlyGift := GetConfig()
 
 	if len(RoomId) < 1 || GuardPrintColor < 1 || GiftPrintColor < 1 || GiftLinePrice < 0.01 || CommonPrintColor < 1 || len(Linekey) < 1 {
 		RoomId = "2233"
@@ -35,6 +37,8 @@ func main() {
 		if len(Linekey) == 0 {
 			Linekey = "排队"
 		}
+		fmt.Println("是否开启仅限礼物及舰长排队（开启后只有礼物用户及舰长会加入队列）输入1启用")
+		_, _ = fmt.Scanln(&IsOnlyGift)
 		fmt.Println("自定义颜色编号")
 		for i := 1; i < 7; i++ {
 			ColorPrint("这是测试字符"+" "+"编号"+strconv.Itoa(i), i)
@@ -63,9 +67,9 @@ func main() {
 			fmt.Println("请输入普通队列颜色[编号]")
 			_, _ = fmt.Scanln(&CommonPrintColor)
 		}
-		fmt.Println("请输入礼物队列触发价格(RMB)，只有单次礼物大于此价格才会被加入")
+		fmt.Println("请输入礼物队列触发价格(RMB)，只有单次或累计礼物大于此价格才会被加入")
 		_, _ = fmt.Scanln(&GiftLinePrice)
-		if SetConfig(RoomId, GuardPrintColor, GiftPrintColor, GiftLinePrice, CommonPrintColor, Linekey) {
+		if SetConfig(RoomId, GuardPrintColor, GiftPrintColor, GiftLinePrice, CommonPrintColor, Linekey, IsOnlyGift) {
 			fmt.Println("配置信息已保存，下次启动将自动读取")
 		} else {
 			fmt.Println("配置文件保存失败")
@@ -151,12 +155,14 @@ func main() {
 				})
 				lineup.GuardLine = removeRepeatElement(lineup.GuardLine)
 			} else {
-				lineup.CommonLine = append(lineup.CommonLine, &Line{
-					Uid:        danmaku.Sender.Uid,
-					UserName:   danmaku.Sender.Uname,
-					PrintColor: CommonPrintColor,
-				})
-				lineup.CommonLine = removeRepeatElement(lineup.CommonLine)
+				if IsOnlyGift != 1 {
+					lineup.CommonLine = append(lineup.CommonLine, &Line{
+						Uid:        danmaku.Sender.Uid,
+						UserName:   danmaku.Sender.Uname,
+						PrintColor: CommonPrintColor,
+					})
+					lineup.CommonLine = removeRepeatElement(lineup.CommonLine)
+				}
 			}
 			SetLine(lineup)