const DOCUMENT_TEMPLATE_IDS = { Certification: '1V0uMuM80BGpjpdt1AmuzlU97tDI_u-y2rOfdl4tkqmc', }; const OUTPUT_FOLDER_ID = '1ROyJXk-QANTHM6Jiw0ne3EQiR1f2UsLr'; declare namespace GoogleAppsScript { namespace Spreadsheet { interface RichTextValue { getLinkUrl(): string | null; getLinkUrl(startOffset: number, endOffset: number): string | null; } } } type DocSection = | GoogleAppsScript.Document.Body | GoogleAppsScript.Document.HeaderSection | GoogleAppsScript.Document.FooterSection; function onOpen() { const ui = SpreadsheetApp.getUi(); ui.createMenu('CMS Document Generation') .addItem('Generate document for current row', 'generateForCurrentRow') .addItem('Generate document for all rows', 'generateForAllRows') // .addItem('CAUTION! Write list of all files to current sheet', 'dumpFiles') .addToUi(); } function findAllFiles(path: string, folder: GoogleAppsScript.Drive.Folder) { const out: string[][] = []; const files = folder.getFiles(); while (files.hasNext()) { const file = files.next(); if (!file.isTrashed()) { out.push([ path, file.getId(), file.getName(), `=HYPERLINK("${file.getUrl()}", "${file.getName()}")`, ]); } } const subfolders = folder.getFolders(); while (subfolders.hasNext()) { const subfolder = subfolders.next(); out.push(...findAllFiles(path + '/' + subfolder.getName(), subfolder)); } return out; } function dumpFiles() { const root = DriveApp.getFolderById('1mWOQDmQjQsi8the09VXpnKnvFUzj4wiv'); const out = findAllFiles('', root); const active_sheet = SpreadsheetApp.getActive(); const range = active_sheet.getRange(`R1C1:R${out.length}C${out[0].length}`); range.setValues(out); } function copyElement( source_element: GoogleAppsScript.Document.Element, dest: DocSection, index?: number ) { const element = source_element.copy(); switch (element.getType()) { case DocumentApp.ElementType.PARAGRAPH: if (index !== undefined) dest.insertParagraph(index, element.asParagraph()); else dest.appendParagraph(element.asParagraph()); break; case DocumentApp.ElementType.TABLE: if (index !== undefined) dest.insertTable(index, element.asTable()); else dest.appendTable(element.asTable()); break; case DocumentApp.ElementType.LIST_ITEM: if (index !== undefined) dest.insertListItem(index, element.asListItem()); else dest.appendListItem(element.asListItem()); break; case DocumentApp.ElementType.INLINE_IMAGE: if (index !== undefined) dest.insertImage(index, element.asInlineImage()); else dest.appendImage(element.asInlineImage()); break; default: throw new Error( "According to the doc this type couldn't appear in the body: " + element.getType() ); } } function copySection(source: DocSection, dest: DocSection, index?: number) { const totalElements = source.getNumChildren(); for (let j = 0; j < totalElements; ++j) { if (index) copyElement(source.getChild(j), dest, index + j); else copyElement(source.getChild(j), dest); } } function spreadsheetRowToObject( sheet: GoogleAppsScript.Spreadsheet.Sheet, rownum: number ): { [key: string]: GoogleAppsScript.Spreadsheet.RichTextValue } { // TODO: could be more efficient const range = sheet.getDataRange(); const headers = range.getValues()[0]; const row_data = range.getRichTextValues()[rownum - 1]; return headers.reduce((acc, header, index) => { acc[String(header)] = row_data[index]; return acc; }, {}); } function trashFiles(files: GoogleAppsScript.Drive.FileIterator) { while (files.hasNext()) { files.next().setTrashed(true); } } function generateForRow( spreadsheet: GoogleAppsScript.Spreadsheet.Spreadsheet, row_num: number ) { const row = spreadsheetRowToObject(spreadsheet.getActiveSheet(), row_num); if (!(row['Type'].getText() in DOCUMENT_TEMPLATE_IDS)) throw new Error(`${row['Type']} is not a valid type of document!`); const template_doc = DocumentApp.openById( DOCUMENT_TEMPLATE_IDS[ row['Type'].getText() as keyof typeof DOCUMENT_TEMPLATE_IDS ] ); const link = row['Document'].getLinkUrl(); if (link === null) { throw new Error(`Link missing for ${row['Document'].getText()}`); } // TODO: should probably handle this better if (!link.includes('document')) return; const source_file = DriveApp.getFileById(DocumentApp.openByUrl(link).getId()); const out_folder = DriveApp.getFolderById(OUTPUT_FOLDER_ID); const out_name = row['Document'].getText() + '_' + (row['Version'].getText() || 'No Version') + '-' + (row['Approved Date'].getText() || 'NOT APPROVED'); // Delete old files with the same name trashFiles(out_folder.getFilesByName(out_name)); trashFiles(out_folder.getFilesByName(out_name + '.pdf')); // Duplicate source document and open const out_file = source_file.makeCopy(out_name, out_folder); const out_doc = DocumentApp.openById(out_file.getId()); // Copy header if (!out_doc.getHeader()) out_doc.addHeader(); copySection(template_doc.getHeader(), out_doc.getHeader().clear()); // Copy footer if (!out_doc.getFooter()) out_doc.addFooter(); copySection(template_doc.getFooter(), out_doc.getFooter().clear()); const insert_point = template_doc .getBody() .findText('{{body}}') ?.getElement(); if (!insert_point) { throw new Error("Could not find insert point '{{body}}'"); } // find the parent element that is a direct descendant of the body let parent = insert_point; while (parent.getParent().getType() != DocumentApp.ElementType.BODY_SECTION) { parent = parent.getParent(); } const index = template_doc.getBody().getChildIndex(parent); // Copy body contents above and below {{body}} tag for (let j = 0; j < template_doc.getBody().getNumChildren(); j++) { if (j < index) copyElement(template_doc.getBody().getChild(j), out_doc.getBody(), j); // don't copy {{body}} tag else if (j == index) continue; else copyElement(template_doc.getBody().getChild(j), out_doc.getBody()); } // do text replacement Object.entries(row).forEach(([header, data]) => { let replacement = data.getText(); if (!replacement) { // default values when empty if (header === 'Version') { replacement = 'No Version'; } else if (header === 'Approved Date') { replacement = 'NOT APPROVED'; } } out_doc.getBody().replaceText(`{{${header}}}`, replacement); out_doc.getHeader().replaceText(`{{${header}}}`, replacement); out_doc.getFooter().replaceText(`{{${header}}}`, replacement); }); out_doc.saveAndClose(); // create PDF file out_folder.createFile(out_file.getAs('application/pdf')); } function generateForCurrentRow() { const spreadsheet = SpreadsheetApp.getActive(); const cell = spreadsheet.getCurrentCell(); if (!cell) throw new Error('No Cell selected for operation on row'); generateForRow(spreadsheet, cell.getRow()); } function generateForAllRows() { const spreadsheet = SpreadsheetApp.getActive(); const row_count = spreadsheet.getActiveSheet().getDataRange().getNumRows(); for (let row_num = 2; row_num < row_count; row_num++) { generateForRow(spreadsheet, row_num); } }