From ef254cead7b9be052ec0211849fb78ae577095dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Mon, 25 Oct 2021 00:27:24 +0200
Subject: [PATCH] fix: fix storage context when using configure

---
 packages/core/src/Extension.ts | 9 +++++++++
 packages/core/src/Mark.ts      | 9 +++++++++
 packages/core/src/Node.ts      | 9 +++++++++
 3 files changed, 27 insertions(+)

diff --git a/packages/core/src/Extension.ts b/packages/core/src/Extension.ts
index 7befb141..51199d02 100644
--- a/packages/core/src/Extension.ts
+++ b/packages/core/src/Extension.ts
@@ -300,6 +300,15 @@ export class Extension<Options = any, Storage = any> {
 
     extension.options = mergeDeep(this.options, options) as Options
 
+    extension.storage = callOrReturn(getExtensionField<AnyConfig['addStorage']>(
+      extension,
+      'addStorage',
+      {
+        name: extension.name,
+        options: extension.options,
+      },
+    ))
+
     return extension
   }
 
diff --git a/packages/core/src/Mark.ts b/packages/core/src/Mark.ts
index 18212aed..0947f69f 100644
--- a/packages/core/src/Mark.ts
+++ b/packages/core/src/Mark.ts
@@ -414,6 +414,15 @@ export class Mark<Options = any, Storage = any> {
 
     extension.options = mergeDeep(this.options, options) as Options
 
+    extension.storage = callOrReturn(getExtensionField<AnyConfig['addStorage']>(
+      extension,
+      'addStorage',
+      {
+        name: extension.name,
+        options: extension.options,
+      },
+    ))
+
     return extension
   }
 
diff --git a/packages/core/src/Node.ts b/packages/core/src/Node.ts
index 1f9b56ce..14e28117 100644
--- a/packages/core/src/Node.ts
+++ b/packages/core/src/Node.ts
@@ -494,6 +494,15 @@ export class Node<Options = any, Storage = any> {
 
     extension.options = mergeDeep(this.options, options) as Options
 
+    extension.storage = callOrReturn(getExtensionField<AnyConfig['addStorage']>(
+      extension,
+      'addStorage',
+      {
+        name: extension.name,
+        options: extension.options,
+      },
+    ))
+
     return extension
   }