import * as fs from 'fs';
import { DocumentProcessorServiceClient } from '@google-cloud/documentai';
import { google } from '@google-cloud/documentai/build/protos/protos';


// Set up the Google Cloud Document AI client
const client = new DocumentProcessorServiceClient();
const projectId = 'your-project';
const location = 'region';
const processorId = 'processor ID';

// read in file from local storage
const testPDF = fs.readFileSync("src/test.pdf");
const docProcessor = new DocumentProcessorServiceClient();

// Define the request object for processing the document
const request:google.cloud.documentai.v1.IProcessRequest = {
    name: `projects/${projectId}/locations/${location}/processors/${processorId}`,
    rawDocument: {
        content: testPDF.toString('base64'),
        mimeType: 'application/pdf',
      }
};

// parse tables
function parseTables(response:Array<google.cloud.documentai.v1.ProcessResponse>):any{
    var tables:any[] = [];
    const pagei = 0;
    const tablei = 0;
    response[0].document?.pages?.forEach((page, index) => {
        page.tables?.forEach((table, index) => {
            table.headerRows?.forEach((row, index) => {
                row.cells?.forEach((cell, index) => {
                    console.log(cell.layout?.textAnchor?.content);
                });
            });
            table.bodyRows?.forEach((row, index) => {
                row.cells?.forEach((cell, index) => {
                    console.log(cell.layout?.textAnchor?.content);
                });
            });
            console.log(index);
        });
    });
}

function processDocument(){
    docProcessor.processDocument(request).then(
        data => {
            parseTables(data);  
        console.log(data);
      }).catch(function(error:any) {
        console.log(error);
      }); 
    
}
processDocument();