From 26b0f6b5c526bb4cff8fb6246eeb3496deecf0f5 Mon Sep 17 00:00:00 2001
From: Jannis R <mail@jannisr.de>
Date: Thu, 31 Oct 2024 14:38:19 +0100
Subject: [PATCH] pass workingState into formatter.{before,after}All()

---
 index.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/index.js b/index.js
index 34e8c12..052291b 100644
--- a/index.js
+++ b/index.js
@@ -224,7 +224,7 @@ LANGUAGE sql;
 		if ('string' === typeof beforeAll && beforeAll) {
 			yield beforeAll
 		} else if ('function' === typeof beforeAll) {
-			yield beforeAll(opt)
+			yield beforeAll(opt, workingState)
 		}
 
 		if (task.file) {
@@ -249,7 +249,7 @@ LANGUAGE sql;
 		if ('string' === typeof afterAll && afterAll) {
 			yield afterAll + ';\n'
 		} else if ('function' === typeof afterAll) {
-			yield afterAll(opt) + ';\n'
+			yield afterAll(opt, workingState) + ';\n'
 		}
 	}