feat: add project metric

Author: ulemons

services/libs/data-access-layer/src/dashboards/base.ts

@@ -6,6 +6,53 @@ export async function getMetrics(
   qx: QueryExecutor,
   segmentId?: string,
 ): Promise<IDashboardMetrics> {
+  try {
+    const [snapshotData, projectsData] = await Promise.all([
+      getSnapshotMetrics(qx, segmentId),
+      getProjectsCount(qx, segmentId),
+    ])
+
+    if (!snapshotData) {
+      // TODO: remove this mock once Tinybird sinks are available
+      const mockMetrics = getMockMetrics()
+      return {
+        ...mockMetrics,
+        projectsTotal: projectsData.projectsTotal,
+        projectsLast30Days: projectsData.projectsLast30Days,
+      }
+    }
+
+    return {
+      ...snapshotData,
+      projectsTotal: projectsData.projectsTotal,
+      projectsLast30Days: projectsData.projectsLast30Days,
+    }
+  } catch (error: unknown) {
+    const msg = error instanceof Error ? error.message : ''
+    const code = error && typeof error === 'object' && 'code' in error ? error.code : null
+
+    // Detect missing table
+    const isMissingTable = code === '42P01' || /does not exist/i.test(msg)
+
+    if (isMissingTable) {
+      // TODO: remove this mock once Tinybird sinks are available
+      const mockMetrics = getMockMetrics()
+      const projectsData = await getProjectsCount(qx, segmentId)
+      return {
+        ...mockMetrics,
+        projectsTotal: projectsData.projectsTotal,
+        projectsLast30Days: projectsData.projectsLast30Days,
+      }
+    }
+
+    throw error
+  }
+}
+
+async function getSnapshotMetrics(
+  qx: QueryExecutor,
+  segmentId?: string,
+): Promise<Omit<IDashboardMetrics, 'projectsTotal' | 'projectsLast30Days'> | null> {
   const tableName = segmentId
     ? 'dashboardMetricsPerSegmentSnapshot'
     : 'dashboardMetricsTotalSnapshot'
@@ -31,29 +78,43 @@ export async function getMetrics(
     `
 
   const params = segmentId ? { segmentId } : {}
+  const [row] = await qx.select(query, params)
 
-  try {
-    const [row] = await qx.select(query, params)
-
-    if (!row) {
-      // TODO: remove this mock once Tinybird sinks are available
-      return getMockMetrics()
-    }
-
-    return row
-  } catch (error: unknown) {
-    const msg = error instanceof Error ? error.message : ''
-    const code = error && typeof error === 'object' && 'code' in error ? error.code : null
+  return row || null
+}
 
-    // Detect missing table
-    const isMissingTable = code === '42P01' || /does not exist/i.test(msg)
+async function getProjectsCount(
+  qx: QueryExecutor,
+  segmentId?: string,
+): Promise<{ projectsTotal: number; projectsLast30Days: number }> {
+  let query: string
+  let params: Record<string, any>
 
-    if (isMissingTable) {
-      // TODO: remove this mock once Tinybird sinks are available
-      return getMockMetrics()
-    }
+  if (!segmentId) {
+    // Count all segments
+    query = `
+      SELECT 
+        COUNT(*) as "projectsTotal",
+        COUNT(CASE WHEN "createdAt" >= NOW() - INTERVAL '30 days' THEN 1 END) as "projectsLast30Days"
+      FROM segments 
+    `
+    params = {}
+  } else {
+    // Count segments where the provided segmentId is current, parent, or grandparent
+    query = `
+      SELECT 
+        COUNT(*) as "projectsTotal",
+        COUNT(CASE WHEN s."createdAt" >= NOW() - INTERVAL '30 days' THEN 1 END) as "projectsLast30Days"
+      FROM segments s
+      WHERE (s.id = $(segmentId) OR s."parentId" = $(segmentId) OR s."grandparentId" = $(segmentId))
+    `
+    params = { segmentId }
+  }
 
-    throw error
+  const [result] = await qx.select(query, params)
+  return {
+    projectsTotal: parseInt(result.projectsTotal) || 0,
+    projectsLast30Days: parseInt(result.projectsLast30Days) || 0,
   }
 }