feat(surveys): add column with responses count

frontend/src/lib/api.ts

@@ -447,6 +447,10 @@ class ApiRequest {
         return this.projectsDetail(teamId).addPathComponent('surveys')
     }
 
+    public surveysResponsesCount(teamId?: TeamType['id']): ApiRequest {
+        return this.projectsDetail(teamId).addPathComponent('surveys/responses_count')
+    }
+
     public survey(id: Survey['id'], teamId?: TeamType['id']): ApiRequest {
         return this.surveys(teamId).addPathComponent(id)
     }
@@ -1440,6 +1444,9 @@ const api = {
         async update(surveyId: Survey['id'], data: Partial<Survey>): Promise<Survey> {
             return await new ApiRequest().survey(surveyId).update({ data })
         },
+        async getResponsesCount(): Promise<{ [key: string]: number }> {
+            return await new ApiRequest().surveysResponsesCount().get()
+        },
     },
 
     dataWarehouseTables: {

frontend/src/scenes/surveys/Surveys.tsx

@@ -94,14 +94,12 @@ export function Surveys(): JSX.Element {
                                         )
                                     },
                                 },
-                                // TODO: add responses count later
-                                // {
-                                //     title: 'Responses',
-                                //     render: function RenderResponses() {
-                                //         // const responsesCount = getResponsesCount(survey)
-                                //         return <div>{0}</div>
-                                //     },
-                                // },
+                                {
+                                    title: 'Responses',
+                                    render: function RenderResponses(_, survey) {
+                                        return <div>{survey.responses_count}</div>
+                                    },
+                                },
                                 {
                                     dataIndex: 'type',
                                     title: 'Type',

frontend/src/scenes/surveys/surveysLogic.tsx

@@ -25,8 +25,8 @@ export const surveysLogic = kea<surveysLogicType>([
         surveys: {
             __default: [] as Survey[],
             loadSurveys: async () => {
-                const response = await api.surveys.list()
-                return response.results
+                const responseSurveys = await api.surveys.list()
+                return responseSurveys.results
             },
             deleteSurvey: async (id) => {
                 await api.surveys.delete(id)
@@ -36,6 +36,13 @@ export const surveysLogic = kea<surveysLogicType>([
                 const updatedSurvey = await api.surveys.update(id, { ...updatePayload })
                 return values.surveys.map((survey) => (survey.id === id ? updatedSurvey : survey))
             },
+            loadResponsesCount: async () => {
+                const surveysResponsesCount = await api.surveys.getResponsesCount()
+                return values.surveys.map((survey) => ({
+                    ...survey,
+                    responses_count: surveysResponsesCount[survey.id] || 0,
+                }))
+            },
         },
     })),
     listeners(() => ({
@@ -72,5 +79,6 @@ export const surveysLogic = kea<surveysLogicType>([
     }),
     afterMount(async ({ actions }) => {
         await actions.loadSurveys()
+        await actions.loadResponsesCount()
     }),
 ])

frontend/src/types.ts

@@ -2083,6 +2083,7 @@ export interface Survey {
     end_date: string | null
     archived: boolean
     remove_targeting_flag?: boolean
+    responses_count?: number
 }
 
 export enum SurveyType {

posthog/api/survey.py

@@ -4,12 +4,14 @@
 
 from posthog.api.shared import UserBasicSerializer
 from posthog.api.utils import get_token
+from posthog.client import sync_execute
 from posthog.exceptions import generate_exception_response
 from posthog.models.feedback.survey import Survey
 from rest_framework.response import Response
+from rest_framework.decorators import action
 from posthog.api.feature_flag import FeatureFlagSerializer, MinimalFeatureFlagSerializer
 from posthog.api.routing import StructuredViewSetMixin
-from rest_framework import serializers, viewsets
+from rest_framework import serializers, viewsets, request
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.request import Request
 from rest_framework import status
@@ -138,7 +140,6 @@ def create(self, validated_data):
         return super().create(validated_data)
 
     def update(self, instance: Survey, validated_data):
-
         if validated_data.get("remove_targeting_flag"):
             if instance.targeting_flag:
                 instance.targeting_flag.delete()
@@ -207,6 +208,24 @@ def destroy(self, request: Request, *args: Any, **kwargs: Any) -> Response:
 
         return super().destroy(request, *args, **kwargs)
 
+    @action(methods=["GET"], detail=False)
+    def responses_count(self, request: request.Request, **kwargs):
+        data = sync_execute(
+            f"""
+            SELECT JSONExtractString(properties, '$survey_id') as survey_id, count()
+            FROM events
+            WHERE event = 'survey sent' AND team_id = %(team_id)s
+            GROUP BY survey_id
+        """,
+            {"team_id": self.team_id},
+        )
+
+        counts = {}
+        for survey_id, count in data:
+            counts[survey_id] = count
+
+        return Response(counts)
+
 
 class SurveyAPISerializer(serializers.ModelSerializer):
     """