add type to this context in extensions

packages/core/src/utils/getSchemaTypeByName.ts

@@ -2,11 +2,11 @@ import { Schema } from 'prosemirror-model'
 
 export default function getSchemaTypeByName(name: string, schema: Schema) {
   if (schema.nodes[name]) {
-    return 'node'
+    return schema.nodes[name]
   }
 
   if (schema.marks[name]) {
-    return 'mark'
+    return schema.marks[name]
   }
 
   return null

packages/core/src/utils/getSchemaTypeNameByName.ts

@@ -0,0 +1,13 @@
+import { Schema } from 'prosemirror-model'
+
+export default function getSchemaTypeNameByName(name: string, schema: Schema) {
+  if (schema.nodes[name]) {
+    return 'node'
+  }
+
+  if (schema.marks[name]) {
+    return 'mark'
+  }
+
+  return null
+}