fix: fix storage context when using configure

ueberdosis · Oct 24, 2021 · ef254ce · ef254ce
1 parent da2d3d4
commit ef254ce
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
   }