analyzer.js

import { getMockResponse, isMockingModeEnabled } from '../../__mocks__/webviewer-redaction-ai.mock.js';

// Send the loaded document text to the server, to be
// analyzed for personal information identification (PII)
const sendTextToServer = async () => {
  // *********************************************
  // MOCKING MODE: Skip actual server call and 
  // return mock document id
  if (isMockingModeEnabled())
    return getMockResponse('documentId');
  // *********************************************

  try {
    const response = await fetch('/api/send-text', {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        documentText: loadedDocument.text,
        timestamp: new Date().toISOString()
      }),
    });

    if (!response.ok)
      throw new Error(`Server error: ${response.status} ${response.statusText}`);

    return await response.json();
  } catch (error) {
    console.error('Error sending document text to server:', error);
    throw error;
  }
}

// Analyze the loaded document text for personal information identification (PII)
const analyzeDocument = async (documentId) => {
  // *********************************************
  // MOCKING MODE: Skip actual server call
  if (isMockingModeEnabled())
    return;
  // *********************************************

  try {
    const response = await fetch('/api/analyze-pii', {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        documentId: documentId,
        timestamp: new Date().toISOString()
      }),
    });

    if (!response.ok)
      throw new Error(`Server error: ${response.status} ${response.statusText}`);

    return await response.json();
  } catch (error) {
    console.error('Error analyzing document for PII:', error);
    throw error;
  }
}

// Receive analysis result from the server
const getAnalysisResult = async (documentId) => {
  // *********************************************
  // MOCKING MODE: Return mock analysis result
  if (isMockingModeEnabled())
    return getMockResponse('analysis');
  // *********************************************

  try {
    const response = await fetch(`/api/get-results/${documentId}`, {
      method: 'GET',
      headers: {
        'Content-Type': 'application/json',
      },
    });

    if (!response.ok)
      throw new Error(`Server error: ${response.status} ${response.statusText}`);

    return await response.json();
  } catch (error) {
    console.error('Error getting analysis result:', error);
    throw error;
  }
}

const analyzeDocumentForPII = async () => {
  // Show WebViewer loading spinner
  WebViewer.getInstance().UI.openElements('loadingModal');

  // *********************************************
  // MOCKING MODE: Keep spinner visible for
  // 2 seconds
  if (isMockingModeEnabled())
    await new Promise((resolve) => setTimeout(resolve, 2000));
  // *********************************************

  try {
    // Step 1: Send document text to server
    const sendResult = await sendTextToServer();
    const documentId = sendResult.documentId;

    // Step 2: Analyze document for PII
    await analyzeDocument(documentId);

    // Step 3: Get analysis result from server
    aiAnalysisResult = await getAnalysisResult(documentId);
  } catch (error) {
    console.error('Failed to analyze document:', error);
  }

  // Hide WebViewer loading spinner
  WebViewer.getInstance().UI.closeElements('loadingModal');
}

export { analyzeDocumentForPII };