small changes

plugins/rp.go

@@ -26,6 +26,7 @@ func RegisterRP(bot *laniakea.Bot) {
 	rp = rp.Command(rpUserPromptGet, "rpuserpget")
 	rp = rp.Command(rpUserPromptSet, "rpuserpset")
 	rp = rp.Command(generate, "g", "gen", "г")
+	rp = rp.Command(chatStat, "chatstat")
 	rp = rp.Payload(debugTokens, "rp.tokens")
 
 	bot.AddPlugins(rp.Build())
@@ -350,3 +351,29 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
 		ctx.Error(err)
 	}
 }
+
+func chatStat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
+	redisRpRep := red.NewRPRepository(db)
+	waifuId := redisRpRep.GetSelectedWaifu(ctx.FromID)
+	if waifuId == 0 {
+		ctx.Answer("Не выбрана вайфу")
+		return
+	}
+	chatId := redisRpRep.GetChatId(ctx.FromID, waifuId)
+	if chatId == "" {
+		chatId = uuid.New().String()
+		err := redisRpRep.SetChatId(ctx.FromID, waifuId, chatId)
+		if err != nil {
+			ctx.Error(err)
+			return
+		}
+	}
+	messageCount, err := mdb.GetChatHistorySize(db, chatId)
+	if err != nil {
+		ctx.Error(err)
+		return
+	}
+
+	tokens := redisRpRep.GetChatTokens(ctx.FromID, waifuId)
+	ctx.Answerf("Кол-во сообщений: %d\nКол-во токенов: %d", messageCount, tokens)
+}