Stirling-PDF/frontend/src/hooks/tools/addPassword/useAddPasswordOperation.ts

import { useTranslation } from 'react-i18next';
import { useToolOperation } from '../shared/useToolOperation';
import { createStandardErrorHandler } from '../../../utils/toolErrorHandler';
import { AddPasswordFullParameters, defaultParameters } from './useAddPasswordParameters';
import { defaultParameters as permissionsDefaults } from '../changePermissions/useChangePermissionsParameters';
import { getFormData } from '../changePermissions/useChangePermissionsOperation';

// Static function that can be used by both the hook and automation executor
export const buildAddPasswordFormData = (parameters: AddPasswordFullParameters, file: File): FormData => {
  const formData = new FormData();
  formData.append("fileInput", file);
  formData.append("password", parameters.password);
  formData.append("ownerPassword", parameters.ownerPassword);
  formData.append("keyLength", parameters.keyLength.toString());
  getFormData(parameters.permissions).forEach(([key, value]) => {
    formData.append(key, value);
  });
  return formData;
};

// Full default parameters including permissions for automation
const fullDefaultParameters: AddPasswordFullParameters = {
  ...defaultParameters,
  permissions: permissionsDefaults,
};

// Static configuration object
export const addPasswordOperationConfig = {
  operationType: 'addPassword',
  endpoint: '/api/v1/security/add-password',
  buildFormData: buildAddPasswordFormData,
  filePrefix: 'encrypted_', // Will be overridden in hook with translation
  multiFileEndpoint: false,
  defaultParameters: fullDefaultParameters,
} as const;

export const useAddPasswordOperation = () => {
  const { t } = useTranslation();

  return useToolOperation<AddPasswordFullParameters>({
    ...addPasswordOperationConfig,
    filePrefix: t('addPassword.filenamePrefix', 'encrypted') + '_',
    getErrorMessage: createStandardErrorHandler(t('addPassword.error.failed', 'An error occurred while encrypting the PDF.'))
  });
};