From: John Runyon <redacted>
Date: Sun, 15 Oct 2023 08:17:59 +0000 (-0600)
Subject: trivia - add max-streak tracking
X-Git-Url: https://jfr.im/git/erebus.git/commitdiff_plain/6df8e71aa82c45c74734c5948628b3df5e681fb4

trivia - add max-streak tracking
---

diff --git a/modules/contrib/trivia/trivia.json.example b/modules/contrib/trivia/trivia.json.example
index d353b0e..3aff6e5 100644
--- a/modules/contrib/trivia/trivia.json.example
+++ b/modules/contrib/trivia/trivia.json.example
@@ -27,5 +27,6 @@
     "maxmissedquestions": 25,
     "hinttimer": 10.0,
     "starttime": 0,
-    "target": 500
+    "target": 500,
+    "streaks": {}
 }
diff --git a/modules/trivia.py b/modules/trivia.py
index b716a59..ff106fa 100644
--- a/modules/trivia.py
+++ b/modules/trivia.py
@@ -419,6 +419,13 @@ class TriviaState(object):
 		else:
 			return len(self.db['users'])+1
 
+	def get_streak(self, user):
+		user = str(user).lower()
+		if user in self.db['streaks']:
+			return self.db['streaks'][user]
+		else:
+			return 0
+
 	def targetuser(self, user):
 		if len(self.db['ranks']) == 0: return "no one is ranked!"
 
@@ -466,6 +473,12 @@ def trivia_checkanswer(bot, user, chan, *args):
 		response += ". New score: %d. Rank: %d. Target: %s %s" % (state.addpoint(user), state.rank(user), state.targetuser(user), state.targetpoints(user))
 		bot.fastmsg(chan, response)
 
+		user_lower = str(user).lower()
+		if user_lower not in state.db['streaks']:
+			state.db['streaks'][user_lower] = 0
+		if state.streak > state.db['streaks'][user_lower]:
+			state.db['streaks'][user_lower] = state.streak
+
 		if state.streak >= 3:
 			bot.msg(chan, "\00312%s\003 is on a streak! \00307%d\003 answers correct in a row!" % (user, state.streak))
 
@@ -488,7 +501,7 @@ def save(bot, user, chan, realtarget, *args):
 @lib.hook(needchan=False, wantchan=True)
 @lib.help("[<user>]", "shows how many points you or someone has")
 def points(bot, user, chan, realtarget, *args):
-	if chan is not None: eplyto = chan
+	if chan is not None: replyto = chan
 	else: replyto = user
 
 	if len(args) != 0: who = args[0]
@@ -874,12 +887,52 @@ def delcat(bot, user, chan, realtarget, *args):
 	else:
 		bot.msg(user, "Category does not exist.")
 
+@lib.hook(needchan=False)
+@lib.help("[<nick>]", "shows a user's (or your own) max streak")
+def streak(bot, user, chan, realtarget, *args):
+	if chan is not None: replyto = chan
+	else: replyto = user
+
+	if len(args) != 0: who = args[0]
+	else: who = user
+
+	bot.msg(replyto, "%s's highest streak is %d" % (who, state.get_streak(who)))
+
+@lib.hook(needchan=False)
+@lib.help(None, "shows top streaks of all time")
+def topstreaks(bot, user, chan, realtarget, *args):
+	db = state.db['streaks']
+	streaks = [(k, db[k]) for k in db.keys()]
+	streaks.sort(key=lambda v: v[1], reverse=True)
+	return "Top streaks of all time: %s (%d), %s (%d), %s (%d)." % (streaks[0][0], streaks[0][1], streaks[1][0], streaks[1][1], streaks[2][0], streaks[2][1])
+
+@lib.hook(glevel=lib.MANAGER, needchan=False)
+@lib.help("<nick> <new streak>", "set a user's max streak")
+@lib.argsEQ(2)
+def setstreak(bot, user, chan, realtarget, *args):
+	temp = 0
+	target = args[0].lower()
+	try:
+		newstreak = int(args[1])
+	except ValueError:
+		return "Syntax: SETSTREAK <nick> <new streak>"
+
+	if target in state.db['streaks']:
+		temp = state.db['streaks'][target]
+		if newstreak == 0:
+			del state.db['streaks'][target]
+	if newstreak > 0:
+		state.db['streaks'][target] = newstreak
+	return "Done. Streak used to be %d" % (temp)
+
 @lib.hook(needchan=False)
 def triviahelp(bot, user, chan, realtarget, *args):
 	bot.slowmsg(user,             "START")
 	bot.slowmsg(user,             "TOP10")
 	bot.slowmsg(user,             "POINTS        [<user>]")
 	bot.slowmsg(user,             "RANK          [<user>]")
+	bot.slowmsg(user,             "STREAK        [<user>]")
+	bot.slowmsg(user,             "TOPSTREAKS")
 	bot.slowmsg(user,             "BADQ          <reason> (include info to identify question)")
 	if user.glevel >= 1:
 		bot.slowmsg(user,         "SKIP                            (KNOWN)")
@@ -902,6 +955,7 @@ def triviahelp(bot, user, chan, realtarget, *args):
 				bot.slowmsg(user, "HINTTIMER     <float seconds>   (ADMIN)")
 				bot.slowmsg(user, "HINTNUM       <hints>           (ADMIN)")
 				bot.slowmsg(user, "QUESTIONPAUSE <float seconds>   (ADMIN)")
+				bot.slowmsg(user, "SETSTREAK     <nick> <streak>   (ADMIN)")
 
 @lib.hooknum(332) # topic is...
 @lib.hooknum(331) # no topic set