package com.acumenvelocity.ath.controller;
   
   import java.io.File;
   import java.io.FileOutputStream;
   import java.io.InputStream;
   
   import javax.ws.rs.core.Response.Status;
   
   import com.acumenvelocity.ath.common.AthUtil;
  import com.acumenvelocity.ath.common.ControllerUtil;
  import com.acumenvelocity.ath.common.Log;
  import com.acumenvelocity.ath.common.PdfUtil;
  import com.acumenvelocity.ath.common.Response;
  import com.acumenvelocity.ath.model.OcrMode;
  
  import io.swagger.oas.inflector.models.RequestContext;
  import io.swagger.oas.inflector.models.ResponseContext;
  import net.sf.okapi.common.LocaleId;
  import net.sf.okapi.common.MimeTypeMapper;
  import net.sf.okapi.common.StreamUtil;
  
  public class UtilityController {
  
    /**
     * Convert PDF to DOCX using Adobe PDF Services
     * 
     * @param request     the request context
     * @param pdfFile     the PDF file to convert
     * @param pdfFileName the original name of the PDF file
     * @param language    the language code (e.g., "en-US", "fr-FR")
     * @param ocrMode     OCR mode: AUTO (default), ENABLED, or DISABLED
     * @return ResponseContext containing the converted DOCX file or error
     */
    public ResponseContext convertPdfToDocx(RequestContext request, File pdfFile,
        String pdfFileName, String language, String ocrModeValue) {
  
      if (!ControllerUtil.checkParam(pdfFile)) {
        return Response.error(400, "Invalid request, pdfFile is not specified");
      }
  
      if (!ControllerUtil.checkParam(pdfFileName)) {
        return Response.error(400, "Invalid request, pdfFileName is not specified");
      }
  
      if (!ControllerUtil.checkParam(language)) {
        return Response.error(400, "Invalid request, language is not specified");
      }
  
      OcrMode ocrMode;
      
      // Default to AUTO if not specified
      if (ocrModeValue == null) {
        ocrMode = OcrMode.AUTO;
        
      } else {
        ocrMode = OcrMode.fromValue(ocrModeValue);
      }
  
      try {
        LocaleId locale = LocaleId.fromString(language);
  
        // Determine OCR mode if set to AUTO
        if (ocrMode == OcrMode.AUTO) {
          boolean needsOcr = PdfUtil.needsOcr(pdfFile);
  
          if (needsOcr) {
            ocrMode = OcrMode.ENABLED;
  
            Log.info(UtilityController.class,
                "PDF '{}' requires OCR, enabling OCR mode", pdfFileName);
  
          } else {
            ocrMode = OcrMode.DISABLED;
  
            Log.info(UtilityController.class,
                "PDF '{}' contains selectable text, disabling OCR mode", pdfFileName);
          }
        }
  
        Log.info(UtilityController.class,
            "Converting PDF '{}' to DOCX with language='{}', ocrMode={}",
            pdfFileName, language, ocrMode);
  
        // Convert PDF to DOCX
        InputStream docxStream = PdfUtil.convertPdfToDocx(
            new java.io.FileInputStream(pdfFile),
            locale,
            ocrMode);
  
        // Generate output filename by replacing .pdf extension with .docx
        String docxFileName = pdfFileName;
  
        if (docxFileName.toLowerCase().endsWith(".pdf")) {
          docxFileName = docxFileName.substring(0, docxFileName.length() - 4) + ".docx";
  
        } else {
          docxFileName = docxFileName + ".docx";
        }
  
       Log.info(UtilityController.class,
           "Successfully converted PDF '{}' to DOCX '{}'",
           pdfFileName, docxFileName);
 
       // Create temp file to hold the converted content
       File outputFile = AthUtil.createTempFile();
 
       try (FileOutputStream fos = new FileOutputStream(outputFile)) {
         StreamUtil.copy(docxStream, fos);
       }
 
       return Response.builder()
           .status(Status.OK)
           .header("Content-Type",
               MimeTypeMapper.DOCX_MIME_TYPE)
           .header("Content-Disposition",
               String.format("attachment; filename=\"%s\"", docxFileName))
           .entity(outputFile)
           .build();
 
     } catch (Exception e) {
       Log.error(UtilityController.class, e,
           "Error converting PDF '{}' to DOCX", pdfFileName);
 
       return Response.error(500, e,
           "Failed to convert PDF to DOCX: " + e.getMessage());
     }
   }
 
   /**
    * Convert DOCX to PDF using Adobe PDF Services
    * 
    * @param request      the request context
    * @param docxFile     the DOCX file to convert
    * @param docxFileName the original name of the DOCX file
    * @param language     the language code (e.g., "en-US", "fr-FR")
    * @return ResponseContext containing the converted PDF file or error
    */
   public ResponseContext convertDocxToPdf(RequestContext request, File docxFile,
       String docxFileName, String language) {
 
     if (!ControllerUtil.checkParam(docxFile)) {
       return Response.error(400, "Invalid request, docxFile is not specified");
     }
 
     if (!ControllerUtil.checkParam(docxFileName)) {
       return Response.error(400, "Invalid request, docxFileName is not specified");
     }
 
     if (!ControllerUtil.checkParam(language)) {
       return Response.error(400, "Invalid request, language is not specified");
     }
 
     try {
       LocaleId locale = LocaleId.fromString(language);
 
       Log.info(UtilityController.class,
           "Converting DOCX '{}' to PDF with language='{}'",
           docxFileName, language);
 
       // Convert DOCX to PDF
       InputStream pdfStream = PdfUtil.convertDocxToPdf(
           new java.io.FileInputStream(docxFile),
           locale);
 
       // Generate output filename by replacing .docx extension with .pdf
       String pdfFileName = docxFileName;
 
       if (pdfFileName.toLowerCase().endsWith(".docx")) {
         pdfFileName = pdfFileName.substring(0, pdfFileName.length() - 5) + ".pdf";
 
       } else {
         pdfFileName = pdfFileName + ".pdf";
       }
 
       Log.info(UtilityController.class,
           "Successfully converted DOCX '{}' to PDF '{}'",
           docxFileName, pdfFileName);
 
       // Create temp file to hold the converted content
       File outputFile = AthUtil.createTempFile();
 
       try (FileOutputStream fos = new FileOutputStream(outputFile)) {
         StreamUtil.copy(pdfStream, fos);
       }
 
       return Response.builder()
           .status(Status.OK)
           .header("Content-Type",
               MimeTypeMapper.PDF_MIME_TYPE)
           .header("Content-Disposition",
               String.format("attachment; filename=\"%s\"", pdfFileName))
           .entity(outputFile)
           .build();
 
     } catch (Exception e) {
       Log.error(UtilityController.class, e,
           "Error converting DOCX '{}' to PDF", docxFileName);
 
       return Response.error(500, e,
           "Failed to convert DOCX to PDF: " + e.getMessage());
     }
   }
 }