zkldi_Tachi/server/src/utils/queries/summary.ts

import db from "external/mongo/db";
import { GetGradeLampDistributionForFolders } from "utils/folder";
import { GetTimeXHoursAgo } from "utils/misc";
import type { integer } from "tachi-common";

// Various utils related to the player summary endpoint.
const REASONABLE_HOURS_AGO = 16;

export function GetRecentPlaycount(userID: integer) {
	const time = GetTimeXHoursAgo(REASONABLE_HOURS_AGO);

	return db.scores.count({ userID, timeAchieved: { $gte: time } });
}

export function GetRecentSessions(userID: integer) {
	const time = GetTimeXHoursAgo(REASONABLE_HOURS_AGO);

	return db.sessions.find({
		userID,
		timeEnded: { $gte: time },
	});
}

export async function GetRecentlyViewedFoldersAnyGPT(userID: integer) {
	const time = GetTimeXHoursAgo(REASONABLE_HOURS_AGO);

	const views = await db["recent-folder-views"].find(
		{
			userID,
			lastViewed: { $gte: time },
		},
		{
			sort: {
				lastViewed: -1,
			},
			limit: 4,
		}
	);

	const folders = await db.folders.find({
		folderID: { $in: views.map((e) => e.folderID) },
	});

	const stats = await GetGradeLampDistributionForFolders(userID, folders);

	// TODO: Sort recently viewed folders based on how recently viewed
	// they were.

	return { folders, stats };
}

export async function GetGoalSummary(userID: integer) {
	const time = GetTimeXHoursAgo(REASONABLE_HOURS_AGO);

	const achievedGoals = await db["goal-subs"].find({
		timeAchieved: { $gte: time },
		userID,
	});

	const improvedGoals = await db["goal-subs"].find({
		lastInteraction: { $gte: time },
		achieved: false,
		userID,
	});

	const goals = await db.goals.find({
		goalID: {
			$in: [...achievedGoals.map((e) => e.goalID), ...improvedGoals.map((e) => e.goalID)],
		},
	});

	return { achievedGoals, improvedGoals, goals };
}