feat(plugins): onbefore and onafter

Author: Novout

scripts/build.js

@@ -46,7 +46,7 @@ const client = () => {
         importPath,
         correctSet,
         vendorSourceMap,
-        dirnameAndFilenameSupport
+        dirnameAndFilenameSupport,
       ],
       external: ['fontkit-next'],
     })
@@ -81,7 +81,7 @@ const node = () => {
         importPath,
         correctSet,
         vendorSourceMap,
-        dirnameAndFilenameSupport
+        dirnameAndFilenameSupport,
       ],
       external: ['fontkit-next', 'pdfkit'], // TODO: Remove external pdfkit
     })

scripts/esbuild-dirname-and-filename.js

@@ -1,24 +1,24 @@
-const fs = require("fs");
-const path = require("path");
+const fs = require('fs')
+const path = require('path')
 
-const nodeModules = new RegExp(/^(?:.*[\\\/])?node_modules(?:[\\\/].*)?$/);
+const nodeModules = new RegExp(/^(?:.*[\\\/])?node_modules(?:[\\\/].*)?$/)
 
 module.exports = {
-  name: "dirname",
+  name: 'dirname',
   setup(build) {
     build.onLoad({ filter: /.*/ }, ({ path: filePath }) => {
       if (!filePath.match(nodeModules)) {
-        let contents = fs.readFileSync(filePath, "utf8");
-        const loader = path.extname(filePath).substring(1);
-        const dirname = path.dirname(filePath);
+        let contents = fs.readFileSync(filePath, 'utf8')
+        const loader = path.extname(filePath).substring(1)
+        const dirname = path.dirname(filePath)
         contents = contents
-          .replace("__dirname", `"${dirname}"`)
-          .replace("__filename", `"${filePath}"`);
+          .replace('__dirname', `"${dirname}"`)
+          .replace('__filename', `"${filePath}"`)
         return {
           contents,
           loader,
-        };
+        }
       }
-    });
+    })
   },
-};
+}

src/runner/pdfeasy.ts

@@ -256,6 +256,8 @@ export default class {
 
     const runType = options?.type || 'client'
 
+    this.options?.plugins?.forEach(({ onBefore }) => onBefore && onBefore())
+
     return new Promise(async (res, rej) => {
       if (!this.pdfkit) {
         this.reset()
@@ -286,6 +288,10 @@ export default class {
           .then(async () => {
             await runPluginBackground(this)
 
+            this.options?.plugins?.forEach(
+              ({ onAfter }) => onAfter && onAfter()
+            )
+
             this.pdfkit?.end()
 
             pageHandler(this).then(() => {
@@ -305,6 +311,10 @@ export default class {
         this.pipeline()
           .then(() => {
             pageHandler(this).then(() => {
+              this.options?.plugins?.forEach(
+                ({ onAfter }) => onAfter && onAfter()
+              )
+
               this.pdfkit?.end()
             })
           })

src/types.ts

@@ -146,7 +146,9 @@ export interface PluginGenerate {
 
 export interface Plugin {
   background?: PluginBackgroundCallback<string | false>
-  page?: PluginPageCallback<void>[]
+  page: PluginPageCallback<void>[]
+  onAfter?: () => void
+  onBefore?: () => void
 }
 
 export type PluginBackgroundCallback<T> = (context: PDFKit.PDFPage) => T