1-3076: you gotta do what you gotta do. impl

Unleash · Nov 6, 2024 · bf9d200 · bf9d200
1 parent 5180319
commit bf9d200
Show file tree

Hide file tree

Showing 5 changed files with 88 additions and 12 deletions.
diff --git a/src/lib/features/project-status/createProjectStatusService.ts b/src/lib/features/project-status/createProjectStatusService.ts
@@ -4,6 +4,10 @@ import EventStore from '../events/event-store';
 import FakeEventStore from '../../../test/fixtures/fake-event-store';
 import ProjectStore from '../project/project-store';
 import FakeProjectStore from '../../../test/fixtures/fake-project-store';
+import FakeApiTokenStore from '../../../test/fixtures/fake-api-token-store';
+import { ApiTokenStore } from '../../db/api-token-store';
+import SegmentStore from '../segment/segment-store';
+import FakeSegmentStore from '../../../test/fixtures/fake-segment-store';
 
 export const createProjectStatusService = (
     db: Db,
@@ -16,15 +20,37 @@ export const createProjectStatusService = (
         config.getLogger,
         config.flagResolver,
     );
-    return new ProjectStatusService({ eventStore, projectStore });
+    const apiTokenStore = new ApiTokenStore(
+        db,
+        config.eventBus,
+        config.getLogger,
+        config.flagResolver,
+    );
+    const segmentStore = new SegmentStore(
+        db,
+        config.eventBus,
+        config.getLogger,
+        config.flagResolver,
+    );
+
+    return new ProjectStatusService({
+        eventStore,
+        projectStore,
+        apiTokenStore,
+        segmentStore,
+    });
 };
 
 export const createFakeProjectStatusService = () => {
     const eventStore = new FakeEventStore();
     const projectStore = new FakeProjectStore();
+    const apiTokenStore = new FakeApiTokenStore();
+    const segmentStore = new FakeSegmentStore();
     const projectStatusService = new ProjectStatusService({
         eventStore,
         projectStore,
+        apiTokenStore,
+        segmentStore,
     });
 
     return {

diff --git a/src/lib/features/project-status/project-status-service.ts b/src/lib/features/project-status/project-status-service.ts
@@ -1,31 +1,66 @@
 import type { ProjectStatusSchema } from '../../openapi';
-import type { IEventStore, IProjectStore, IUnleashStores } from '../../types';
+import type {
+    IApiTokenStore,
+    IEventStore,
+    IProjectStore,
+    ISegmentStore,
+    IUnleashStores,
+} from '../../types';
 
 export class ProjectStatusService {
     private eventStore: IEventStore;
     private projectStore: IProjectStore;
+    private apiTokenStore: IApiTokenStore;
+    private segmentStore: ISegmentStore;
 
     constructor({
         eventStore,
         projectStore,
-    }: Pick<IUnleashStores, 'eventStore' | 'projectStore'>) {
+        apiTokenStore,
+        segmentStore,
+    }: Pick<
+        IUnleashStores,
+        'eventStore' | 'projectStore' | 'apiTokenStore' | 'segmentStore'
+    >) {
         this.eventStore = eventStore;
         this.projectStore = projectStore;
+        this.apiTokenStore = apiTokenStore;
+        this.segmentStore = segmentStore;
     }
 
     async getProjectStatus(projectId: string): Promise<ProjectStatusSchema> {
+        const [
+            connectedEnvironments,
+            members,
+            apiTokens,
+            segments,
+            activityCountByDate,
+        ] = await Promise.all([
+            this.projectStore.getConnectedEnvironmentCountForProject(projectId),
+            this.projectStore.getMembersCountByProject(projectId),
+            this.apiTokenStore
+                .getAll()
+                .then(
+                    (tokens) =>
+                        tokens.filter(
+                            (token) =>
+                                token.project === projectId ||
+                                token.projects.includes(projectId),
+                        ).length,
+                ),
+
+            this.segmentStore.getProjectSegmentCount(projectId),
+            this.eventStore.getProjectEventActivity(projectId),
+        ]);
+
         return {
             resources: {
-                connectedEnvironments:
-                    await this.projectStore.getConnectedEnvironmentCountForProject(
-                        projectId,
-                    ),
-                members: 0,
-                apiTokens: 0,
-                segments: 0,
+                connectedEnvironments,
+                members,
+                apiTokens,
+                segments,
             },
-            activityCountByDate:
-                await this.eventStore.getProjectEventActivity(projectId),
+            activityCountByDate,
         };
     }
 }
diff --git a/src/lib/features/segment/segment-store-type.ts b/src/lib/features/segment/segment-store-type.ts
@@ -25,4 +25,6 @@ export interface ISegmentStore extends Store<ISegment, number> {
     existsByName(name: string): Promise<boolean>;
 
     count(): Promise<number>;
+
+    getProjectSegmentCount(projectId: string): Promise<number>;
 }
diff --git a/src/lib/features/segment/segment-store.ts b/src/lib/features/segment/segment-store.ts
@@ -370,6 +370,15 @@ export default class SegmentStore implements ISegmentStore {
         return Boolean(rows[0]);
     }
 
+    async getProjectSegmentCount(projectId: string): Promise<number> {
+        const result = await this.db.raw(
+            `SELECT COUNT(*) FROM ${T.segments} WHERE segment_project_id = ?`,
+            [projectId],
+        );
+
+        return Number(result.rows[0].count);
+    }
+
     prefixColumns(): string[] {
         return COLUMNS.map((c) => `${T.segments}.${c}`);
     }

diff --git a/src/test/fixtures/fake-segment-store.ts b/src/test/fixtures/fake-segment-store.ts
@@ -62,4 +62,8 @@ export default class FakeSegmentStore implements ISegmentStore {
     }
 
     destroy(): void {}
+
+    async getProjectSegmentCount(): Promise<number> {
+        return 0;
+    }
 }