From af91b811bf715aa1802ff5d2cfc75ab05b19d130 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Fri, 4 Jun 2021 15:55:05 +0200
Subject: [PATCH] refactoring

---
 docs/src/demos/Experiments/Figure/figure.ts | 25 +++------------------
 1 file changed, 3 insertions(+), 22 deletions(-)

diff --git a/docs/src/demos/Experiments/Figure/figure.ts b/docs/src/demos/Experiments/Figure/figure.ts
index d8dae282..980d8d13 100644
--- a/docs/src/demos/Experiments/Figure/figure.ts
+++ b/docs/src/demos/Experiments/Figure/figure.ts
@@ -4,9 +4,6 @@ import {
   nodeInputRule,
   mergeAttributes,
   findChildrenInRange,
-  isNodeSelection,
-  Predicate,
-  NodeWithPos,
 } from '@tiptap/core'
 
 export interface FigureOptions {
@@ -127,25 +124,9 @@ export const Figure = Node.create<FigureOptions>({
 
       imageToFigure: () => ({ tr }) => {
         const { doc, selection } = tr
-        const nodes: NodeWithPos[] = []
-        const predicate: Predicate = node => node.type.name === 'image'
-
-        if (isNodeSelection(selection)) {
-          const node = doc.nodeAt(selection.from)
-
-          if (!node || !predicate(node)) {
-            return false
-          }
-
-          nodes.push({ node, pos: selection.from })
-        } else {
-          const range = {
-            from: selection.from,
-            to: selection.to,
-          }
-
-          nodes.push(...findChildrenInRange(doc, range, predicate))
-        }
+        const { from, to } = selection
+        const range = { from, to }
+        const nodes = findChildrenInRange(doc, range, node => node.type.name === 'image')
 
         if (!nodes.length) {
           return false