include more bot data

storage/tokens.go

@@ -164,7 +164,7 @@ func (t TokenBot) WithUser() TokenUserBot {
 	return TokenUserBot{
 		ID:   t.ID,
 		End:  t.End,
-		User: GetBot(t.BotID),
+		User: GetBot(t.BotID).User,
 	}
 }
 
@@ -226,4 +226,25 @@ func BotTokenFromToken(token string) (bool, string) {
 		return false, ""
 	}
 	return true, getBotToken(tokenBot.BotID)
+}
+
+func TokensUnderBot(botID string) []SimpleTokenUserBot {
+	var tokens []TokenTiny
+	err := client.SMembers(ctx, "tokens").ScanSlice(&tokens)
+	if err != nil {
+		panic(err)
+	}
+
+	simpleTokens := make([]SimpleTokenUserBot, 0)
+	for _, t := range tokens {
+		var token TokenBot
+		err := client.Get(ctx, "token:" + t.ID).Scan(&token)
+		if err != nil {
+			panic(err)
+		}
+		if token.BotID == botID {
+			simpleTokens = append(simpleTokens, token.WithUser().Simplify())
+		}
+	}
+	return simpleTokens
 }