feat(security-questionnaire): add action to answer a single question (#1826)

Co-authored-by: Tofik Hasanov <[email protected]>

Author: github-actions[bot]

apps/app/src/app/(app)/[orgId]/security-questionnaire/actions/answer-single-question.ts

@@ -0,0 +1,71 @@
+'use server';
+
+import { authActionClient } from '@/actions/safe-action';
+import { answerQuestion } from '@/jobs/tasks/vendors/answer-question';
+import { z } from 'zod';
+import { headers } from 'next/headers';
+import { revalidatePath } from 'next/cache';
+
+const inputSchema = z.object({
+  question: z.string(),
+  questionIndex: z.number(),
+  totalQuestions: z.number(),
+});
+
+export const answerSingleQuestionAction = authActionClient
+  .inputSchema(inputSchema)
+  .metadata({
+    name: 'answer-single-question',
+    track: {
+      event: 'answer-single-question',
+      channel: 'server',
+    },
+  })
+  .action(async ({ parsedInput, ctx }) => {
+    const { question, questionIndex, totalQuestions } = parsedInput;
+    const { session } = ctx;
+
+    if (!session?.activeOrganizationId) {
+      throw new Error('No active organization');
+    }
+
+    const organizationId = session.activeOrganizationId;
+
+    try {
+      // Call answerQuestion function directly
+      const result = await answerQuestion(
+        {
+          question,
+          organizationId,
+          questionIndex,
+          totalQuestions,
+        },
+        {
+          useMetadata: false,
+        },
+      );
+
+      // Revalidate the page to show updated answer
+      const headersList = await headers();
+      let path = headersList.get('x-pathname') || headersList.get('referer') || '';
+      path = path.replace(/\/[a-z]{2}\//, '/');
+      revalidatePath(path);
+
+      return {
+        success: result.success,
+        data: {
+          questionIndex: result.questionIndex,
+          question: result.question,
+          answer: result.answer,
+          sources: result.sources,
+          error: result.error,
+        },
+      };
+    } catch (error) {
+      return {
+        success: false,
+        error: error instanceof Error ? error.message : 'Failed to answer question',
+      };
+    }
+  });
+

apps/app/src/app/(app)/[orgId]/security-questionnaire/hooks/useQuestionnaireDetail/useQuestionnaireDetail.ts

@@ -94,7 +94,6 @@ export function useQuestionnaireDetail({
 
   // Single answer hook
   const singleAnswer = useQuestionnaireSingleAnswer({
-    singleAnswerToken: state.singleAnswerToken,
     results: state.results.map((r) => ({
       question: r.question,
       answer: r.answer,

apps/app/src/app/(app)/[orgId]/security-questionnaire/hooks/useQuestionnaireDetail/useQuestionnaireDetailState.ts

@@ -100,23 +100,17 @@ export function useQuestionnaireDetailState({
 
   const deleteAnswerAction = useAction(deleteQuestionnaireAnswer);
 
-  // Create trigger tokens
+  // Create trigger token for auto-answer (single question answers now use server action)
   useEffect(() => {
-    const fetchTokens = async () => {
-      const [autoTokenResult, singleTokenResult] = await Promise.all([
-        createTriggerToken('vendor-questionnaire-orchestrator'),
-        createTriggerToken('answer-question'),
-      ]);
+    const fetchToken = async () => {
+      const autoTokenResult = await createTriggerToken('vendor-questionnaire-orchestrator');
 
       if (autoTokenResult.success && autoTokenResult.token) {
         setAutoAnswerToken(autoTokenResult.token);
       }
-      if (singleTokenResult.success && singleTokenResult.token) {
-        setSingleAnswerToken(singleTokenResult.token);
-      }
     };
 
-    fetchTokens();
+    fetchToken();
   }, []);
 
   // Sync queue ref with state

apps/app/src/app/(app)/[orgId]/security-questionnaire/hooks/useQuestionnaireParse.ts

@@ -16,8 +16,6 @@ interface UseQuestionnaireParseProps {
   parseToken: string | null;
   autoAnswerToken: string | null;
   setAutoAnswerToken: (token: string | null) => void;
-  singleAnswerToken: string | null;
-  setSingleAnswerToken: (token: string | null) => void;
   setIsParseProcessStarted: (started: boolean) => void;
   setParseTaskId: (id: string | null) => void;
   setParseToken: (token: string | null) => void;
@@ -36,8 +34,6 @@ export function useQuestionnaireParse({
   parseToken,
   autoAnswerToken,
   setAutoAnswerToken,
-  singleAnswerToken,
-  setSingleAnswerToken,
   setIsParseProcessStarted,
   setParseTaskId,
   setParseToken,
@@ -62,18 +58,6 @@ export function useQuestionnaireParse({
     }
   }, [autoAnswerToken, setAutoAnswerToken]);
 
-  // Get trigger token for single answer (can trigger and read)
-  useEffect(() => {
-    async function getSingleAnswerToken() {
-      const result = await createTriggerToken('answer-question');
-      if (result.success && result.token) {
-        setSingleAnswerToken(result.token);
-      }
-    }
-    if (!singleAnswerToken) {
-      getSingleAnswerToken();
-    }
-  }, [singleAnswerToken, setSingleAnswerToken]);
 
   // Track parse task with realtime hook
   const { run: parseRun, error: parseError } = useRealtimeRun<typeof parseQuestionnaireTask>(

apps/app/src/app/(app)/[orgId]/security-questionnaire/hooks/useQuestionnaireParser.ts

@@ -15,8 +15,6 @@ export function useQuestionnaireParser() {
     parseToken: state.parseToken,
     autoAnswerToken: state.autoAnswerToken,
     setAutoAnswerToken: state.setAutoAnswerToken,
-    singleAnswerToken: state.singleAnswerToken,
-    setSingleAnswerToken: state.setSingleAnswerToken,
     setIsParseProcessStarted: state.setIsParseProcessStarted,
     setParseTaskId: state.setParseTaskId,
     setParseToken: state.setParseToken,
@@ -44,7 +42,6 @@ export function useQuestionnaireParser() {
   });
 
   const singleAnswer = useQuestionnaireSingleAnswer({
-    singleAnswerToken: state.singleAnswerToken,
     results: state.results,
     answeringQuestionIndex: state.answeringQuestionIndex,
     setResults: state.setResults,