litlyx/dashboard/server/api/metrics/[project_id]/data/countries.ts

import { getUserProjectFromId } from "~/server/LIVE_DEMO_DATA";
import { VisitModel } from "@schema/metrics/VisitSchema";
import { DATA_EXPIRE_TIME, Redis } from "~/server/services/CacheService";


export type CountriesAggregated = {
    _id: string,
    count: number
}

export default defineEventHandler(async event => {

    const project_id = getRequestProjectId(event);
    if (!project_id) return;

    const user = getRequestUser(event);
    const project = await getUserProjectFromId(project_id, user);
    if (!project) return;

    const limit = getRequestHeader(event, 'x-query-limit');
    const numLimit = parseInt(limit || '10');

    const from = getRequestHeader(event, 'x-from');
    const to = getRequestHeader(event, 'x-to');

    if (!from || !to) return setResponseStatus(event, 400, 'x-from and x-to headers missing');

    return await Redis.useCache({
        key: `countries:${project_id}:${numLimit}:${from}:${to}`,
        exp: DATA_EXPIRE_TIME
    }, async () => {
        const countries: CountriesAggregated[] = await VisitModel.aggregate([
            {
                $match: {
                    project_id: project._id,
                    country: { $ne: null },
                    created_at: {
                        $gte: new Date(from),
                        $lte: new Date(to)
                    }
                },
            },
            { $group: { _id: "$country", count: { $sum: 1, } } },
            { $sort: { count: -1 } },
            { $limit: numLimit }
        ]);

        return countries;
    });



});