feat: 添加21点游戏

handler/blackjack/blackjack.go

@@ -0,0 +1,104 @@
+package blackjack
+
+import (
+	"git.lxtend.com/qqbot/constants"
+	"git.lxtend.com/qqbot/handler"
+	"git.lxtend.com/qqbot/handler/blackjack/controller"
+	"git.lxtend.com/qqbot/handler/blackjack/core"
+	"git.lxtend.com/qqbot/handler/blackjack/view"
+	"git.lxtend.com/qqbot/model"
+	"git.lxtend.com/qqbot/util"
+)
+
+var userGameMap map[string]*core.BackJackGame
+
+func init() {
+	handler.RegisterHandler("bj", blackJack, constants.LEVEL_USER)
+	userGameMap = make(map[string]*core.BackJackGame)
+}
+
+func blackJack(msg model.Message) model.Reply {
+	tokens := util.SplitN(msg.RawMsg, 2)
+	if len(tokens) < 2 {
+		return model.Reply{
+			ReplyMsg:       "Invalid command",
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+
+	if tokens[1] == "exit" {
+		delete(userGameMap, util.From(msg.GroupInfo.GroupId, msg.UserId))
+		return model.Reply{
+			ReplyMsg:       "Bye",
+			ReferOriginMsg: false,
+			FromMsg:        msg,
+		}
+	}
+
+	if tokens[1] == "start" {
+		userGameMap[util.From(msg.GroupInfo.GroupId, msg.UserId)] = core.NewBlackJackGame(controller.NewBlackJackSimulator(), view.NewLiteralViewer(controller.NewBlackJackSimulator()))
+	}
+	if _, ok := userGameMap[util.From(msg.GroupInfo.GroupId, msg.UserId)]; !ok {
+		return model.Reply{
+			ReplyMsg:       "Please start a game first",
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+	response, _ := userGameMap[util.From(msg.GroupInfo.GroupId, msg.UserId)].AddCommand(tokens[1])
+	handler.RegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId, blackJackWithNoBj)
+	if response[len(response)-1:] == "\n" {
+		response = response[:len(response)-1]
+	}
+	return model.Reply{
+		ReplyMsg:       response,
+		ReferOriginMsg: false,
+		FromMsg:        msg,
+	}
+}
+
+func blackJackWithNoBj(msg model.Message) (model.Reply, bool) {
+	if msg.RawMsg == "exit" {
+		delete(userGameMap, util.From(msg.GroupInfo.GroupId, msg.UserId))
+		handler.UnRegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId)
+		return model.Reply{
+			ReplyMsg:       "Bye",
+			ReferOriginMsg: false,
+			FromMsg:        msg,
+		}, true
+	}
+
+	if msg.RawMsg == "start" {
+		userGameMap[util.From(msg.GroupInfo.GroupId, msg.UserId)] = core.NewBlackJackGame(controller.NewBlackJackSimulator(), view.NewLiteralViewer(controller.NewBlackJackSimulator()))
+	}
+
+	if _, ok := userGameMap[util.From(msg.GroupInfo.GroupId, msg.UserId)]; !ok {
+		return model.Reply{
+			ReplyMsg:       "Please start a game first",
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}, true
+	}
+	response, err := userGameMap[util.From(msg.GroupInfo.GroupId, msg.UserId)].AddCommand(msg.RawMsg)
+	if err != nil {
+		if err.Error() == "invalid command" {
+			return model.Reply{
+				ReplyMsg:       "",
+				ReferOriginMsg: false,
+				FromMsg:        msg,
+			}, false
+		} else if err.Error() == "game over" {
+			delete(userGameMap, util.From(msg.GroupInfo.GroupId, msg.UserId))
+			handler.UnRegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId)
+		}
+	}
+	if len(response) > 1 && response[len(response)-1:] == "\n" {
+		response = response[:len(response)-1]
+	}
+	return model.Reply{
+		ReplyMsg:       response,
+		ReferOriginMsg: false,
+		FromMsg:        msg,
+	}, true
+}