-
Notifications
You must be signed in to change notification settings - Fork 298
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Co-authored-by: Konsti Wohlwend <[email protected]>
- Loading branch information
1 parent
ac0d657
commit cd35e8c
Showing
29 changed files
with
78,806 additions
and
36 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,252 @@ | ||
| import { prismaClient } from "@/prisma-client"; | ||
| import { createSmartRouteHandler } from "@/route-handlers/smart-route-handler"; | ||
| import { KnownErrors } from "@stackframe/stack-shared"; | ||
| import { ProjectsCrud } from "@stackframe/stack-shared/dist/interface/crud/projects"; | ||
| import { UsersCrud } from "@stackframe/stack-shared/dist/interface/crud/users"; | ||
| import { adaptSchema, adminAuthTypeSchema, yupArray, yupMixed, yupNumber, yupObject, yupString } from "@stackframe/stack-shared/dist/schema-fields"; | ||
| import yup from 'yup'; | ||
| import { usersCrudHandlers } from "../../users/crud"; | ||
|
|
||
| type DataPoints = yup.InferType<typeof DataPointsSchema>; | ||
|
|
||
| const DataPointsSchema = yupArray(yupObject({ | ||
| date: yupString().defined(), | ||
| activity: yupNumber().defined(), | ||
| }).defined()).defined(); | ||
|
|
||
|
|
||
| async function loadUsersByCountry(projectId: string): Promise<Record<string, number>> { | ||
| const a = await prismaClient.$queryRaw<{countryCode: string|null, userCount: bigint}[]>` | ||
| WITH LatestEventWithCountryCode AS ( | ||
| SELECT DISTINCT ON ("userId") | ||
| "data"->'userId' AS "userId", | ||
| "countryCode", | ||
| "eventStartedAt" AS latest_timestamp | ||
| FROM "Event" | ||
| LEFT JOIN "EventIpInfo" eip | ||
| ON "Event"."endUserIpInfoGuessId" = eip.id | ||
| WHERE '$user-activity' = ANY("systemEventTypeIds"::text[]) | ||
| AND "data"->>'projectId' = ${projectId} | ||
| AND "countryCode" IS NOT NULL | ||
| ORDER BY "userId", "eventStartedAt" DESC | ||
| ) | ||
| SELECT "countryCode", COUNT("userId") AS "userCount" | ||
| FROM LatestEventWithCountryCode | ||
| GROUP BY "countryCode" | ||
| ORDER BY "userCount" DESC; | ||
| `; | ||
| console.log("AAAAAAAAAA", a); | ||
|
|
||
| const rec = Object.fromEntries( | ||
| a.map(({ userCount, countryCode }) => [countryCode, Number(userCount)]) | ||
| .filter(([countryCode, userCount]) => countryCode) | ||
| ); | ||
| return rec; | ||
| } | ||
|
|
||
| async function loadTotalUsers(projectId: string, now: Date): Promise<DataPoints> { | ||
| return (await prismaClient.$queryRaw<{date: Date, dailyUsers: bigint, cumUsers: bigint}[]>` | ||
| WITH date_series AS ( | ||
| SELECT GENERATE_SERIES( | ||
| ${now}::date - INTERVAL '1 month', | ||
| ${now}::date, | ||
| '1 day' | ||
| ) | ||
| AS registration_day | ||
| ) | ||
| SELECT | ||
| ds.registration_day AS "date", | ||
| COALESCE(COUNT(pu."projectUserId"), 0) AS "dailyUsers", | ||
| SUM(COALESCE(COUNT(pu."projectUserId"), 0)) OVER (ORDER BY ds.registration_day) AS "cumUsers" | ||
| FROM date_series ds | ||
| LEFT JOIN "ProjectUser" pu | ||
| ON DATE(pu."createdAt") = ds.registration_day AND pu."projectId" = ${projectId} | ||
| GROUP BY ds.registration_day | ||
| ORDER BY ds.registration_day | ||
| `).map((x) => ({ | ||
| date: x.date.toISOString().split('T')[0], | ||
| activity: Number(x.dailyUsers), | ||
| })); | ||
| } | ||
|
|
||
| async function loadDailyActiveUsers(projectId: string, now: Date) { | ||
| const res = await prismaClient.$queryRaw<{day: Date, dau: bigint}[]>` | ||
| WITH date_series AS ( | ||
| SELECT GENERATE_SERIES( | ||
| ${now}::date - INTERVAL '1 month', | ||
| ${now}::date, | ||
| '1 day' | ||
| ) | ||
| AS "day" | ||
| ), | ||
| daily_users AS ( | ||
| SELECT | ||
| DATE_TRUNC('day', "eventStartedAt") AS "day", | ||
| COUNT(DISTINCT "data"->'userId') AS "dau" | ||
| FROM "Event" | ||
| WHERE "eventStartedAt" >= ${now} - INTERVAL '1 month' | ||
| AND "eventStartedAt" < ${now} | ||
| AND '$user-activity' = ANY("systemEventTypeIds"::text[]) | ||
| AND "data"->>'projectId' = ${projectId} | ||
| GROUP BY DATE_TRUNC('day', "eventStartedAt") | ||
| ) | ||
| SELECT ds."day", COALESCE(du.dau, 0) AS dau | ||
| FROM date_series ds | ||
| LEFT JOIN daily_users du | ||
| ON ds."day" = du."day" | ||
| ORDER BY ds."day" | ||
| `; | ||
|
|
||
| return res.map(x => ({ | ||
| date: x.day.toISOString().split('T')[0], | ||
| activity: Number(x.dau), | ||
| })); | ||
| } | ||
|
|
||
| async function loadLoginMethods(projectId: string): Promise<{method: string, count: number }[]> { | ||
| return await prismaClient.$queryRaw<{ method: string, count: number }[]>` | ||
| WITH tab AS ( | ||
| SELECT | ||
| COALESCE( | ||
| soapc."type"::text, | ||
| poapc."type"::text, | ||
| CASE WHEN pam IS NOT NULL THEN 'password' ELSE NULL END, | ||
| CASE WHEN pkm IS NOT NULL THEN 'passkey' ELSE NULL END, | ||
| CASE WHEN oam IS NOT NULL THEN 'otp' ELSE NULL END, | ||
| 'other' | ||
| ) AS "method", | ||
| method.id AS id | ||
| FROM | ||
| "AuthMethod" method | ||
| LEFT JOIN "OAuthAuthMethod" oaam ON method.id = oaam."authMethodId" | ||
| LEFT JOIN "OAuthProviderConfig" oapc | ||
| ON oaam."projectConfigId" = oapc."projectConfigId" AND oaam."oauthProviderConfigId" = oapc.id | ||
| LEFT JOIN "StandardOAuthProviderConfig" soapc | ||
| ON oapc."projectConfigId" = soapc."projectConfigId" AND oapc.id = soapc.id | ||
| LEFT JOIN "ProxiedOAuthProviderConfig" poapc | ||
| ON oapc."projectConfigId" = poapc."projectConfigId" AND oapc.id = poapc.id | ||
| LEFT JOIN "PasswordAuthMethod" pam ON method.id = pam."authMethodId" | ||
| LEFT JOIN "PasskeyAuthMethod" pkm ON method.id = pkm."authMethodId" | ||
| LEFT JOIN "OtpAuthMethod" oam ON method.id = oam."authMethodId" | ||
| WHERE method."projectId" = ${projectId}) | ||
| SELECT LOWER("method") AS method, COUNT(id)::int AS "count" FROM tab | ||
| GROUP BY "method" | ||
| `; | ||
| } | ||
|
|
||
| async function loadRecentlyActiveUsers(project: ProjectsCrud["Admin"]["Read"]): Promise<UsersCrud["Admin"]["Read"][]> { | ||
| // use the Events table to get the most recent activity | ||
| const events = await prismaClient.$queryRaw<{ data: any, eventStartedAt: Date }[]>` | ||
| WITH RankedEvents AS ( | ||
| SELECT | ||
| "data", "eventStartedAt", | ||
| ROW_NUMBER() OVER ( | ||
| PARTITION BY "data"->>'userId' | ||
| ORDER BY "eventStartedAt" DESC | ||
| ) as rn | ||
| FROM "Event" | ||
| WHERE "data"->>'projectId' = ${project.id} | ||
| AND '$user-activity' = ANY("systemEventTypeIds"::text[]) | ||
| ) | ||
| SELECT "data", "eventStartedAt" | ||
| FROM RankedEvents | ||
| WHERE rn = 1 | ||
| ORDER BY "eventStartedAt" DESC | ||
| LIMIT 5 | ||
| `; | ||
| const userObjects: UsersCrud["Admin"]["Read"][] = []; | ||
| for (const event of events) { | ||
| let user; | ||
| try { | ||
| user = await usersCrudHandlers.adminRead({ | ||
| project, | ||
| user_id: event.data.userId, | ||
| allowedErrorTypes: [ | ||
| KnownErrors.UserNotFound, | ||
| ], | ||
| }); | ||
| } catch (e) { | ||
| if (e instanceof KnownErrors.UserNotFound) { | ||
| // user probably deleted their account, skip | ||
| continue; | ||
| } | ||
| throw e; | ||
| } | ||
| userObjects.push(user); | ||
| } | ||
| return userObjects; | ||
| } | ||
|
|
||
| export const GET = createSmartRouteHandler({ | ||
| metadata: { | ||
| hidden: true, | ||
| }, | ||
| request: yupObject({ | ||
| auth: yupObject({ | ||
| type: adminAuthTypeSchema.defined(), | ||
| project: adaptSchema.defined(), | ||
| }), | ||
| }), | ||
| response: yupObject({ | ||
| statusCode: yupNumber().oneOf([200]).defined(), | ||
| bodyType: yupString().oneOf(["json"]).defined(), | ||
| body: yupObject({ | ||
| total_users: yupNumber().integer().defined(), | ||
| daily_users: DataPointsSchema, | ||
| daily_active_users: DataPointsSchema, | ||
| // TODO: Narrow down the types further | ||
| users_by_country: yupMixed().defined(), | ||
| recently_registered: yupMixed().defined(), | ||
| recently_active: yupMixed().defined(), | ||
| login_methods: yupMixed().defined(), | ||
| }).defined(), | ||
| }), | ||
| handler: async (req) => { | ||
| const now = new Date(); | ||
|
|
||
| const [ | ||
| totalUsers, | ||
| dailyUsers, | ||
| dailyActiveUsers, | ||
| usersByCountry, | ||
| recentlyRegistered, | ||
| recentlyActive, | ||
| loginMethods | ||
| ] = await Promise.all([ | ||
| prismaClient.projectUser.count({ | ||
| where: { projectId: req.auth.project.id, }, | ||
| }), | ||
| loadTotalUsers(req.auth.project.id, now), | ||
| loadDailyActiveUsers(req.auth.project.id, now), | ||
| loadUsersByCountry(req.auth.project.id), | ||
| (await usersCrudHandlers.adminList({ | ||
| project: req.auth.project, | ||
| query: { | ||
| order_by: 'signed_up_at', | ||
| desc: true, | ||
| limit: 5, | ||
| }, | ||
| allowedErrorTypes: [ | ||
| KnownErrors.UserNotFound, | ||
| ], | ||
| })).items, | ||
| loadRecentlyActiveUsers(req.auth.project), | ||
| loadLoginMethods(req.auth.project.id), | ||
| ] as const); | ||
|
|
||
| return { | ||
| statusCode: 200, | ||
| bodyType: "json", | ||
| body: { | ||
| total_users: totalUsers, | ||
| daily_users: dailyUsers, | ||
| daily_active_users: dailyActiveUsers, | ||
| users_by_country: usersByCountry, | ||
| recently_registered: recentlyRegistered, | ||
| recently_active: recentlyActive, | ||
| login_methods: loginMethods, | ||
| } | ||
| }; | ||
| }, | ||
| }); | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
1 change: 1 addition & 0 deletions
1
...d/src/app/(main)/(protected)/projects/[projectId]/metrics/country-data.geo.json
Large diffs are not rendered by default.
Oops, something went wrong.
Oops, something went wrong.